bot-forge 1.0.2

Rust CLI for installing agent skills and developer tools from configurable forms.
Documentation 
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

use std::path::PathBuf;

use crate::backends::apt_mirror::{
    AptMirrorPreview, apply_apt_mirror, apt_mirror_preview, check_apt_mirror, restore_apt_mirror,
};
use crate::cli::actions::{confirm_action, print_human, value_after};
use crate::config::load_config_with_overlays;
use crate::ui::{self, Document, StatusKind};

pub(crate) fn cmd_apt_mirror(args: &[String]) -> Result<(), String> {
    let Some(action) = args.first().map(String::as_str) else {
        return Err(
            "apt-mirror requires the show, check, apply, or restore subcommand".to_string(),
        );
    };
    if !matches!(action, "show" | "check" | "apply" | "restore") {
        return Err(format!(
            "unknown apt-mirror subcommand: {action}; expected show, check, apply, or restore"
        ));
    }

    let mut config_path = None;
    let mut overlays = Vec::new();
    let mut yes = false;
    let mut index = 1;
    while index < args.len() {
        match args[index].as_str() {
            "-y" | "--yes" if matches!(action, "apply" | "restore") => yes = true,
            "--config" => {
                index += 1;
                config_path = Some(PathBuf::from(value_after(args, index, "--config")?));
            }
            "--overlay" => {
                index += 1;
                overlays.push(PathBuf::from(value_after(args, index, "--overlay")?));
            }
            value => return Err(format!("unknown apt-mirror option: {value}")),
        }
        index += 1;
    }

    let loaded = load_config_with_overlays(config_path.as_deref(), &overlays)
        .map_err(|error| error.to_string())?;
    let preview = apt_mirror_preview(loaded.document.apt_mirror.as_ref())
        .map_err(|error| error.to_string())?;
    if action == "restore" {
        print_apt_mirror_preview(&preview)?;
        if !yes && !confirm_action("Replace the APT source file with the bot-forge backup?")? {
            return print_human(
                Document::with_subtitle("bot-forge", "apt mirror")
                    .status(StatusKind::Info, "Restore cancelled"),
            );
        }
        restore_apt_mirror(&preview.source_file).map_err(|error| error.to_string())?;
        return print_human(
            Document::with_subtitle("bot-forge", "apt mirror")
                .status(StatusKind::Success, "Restored from backup")
                .labeled_path("Source", preview.source_file.display().to_string()),
        );
    }
    print_apt_mirror_preview(&preview)?;

    match action {
        "show" => Ok(()),
        "check" => {
            ui::stdout_status(
                StatusKind::Info,
                "Validating APT mirror in a temporary directory.",
            );
            check_apt_mirror(&preview, |message| {
                ui::stderr_status(StatusKind::Warning, message)
            })
            .map_err(|error| error.to_string())?;
            print_human(
                Document::with_subtitle("bot-forge", "apt mirror check")
                    .status(StatusKind::Success, "Validation passed"),
            )
        }
        "apply" => {
            ui::stdout_status(
                StatusKind::Info,
                "Validating APT mirror in a temporary directory.",
            );
            check_apt_mirror(&preview, |message| {
                ui::stderr_status(StatusKind::Warning, message)
            })
            .map_err(|error| error.to_string())?;
            ui::stdout_status(StatusKind::Success, "APT mirror validation passed.");
            if !yes
                && !confirm_action(
                    "Write the source file after validation? Existing system sources will not be removed or disabled.",
                )?
            {
                return print_human(
                    Document::with_subtitle("bot-forge", "apt mirror")
                        .status(StatusKind::Info, "Write cancelled"),
                );
            }
            apply_apt_mirror(&preview).map_err(|error| error.to_string())?;
            print_human(
                Document::with_subtitle("bot-forge", "apt mirror")
                    .status(StatusKind::Success, "Write complete")
                    .labeled_path("Source", preview.source_file.display().to_string()),
            )
        }
        _ => unreachable!(),
    }
}

fn print_apt_mirror_preview(preview: &AptMirrorPreview) -> Result<(), String> {
    print_human(
        Document::with_subtitle("bot-forge", "apt mirror")
            .field(
                "Platform",
                format!(
                    "{} {} ({})",
                    preview.distribution, preview.codename, preview.architecture
                ),
            )
            .labeled_path("Source", preview.source_file.display().to_string())
            .blank()
            .section("bot-forge.sources")
            .code(preview.source_contents.trim_end()),
    )
}