feat(rustup-mode)!: add ensure_active_toolchain flag to update()

Author: rami3l

src/cli/rustup_mode.rs

@@ -584,7 +584,7 @@ pub async fn main(current_dir: PathBuf, process: &Process) -> Result<utils::Exit
 
     match subcmd {
         RustupSubcmd::DumpTestament => common::dump_testament(process),
-        RustupSubcmd::Install { opts } => update(cfg, opts).await,
+        RustupSubcmd::Install { opts } => update(cfg, opts, false).await,
         RustupSubcmd::Uninstall { opts } => toolchain_remove(cfg, opts),
         RustupSubcmd::Show { verbose, subcmd } => handle_epipe(match subcmd {
             None => show(cfg, verbose),
@@ -610,11 +610,12 @@ pub async fn main(current_dir: PathBuf, process: &Process) -> Result<utils::Exit
                     force_non_host,
                     ..UpdateOpts::default()
                 },
+                false,
             )
             .await
         }
         RustupSubcmd::Toolchain { subcmd } => match subcmd {
-            ToolchainSubcmd::Install { opts } => update(cfg, opts).await,
+            ToolchainSubcmd::Install { opts } => update(cfg, opts, false).await,
             ToolchainSubcmd::List { verbose, quiet } => {
                 handle_epipe(common::list_toolchains(cfg, verbose, quiet))
             }
@@ -791,7 +792,11 @@ async fn check_updates(cfg: &Cfg<'_>) -> Result<utils::ExitCode> {
     Ok(utils::ExitCode(0))
 }
 
-async fn update(cfg: &mut Cfg<'_>, opts: UpdateOpts) -> Result<utils::ExitCode> {
+async fn update(
+    cfg: &mut Cfg<'_>,
+    opts: UpdateOpts,
+    ensure_active_toolchain: bool,
+) -> Result<utils::ExitCode> {
     let mut exit_code = utils::ExitCode(0);
 
     common::warn_if_host_is_emulated(cfg.process);
@@ -861,6 +866,13 @@ async fn update(cfg: &mut Cfg<'_>, opts: UpdateOpts) -> Result<utils::ExitCode>
         if self_update {
             exit_code &= common::self_update(|| Ok(()), cfg.process).await?;
         }
+    } else if ensure_active_toolchain {
+        let (toolchain, reason) = cfg.find_or_install_active_toolchain(true).await?;
+        info!(
+            "the active toolchain `{}` has been installed",
+            toolchain.name()
+        );
+        info!("it's active because: {reason}");
     } else {
         exit_code &= common::update_all_channels(cfg, self_update, opts.force).await?;
         info!("cleaning up downloads & tmp directories");

src/config.rs

@@ -737,7 +737,7 @@ impl<'a> Cfg<'a> {
     }
 
     #[tracing::instrument(level = "trace", skip_all)]
-    async fn find_or_install_active_toolchain(
+    pub(crate) async fn find_or_install_active_toolchain(
         &'a self,
         verbose: bool,
     ) -> Result<(Toolchain<'a>, ActiveReason)> {