From 8fd29f834b8875f51f7a6ac986d58fc5791d44b9 Mon Sep 17 00:00:00 2001
From: Miles Johnson <mileswjohnson@gmail.com>
Date: Mon, 2 Oct 2023 10:42:40 -0700
Subject: [PATCH] Enable code.

---
 CHANGELOG.md |  4 ++++
 src/proto.rs | 62 ++++++++++++++++++++++++----------------------------
 2 files changed, 32 insertions(+), 34 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 59675f3..80e3107 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## 0.3.0
 
+#### 🚀 Updates
+
+- Will now attempt to install `rustup` if it does not exist on the current machine.
+
 #### 🐞 Fixes
 
 - Will now respect the `RUSTUP_HOME` environment variable when locating the `.rustup` store.
diff --git a/src/proto.rs b/src/proto.rs
index c126e23..27be77b 100644
--- a/src/proto.rs
+++ b/src/proto.rs
@@ -45,39 +45,33 @@ pub fn native_install(
 
     // Install rustup if it does not exist
     if !command_exists(&env, "rustup") {
-        // host_log!("Installing rustup");
-
-        // NOTE: this won't work until proto v0.19.1, how to allow it otherwise???
-
-        // let is_windows = env.os.is_windows();
-        // let script_path = PathBuf::from("/proto/temp").join(if is_windows {
-        //     "rustup-init.exe"
-        // } else {
-        //     "rustup-init.sh"
-        // });
-
-        // if !script_path.exists() {
-        //     fs::write(
-        //         &script_path,
-        //         fetch_url_text(if is_windows {
-        //             "https://win.rustup.rs"
-        //         } else {
-        //             "https://sh.rustup.rs"
-        //         })?,
-        //     )?;
-        // }
-
-        // exec_command!(ExecCommandInput {
-        //     command: script_path.to_string_lossy().to_string(),
-        //     args: vec!["--default-toolchain".into(), "none".into(), "-y".into()],
-        //     set_executable: true,
-        //     stream: true,
-        //     ..ExecCommandInput::default()
-        // });
-
-        return err!(
-            "proto requires `rustup` to be installed and available on PATH to use Rust. Please install it and try again."
-        );
+        host_log!("Installing rustup");
+
+        let is_windows = env.os.is_windows();
+        let script_path = PathBuf::from("/proto/temp").join(if is_windows {
+            "rustup-init.exe"
+        } else {
+            "rustup-init.sh"
+        });
+
+        if !script_path.exists() {
+            fs::write(
+                &script_path,
+                fetch_url_text(if is_windows {
+                    "https://win.rustup.rs"
+                } else {
+                    "https://sh.rustup.rs"
+                })?,
+            )?;
+        }
+
+        exec_command!(ExecCommandInput {
+            command: script_path.to_string_lossy().to_string(),
+            args: vec!["--default-toolchain".into(), "none".into(), "-y".into()],
+            set_executable: true,
+            stream: true,
+            ..ExecCommandInput::default()
+        });
     }
 
     let channel = if input.context.version == "canary" {
@@ -88,7 +82,7 @@ pub fn native_install(
 
     let triple = format!("{}-{}", channel, get_target_triple(&env, NAME)?);
 
-    host_log!("Installing target \"{}\" with rustup", triple);
+    host_log!("Installing target {} with rustup", triple);
 
     // Install if not already installed
     let installed_list = exec_command!(pipe, "rustup", ["toolchain", "list"]);