capulus 0.6.5

Shared support for Rust command-line tools.
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

use std::process::{ExitCode, Termination};

use anyhow::Result;

use crate::{cancel::INTERRUPTED_EXIT_CODE, error_is_cancelled, progress::Ui};

pub struct CliTermination {
    result: Result<i32>,
    ui: Option<Ui>,
}

impl CliTermination {
    pub fn without_ui(result: Result<i32>) -> Self {
        Self { result, ui: None }
    }

    pub fn with_ui(ui: &Ui, result: Result<i32>) -> Self {
        Self {
            result,
            ui: Some(ui.clone()),
        }
    }

    fn report_error(ui: Option<&Ui>, message: &str) {
        match ui {
            Some(ui) => ui.error(message),
            None => crate::ui::print_error(message),
        }
    }
}

impl Termination for CliTermination {
    fn report(self) -> ExitCode {
        let Self { result, ui } = self;
        match result {
            Ok(code) => match u8::try_from(code) {
                Ok(code) => ExitCode::from(code),
                Err(_) => {
                    Self::report_error(ui.as_ref(), &format!("invalid command exit code {code}"));
                    ExitCode::FAILURE
                }
            },
            Err(error) if error_is_cancelled(&error) => ExitCode::from(
                u8::try_from(INTERRUPTED_EXIT_CODE).expect("SIGINT exit code fits in one byte"),
            ),
            Err(error) => {
                Self::report_error(ui.as_ref(), &error.to_string());
                ExitCode::FAILURE
            }
        }
    }
}