Enable to set multiple actions for completion

Author: ryuichiueda

src/core/builtins/complete.rs

@@ -59,11 +59,11 @@ fn print_each_complete(name: &str, info: &CompletionEntry) -> i32 {
         }
     } else if !info.function.is_empty() {
         print!("complete {}-F {} ", &o_options, &info.function);
-    } else if !info.action.is_empty() {
-        let symbol = action_to_reduce_symbol(&info.action);
+    } else if !info.actions.is_empty() {
+        let symbol = action_to_reduce_symbol(&info.actions[0]);
 
         if symbol.is_empty() {
-            print!("complete {}-A {} ", &o_options, &info.action);
+            print!("complete {}-A {} ", &o_options, &info.actions[0]);
         } else {
             print!("complete {}-{} ", &o_options, &symbol);
         }
@@ -195,17 +195,27 @@ pub fn complete(core: &mut ShellCore, args: &[String]) -> i32 {
         }
     }
 
-    let action = opt_to_action(&args[1]);
-    if !action.is_empty() {
-        for command in &args[2..] {
+    let mut actions = vec![];
+    for i in 1..args.len() {
+        let action = opt_to_action(&args[i]);
+        if action.is_empty() {
+            break;
+        }
+        actions.push(action);
+    }
+
+    //let action = opt_to_action(&args[1]);
+    if !actions.is_empty() {
+        let start = actions.len() + 1;
+        for command in &args[start..] {
             if !core.completion.entries.contains_key(command) {
                 core.completion
                     .entries
                     .insert(command.clone(), CompletionEntry::default());
             }
 
             let info = &mut core.completion.entries.get_mut(command).unwrap();
-            info.action = action.clone();
+            info.actions = actions.clone();
             info.options = options.clone();
             info.o_options = o_options.clone();
         }
@@ -221,7 +231,7 @@ pub fn complete(core: &mut ShellCore, args: &[String]) -> i32 {
             }
 
             let info = &mut core.completion.entries.get_mut(command).unwrap();
-            info.action = args[2].clone();
+            info.actions.push(args[2].clone());
             info.options = options.clone();
         }
 

src/core/completion.rs

@@ -14,7 +14,7 @@ pub struct Completion {
 pub struct CompletionEntry {
     pub function: String,
     pub o_options: Vec<String>,
-    pub action: String,
+    pub actions: Vec<String>,
     pub options: HashMap<String, String>,
     pub large_w_cands: String,
 }

src/feeder/terminal/completion.rs

@@ -135,8 +135,8 @@ impl Terminal {
 
         let command = format!(
             "COMPREPLY=($(compgen -A \"{}\" \"{}\"))",
-            &info.action, &target_word
-        );
+            &info.actions[0], &target_word
+        ); //TODO: use actions other than the first one
         let mut feeder = Feeder::new(&command);
 
         if let Ok(Some(mut a)) = SimpleCommand::parse(&mut feeder, core) {
@@ -168,7 +168,7 @@ impl Terminal {
         core.completion.current = info.clone();
         if !info.function.is_empty() {
             Self::exec_complete_function(&org_word, prev_pos, cur_pos, core)?;
-        } else if !info.action.is_empty() {
+        } else if !info.actions.is_empty() {
             Self::exec_action(cur_pos, core)?;
         }
 
@@ -221,11 +221,11 @@ impl Terminal {
         pos: &str,
     ) -> Vec<String> {
         if core.completion.entries.contains_key(com) {
-            let action = core.completion.entries[com].action.clone();
+            let action = core.completion.entries[com].actions.clone();
             let options = core.completion.entries[com].options.clone();
 
             if !action.is_empty() {
-                let mut cands = match action.as_ref() {
+                let mut cands = match action[0].as_ref() {
                     "alias" => compgen::compgen_a(core, args),
                     "command" => compgen::compgen_c(core, args),
                     "job" => compgen::compgen_j(core, args),