use std::collections::HashMap; use std::fmt::Write; use std::path::PathBuf; use std::{env, fs}; static CONFIGS: &[(&str, usize)] = &[ // BEGIN AUTOGENERATED CONFIG FEATURES // Generated by gen_config.py. DO NOT EDIT. ("TASK_ARENA_SIZE", 4096), // END AUTOGENERATED CONFIG FEATURES ]; struct ConfigState { value: usize, seen_feature: bool, seen_env: bool, } fn main() { let crate_name = env::var("CARGO_PKG_NAME") .unwrap() .to_ascii_uppercase() .replace('-', "_"); // only rebuild if build.rs changed. Otherwise Cargo will rebuild if any // other file changed. println!("cargo:rerun-if-changed=build.rs"); // Rebuild if config envvar changed. for (name, _) in CONFIGS { println!("cargo:rerun-if-env-changed={crate_name}_{name}"); } let mut configs = HashMap::new(); for (name, default) in CONFIGS { configs.insert( *name, ConfigState { value: *default, seen_env: false, seen_feature: false, }, ); } let prefix = format!("{crate_name}_"); for (var, value) in env::vars() { if let Some(name) = var.strip_prefix(&prefix) { let Some(cfg) = configs.get_mut(name) else { panic!("Unknown env var {name}") }; let Ok(value) = value.parse::() else { panic!("Invalid value for env var {name}: {value}") }; cfg.value = value; cfg.seen_env = true; } if let Some(feature) = var.strip_prefix("CARGO_FEATURE_") { if let Some(i) = feature.rfind('_') { let name = &feature[..i]; let value = &feature[i + 1..]; if let Some(cfg) = configs.get_mut(name) { let Ok(value) = value.parse::() else { panic!("Invalid value for feature {name}: {value}") }; // envvars take priority. if !cfg.seen_env { if cfg.seen_feature { panic!("multiple values set for feature {}: {} and {}", name, cfg.value, value); } cfg.value = value; cfg.seen_feature = true; } } } } } let mut data = String::new(); for (name, cfg) in &configs { writeln!(&mut data, "pub const {}: usize = {};", name, cfg.value).unwrap(); } let out_dir = PathBuf::from(env::var_os("OUT_DIR").unwrap()); let out_file = out_dir.join("config.rs").to_string_lossy().to_string(); fs::write(out_file, data).unwrap(); // cortex-m targets let target = env::var("TARGET").unwrap(); if target.starts_with("thumbv6m-") { println!("cargo:rustc-cfg=cortex_m"); println!("cargo:rustc-cfg=armv6m"); } else if target.starts_with("thumbv7m-") { println!("cargo:rustc-cfg=cortex_m"); println!("cargo:rustc-cfg=armv7m"); } else if target.starts_with("thumbv7em-") { println!("cargo:rustc-cfg=cortex_m"); println!("cargo:rustc-cfg=armv7m"); println!("cargo:rustc-cfg=armv7em"); // (not currently used) } else if target.starts_with("thumbv8m.base") { println!("cargo:rustc-cfg=cortex_m"); println!("cargo:rustc-cfg=armv8m"); println!("cargo:rustc-cfg=armv8m_base"); } else if target.starts_with("thumbv8m.main") { println!("cargo:rustc-cfg=cortex_m"); println!("cargo:rustc-cfg=armv8m"); println!("cargo:rustc-cfg=armv8m_main"); } if target.ends_with("-eabihf") { println!("cargo:rustc-cfg=has_fpu"); } }