jotdown/bench/input/build.rs

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 input = std::fs::read_to_string(
                            std::path::Path::new(".").join(entry.file_name()),
                        )
                        .ok()?;
                        return Some((name.to_string(), 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");
}