valence/crates/valence_protocol/build/main.rs

use std::path::Path;
use std::process::Command;
use std::{env, fs};

use anyhow::Context;
use proc_macro2::{Ident, Span};

mod block;
mod enchant;
mod item;
mod translation_key;

pub fn main() -> anyhow::Result<()> {
    println!("cargo:rerun-if-changed=../../extracted/");

    let generators = [
        (block::build as fn() -> _, "block.rs"),
        (enchant::build, "enchant.rs"),
        (item::build, "item.rs"),
        (translation_key::build, "translation_key.rs"),
    ];

    let out_dir = env::var_os("OUT_DIR").context("failed to get OUT_DIR env var")?;

    for (gen, file_name) in generators {
        let path = Path::new(&out_dir).join(file_name);
        let code = gen()?.to_string();
        fs::write(&path, code)?;

        // Format the output for debugging purposes.
        // Doesn't matter if rustfmt is unavailable.
        let _ = Command::new("rustfmt").arg(path).output();
    }

    Ok(())
}

fn ident(s: impl AsRef<str>) -> Ident {
    let s = s.as_ref().trim();

    match s.as_bytes() {
        // TODO: check for the other rust keywords.
        [b'0'..=b'9', ..] | b"type" => Ident::new(&format!("_{s}"), Span::call_site()),
        _ => Ident::new(s, Span::call_site()),
    }
}