ws-clean 0.5.0

A simple cleanup tool for code workspaces
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use crate::init::init;
use crate::parsing::get_filter;
use clap::Parser;
use parsing::load_file_as_sections;
use std::fs::remove_file;
use std::io::stdout;
use std::io::Write;
use std::process::exit;
use walkdir::WalkDir;

mod init;
mod parsing;

#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[clap(long = "what-if", action, required = false, default_value_t = false)]
    /// Show the result of the command if it was performed
    what_if: bool,

    #[clap(
        long = "verbose",
        short = 'v',
        required = false,
        default_value_t = false
    )]
    /// Show more information
    verbose: bool,

    #[clap(long = "init", action, required = false, default_value_t = false)]
    /// Prompts to init cleanup in the working directory
    init: bool,
}

fn main() {
    let args = Args::parse();

    if args.init {
        init();
        exit(0);
    }

    let mut seen = vec![];

    if args.what_if {
        println!("Following files would be deleted:");
    }

    if args.verbose {
        writeln!(stdout(), "Deleted files:").expect("Failed to write to stdout");
    }

    for (_, val) in load_file_as_sections() {
        let filter = get_filter(&val);

        WalkDir::new(".")
            .into_iter()
            .filter_map(|e| e.ok())
            .filter(|e| filter(e))
            .for_each(|e| {
                let path = e.path().display().to_string();

                if seen.contains(&path) {
                    return;
                }

                if args.what_if || args.verbose {
                    println!("{}", e.path().display());
                }

                seen.push(path);

                if !args.what_if {
                    remove_file(e.path()).unwrap_or_else(|err| {
                        eprintln!("Couldn't delete {}: {}", e.path().display(), err);
                    });
                }
            })
    }

    if (args.verbose || args.what_if) && seen.is_empty() {
        println!("None");
    }
}