progenitor-cli 0.3.0

A CLI tool for generating custom code templates
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

use clap::Args;
use colored::Colorize;
use std::path::PathBuf;
use std::fs;

#[derive(Args, Debug)]
pub struct Remove {
    /// Name of the template to remove
    #[arg(short, long)]
    pub name: Option<String>,
    /// Remove all templates
    #[arg(long)]
    pub all: bool,
}

impl Remove {
    pub fn execute(&self) {
        let home_dir = match dirs::home_dir() {
            Some(path) => path,
            None => {
                eprintln!("{}: Could not find home directory.", "Error".red());
                return;
            }
        };
        let progenitor_templates_dir = home_dir.join(".progenitor").join("templates");

        if self.all {
            self.remove_all_templates(&progenitor_templates_dir);
        } else if let Some(name) = &self.name {
            self.remove_single_template(&progenitor_templates_dir, name);
        } else {
            eprintln!("{}: Please specify a template name to remove or use the --all flag.", "Error".red());
        }
    }

    fn remove_all_templates(&self, templates_dir: &PathBuf) {
        if !templates_dir.exists() {
            println!("{}: No templates directory found at {}. Nothing to remove.", "Info".blue(), templates_dir.display());
            return;
        }

        match fs::remove_dir_all(templates_dir) {
            Ok(_) => println!("{}: All templates removed successfully from {}!", "Success".green(), templates_dir.display()),
            Err(e) => eprintln!("{}: Failed to remove all templates from {}: {}", "Error".red(), templates_dir.display(), e),
        }
    }

    fn remove_single_template(&self, templates_dir: &PathBuf, name: &str) {
        let template_path = templates_dir.join(name);

        if !template_path.exists() {
            eprintln!("{}: Template '{}' not found at {}.", "Error".red(), name, template_path.display());
            return;
        }

        match fs::remove_dir_all(&template_path) {
            Ok(_) => println!("{}: Template '{}' removed successfully!", "Success".green(), name),
            Err(e) => eprintln!("{}: Failed to remove template '{}': {}", "Error".red(), name, e),
        }
    }
}