use and remove commands: restore .env from vault, delete projects

Author: tusharkhatriofficial

src/cli/mod.rs

@@ -2,6 +2,8 @@ pub mod init;
 pub mod add;
 pub mod add_auto;
 pub mod list;
+pub mod use_project;
+pub mod remove;
 use clap::{Parser, Subcommand};
 
 #[derive(Parser)]

src/cli/remove.rs

@@ -0,0 +1,43 @@
+use anyhow::{Context, Result};
+use colored::Colorize;
+use std::io::{self, Write};
+
+use crate::crypto;
+use crate::vault;
+
+pub fn handle_remove(name: &str) -> Result<()> {
+    let password = crypto::prompt_existing_password()?;
+    let conn = vault::open_vault(&password)
+        .context("Failed to open vault")?;
+
+    // Check project exists
+    let project = vault::project::get_project(&conn, name)?;
+    let var_count = vault::project::count_variables(&conn, &project.id)?;
+
+    // Confirm deletion
+    print!(
+        "Delete project {} ({} variables)? [y/N]: ",
+        name.cyan().bold(),
+        var_count
+    );
+    io::stdout().flush()?;
+
+    let mut input = String::new();
+    io::stdin().read_line(&mut input)?;
+
+    if input.trim().to_lowercase() != "y" {
+        println!("Cancelled.");
+        return Ok(());
+    }
+
+    vault::project::delete_project(&conn, name)?;
+
+    println!(
+        "{} Removed project {} and {} variables",
+        "Done.".green().bold(),
+        name.cyan(),
+        var_count
+    );
+
+    Ok(())
+}

src/cli/use_project.rs

@@ -0,0 +1,63 @@
+use anyhow::{Context, Result};
+use colored::Colorize;
+use std::collections::BTreeMap;
+use std::env;
+
+use crate::crypto;
+use crate::env_parser;
+use crate::vault;
+
+pub fn handle_use(project_name: &str) -> Result<()> {
+    let password = crypto::prompt_existing_password()?;
+    let conn = vault::open_vault(&password)
+        .context("Failed to open vault")?;
+    let enc_key = vault::get_encryption_key(&conn, &password)?;
+
+    // Get the project
+    let project = vault::project::get_project(&conn, project_name)?;
+
+    // Get all variables
+    let variables = vault::variable::get_variables(&conn, &project.id)?;
+
+    if variables.is_empty() {
+        println!(
+            "{} Project {} has no variables.",
+            "Warning:".yellow(),
+            project_name.cyan()
+        );
+        return Ok(());
+    }
+
+    // Decrypt all values
+    let mut vars = BTreeMap::new();
+    for var in &variables {
+        let decrypted = crypto::decrypt_value(&enc_key, &var.encrypted_value)
+            .context(format!("Failed to decrypt variable: {}", var.key))?;
+        vars.insert(var.key.clone(), decrypted);
+    }
+
+    // Write .env to current directory
+    let cwd = env::current_dir().context("Could not determine current directory")?;
+    let env_path = cwd.join(".env");
+
+    // Warn if .env already exists
+    if env_path.exists() {
+        println!(
+            "{} Existing .env file will be overwritten.",
+            "Warning:".yellow()
+        );
+    }
+
+    env_parser::write_env_file(&env_path, &vars)?;
+
+    // Update last_used_at
+    vault::project::touch_project(&conn, project_name)?;
+
+    println!(
+        "{} Wrote {} variables to .env",
+        "Done.".green().bold(),
+        variables.len()
+    );
+
+    Ok(())
+}

src/main.rs

@@ -18,6 +18,8 @@ fn main() -> Result<()> {
         Commands::Add { name } => cli::add::handle_add(&name)?,
         Commands::AddAuto => cli::add_auto::handle_add_auto()?,
         Commands::List => cli::list::handle_list()?,
+        Commands::Use { project } => cli::use_project::handle_use(&project)?,
+        Commands::Remove { name } => cli::remove::handle_remove(&name)?,
 
         // placeholder commands for now - will implement later
         Commands::Migrate => println!("dotkeep migrate: upgrading schema..."),
@@ -26,12 +28,12 @@ fn main() -> Result<()> {
         // Commands::Add { name } => println!("dotkeep add: scanning .env for {}...", name),
         // Commands::AddAuto => println!("dotkeep add-auto: detecting project..."),
         // Commands::List => println!("dotkeep list: showing all projects..."),
-        Commands::Remove { name } => println!("dotkeep remove: deleting {}...", name),
+        // Commands::Remove { name } => println!("dotkeep remove: deleting {}...", name),
         Commands::Inspect { name } => println!("dotkeep inspect: showing {}...", name),
         Commands::Diff { project1, project2 } => {
             println!("dotkeep diff: comparing {} vs {}...", project1, project2);
         }
-        Commands::Use { project } => println!("dotkeep use: writing .env for {}...", project),
+        // Commands::Use { project } => println!("dotkeep use: writing .env for {}...", project),
         Commands::Status => println!("dotkeep status: showing active project..."),
         Commands::Recent => println!("dotkeep recent: showing recent projects..."),
         Commands::Search { key } => println!("dotkeep search: finding {}...", key),