refactor(cli): rewrite rustup component with clap-derive

Author: rami3l

src/cli/rustup_mode.rs

@@ -150,6 +150,12 @@ enum RustupSubcmd {
         #[command(subcommand)]
         subcmd: TargetSubcmd,
     },
+
+    /// Modify a toolchain's installed components
+    Component {
+        #[command(subcommand)]
+        subcmd: ComponentSubcmd,
+    },
 }
 
 #[derive(Debug, Subcommand)]
@@ -294,6 +300,44 @@ enum TargetSubcmd {
     },
 }
 
+#[derive(Debug, Subcommand)]
+#[command(arg_required_else_help = true, subcommand_required = true)]
+enum ComponentSubcmd {
+    /// List installed and available components
+    List {
+        #[arg(long, help = OFFICIAL_TOOLCHAIN_ARG_HELP)]
+        toolchain: Option<PartialToolchainDesc>,
+
+        /// List only installed components
+        #[arg(long)]
+        installed: bool,
+    },
+
+    /// Add a component to a Rust toolchain
+    Add {
+        #[arg(required = true, num_args = 1..)]
+        component: Vec<String>,
+
+        #[arg(long, help = OFFICIAL_TOOLCHAIN_ARG_HELP)]
+        toolchain: Option<PartialToolchainDesc>,
+
+        #[arg(long)]
+        target: Option<String>,
+    },
+
+    /// Remove a component from a Rust toolchain
+    Remove {
+        #[arg(required = true, num_args = 1..)]
+        component: Vec<String>,
+
+        #[arg(long, help = OFFICIAL_TOOLCHAIN_ARG_HELP)]
+        toolchain: Option<PartialToolchainDesc>,
+
+        #[arg(long)]
+        target: Option<String>,
+    },
+}
+
 impl Rustup {
     fn dispatch(self, cfg: &mut Cfg) -> Result<utils::ExitCode> {
         match self.subcmd {
@@ -338,6 +382,22 @@ impl Rustup {
                 TargetSubcmd::Add { target, toolchain } => target_add(cfg, target, toolchain),
                 TargetSubcmd::Remove { target, toolchain } => target_remove(cfg, target, toolchain),
             },
+            RustupSubcmd::Component { subcmd } => match subcmd {
+                ComponentSubcmd::List {
+                    toolchain,
+                    installed,
+                } => handle_epipe(component_list(cfg, toolchain, installed)),
+                ComponentSubcmd::Add {
+                    component,
+                    toolchain,
+                    target,
+                } => component_add(cfg, component, toolchain, target.as_deref()),
+                ComponentSubcmd::Remove {
+                    component,
+                    toolchain,
+                    target,
+                } => component_remove(cfg, component, toolchain, target.as_deref()),
+            },
         }
     }
 }
@@ -416,18 +476,9 @@ pub fn main() -> Result<utils::ExitCode> {
             ("dump-testament", _) => common::dump_testament()?,
             (
                 "show" | "update" | "install" | "uninstall" | "toolchain" | "check" | "default"
-                | "target",
+                | "target" | "component",
                 _,
             ) => Rustup::from_arg_matches(&matches)?.dispatch(cfg)?,
-            ("component", c) => match c.subcommand() {
-                Some(s) => match s {
-                    ("list", m) => handle_epipe(component_list(cfg, m))?,
-                    ("add", m) => component_add(cfg, m)?,
-                    ("remove", m) => component_remove(cfg, m)?,
-                    _ => unreachable!(),
-                },
-                None => unreachable!(),
-            },
             ("override", c) => match c.subcommand() {
                 Some(s) => match s {
                     ("list", _) => handle_epipe(common::list_overrides(cfg))?,
@@ -515,55 +566,6 @@ pub(crate) fn cli() -> Command {
                 .about("Dump information about the build")
                 .hide(true), // Not for users, only CI
         )
-        .subcommand(
-            Command::new("component")
-                .about("Modify a toolchain's installed components")
-                .subcommand_required(true)
-                .arg_required_else_help(true)
-                .subcommand(
-                    Command::new("list")
-                        .about("List installed and available components")
-                        .arg(
-                            Arg::new("toolchain")
-                                .help(OFFICIAL_TOOLCHAIN_ARG_HELP)
-                                .long("toolchain")
-                                .num_args(1)
-                                .value_parser(partial_toolchain_desc_parser),
-                        )
-                        .arg(
-                            Arg::new("installed")
-                                .long("installed")
-                                .help("List only installed components")
-                                .action(ArgAction::SetTrue),
-                        ),
-                )
-                .subcommand(
-                    Command::new("add")
-                        .about("Add a component to a Rust toolchain")
-                        .arg(Arg::new("component").required(true).num_args(1..))
-                        .arg(
-                            Arg::new("toolchain")
-                                .help(OFFICIAL_TOOLCHAIN_ARG_HELP)
-                                .long("toolchain")
-                                .num_args(1)
-                                .value_parser(partial_toolchain_desc_parser),
-                        )
-                        .arg(Arg::new("target").long("target").num_args(1)),
-                )
-                .subcommand(
-                    Command::new("remove")
-                        .about("Remove a component from a Rust toolchain")
-                        .arg(Arg::new("component").required(true).num_args(1..))
-                        .arg(
-                            Arg::new("toolchain")
-                                .help(OFFICIAL_TOOLCHAIN_ARG_HELP)
-                                .long("toolchain")
-                                .num_args(1)
-                                .value_parser(partial_toolchain_desc_parser),
-                        )
-                        .arg(Arg::new("target").long("target").num_args(1)),
-                ),
-        )
         .subcommand(
             Command::new("override")
                 .about("Modify toolchain overrides for directories")
@@ -1260,40 +1262,56 @@ fn target_remove(
     Ok(utils::ExitCode(0))
 }
 
-fn component_list(cfg: &Cfg, m: &ArgMatches) -> Result<utils::ExitCode> {
-    let toolchain = explicit_desc_or_dir_toolchain_old(cfg, m)?;
+fn component_list(
+    cfg: &Cfg,
+    toolchain: Option<PartialToolchainDesc>,
+    installed_only: bool,
+) -> Result<utils::ExitCode> {
+    let toolchain = explicit_desc_or_dir_toolchain(cfg, toolchain)?;
     // downcasting required because the toolchain files can name any toolchain
     let distributable = (&toolchain).try_into()?;
-    common::list_components(distributable, m.get_flag("installed"))?;
+    common::list_components(distributable, installed_only)?;
     Ok(utils::ExitCode(0))
 }
 
-fn component_add(cfg: &Cfg, m: &ArgMatches) -> Result<utils::ExitCode> {
-    let toolchain = explicit_desc_or_dir_toolchain_old(cfg, m)?;
+fn component_add(
+    cfg: &Cfg,
+    components: Vec<String>,
+    toolchain: Option<PartialToolchainDesc>,
+    target: Option<&str>,
+) -> Result<utils::ExitCode> {
+    let toolchain = explicit_desc_or_dir_toolchain(cfg, toolchain)?;
     let distributable = DistributableToolchain::try_from(&toolchain)?;
-    let target = get_target(m, &distributable);
+    let target = get_target(target, &distributable);
 
-    for component in m.get_many::<String>("component").unwrap() {
+    for component in &components {
         let new_component = Component::try_new(component, &distributable, target.as_ref())?;
         distributable.add_component(new_component)?;
     }
 
     Ok(utils::ExitCode(0))
 }
 
-fn get_target(m: &ArgMatches, distributable: &DistributableToolchain<'_>) -> Option<TargetTriple> {
-    m.get_one::<String>("target")
-        .map(|s| &**s)
+fn get_target(
+    target: Option<&str>,
+    distributable: &DistributableToolchain<'_>,
+) -> Option<TargetTriple> {
+    target
         .map(TargetTriple::new)
         .or_else(|| Some(distributable.desc().target.clone()))
 }
 
-fn component_remove(cfg: &Cfg, m: &ArgMatches) -> Result<utils::ExitCode> {
-    let toolchain = explicit_desc_or_dir_toolchain_old(cfg, m)?;
+fn component_remove(
+    cfg: &Cfg,
+    components: Vec<String>,
+    toolchain: Option<PartialToolchainDesc>,
+    target: Option<&str>,
+) -> Result<utils::ExitCode> {
+    let toolchain = explicit_desc_or_dir_toolchain(cfg, toolchain)?;
     let distributable = DistributableToolchain::try_from(&toolchain)?;
-    let target = get_target(m, &distributable);
+    let target = get_target(target, &distributable);
 
-    for component in m.get_many::<String>("component").unwrap() {
+    for component in &components {
         let new_component = Component::try_new(component, &distributable, target.as_ref())?;
         distributable.remove_component(new_component)?;
     }

tests/suite/cli-ui/rustup/rustup_component_cmd_add_cmd_help_flag_stdout.toml

@@ -4,15 +4,15 @@ stdout = """
 ...
 Add a component to a Rust toolchain
 
-Usage: rustup[EXE] component add [OPTIONS] <component>...
+Usage: rustup[EXE] component add [OPTIONS] <COMPONENT>...
 
 Arguments:
-  <component>...  
+  <COMPONENT>...  
 
 Options:
-      --toolchain <toolchain>  Toolchain name, such as 'stable', 'nightly', or '1.8.0'. For more
+      --toolchain <TOOLCHAIN>  Toolchain name, such as 'stable', 'nightly', or '1.8.0'. For more
                                information see `rustup help toolchain`
-      --target <target>        
+      --target <TARGET>        
   -h, --help                   Print help
 """
 stderr = ""

tests/suite/cli-ui/rustup/rustup_component_cmd_list_cmd_help_flag_stdout.toml

@@ -1,13 +1,13 @@
 bin.name = "rustup"
-args = ["component","list","--help"]
+args = ["component", "list", "--help"]
 stdout = """
 ...
 List installed and available components
 
 Usage: rustup[EXE] component list [OPTIONS]
 
 Options:
-      --toolchain <toolchain>  Toolchain name, such as 'stable', 'nightly', or '1.8.0'. For more
+      --toolchain <TOOLCHAIN>  Toolchain name, such as 'stable', 'nightly', or '1.8.0'. For more
                                information see `rustup help toolchain`
       --installed              List only installed components
   -h, --help                   Print help

tests/suite/cli-ui/rustup/rustup_component_cmd_remove_cmd_help_flag_stdout.toml

@@ -4,15 +4,15 @@ stdout = """
 ...
 Remove a component from a Rust toolchain
 
-Usage: rustup[EXE] component remove [OPTIONS] <component>...
+Usage: rustup[EXE] component remove [OPTIONS] <COMPONENT>...
 
 Arguments:
-  <component>...  
+  <COMPONENT>...  
 
 Options:
-      --toolchain <toolchain>  Toolchain name, such as 'stable', 'nightly', or '1.8.0'. For more
+      --toolchain <TOOLCHAIN>  Toolchain name, such as 'stable', 'nightly', or '1.8.0'. For more
                                information see `rustup help toolchain`
-      --target <target>        
+      --target <TARGET>        
   -h, --help                   Print help
 """
 stderr = ""

tests/suite/cli-ui/rustup/rustup_help_cmd_stdout.toml

@@ -9,7 +9,6 @@ The Rust toolchain installer
 Usage: rustup[EXE] [OPTIONS] [+toolchain] [COMMAND]
 
 Commands:
-  component    Modify a toolchain's installed components
   override     Modify toolchain overrides for directories
   run          Run a command with an environment configured for a given toolchain
   which        Display which binary will be run for a given command
@@ -24,6 +23,7 @@ Commands:
   default      Set the default toolchain
   toolchain    Modify or query the installed toolchains
   target       Modify a toolchain's supported targets
+  component    Modify a toolchain's installed components
   help         Print this message or the help of the given subcommand(s)
 
 Arguments:

tests/suite/cli-ui/rustup/rustup_help_flag_stdout.toml

@@ -9,7 +9,6 @@ The Rust toolchain installer
 Usage: rustup[EXE] [OPTIONS] [+toolchain] [COMMAND]
 
 Commands:
-  component    Modify a toolchain's installed components
   override     Modify toolchain overrides for directories
   run          Run a command with an environment configured for a given toolchain
   which        Display which binary will be run for a given command
@@ -24,6 +23,7 @@ Commands:
   default      Set the default toolchain
   toolchain    Modify or query the installed toolchains
   target       Modify a toolchain's supported targets
+  component    Modify a toolchain's installed components
   help         Print this message or the help of the given subcommand(s)
 
 Arguments:

tests/suite/cli-ui/rustup/rustup_only_options_stdout.toml

@@ -9,7 +9,6 @@ The Rust toolchain installer
 Usage: rustup[EXE] [OPTIONS] [+toolchain] [COMMAND]
 
 Commands:
-  component    Modify a toolchain's installed components
   override     Modify toolchain overrides for directories
   run          Run a command with an environment configured for a given toolchain
   which        Display which binary will be run for a given command
@@ -24,6 +23,7 @@ Commands:
   default      Set the default toolchain
   toolchain    Modify or query the installed toolchains
   target       Modify a toolchain's supported targets
+  component    Modify a toolchain's installed components
   help         Print this message or the help of the given subcommand(s)
 
 Arguments: