mirror of
https://github.com/Pumpkin-MC/Pumpkin.git
synced 2026-08-31 08:22:33 +00:00
110 lines
3.5 KiB
Rust
110 lines
3.5 KiB
Rust
use heck::ToPascalCase;
|
|
use proc_macro2::TokenStream;
|
|
use quote::{format_ident, quote};
|
|
use std::{fs, io::Write, path::Path, process::Command};
|
|
|
|
mod biome;
|
|
mod block;
|
|
mod chunk_status;
|
|
mod composter_increase_chance;
|
|
mod damage_type;
|
|
mod enchantments;
|
|
mod entity_pose;
|
|
mod entity_status;
|
|
mod entity_type;
|
|
mod flower_pot_transformations;
|
|
mod fluid;
|
|
mod fuels;
|
|
mod game_event;
|
|
mod game_rules;
|
|
mod item;
|
|
pub mod loot;
|
|
mod message_type;
|
|
mod noise_parameter;
|
|
mod noise_router;
|
|
mod packet;
|
|
mod particle;
|
|
mod recipes;
|
|
mod scoreboard_slot;
|
|
mod screen;
|
|
mod sound;
|
|
mod sound_category;
|
|
mod spawn_egg;
|
|
mod status_effect;
|
|
mod tag;
|
|
mod world_event;
|
|
|
|
pub const OUT_DIR: &str = "src/generated";
|
|
|
|
pub fn main() {
|
|
let path = Path::new(OUT_DIR);
|
|
if !path.exists() {
|
|
let _ = fs::create_dir(OUT_DIR);
|
|
}
|
|
write_generated_file(packet::build(), "packet.rs");
|
|
write_generated_file(screen::build(), "screen.rs");
|
|
write_generated_file(particle::build(), "particle.rs");
|
|
write_generated_file(sound::build(), "sound.rs");
|
|
write_generated_file(chunk_status::build(), "chunk_status.rs");
|
|
write_generated_file(game_event::build(), "game_event.rs");
|
|
write_generated_file(game_rules::build(), "game_rules.rs");
|
|
write_generated_file(sound_category::build(), "sound_category.rs");
|
|
write_generated_file(entity_pose::build(), "entity_pose.rs");
|
|
write_generated_file(scoreboard_slot::build(), "scoreboard_slot.rs");
|
|
write_generated_file(world_event::build(), "world_event.rs");
|
|
write_generated_file(entity_type::build(), "entity_type.rs");
|
|
write_generated_file(noise_parameter::build(), "noise_parameter.rs");
|
|
write_generated_file(biome::build(), "biome.rs");
|
|
write_generated_file(damage_type::build(), "damage_type.rs");
|
|
write_generated_file(message_type::build(), "message_type.rs");
|
|
write_generated_file(spawn_egg::build(), "spawn_egg.rs");
|
|
write_generated_file(item::build(), "item.rs");
|
|
write_generated_file(fluid::build(), "fluid.rs");
|
|
write_generated_file(status_effect::build(), "status_effect.rs");
|
|
write_generated_file(entity_status::build(), "entity_status.rs");
|
|
write_generated_file(block::build(), "block.rs");
|
|
write_generated_file(tag::build(), "tag.rs");
|
|
write_generated_file(noise_router::build(), "noise_router.rs");
|
|
write_generated_file(
|
|
flower_pot_transformations::build(),
|
|
"flower_pot_transformations.rs",
|
|
);
|
|
write_generated_file(
|
|
composter_increase_chance::build(),
|
|
"composter_increase_chance.rs",
|
|
);
|
|
write_generated_file(recipes::build(), "recipes.rs");
|
|
write_generated_file(enchantments::build(), "enchantment.rs");
|
|
write_generated_file(fuels::build(), "fuels.rs");
|
|
}
|
|
|
|
pub fn array_to_tokenstream(array: &[String]) -> TokenStream {
|
|
let mut variants = TokenStream::new();
|
|
|
|
for item in array.iter() {
|
|
let name = format_ident!("{}", item.to_pascal_case());
|
|
variants.extend([quote! {
|
|
#name,
|
|
}]);
|
|
}
|
|
variants
|
|
}
|
|
|
|
pub fn write_generated_file(content: TokenStream, out_file: &str) {
|
|
let path = Path::new(OUT_DIR).join(out_file);
|
|
let code = content.to_string();
|
|
|
|
let mut file = fs::File::create(&path).unwrap();
|
|
if let Err(e) = file.write_all(code.as_bytes()) {
|
|
println!("cargo::error={e}");
|
|
}
|
|
|
|
// Try to format the output for debugging purposes.
|
|
// Doesn't matter if rustfmt is unavailable.
|
|
let _ = Command::new("rustfmt").arg(&path).output();
|
|
// Try to auto optimize using clippy.
|
|
let _ = Command::new("cargo clippy --fix --allow-dirty")
|
|
.arg(&path)
|
|
.output();
|
|
}
|