From eedfe97e7d3cbacc590fda3a03f65314d002e6c7 Mon Sep 17 00:00:00 2001
From: Roland Schaer <roele33@hotmail.com>
Date: Tue, 14 Jan 2025 11:46:50 -0300
Subject: [PATCH] feat: allow to specify Rust profile

---
 docs/lang/rust.md        | 19 +++++++++++++++++++
 src/cmd.rs               |  7 +++++++
 src/plugins/core/rust.rs |  8 ++++++++
 3 files changed, 34 insertions(+)

diff --git a/docs/lang/rust.md b/docs/lang/rust.md
index 2520cabbb8..7a75361f9e 100644
--- a/docs/lang/rust.md
+++ b/docs/lang/rust.md
@@ -32,6 +32,25 @@ mise use -g rust@1.82
 cargo build
 ```
 
+## Tool Options
+
+The following [tool-options](/dev-tools/#tool-options) are available for the `rust` backend—these
+go in `[tools]` in `mise.toml`.
+
+### `profile`
+
+The `profile` option allows you to specify the type of release to install. The following values
+are supported:
+
+- `minimal`: Includes as few components as possible to get a working compiler (rustc, rust-std, and cargo)
+- `default` (default): Includes all of components in the minimal profile, and adds rust-docs, rustfmt, and clippy
+- `complete`: Includes all the components available through rustup. This should never be used, as it includes every component ever included in the metadata and thus will almost always fail.
+
+```toml
+[tools]
+"rust" = { version = "1.83.0", profile = "minimal" }
+```
+
 ## Settings
 
 <script setup>
diff --git a/src/cmd.rs b/src/cmd.rs
index 78bbda6712..f2738d145b 100644
--- a/src/cmd.rs
+++ b/src/cmd.rs
@@ -248,6 +248,13 @@ impl<'a> CmdLineRunner<'a> {
         None
     }
 
+    pub fn opt_arg<S: AsRef<OsStr>>(mut self, arg: Option<S>) -> Self {
+        if let Some(arg) = arg {
+            self.cmd.arg(arg);
+        }
+        self
+    }
+
     pub fn arg<S: AsRef<OsStr>>(mut self, arg: S) -> Self {
         self.cmd.arg(arg.as_ref());
         self
diff --git a/src/plugins/core/rust.rs b/src/plugins/core/rust.rs
index 9431c50851..e56c34a77a 100644
--- a/src/plugins/core/rust.rs
+++ b/src/plugins/core/rust.rs
@@ -59,6 +59,10 @@ impl RustPlugin {
     fn target_triple(&self, tv: &ToolVersion) -> String {
         format!("{}-{}", tv.version, TARGET)
     }
+
+    fn tv_profile(&self, tv: &ToolVersion) -> Option<String> {
+        tv.request.options().get("profile").cloned()
+    }
 }
 
 impl Backend for RustPlugin {
@@ -98,10 +102,14 @@ impl Backend for RustPlugin {
     fn install_version_(&self, ctx: &InstallContext, tv: ToolVersion) -> Result<ToolVersion> {
         self.setup_rustup(ctx, &tv)?;
 
+        let profile = self.tv_profile(&tv);
+
         CmdLineRunner::new(RUSTUP_BIN)
             .with_pr(&ctx.pr)
             .arg("toolchain")
             .arg("install")
+            .opt_arg(profile.as_ref().and_then(|_| Some("--profile".to_string())))
+            .opt_arg(profile)
             .arg(&tv.version)
             .prepend_path(self.list_bin_paths(&tv)?)?
             .envs(self.exec_env(&Config::get(), Config::get().get_toolset()?, &tv)?)