feat: uninstall-toolchain command (#487)

* feat: uninstall-toolchain command

* fix: more verbose installs and clean up downloaded archive

jolt-core/src/host/toolchain.rs

@@ -28,8 +28,13 @@ pub fn install_toolchain() -> Result<()> {
             download_toolchain(&client, &toolchain_url)
         }))?;
         unpack_toolchain()?;
+        remove_archive()?;
         link_toolchain()?;
         write_tag_file()?;
+        println!(
+            "\"riscv32im-jolt-zkvm-elf\" toolchain installed successfully at {:?}",
+            jolt_dir()
+        );
     }
     Ok(())
 }
@@ -39,7 +44,6 @@ pub fn install_no_std_toolchain() -> Result<()> {
     std::process::Command::new("rustup")
         .args(["target", "add", "riscv32im-unknown-none-elf"])
         .output()?;
-
     Ok(())
 }
 
@@ -114,7 +118,6 @@ async fn download_toolchain(client: &Client, url: &str) -> Result<()> {
         fs::create_dir(&jolt_dir)?;
     }
 
-    println!("Downloading toolchain");
     let mut response = client.get(url).send().await?;
     if response.status().is_success() {
         let mut file = File::create(output_path)?;
@@ -146,6 +149,14 @@ async fn download_toolchain(client: &Client, url: &str) -> Result<()> {
     }
 }
 
+fn remove_archive() -> Result<()> {
+    let toolchain_archive = jolt_dir().join("rust-toolchain.tar.gz");
+    if toolchain_archive.exists() {
+        fs::remove_file(&toolchain_archive)?;
+    }
+    Ok(())
+}
+
 fn toolchain_url() -> String {
     let target = target_lexicon::HOST;
     format!(
@@ -154,6 +165,55 @@ fn toolchain_url() -> String {
     )
 }
 
+#[cfg(not(target_arch = "wasm32"))]
+pub fn uninstall_no_std_toolchain() -> Result<()> {
+    std::process::Command::new("rustup")
+        .args(["target", "remove", "riscv32im-unknown-none-elf"])
+        .output()?;
+
+    println!("\"riscv32im-unknown-none-elf\" toolchain uninstalled successfully");
+    Ok(())
+}
+
+#[cfg(not(target_arch = "wasm32"))]
+/// Uninstalls the toolchain if it is already installed
+pub fn uninstall_toolchain() -> Result<()> {
+    if !has_toolchain() {
+        println!("Toolchain is not installed");
+        return Ok(());
+    }
+
+    // Remove the linked toolchain from rustup
+    let output = std::process::Command::new("rustup")
+        .args(["toolchain", "remove", "riscv32im-jolt-zkvm-elf"])
+        .output()?;
+
+    if !output.status.success() {
+        bail!(
+            "Failed to remove toolchain: {}",
+            String::from_utf8(output.stderr)?
+        );
+    }
+
+    // Remove the unpacked toolchain directory
+    let link_path = jolt_dir().join("rust/build/host/stage2");
+    if link_path.exists() {
+        fs::remove_dir_all(&link_path)?;
+    }
+
+    // Remove the downloaded toolchain archive
+    remove_archive()?;
+
+    // Remove the toolchain tag file
+    let tag_file = toolchain_tag_file();
+    if tag_file.exists() {
+        fs::remove_file(&tag_file)?;
+    }
+
+    println!("\"riscv32im-jolt-zkvm-elf\" toolchain uninstalled successfully");
+    Ok(())
+}
+
 fn has_toolchain() -> bool {
     let tag_path = toolchain_tag_file();
     if let Ok(tag) = read_to_string(tag_path) {

src/main.rs

@@ -32,6 +32,8 @@ enum Command {
     },
     /// Installs the required RISC-V toolchains for Rust
     InstallToolchain,
+    /// Uninstalls the RISC-V toolchains for Rust
+    UninstallToolchain,
     /// Handles preprocessing and generates WASM compatible files
     BuildWasm,
 }
@@ -41,6 +43,7 @@ fn main() {
     match cli.command {
         Command::New { name, wasm } => create_project(name, wasm),
         Command::InstallToolchain => install_toolchain(),
+        Command::UninstallToolchain => uninstall_toolchain(),
         Command::BuildWasm => build_wasm(),
     }
 }
@@ -61,6 +64,12 @@ fn install_toolchain() {
     display_welcome();
 }
 
+fn uninstall_toolchain() {
+    if let Err(err) = toolchain::uninstall_toolchain() {
+        panic!("toolchain uninstall failed: {}", err);
+    }
+}
+
 fn create_folder_structure(name: &str) -> Result<()> {
     fs::create_dir(name)?;
     fs::create_dir(format!("{}/src", name))?;