feat: Add automation based on config file

Author: jeevithakannan2

Cargo.lock

@@ -21,6 +21,8 @@ ego-tree = { workspace = true }
 oneshot = "0.1.8"
 portable-pty = "0.8.1"
 ratatui = "0.28.1"
+serde = { version = "1.0.205", features = ["derive"], default-features = false }
+toml = { version = "0.8.19", features = ["parse"], default-features = false }
 tui-term = "0.1.12"
 temp-dir = "0.1.14"
 unicode-width = "0.2.0"

tui/Cargo.toml

@@ -0,0 +1,15 @@
+use serde::Deserialize;
+use std::path::Path;
+
+#[derive(Deserialize)]
+pub struct Config {
+    pub auto_select: Vec<String>,
+}
+
+impl Config {
+    pub fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {
+        let content = std::fs::read_to_string(path)?;
+        let config: Config = toml::from_str(&content)?;
+        Ok(config)
+    }
+}

tui/src/config.rs

@@ -1,3 +1,4 @@
+mod config;
 mod confirmation;
 mod filter;
 mod float;
@@ -9,6 +10,7 @@ mod theme;
 
 use std::{
     io::{self, stdout},
+    path::PathBuf,
     time::Duration,
 };
 
@@ -26,6 +28,11 @@ use state::AppState;
 // Linux utility toolbox
 #[derive(Debug, Parser)]
 struct Args {
+    #[arg(
+        long,
+        help = "Path to the configuration file for automatic command selection"
+    )]
+    config: Option<PathBuf>,
     #[arg(short, long, value_enum)]
     #[arg(default_value_t = Theme::Default)]
     #[arg(help = "Set the theme to use in the application")]
@@ -38,7 +45,7 @@ struct Args {
 fn main() -> io::Result<()> {
     let args = Args::parse();
 
-    let mut state = AppState::new(args.theme, args.override_validation);
+    let mut state = AppState::new(args.theme, args.override_validation, args.config);
 
     stdout().execute(EnterAlternateScreen)?;
     enable_raw_mode()?;

tui/src/main.rs

@@ -1,3 +1,4 @@
+use crate::config::Config;
 use crate::{
     confirmation::{ConfirmPrompt, ConfirmStatus},
     filter::{Filter, SearchAction},
@@ -19,6 +20,7 @@ use ratatui::{
     widgets::{Block, Borders, List, ListState, Paragraph},
     Frame,
 };
+use std::path::PathBuf;
 use std::rc::Rc;
 use temp_dir::TempDir;
 
@@ -60,6 +62,7 @@ pub struct AppState {
     multi_select: bool,
     selected_commands: Vec<Rc<ListNode>>,
     drawable: bool,
+    auto_select: Option<Vec<String>>,
     #[cfg(feature = "tips")]
     tip: &'static str,
 }
@@ -79,10 +82,16 @@ pub struct ListEntry {
 }
 
 impl AppState {
-    pub fn new(theme: Theme, override_validation: bool) -> Self {
+    pub fn new(theme: Theme, override_validation: bool, config_path: Option<PathBuf>) -> Self {
         let (temp_dir, tabs) = linutil_core::get_tabs(!override_validation);
         let root_id = tabs[0].tree.root().id();
 
+        let auto_select = config_path.and_then(|path| {
+            Config::from_file(&path)
+                .map(|config| config.auto_select)
+                .ok()
+        });
+
         let mut state = Self {
             _temp_dir: temp_dir,
             theme,
@@ -95,14 +104,44 @@ impl AppState {
             multi_select: false,
             selected_commands: Vec::new(),
             drawable: false,
+            auto_select,
             #[cfg(feature = "tips")]
             tip: get_random_tip(),
         };
 
         state.update_items();
+
+        if let Some(auto_select) = state.auto_select.clone() {
+            state.handle_initial_auto_select(&auto_select);
+        }
+
         state
     }
 
+    fn handle_initial_auto_select(&mut self, auto_select: &[String]) {
+        let item_list = self.filter.item_list();
+        self.selected_commands = auto_select
+            .iter()
+            .filter_map(|name| {
+                item_list
+                    .iter()
+                    .find(|item| item.node.name == *name)
+                    .map(|item| item.node.clone())
+            })
+            .collect();
+
+        if !self.selected_commands.is_empty() {
+            let cmd_names = self
+                .selected_commands
+                .iter()
+                .map(|node| node.name.as_str())
+                .collect::<Vec<_>>();
+
+            let prompt = ConfirmPrompt::new(&cmd_names);
+            self.focus = Focus::ConfirmationPrompt(Float::new(Box::new(prompt), 40, 40));
+        }
+    }
+
     fn get_list_item_shortcut(&self) -> Box<[Shortcut]> {
         if self.selected_item_is_dir() {
             Box::new([Shortcut::new("Go to selected dir", ["l", "Right", "Enter"])])