53 lines
1.6 KiB
Rust
53 lines
1.6 KiB
Rust
use std::{fs, path, process};
|
|
use clap::Parser;
|
|
use colored::Colorize;
|
|
|
|
#[derive(Parser, Debug)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Args {
|
|
#[arg(short, long)]
|
|
recursive: bool,
|
|
|
|
#[arg(long)]
|
|
no_preserve: bool,
|
|
|
|
#[arg(trailing_var_arg = true, allow_hyphen_values = false)]
|
|
targets: Vec<String>,
|
|
}
|
|
|
|
fn main() -> std::io::Result<()> {
|
|
let args = Args::parse();
|
|
|
|
if args.targets.len() == 0 {
|
|
println!("{} {}", "error:".red().bold(), "no arguents passed");
|
|
println!("{} {}", "note:".cyan().bold(), "try 'vpr -h' for more information");
|
|
process::exit(0);
|
|
}
|
|
|
|
for target in args.targets.iter() {
|
|
if args.no_preserve == false {
|
|
if target == "/" || target == "~" {
|
|
println!("{} you're trying to delete an important directory ({})! specify '--no-preserve-root' if you really want to do this", "error:".red().bold(), target);
|
|
process::exit(0);
|
|
}
|
|
}
|
|
|
|
if path::Path::new(target).exists() {
|
|
if fs::metadata(target).unwrap().is_dir() {
|
|
if args.recursive == false {
|
|
println!("{} could not remove directory {}, please specify '-r'", "error:".red().bold(), target);
|
|
process::exit(0);
|
|
} else {
|
|
let _ = fs::remove_dir_all(target);
|
|
}
|
|
} else {
|
|
let _ = fs::remove_file(target);
|
|
}
|
|
} else {
|
|
println!("{} {} {}", "error:".red().bold(), "the specified target does not exist:", target);
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|