feat: Add custom completer for cargo help <TAB>

rust-lang · Sep 17, 2024 · ef028f7 · ef028f7
1 parent 25456ea
commit ef028f7
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/bin/cargo/commands/help.rs b/src/bin/cargo/commands/help.rs
@@ -15,7 +15,20 @@ const COMPRESSED_MAN: &[u8] = include_bytes!(concat!(env!("OUT_DIR"), "/man.tgz"
 pub fn cli() -> Command {
     subcommand("help")
         .about("Displays help for a cargo subcommand")
-        .arg(Arg::new("COMMAND").action(ArgAction::Set))
+        .arg(Arg::new("COMMAND").action(ArgAction::Set).add(
+            clap_complete::ArgValueCandidates::new(|| {
+                // TODO: Need a faster way to get all subcommand names.
+                super::builtin()
+                    .iter()
+                    .map(|cmd| {
+                        let name = cmd.get_name();
+                        clap_complete::CompletionCandidate::new(name)
+                            .help(cmd.get_about().cloned())
+                            .hide(cmd.is_hide_set())
+                    })
+                    .collect()
+            }),
+        ))
 }
 
 pub fn exec(gctx: &mut GlobalContext, args: &ArgMatches) -> CliResult {