vaporise/src/main.rs

use anyhow::{Context, Result};
use clap::Parser;
use colored::Colorize;
use std::{fs, path, process};
use std::{io, io::Write};

#[derive(Parser, Debug)]
#[command(version, about, long_about = None, author)]
/// A simple, featureful and blazingly fast memory-safe alternative to 'rm' written in Rust.
struct Args {
    /// Don't try to preserve '~' or '/'
    #[arg(long)]
    no_preserve: bool,

    /// Ask once before removing all
    #[arg(short, long)]
    ask_once: bool,

    /// Ask before removing each target
    #[arg(short = 'x', long, conflicts_with = "ask_once")]
    ask_each: bool,

    #[arg(trailing_var_arg = true, allow_hyphen_values = false)]
    targets: Vec<String>,
}

fn confirm_parse() {
    io::stdout().flush().unwrap();
    
    let mut confirm = String::new();
    io::stdin().read_line(&mut confirm).expect("failed to read input");
    
    if confirm != "y\n" {
        process::exit(0);
    }
}

fn confirm_once() {
    print!("Are you sure you want to delete the specified targets? [y/N]: ");
    confirm_parse();
}

fn confirm_each(target: &String) {
    print!("Are you sure you want to delete {}? [y/N]: ", target);
    confirm_parse();
}

fn vaporise() -> Result<()> {
    let args = Args::parse();

    if args.targets.is_empty() {
        println!("{}: no arguments passed", "error".red().bold());
        println!(
            "{}: try 'vpr -h' for more information",
            "note".cyan().bold()
        );
        process::exit(0);
    }

    if args.ask_once {
        confirm_once();
    }

    for target in args.targets.iter() {
        if !args.no_preserve && (target == "/" || target == "~") {
            println!("{}: you're trying to delete an important directory ({})! specify '{}' if you really want to do this", "error".red().bold(), "--no-preserve".yellow(), target);
            process::exit(0);
        }

        if args.ask_each {
            confirm_each(target);
        }

        if path::Path::new(target).exists() {
            if fs::metadata(target).unwrap().is_dir() {
                fs::remove_dir_all(target)
                    .with_context(|| format!("could not remove directory: {}", target.bold()))?;
            } else {
                fs::remove_file(target)
                    .with_context(|| format!("could not remove file: {}", target.bold()))?;
            }
        } else {
            println!(
                "{}: the specified target does not exist {}",
                "error".red().bold(),
                target.yellow()
            );
        }
    }

    Ok(())
}

fn main() {
    if let Err(error) = vaporise() {
        println!("{}: {:?}", "error".red().bold(), error);
    }
}