mod common;
use json_eval_rs::{JSONEval, ParsedSchema};
use serde_json::{Map, Value};
use std::fs;
use std::path::Path;
use std::sync::Arc;
use std::time::Instant;
fn print_help(program_name: &str) {
println!("\n🚀 JSON Evaluation - Basic Example (Parsed / JSON & MsgPack)\n");
println!("USAGE:");
println!(" {} [OPTIONS] [FILTER]\n", program_name);
println!("OPTIONS:");
println!(" -h, --help Show this help message");
println!(" --compare Enable comparison with expected results");
println!(" --timing Show detailed internal timing breakdown\n");
println!("ARGUMENTS:");
println!(" [FILTER] Optional filter to match scenario names\n");
println!("DESCRIPTION:");
println!(" Evaluates schemas using ParsedSchema for efficient caching.");
println!(" Supports JSON (.json) and MessagePack (.bform) schemas.\n");
println!("EXAMPLES:");
println!(
" {} # Run all scenarios with ParsedSchema",
program_name
);
println!(
" {} zcc # Run scenarios matching 'zcc'",
program_name
);
println!(
" {} --compare # Run with comparison enabled",
program_name
);
println!(
" {} zcc --timing # Run with detailed timing breakdown",
program_name
);
}
fn main() {
let args: Vec<String> = std::env::args().collect();
let program_name = args.get(0).map(|s| s.as_str()).unwrap_or("basic_parsed");
let mut scenario_filter: Option<String> = None;
let mut enable_comparison = false;
let mut show_timing = false;
let mut i = 1;
while i < args.len() {
let arg = &args[i];
if arg == "-h" || arg == "--help" {
print_help(program_name);
return;
} else if arg == "--compare" {
enable_comparison = true;
} else if arg == "--timing" {
show_timing = true;
} else if !arg.starts_with('-') {
scenario_filter = Some(arg.clone());
} else {
eprintln!("Error: unknown option '{}'", arg);
print_help(program_name);
return;
}
i += 1;
}
println!("\n🚀 JSON Evaluation - Basic Example (Parsed / JSON & MsgPack)\n");
println!("📦 Using Arc<ParsedSchema> for efficient caching\n");
if enable_comparison {
println!("🔍 Comparison: enabled");
}
if show_timing {
println!("⏱️ Internal timing: enabled");
}
if enable_comparison || show_timing {
println!();
}
let samples_dir = Path::new("samples");
let mut scenarios = common::discover_scenarios(samples_dir);
if let Some(ref filter) = scenario_filter {
scenarios.retain(|s| s.name.contains(filter));
println!("📋 Filtering scenarios matching: '{}'\n", filter);
}
if scenarios.is_empty() {
if let Some(filter) = scenario_filter {
println!(
"ℹ️ No scenarios found matching '{}' in `{}`.",
filter,
samples_dir.display()
);
} else {
println!(
"ℹ️ No scenarios discovered in `{}`. Add files like `name.json` and `name-data.json`.",
samples_dir.display()
);
}
return;
}
println!("📊 Found {} scenario(s)\n", scenarios.len());
let mut total_parse_time = std::time::Duration::ZERO;
let mut total_eval_time = std::time::Duration::ZERO;
let mut successful_scenarios = 0;
let mut comparison_failures = 0;
for scenario in &scenarios {
println!("==============================");
println!("Scenario: {}", scenario.name);
println!(
"Schema: {} ({})",
scenario.schema_path.display(),
if scenario.is_msgpack {
"MessagePack"
} else {
"JSON"
}
);
println!("Data: {}\n", scenario.data_path.display());
if show_timing {
json_eval_rs::enable_timing();
json_eval_rs::clear_timing_data();
}
let data_str = fs::read_to_string(&scenario.data_path)
.unwrap_or_else(|e| panic!("failed to read {}: {}", scenario.data_path.display(), e));
let parse_start = Instant::now();
let parsed_schema = if scenario.is_msgpack {
let schema_msgpack = fs::read(&scenario.schema_path).unwrap_or_else(|e| {
panic!("failed to read {}: {}", scenario.schema_path.display(), e)
});
println!(
" 📦 MessagePack schema size: {} bytes",
schema_msgpack.len()
);
Arc::new(
ParsedSchema::parse_msgpack(&schema_msgpack)
.unwrap_or_else(|e| panic!("failed to parse MessagePack schema: {}", e)),
)
} else {
let schema_str = fs::read_to_string(&scenario.schema_path).unwrap_or_else(|e| {
panic!("failed to read {}: {}", scenario.schema_path.display(), e)
});
Arc::new(
ParsedSchema::parse(&schema_str)
.unwrap_or_else(|e| panic!("failed to parse schema: {}", e)),
)
};
let parse_time = parse_start.elapsed();
println!(" 📝 Schema parsing: {:?}", parse_time);
let eval_start = Instant::now();
let mut eval = JSONEval::with_parsed_schema(
parsed_schema.clone(), Some("{}"),
Some(&data_str),
)
.unwrap_or_else(|e| panic!("failed to create JSONEval: {}", e));
eval.evaluate(&data_str, Some("{}"), None, None)
.unwrap_or_else(|e| panic!("evaluation failed: {}", e));
let evaluated_schema = eval.get_evaluated_schema(false);
let eval_time = eval_start.elapsed();
println!(" ⚡ Eval: {:?}", eval_time);
println!(" ⏱️ Total: {:?}\n", parse_time + eval_time);
if show_timing {
json_eval_rs::print_timing_summary();
}
total_parse_time += parse_time;
total_eval_time += eval_time;
successful_scenarios += 1;
let evaluated_path = samples_dir.join(format!("{}-evaluated-schema.json", scenario.name));
let parsed_path = samples_dir.join(format!("{}-parsed-schema.json", scenario.name));
fs::write(&evaluated_path, common::pretty_json(&evaluated_schema))
.unwrap_or_else(|e| panic!("failed to write {}: {}", evaluated_path.display(), e));
let mut metadata_obj = Map::new();
metadata_obj.insert(
"dependencies".to_string(),
serde_json::to_value(&*eval.dependencies).unwrap(),
);
metadata_obj.insert(
"evaluations".to_string(),
serde_json::to_value(&*eval.evaluations).unwrap(),
);
metadata_obj.insert(
"sorted_evaluations".to_string(),
serde_json::to_value(&*eval.sorted_evaluations).unwrap(),
);
fs::write(
&parsed_path,
common::pretty_json(&Value::Object(metadata_obj)),
)
.unwrap_or_else(|e| panic!("failed to write {}: {}", parsed_path.display(), e));
println!("✅ Results saved:");
println!(" - {}", evaluated_path.display());
println!(" - {}\n", parsed_path.display());
if enable_comparison {
if let Some(comp_path) = &scenario.comparison_path {
if common::compare_with_expected(&evaluated_schema, comp_path).is_err() {
comparison_failures += 1;
}
println!();
}
}
}
println!("{}", "=".repeat(50));
println!("📊 Summary");
println!("{}", "=".repeat(50));
println!("Total scenarios run: {}", successful_scenarios);
println!("Total parsing time: {:?}", total_parse_time);
println!("Total evaluation time: {:?}", total_eval_time);
println!("Total time: {:?}", total_parse_time + total_eval_time);
if successful_scenarios > 1 {
println!("\nAverage per scenario:");
println!(
" Parsing: {:?}",
total_parse_time / successful_scenarios as u32
);
println!(
" Evaluation: {:?}",
total_eval_time / successful_scenarios as u32
);
}
if enable_comparison {
println!("\nComparison failures: {}", comparison_failures);
}
println!("\n✅ All scenarios completed!\n");
}