113 lines
2.8 KiB
Rust
113 lines
2.8 KiB
Rust
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!("Delete the specified targets? [y/N]: ");
|
|
confirm_parse();
|
|
}
|
|
|
|
fn confirm_each(target: &String) {
|
|
print!("Delete {} ? [y/N]: ", target.bold());
|
|
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 {
|
|
println!();
|
|
for target in args.targets.iter() {
|
|
println!(" {}", target.bold());
|
|
}
|
|
println!();
|
|
confirm_once();
|
|
}
|
|
|
|
let root: &str;
|
|
if cfg!(windows) {
|
|
root = "C:\\";
|
|
} else if cfg!(unix) {
|
|
root = "/";
|
|
} else {
|
|
root = "N/A";
|
|
}
|
|
|
|
for target in args.targets.iter() {
|
|
if !args.no_preserve && target == root {
|
|
println!("{} you're trying to delete the root 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);
|
|
}
|
|
}
|