use crate::commands::registry_quota::{
classify_pull_against_quota, render_quota_error_json, QuotaOutcome,
};
use serde_json::Value;
use std::fs;
use std::path::Path;
#[derive(Debug, Clone)]
pub struct RegistryQuotaLintArgs {
pub observation_file: String,
pub json: bool,
}
#[derive(Debug, Clone, serde::Serialize)]
struct GateReport {
gate: &'static str,
falsify_id: &'static str,
outcome: String,
passed: bool,
}
pub fn run(args: RegistryQuotaLintArgs) -> Result<(), String> {
let path = Path::new(&args.observation_file);
if !path.exists() {
return Err(format!(
"FALSIFY-CRUX-A-22: observation file not found: {}",
args.observation_file
));
}
let raw = fs::read_to_string(path)
.map_err(|e| format!("FALSIFY-CRUX-A-22: failed to read observation: {e}"))?;
if raw.trim().is_empty() {
return Err("FALSIFY-CRUX-A-22: observation file is empty".to_string());
}
let obs: Value = serde_json::from_str(&raw)
.map_err(|e| format!("FALSIFY-CRUX-A-22: observation is not valid JSON: {e}"))?;
let mut reports: Vec<GateReport> = Vec::new();
let mut failures: Vec<String> = Vec::new();
if let Some(v) = obs.get("quota") {
let (r, err) = run_quota_gate(v);
reports.push(r);
if let Some(e) = err {
failures.push(e);
}
}
if let Some(v) = obs.get("atomic") {
let (r, err) = run_atomic_gate(v);
reports.push(r);
if let Some(e) = err {
failures.push(e);
}
}
if let Some(v) = obs.get("ceiling") {
let (r, err) = run_ceiling_gate(v);
reports.push(r);
if let Some(e) = err {
failures.push(e);
}
}
if reports.is_empty() {
return Err("FALSIFY-CRUX-A-22: observation has none of quota/atomic/ceiling".into());
}
if args.json {
let payload = serde_json::json!({
"contract": "CRUX-A-22",
"gates": reports,
});
println!("{}", serde_json::to_string_pretty(&payload).unwrap());
} else {
for r in &reports {
let tag = if r.passed { "PASS" } else { "FAIL" };
println!("[{tag}] {} ({}): {}", r.gate, r.falsify_id, r.outcome);
}
}
if !failures.is_empty() {
return Err(failures.join("\n"));
}
Ok(())
}
fn parse_u64(v: Option<&Value>, field: &str) -> Result<u64, String> {
v.and_then(|x| x.as_u64())
.ok_or_else(|| format!("{field} must be a non-negative integer"))
}
fn parse_outcome_tag(v: Option<&Value>) -> Option<String> {
v.and_then(|x| x.as_str()).map(|s| s.to_ascii_lowercase())
}
fn outcome_tag(o: &QuotaOutcome) -> &'static str {
match o {
QuotaOutcome::Allow { .. } => "allow",
QuotaOutcome::Reject { .. } => "reject",
}
}
fn run_quota_gate(v: &Value) -> (GateReport, Option<String>) {
let parse_inputs = || -> Result<(u64, u64, u64, Option<String>), String> {
Ok((
parse_u64(v.get("quota"), "quota")?,
parse_u64(v.get("used"), "used")?,
parse_u64(v.get("incoming"), "incoming")?,
parse_outcome_tag(v.get("expected_outcome")),
))
};
let (quota, used, incoming, expected) = match parse_inputs() {
Ok(x) => x,
Err(e) => {
let desc = format!("parse error: {e}");
return (
GateReport {
gate: "quota",
falsify_id: "FALSIFY-CRUX-A-22-001",
outcome: desc.clone(),
passed: false,
},
Some(format!("FALSIFY-CRUX-A-22-001 quota gate failed: {desc}")),
);
}
};
let result = classify_pull_against_quota(quota, used, incoming);
let got = outcome_tag(&result);
let body_invariant_ok = match &result {
QuotaOutcome::Reject { used, free, needed } => {
let body = render_quota_error_json(*used, *free, *needed);
serde_json::from_str::<Value>(&body)
.ok()
.and_then(|v| Some((v["needed"].as_u64()?, v["free"].as_u64()?)))
.map(|(n, f)| n > f)
.unwrap_or(false)
}
QuotaOutcome::Allow { .. } => true,
};
let passed = expected.as_deref() == Some(got) && body_invariant_ok;
let desc = format!(
"classified={got} expected={} body_invariant_ok={body_invariant_ok}",
expected.as_deref().unwrap_or("(unspecified)")
);
let err = if passed {
None
} else {
Some(format!("FALSIFY-CRUX-A-22-001 quota gate failed: {desc}"))
};
(
GateReport {
gate: "quota",
falsify_id: "FALSIFY-CRUX-A-22-001",
outcome: desc,
passed,
},
err,
)
}
fn run_atomic_gate(v: &Value) -> (GateReport, Option<String>) {
let parse_inputs = || -> Result<(u64, u64, u64, Option<String>), String> {
Ok((
parse_u64(v.get("quota"), "quota")?,
parse_u64(v.get("used"), "used")?,
parse_u64(v.get("incoming"), "incoming")?,
parse_outcome_tag(v.get("expected_outcome")),
))
};
let (quota, used, incoming, expected) = match parse_inputs() {
Ok(x) => x,
Err(e) => {
let desc = format!("parse error: {e}");
return (
GateReport {
gate: "atomic",
falsify_id: "FALSIFY-CRUX-A-22-002",
outcome: desc.clone(),
passed: false,
},
Some(format!("FALSIFY-CRUX-A-22-002 atomic gate failed: {desc}")),
);
}
};
let r1 = classify_pull_against_quota(quota, used, incoming);
let r2 = classify_pull_against_quota(quota, used, incoming);
let deterministic = r1 == r2;
let got = outcome_tag(&r1);
let outcome_matches = expected.as_deref() == Some(got);
let passed = deterministic && outcome_matches;
let desc = format!(
"deterministic={deterministic} classified={got} expected={}",
expected.as_deref().unwrap_or("(unspecified)")
);
let err = if passed {
None
} else {
Some(format!("FALSIFY-CRUX-A-22-002 atomic gate failed: {desc}"))
};
(
GateReport {
gate: "atomic",
falsify_id: "FALSIFY-CRUX-A-22-002",
outcome: desc,
passed,
},
err,
)
}
fn run_ceiling_gate(v: &Value) -> (GateReport, Option<String>) {
let parse_inputs = || -> Result<(u64, u64, u64, Option<String>, Option<bool>), String> {
Ok((
parse_u64(v.get("quota"), "quota")?,
parse_u64(v.get("used"), "used")?,
parse_u64(v.get("incoming"), "incoming")?,
parse_outcome_tag(v.get("expected_outcome")),
v.get("expected_post_used_le_quota")
.and_then(|x| x.as_bool()),
))
};
let (quota, used, incoming, expected_outcome, expected_invariant) = match parse_inputs() {
Ok(x) => x,
Err(e) => {
let desc = format!("parse error: {e}");
return (
GateReport {
gate: "ceiling",
falsify_id: "FALSIFY-CRUX-A-22-003",
outcome: desc.clone(),
passed: false,
},
Some(format!("FALSIFY-CRUX-A-22-003 ceiling gate failed: {desc}")),
);
}
};
let result = classify_pull_against_quota(quota, used, incoming);
let got = outcome_tag(&result);
let post_used = used.saturating_add(incoming);
let post_used_le_quota = matches!(result, QuotaOutcome::Allow { .. }) && post_used <= quota;
let outcome_ok = expected_outcome.as_deref() == Some(got);
let invariant_ok = expected_invariant
.map(|want| want == post_used_le_quota)
.unwrap_or(true);
let passed = outcome_ok && invariant_ok;
let desc = format!(
"classified={got} expected={} post_used={post_used} post_used_le_quota={post_used_le_quota}",
expected_outcome.as_deref().unwrap_or("(unspecified)"),
);
let err = if passed {
None
} else {
Some(format!("FALSIFY-CRUX-A-22-003 ceiling gate failed: {desc}"))
};
(
GateReport {
gate: "ceiling",
falsify_id: "FALSIFY-CRUX-A-22-003",
outcome: desc,
passed,
},
err,
)
}