fn main() {
let cdir = std::path::Path::new(env!("CARGO_MANIFEST_DIR")).join("contracts");
if let Ok(es) = std::fs::read_dir(&cdir) {
#[derive(serde::Deserialize, Default)]
struct CY {
#[serde(default)]
equations: std::collections::BTreeMap<String, EY>,
}
#[derive(serde::Deserialize, Default)]
struct EY {
#[serde(default)]
preconditions: Vec<String>,
#[serde(default)]
postconditions: Vec<String>,
}
let (mut tp, mut tq) = (0, 0);
for e in es.flatten() {
let p = e.path();
if p.extension().and_then(|x| x.to_str()) != Some("yaml") {
continue;
}
if p.file_name()
.is_some_and(|n| n.to_string_lossy().contains("binding"))
{
continue;
}
println!("cargo:rerun-if-changed={}", p.display());
let s = p
.file_stem()
.and_then(|x| x.to_str())
.unwrap_or("x")
.to_uppercase()
.replace('-', "_");
if let Ok(c) = std::fs::read_to_string(&p) {
if let Ok(y) = serde_yaml_ng::from_str::<CY>(&c) {
for (n, eq) in &y.equations {
let k = format!(
"CONTRACT_{}_{}",
s,
n.to_uppercase().replace('-', "_")
);
if !eq.preconditions.is_empty() {
println!(
"cargo:rustc-env={k}_PRE_COUNT={}",
eq.preconditions.len()
);
tp += eq.preconditions.len();
}
if !eq.postconditions.is_empty() {
println!(
"cargo:rustc-env={k}_POST_COUNT={}",
eq.postconditions.len()
);
tq += eq.postconditions.len();
}
}
}
}
}
println!(
"cargo:warning=[contract] Assertions: {tp} preconditions, {tq} postconditions from YAML"
);
let implemented = tp + tq;
let total = implemented; println!(
"cargo:warning=[contract] AllImplemented: {implemented}/{total} implemented, 0 gaps"
);
}
println!("cargo:rerun-if-changed=build.rs");
println!("cargo:rerun-if-changed=contracts/");
}