ere/crates/dockerized/build.rs

use ere_build_utils::{detect_sdk_version, detect_self_crate_version};
use std::{env, fs, path::Path};

fn main() {
    generate_crate_version();
    generate_zkvm_sdk_version_impl();
    println!("cargo:rerun-if-changed=Cargo.lock");
}

fn generate_crate_version() {
    let crate_version = format!(
        "const CRATE_VERSION: &str = \"{}\";",
        detect_self_crate_version()
    );

    let out_dir = env::var("OUT_DIR").unwrap();
    let dst = Path::new(&out_dir).join("crate_version.rs");
    fs::write(dst, crate_version).unwrap();
}

fn generate_zkvm_sdk_version_impl() {
    let [
        airbender_version,
        jolt_version,
        miden_version,
        nexus_version,
        openvm_version,
        pico_version,
        risc0_version,
        sp1_version,
        ziren_version,
        zisk_version,
    ] = [
        "execution_utils",
        "jolt-sdk",
        "miden-core",
        "nexus-sdk",
        "openvm-sdk",
        "pico-vm",
        "risc0-zkvm",
        "sp1-sdk",
        "zkm-sdk",
        "ziskos",
    ]
    .map(detect_sdk_version);

    let zkvm_sdk_version_impl = format!(
        r#"impl crate::zkVMKind {{
    pub fn sdk_version(&self) -> &'static str {{
        match self {{
            Self::Airbender => "{airbender_version}",
            Self::Jolt => "{jolt_version}",
            Self::Miden => "{miden_version}",
            Self::Nexus => "{nexus_version}",
            Self::OpenVM => "{openvm_version}",
            Self::Pico => "{pico_version}",
            Self::Risc0 => "{risc0_version}",
            Self::SP1 => "{sp1_version}",
            Self::Ziren => "{ziren_version}",
            Self::Zisk => "{zisk_version}",
        }}
    }}
}}"#,
    );

    let out_dir = env::var("OUT_DIR").unwrap();
    let dst = Path::new(&out_dir).join("zkvm_sdk_version_impl.rs");
    fs::write(dst, zkvm_sdk_version_impl).unwrap();
}