Skip to content

Commit

Permalink
feat: Add automation based on config file
Browse files Browse the repository at this point in the history
  • Loading branch information
jeevithakannan2 committed Oct 15, 2024
1 parent 79eb752 commit 4dbcca7
Show file tree
Hide file tree
Showing 4 changed files with 77 additions and 2 deletions.
28 changes: 28 additions & 0 deletions core/src/config.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
use serde::Deserialize;
use std::path::Path;
use std::process;

#[derive(Deserialize)]
pub struct Config {
pub auto_execute: Vec<String>,
}

impl Config {
pub fn from_file(path: &Path) -> Self {
let content = match std::fs::read_to_string(path) {
Ok(content) => content,
Err(e) => {
eprintln!("Failed to read config file {}: {}", path.display(), e);
process::exit(1);
}
};

match toml::from_str(&content) {
Ok(config) => config,
Err(e) => {
eprintln!("Failed to parse config file: {}", e);
process::exit(1);
}
}
}
}
15 changes: 15 additions & 0 deletions core/src/lib.rs
Original file line number Diff line number Diff line change
@@ -1,10 +1,12 @@
mod config;
mod inner;

use std::rc::Rc;

use ego_tree::Tree;
use std::path::PathBuf;

pub use config::Config;
pub use inner::get_tabs;

#[derive(Clone, Hash, Eq, PartialEq)]
Expand Down Expand Up @@ -33,3 +35,16 @@ pub struct ListNode {
pub command: Command,
pub task_list: String,
}

impl Tab {
pub fn find_command(&self, name: &str) -> Option<Rc<ListNode>> {
self.tree.root().descendants().find_map(|node| {
let value = node.value();
if value.name == name && !node.has_children() {
Some(value.clone())
} else {
None
}
})
}
}
8 changes: 7 additions & 1 deletion tui/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ mod theme;

use std::{
io::{self, stdout},
path::PathBuf,
time::Duration,
};

Expand All @@ -26,6 +27,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")]
Expand All @@ -38,7 +44,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()?;
Expand Down
28 changes: 27 additions & 1 deletion tui/src/state.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ use crate::{
};
use crossterm::event::{KeyCode, KeyEvent, KeyEventKind, KeyModifiers};
use ego_tree::NodeId;
use linutil_core::Config;
use linutil_core::{ListNode, Tab};
#[cfg(feature = "tips")]
use rand::Rng;
Expand All @@ -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;

Expand Down Expand Up @@ -79,10 +81,12 @@ 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_execute_commands = config_path.map(|path| Config::from_file(&path).auto_execute);

let mut state = Self {
_temp_dir: temp_dir,
theme,
Expand All @@ -100,9 +104,31 @@ impl AppState {
};

state.update_items();
if let Some(auto_execute_commands) = auto_execute_commands {
state.handle_initial_auto_execute(&auto_execute_commands);
}

state
}

fn handle_initial_auto_execute(&mut self, auto_execute_commands: &[String]) {
self.selected_commands = auto_execute_commands
.iter()
.filter_map(|name| self.tabs.iter().find_map(|tab| tab.find_command(name)))
.collect();

if !self.selected_commands.is_empty() {
let cmd_names: Vec<_> = self
.selected_commands
.iter()
.map(|node| node.name.as_str())
.collect();

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"])])
Expand Down

0 comments on commit 4dbcca7

Please sign in to comment.