wickra-benchmark-cli 0.1.2

Reference command-line runner over wickra-benchmark-core: recompute a curated case or suite against its datasets and report pass/hash-match, exiting non-zero on any failure for use in CI pipelines.
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

//! Load cases and datasets, run them, and render the result.

use crate::args::{Cli, Command, Format};
use std::fmt::Write as _;
use std::path::Path;
use wickra_benchmark_core::{load_candles, run_case, run_suite, BenchmarkCase, CaseResult, Suite};

/// The rendered output and the process exit code (`0` = all passed).
pub struct Output {
    /// The text to print to stdout.
    pub text: String,
    /// The process exit code.
    pub code: u8,
}

/// Dispatch the parsed CLI to its handler.
pub fn run(cli: &Cli) -> Result<Output, String> {
    match &cli.command {
        Command::RunCase {
            case,
            data_root,
            format,
        } => run_case_cmd(case, data_root, *format),
        Command::RunSuite {
            suite,
            data_root,
            format,
        } => run_suite_cmd(suite, data_root, *format),
        Command::ListCases { suite } => list_cases_cmd(suite),
    }
}

fn load_case(path: &Path) -> Result<BenchmarkCase, String> {
    let text = std::fs::read_to_string(path).map_err(|e| format!("{}: {e}", path.display()))?;
    let parsed = if is_toml(path) {
        BenchmarkCase::from_toml(&text)
    } else {
        BenchmarkCase::from_json(&text)
    };
    parsed.map_err(|e| format!("{}: {e}", path.display()))
}

fn load_suite(path: &Path) -> Result<Suite, String> {
    let text = std::fs::read_to_string(path).map_err(|e| format!("{}: {e}", path.display()))?;
    let parsed = if is_toml(path) {
        Suite::from_toml(&text)
    } else {
        Suite::from_json(&text)
    };
    parsed.map_err(|e| format!("{}: {e}", path.display()))
}

fn is_toml(path: &Path) -> bool {
    path.extension()
        .is_some_and(|e| e.eq_ignore_ascii_case("toml"))
}

fn run_case_cmd(case_path: &Path, data_root: &Path, format: Format) -> Result<Output, String> {
    let case = load_case(case_path)?;
    let candles = load_candles(&data_root.join(&case.dataset_ref)).map_err(|e| e.to_string())?;
    let result = run_case(&case, &candles).map_err(|e| e.to_string())?;
    let passed = result.passed && result.hash_match;
    let text = match format {
        Format::Json => serde_json::to_string(&result).map_err(|e| e.to_string())?,
        Format::Text => {
            let mut out = render_table(std::slice::from_ref(&result));
            let _ = writeln!(out, "{}/1 passed", usize::from(passed));
            out
        }
    };
    Ok(Output {
        text,
        code: u8::from(!passed),
    })
}

fn run_suite_cmd(suite_path: &Path, data_root: &Path, format: Format) -> Result<Output, String> {
    let suite = load_suite(suite_path)?;
    let report = run_suite(&suite, data_root).map_err(|e| e.to_string())?;
    let text = match format {
        Format::Json => serde_json::to_string(&report).map_err(|e| e.to_string())?,
        Format::Text => {
            let mut out = render_table(&report.results);
            let _ = writeln!(out, "{}/{} passed", report.passed, report.results.len());
            out
        }
    };
    Ok(Output {
        text,
        code: u8::from(report.failed > 0),
    })
}

fn list_cases_cmd(suite_path: &Path) -> Result<Output, String> {
    let suite = load_suite(suite_path)?;
    let mut text = String::new();
    for id in suite.case_ids() {
        let _ = writeln!(text, "{id}");
    }
    Ok(Output { text, code: 0 })
}

/// A left-aligned `id | passed | hash_match | hash` table with a header. The
/// hash is shown as a short prefix for readability; the JSON output carries the
/// full digest.
fn render_table(results: &[CaseResult]) -> String {
    let id_width = results
        .iter()
        .map(|r| r.id.len())
        .chain(std::iter::once("id".len()))
        .max()
        .unwrap_or(2);
    let mut out = String::new();
    let _ = writeln!(
        out,
        "{:<id_width$}  {:<6}  {:<10}  hash",
        "id", "passed", "hash_match"
    );
    for r in results {
        let _ = writeln!(
            out,
            "{:<id_width$}  {:<6}  {:<10}  {}",
            r.id,
            r.passed,
            r.hash_match,
            &r.hash[..r.hash.len().min(12)]
        );
    }
    out
}