Add uninstall

Author: HaykWD

README.md

@@ -31,6 +31,16 @@ curl --proto '=https' --tlsv1.2 -LsSf \
 
 The installer places `rgpt` in `$CARGO_HOME/bin` (normally `~/.cargo/bin`). Add that directory to your `PATH` if it is not already present. You can also download an archive for a specific release from the [Releases page](https://github.com/ShaderCompilation/rgpt/releases).
 
+### Uninstall
+
+Run the installed command and confirm the prompt:
+
+```sh
+rgpt --uninstall
+```
+
+This removes the executable plus rgpt's configuration, saved chats, roles, and response cache. For a non-interactive uninstall, use `rgpt --uninstall --yes`.
+
 ### Build from source
 
 To build locally, install a current stable Rust toolchain, then run:

src/cli.rs

@@ -27,6 +27,14 @@ pub struct Cli {
     #[arg(long, help_heading = "Assistance Options")]
     pub editor: bool,
 
+    /// Remove rgpt, including its saved configuration, chats, roles, and cache.
+    #[arg(long, help_heading = "Application Options")]
+    pub uninstall: bool,
+
+    /// Do not ask for confirmation when uninstalling.
+    #[arg(long, requires = "uninstall", help_heading = "Application Options")]
+    pub yes: bool,
+
     /// Large language model to use.
     #[arg(long)]
     pub model: Option<String>,

src/config.rs

@@ -134,7 +134,11 @@ impl Config {
     }
 }
 
-fn config_path() -> Result<PathBuf> {
+pub fn app_dir() -> Result<PathBuf> {
     let base = dirs::config_dir().context("could not determine config directory")?;
-    Ok(base.join("rgpt").join(".rgptrc"))
+    Ok(base.join("rgpt"))
+}
+
+fn config_path() -> Result<PathBuf> {
+    Ok(app_dir()?.join(".rgptrc"))
 }

src/main.rs

@@ -9,6 +9,7 @@ mod render;
 mod role;
 mod shell_cmd;
 mod tools;
+mod uninstall;
 
 use anyhow::{Context, Result};
 use clap::Parser;
@@ -23,6 +24,10 @@ fn main() -> Result<()> {
     let cli = cli::Cli::parse();
     cli.validate()?;
 
+    if cli.uninstall {
+        return uninstall::run(cli.yes);
+    }
+
     let config = Config::load(cli.ollama).context("loading config")?;
     SystemRole::ensure_defaults(&config).context("creating default roles")?;
 

src/uninstall.rs

@@ -0,0 +1,61 @@
+use std::fs;
+use std::io::{self, Write};
+use std::path::Path;
+
+use anyhow::{Context, Result, bail};
+
+use crate::config;
+
+pub fn run(assume_yes: bool) -> Result<()> {
+    let binary = std::env::current_exe().context("locating the rgpt executable")?;
+    let data_dir = config::app_dir()?;
+
+    if !assume_yes {
+        println!("This will remove:");
+        println!("  {}", binary.display());
+        println!(
+            "  {} (configuration, chats, roles, and cache)",
+            data_dir.display()
+        );
+        print!("Continue? [y/N] ");
+        io::stdout().flush().ok();
+
+        let mut answer = String::new();
+        io::stdin()
+            .read_line(&mut answer)
+            .context("reading uninstall confirmation")?;
+        if !matches!(answer.trim().to_ascii_lowercase().as_str(), "y" | "yes") {
+            bail!("Uninstall cancelled.");
+        }
+    }
+
+    remove_data_dir(&data_dir)?;
+    fs::remove_file(&binary)
+        .with_context(|| format!("removing executable {}", binary.display()))?;
+    println!("rgpt has been uninstalled.");
+    Ok(())
+}
+
+fn remove_data_dir(path: &Path) -> Result<()> {
+    if path.exists() {
+        fs::remove_dir_all(path)
+            .with_context(|| format!("removing application data directory {}", path.display()))?;
+    }
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::remove_data_dir;
+
+    #[test]
+    fn removes_application_data_directory() {
+        let dir = std::env::temp_dir().join(format!("rgpt-uninstall-test-{}", std::process::id()));
+        std::fs::create_dir_all(&dir).unwrap();
+        std::fs::write(dir.join("config"), "secret").unwrap();
+
+        remove_data_dir(&dir).unwrap();
+
+        assert!(!dir.exists());
+    }
+}