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

use clap::{Parser, Subcommand};
use colored::Colorize;
use once_cell::sync::Lazy;

mod commands;
mod utils;

pub mod configs;

/// The Precise and Reproducible Benchmarking Harness CLI
#[derive(Parser)]
pub struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    Run(commands::run::RunArgs),
    Report(commands::report::ReportArgs),
}

/// Plot benchmark results
#[derive(Parser)]
struct PlotArgs {}

static CMD_ARGS: Lazy<Cli> = Lazy::new(|| {
    let mut args = std::env::args().collect::<Vec<_>>();
    if args.len() > 1 && args[1] == "harness" {
        args = args[1..].to_vec();
    }
    Cli::parse_from(args)
});

pub fn dump_backtrace(e: &anyhow::Error) {
    let env = std::env::var("RUST_BACKTRACE");
    if env.is_ok() && env != Ok("0".to_string()) {
        eprintln!("BACKTRACE:");
        eprintln!("{}", e.backtrace());
    }
}

#[doc(hidden)]
pub fn main() -> anyhow::Result<()> {
    let args = &*CMD_ARGS;
    let result = entey(args);
    if result.is_err() {
        std::process::exit(1);
    }
    Ok(())
}

#[doc(hidden)]
pub fn entey(args: &Cli) -> anyhow::Result<()> {
    let git = git_info2::get();
    let run_result = match &args.command {
        Commands::Run(cmd) => cmd.run(),
        Commands::Report(cmd) => cmd.run(),
    };
    if let Err(err) = run_result.as_ref() {
        eprintln!("❌ {}: {}", "ERROR".red().bold(), err.to_string().red());
        dump_backtrace(err);
    }
    let restore_result = utils::git::restore_git_state(&git);
    if let Err(err) = restore_result.as_ref() {
        eprintln!("❌ {}: {}", "ERROR".red().bold(), err.to_string().red());
        dump_backtrace(err);
    }
    Ok(())
}