cu-profiler-cli 0.1.1

Command-line interface for cu-profiler
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

//! `cu-profiler` CLI entry point — a thin wrapper over `cu-profiler-core` and
//! `cu-profiler-report`. All real logic lives in the library crates; this binary
//! only parses arguments, configures tracing, dispatches, and translates results
//! into the documented exit codes.
#![forbid(unsafe_code)]

mod args;
mod commands;
mod exit;

use clap::Parser;
use tracing::Level;

use args::{BaselineCommand, Cli, Command};
use cu_profiler_core::Result;
use exit::{ExitCode, code_for_error};

fn main() {
    let cli = Cli::parse();
    init_tracing(&cli);

    let code = match dispatch(&cli) {
        Ok(exit_code) => exit_code.code(),
        Err(err) => {
            eprintln!("error: {err}");
            code_for_error(&err).code()
        }
    };
    std::process::exit(code);
}

fn dispatch(cli: &Cli) -> Result<ExitCode> {
    let quiet = cli.quiet;
    match &cli.command {
        Command::Init(args) => commands::init(args, quiet),
        Command::Run(args) => commands::run(args, quiet),
        Command::Compare(args) => commands::compare(args, quiet),
        Command::Ci(args) => commands::ci(args, quiet),
        Command::Explain(args) => commands::explain(args, quiet),
        Command::Inspect(args) => commands::inspect(args, quiet),
        Command::Import(args) => commands::import(args, quiet),
        Command::Baseline(args) => match &args.command {
            BaselineCommand::Save(a) => commands::baseline_save(a, quiet),
            BaselineCommand::Approve(a) => commands::baseline_approve(a, quiet),
        },
    }
}

/// Configure tracing for the tool itself. Events go to stderr so stdout stays
/// reserved for report output.
fn init_tracing(cli: &Cli) {
    let level = if cli.trace {
        Level::TRACE
    } else if cli.quiet {
        Level::ERROR
    } else {
        match cli.verbose {
            0 => Level::WARN,
            1 => Level::INFO,
            2 => Level::DEBUG,
            _ => Level::TRACE,
        }
    };
    // Ignore the error if a subscriber is already set (e.g. in tests).
    let _ = tracing_subscriber::fmt()
        .with_max_level(level)
        .with_writer(std::io::stderr)
        .try_init();
}