use std::io::Write; fn main() { let inputs = std::fs::read_dir("..") .unwrap() .filter_map(|entry| { let entry = entry.ok()?; if let Some(name) = entry.file_name().to_str() { if let Some(name) = name.strip_suffix(".dj") { if entry.file_type().map_or(false, |ty| !ty.is_dir()) { let name_snake = name.replace('-', "_"); let input = std::fs::read_to_string( std::path::Path::new("..").join(entry.file_name()), ) .ok()?; return Some((name_snake, input)); } } } None }) .collect::>(); let out_dir = std::env::var_os("OUT_DIR").unwrap(); let mut out = std::fs::File::create(std::path::Path::new(&out_dir).join("lib.rs")).unwrap(); inputs.iter().for_each(|(name, input)| { write!( out, "#[allow(dead_code)]\nconst {}: &str = r###\"{}\"###;", name.to_uppercase(), input, ) .unwrap() }); write!( out, "#[allow(dead_code)]\npub const ALL: &str = r###\"{}\"###;", inputs.iter().map(|(_, s)| s.as_str()).collect::(), ) .unwrap(); write!( out, "#[allow(dead_code)]\npub const INPUTS: &[(&str, &str)] = &[{}];", inputs .iter() .map(|(n, _)| n.as_ref()) .chain(std::iter::once("all")) .map(|n| format!("(\"{}\", {}),", n, n.to_uppercase())) .collect::(), ) .unwrap(); println!("cargo:rerun-if-change=always_rerun"); }