percli 1.0.0

Offline CLI simulator for the Percolator risk engine
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

pub mod json;
pub mod status;
pub mod table;

use anyhow::Result;
use clap::ValueEnum;
use owo_colors::OwoColorize;
use percli_core::EngineSnapshot;

use percli_core::scenario::runner::{RunResult, StepOutcome};

#[derive(Debug, Clone, Copy, ValueEnum)]
pub enum OutputFormat {
    Table,
    Json,
}

impl OutputFormat {
    pub fn from_env() -> Self {
        match std::env::var("PERC_FORMAT").ok().as_deref() {
            Some("json") => OutputFormat::Json,
            _ => OutputFormat::Table,
        }
    }
}

pub fn print_run_result(result: &RunResult, format: OutputFormat) -> Result<()> {
    match format {
        OutputFormat::Table => {
            let total = result.step_results.len();
            for sr in &result.step_results {
                let status = match &sr.outcome {
                    StepOutcome::Ok => format!(
                        "{} {}",
                        "\u{2713}".if_supports_color(owo_colors::Stream::Stdout, |t| t.green()),
                        "ok".if_supports_color(owo_colors::Stream::Stdout, |t| t.green())
                    ),
                    StepOutcome::Warning(w) => format!(
                        "{} {}: {w}",
                        "\u{25b2}".if_supports_color(owo_colors::Stream::Stdout, |t| t.yellow()),
                        "warn".if_supports_color(owo_colors::Stream::Stdout, |t| t.yellow())
                    ),
                    StepOutcome::QueryResult(_) => format!(
                        "{} {}",
                        "\u{25cf}".if_supports_color(owo_colors::Stream::Stdout, |t| t.cyan()),
                        "query".if_supports_color(owo_colors::Stream::Stdout, |t| t.cyan())
                    ),
                    StepOutcome::AssertPassed => format!(
                        "{} {}",
                        "\u{2713}".if_supports_color(owo_colors::Stream::Stdout, |t| t.green()),
                        "pass".if_supports_color(owo_colors::Stream::Stdout, |t| t.green())
                    ),
                    StepOutcome::AssertFailed(msg) => format!(
                        "{} {}: {msg}",
                        "\u{2717}".if_supports_color(owo_colors::Stream::Stdout, |t| t.red()),
                        "FAIL".if_supports_color(owo_colors::Stream::Stdout, |t| t.red())
                    ),
                };

                let prefix = format!(
                    "{:>8}",
                    format!("{}/{total}", sr.step_num)
                        .if_supports_color(owo_colors::Stream::Stdout, |t| t.dimmed())
                );
                println!("{prefix}  {:<52} {status}", sr.description);

                if let Some(delta) = &sr.delta {
                    table::print_delta(delta);
                }

                if let StepOutcome::QueryResult(snap) = &sr.outcome {
                    println!();
                    table::print_snapshot(snap);
                    println!();
                } else if let Some(snap) = &sr.snapshot {
                    println!();
                    table::print_snapshot(snap);
                    println!();
                }
            }

            println!();
            table::print_snapshot(&result.final_snapshot);
            Ok(())
        }
        OutputFormat::Json => json::print_run_result(result),
    }
}

pub fn print_snapshot(snap: &EngineSnapshot, format: OutputFormat) -> Result<()> {
    match format {
        OutputFormat::Table => {
            table::print_snapshot(snap);
            Ok(())
        }
        OutputFormat::Json => json::print_snapshot(snap),
    }
}