use std::fmt; use cursive::event::{Event, EventResult, Key}; use cursive::theme::{BaseColor, Color, ColorStyle}; use cursive::view::Resizable; use cursive::views::{EditView, LinearLayout, OnEventView, TextView}; use cursive::Cursive; use crate::app::App; use crate::utils::{GRID_WIDTH, VIEW_WIDTH}; static COMMANDS: &'static [&'static str] = &[ "add", "add-auto", "delete", "track-up", "track-down", "month-prev", "month-next", "quit", "write", "help", ]; fn get_command_completion(prefix: &str) -> Option { let first_match = COMMANDS.iter().filter(|&x| x.starts_with(prefix)).next(); return first_match.map(|&x| x.into()); } fn get_habit_completion(prefix: &str, habit_names: &[String]) -> Option { let first_match = habit_names.iter().filter(|&x| x.starts_with(prefix)).next(); eprintln!("{:?}| {:?}", prefix, first_match); return first_match.map(|x| x.into()); } pub fn open_command_window(s: &mut Cursive) { let habit_list: Vec = s .call_on_name("Main", |view: &mut App| { return view.list_habits(); }) .unwrap(); let style = ColorStyle::new(Color::Dark(BaseColor::Black), Color::Dark(BaseColor::White)); let command_window = OnEventView::new( EditView::new() .filler(" ") .on_submit(call_on_app) .style(style), ) .on_event_inner( Event::Key(Key::Tab), move |view: &mut EditView, _: &Event| { let contents = view.get_content(); if !contents.contains(" ") { let completion = get_command_completion(&*contents); if let Some(c) = completion { let cb = view.set_content(c); return Some(EventResult::Consumed(Some(cb))); }; return None; } else { let word = contents.split(' ').last().unwrap(); let completion = get_habit_completion(word, &habit_list); eprintln!("{:?} | {:?}", completion, contents); if let Some(c) = completion { let cb = view.set_content(format!("{}", contents) + &c[word.len()..]); return Some(EventResult::Consumed(Some(cb))); }; return None; } }, ) .fixed_width(VIEW_WIDTH * GRID_WIDTH); s.call_on_name("Frame", |view: &mut LinearLayout| { let mut commandline = LinearLayout::horizontal() .child(TextView::new(":")) .child(command_window); commandline.set_focus_index(1); view.add_child(commandline); view.set_focus_index(1); }); } fn call_on_app(s: &mut Cursive, input: &str) { // things to do after recieving the command // 1. parse the command // 2. clean existing command messages // 3. remove the command window // 4. handle quit command s.call_on_name("Main", |view: &mut App| { let cmd = Command::from_string(input); view.clear_message(); view.parse_command(cmd); }); s.call_on_name("Frame", |view: &mut LinearLayout| { view.set_focus_index(0); view.remove_child(view.get_focus_index()); }); // special command that requires access to // our main cursive object, has to be parsed again // here // TODO: fix this somehow if let Ok(Command::Quit) = Command::from_string(input) { s.quit(); } } #[derive(PartialEq, Debug)] pub enum Command { Add(String, Option, bool), MonthPrev, MonthNext, Delete(String), TrackUp(String), TrackDown(String), Help(Option), Write, Quit, Blank, } #[derive(Debug)] pub enum CommandLineError { InvalidCommand(String), // command name NotEnoughArgs(String, u32), // command name, required no. of args } impl std::error::Error for CommandLineError {} impl fmt::Display for CommandLineError { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { match self { CommandLineError::InvalidCommand(s) => write!(f, "Invalid command: `{}`", s), CommandLineError::NotEnoughArgs(s, n) => { write!(f, "Command `{}` requires atleast {} argument(s)!", s, n) } } } } type Result = std::result::Result; impl Command { pub fn from_string>(input: P) -> Result { let mut strings: Vec<&str> = input.as_ref().trim().split(' ').collect(); if strings.is_empty() { return Ok(Command::Blank); } let first = strings.first().unwrap().to_string(); let args: Vec = strings.iter_mut().skip(1).map(|s| s.to_string()).collect(); let mut _add = |auto: bool, first: String| { return parse_add(first, args.clone(), auto); }; match first.as_ref() { "add" | "a" => _add(false, first), "add-auto" | "aa" => _add(true, first), "delete" | "d" => { if args.is_empty() { return Err(CommandLineError::NotEnoughArgs(first, 1)); } return Ok(Command::Delete(args[0].to_string())); } "track-up" | "tup" => { if args.is_empty() { return Err(CommandLineError::NotEnoughArgs(first, 1)); } return Ok(Command::TrackUp(args[0].to_string())); } "track-down" | "tdown" => { if args.is_empty() { return Err(CommandLineError::NotEnoughArgs(first, 1)); } return Ok(Command::TrackDown(args[0].to_string())); } "h" | "?" | "help" => { if args.is_empty() { return Ok(Command::Help(None)); } return Ok(Command::Help(Some(args[0].to_string()))); } "mprev" | "month-prev" => return Ok(Command::MonthPrev), "mnext" | "month-next" => return Ok(Command::MonthNext), "q" | "quit" => return Ok(Command::Quit), "w" | "write" => return Ok(Command::Write), "" => return Ok(Command::Blank), s => return Err(CommandLineError::InvalidCommand(s.into())), } } } fn parse_add(verb: String, args: Vec, auto: bool) -> Result { if args.is_empty() { return Err(CommandLineError::NotEnoughArgs(verb, 1)); } let mut pos = 1; let mut acc = "".to_owned(); let mut new_goal: Option = None; for s1 in args { if pos == 1 { acc.push_str(&s1); } else { if let Ok(n) = s1.parse::() { new_goal = Some(n); } else { acc.push(' '); acc.push_str(&s1); } } pos = pos + 1; } return Ok(Command::Add(acc, new_goal, auto)); } #[cfg(test)] mod tests { use super::*; #[test] fn parse_add_command() { let result = Command::from_string("add eat 2"); assert!(result.is_ok()); match result.unwrap() { Command::Add(name, goal, auto) => { assert_eq!(name, "eat"); assert_eq!(goal.unwrap(), 2); assert_eq!(auto, false); } _ => panic!(), } } #[test] fn parse_add_command_without_goal() { let result = Command::from_string("add eat"); assert!(result.is_ok()); match result.unwrap() { Command::Add(name, goal, auto) => { assert_eq!(name, "eat"); assert!(goal.is_none()); assert_eq!(auto, false); } _ => panic!(), } } // #[test] fn parse_add_command_with_long_name() { let result = Command::from_string("add \"eat healthy\" 5"); assert!(result.is_ok()); match result.unwrap() { Command::Add(name, goal, auto) => { assert_eq!(name, "eat healthy"); assert_eq!(goal.unwrap(), 5); assert_eq!(auto, false); } _ => panic!(), } } #[test] fn parse_add_auto_command() { let result = Command::from_string("add-auto eat 2"); assert!(result.is_ok()); match result.unwrap() { Command::Add(name, goal, auto) => { assert_eq!(name, "eat"); assert_eq!(goal.unwrap(), 2); assert_eq!(auto, true); } _ => panic!(), } } #[test] fn parse_delete_command() { let result = Command::from_string("delete eat"); assert!(result.is_ok()); match result.unwrap() { Command::Delete(name) => { assert_eq!(name, "eat"); } _ => panic!(), } } #[test] fn parse_track_up_command() { let result = Command::from_string("track-up eat"); assert!(result.is_ok()); match result.unwrap() { Command::TrackUp(name) => { assert_eq!(name, "eat"); } _ => panic!(), } } #[test] fn parse_track_down_command() { let result = Command::from_string("track-down eat"); assert!(result.is_ok()); match result.unwrap() { Command::TrackDown(name) => { assert_eq!(name, "eat"); } _ => panic!(), } } #[test] fn parse_help_command() { let result = Command::from_string("help add"); assert!(result.is_ok()); match result.unwrap() { Command::Help(name) => { assert_eq!(name.unwrap(), "add"); } _ => panic!(), } } #[test] fn parse_month_prev_command() { let result = Command::from_string("mprev"); assert!(result.is_ok()); assert_eq!(result.unwrap(), Command::MonthPrev); } #[test] fn parse_month_next_command() { let result = Command::from_string("mnext"); assert!(result.is_ok()); assert_eq!(result.unwrap(), Command::MonthNext); } #[test] fn parse_quit_command() { let result = Command::from_string("q"); assert!(result.is_ok()); assert_eq!(result.unwrap(), Command::Quit); } #[test] fn parse_write_command() { let result = Command::from_string("w"); assert!(result.is_ok()); assert_eq!(result.unwrap(), Command::Write); } #[test] fn parse_no_command() { let result = Command::from_string(""); assert!(result.is_ok()); assert_eq!(result.unwrap(), Command::Blank); } }