58 lines
1.7 KiB
Rust
58 lines
1.7 KiB
Rust
|
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::<Vec<_>>();
|
||
|
|
|
||
|
|
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::<String>(),
|
||
|
|
)
|
||
|
|
.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::<String>(),
|
||
|
|
)
|
||
|
|
.unwrap();
|
||
|
|
|
||
|
|
println!("cargo:rerun-if-change=always_rerun");
|
||
|
|
}
|