embassy/embassy-stm32/build.rs

use std::env;
use std::path::PathBuf;
use std::process::Command;

fn main() {
    let chip_name = env::vars_os()
        .map(|(a, _)| a.to_string_lossy().to_string())
        .find(|x| x.starts_with("CARGO_FEATURE_STM32"))
        .expect("No stm32xx Cargo feature enabled")
        .strip_prefix("CARGO_FEATURE_")
        .unwrap()
        .to_ascii_uppercase();

    let out_dir = &PathBuf::from(env::var_os("OUT_DIR").unwrap());
    let out_file = out_dir.join("generated.rs").to_string_lossy().to_string();

    let exit_code = Command::new("python3")
        .args(&["gen.py", &chip_name, &out_file])
        .status()
        .expect("failed to execute gen.py");

    if !exit_code.success() {
        panic!("gen.py exited with {:?}", exit_code)
    }

    for s in env::var("DEP_STM32_METAPAC_V0.1_CFGS").unwrap().split(",") {
        println!("cargo:rustc-cfg={}", s);
    }
    println!("cargo:rerun-if-env-changed=DEP_STM32_METAPAC_V0.1_CFGS");
    println!("cargo:rerun-if-changed=build.rs");
    println!("cargo:rerun-if-changed=gen.py");
}