kabu 0.7.1

CLI tool to enhance git worktree and jj workspace with automated setup
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
126
127
128
129
130
131
132
133

mod cli;
mod color;
mod command;
mod config;
mod error;
mod hook;
mod init;
mod interactive;
mod operation;
mod output;
mod prompt;
mod trust;
mod vcs;

use crate::color::ColorScheme;
use std::process::ExitCode;
use std::sync::atomic::{AtomicBool, Ordering};

// Flag to indicate if a termination signal was received
static SIGNAL_RECEIVED: AtomicBool = AtomicBool::new(false);

fn setup_signal_handlers() {
    // ctrlc handles SIGINT (Ctrl+C) on all platforms
    // With `termination` feature, also handles SIGTERM and SIGHUP on Unix
    let _ = ctrlc::set_handler(|| {
        SIGNAL_RECEIVED.store(true, Ordering::SeqCst);
    });
}

fn main() -> ExitCode {
    setup_signal_handlers();

    let args = cli::parse();

    let result = match args.command {
        cli::Command::Add(add_args) => {
            let color_choice = if add_args.no_color {
                clap::ColorChoice::Never
            } else {
                add_args.color
            };
            let color_config = color::ColorConfig::new(color_choice);
            command::add(add_args, color_config)
        }
        cli::Command::Remove(remove_args) => {
            let color_choice = if remove_args.no_color {
                clap::ColorChoice::Never
            } else {
                remove_args.color
            };
            let color_config = color::ColorConfig::new(color_choice);
            command::remove(remove_args, color_config)
        }
        cli::Command::List(list_args) => {
            let color_choice = if list_args.no_color {
                clap::ColorChoice::Never
            } else {
                list_args.color
            };
            let color_config = color::ColorConfig::new(color_choice);
            command::list(list_args, color_config)
        }
        cli::Command::Path(path_args) => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::path(path_args)
        }
        cli::Command::Cd => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::cd()
        }
        cli::Command::Config(config_args) => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::config(config_args.command)
        }
        cli::Command::Trust(trust_args) => {
            let color_choice = if trust_args.no_color {
                clap::ColorChoice::Never
            } else {
                trust_args.color
            };
            let color_config = color::ColorConfig::new(color_choice);
            command::trust(trust_args, color_config)
        }
        cli::Command::Untrust(untrust_args) => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::untrust(untrust_args)
        }
        cli::Command::Completions { shell } => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::completions(shell)
        }
        cli::Command::Init(init_args) => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::init(init_args)
        }
        cli::Command::Man => {
            color::ColorConfig::new(clap::ColorChoice::Auto);
            command::man()
        }
    };

    // Check if a signal was received
    if SIGNAL_RECEIVED.load(Ordering::SeqCst) {
        // Exit with 130 (128 + SIGINT) - standard for Ctrl+C termination
        // This is the most common case across platforms
        return ExitCode::from(130);
    }

    match result {
        Ok(()) => ExitCode::SUCCESS,
        Err(error::Error::Aborted) => {
            // User-initiated cancellation (e.g., Escape key) - not an error
            ExitCode::SUCCESS
        }
        Err(error::Error::HooksNotTrusted) => {
            // Detailed message already displayed by command
            ExitCode::FAILURE
        }
        Err(error::Error::TrustCheckFailed) => ExitCode::from(1),
        Err(error::Error::CdRequiresShellIntegration) => {
            eprintln!(
                "{}",
                ColorScheme::error(&error::Error::CdRequiresShellIntegration.to_string())
            );
            eprintln!("\nFor setup instructions, run: kabu init --help");
            ExitCode::FAILURE
        }
        Err(e) => {
            eprintln!("{}", ColorScheme::error(&e.to_string()));
            ExitCode::FAILURE
        }
    }
}