2022-08-28 01:31:21 +00:00
|
|
|
from lib.utils import to_snake_case, upper_first_letter, get_dir_location, to_camel_case
|
|
|
|
from ..mappings import Mappings
|
2022-11-06 20:05:01 +00:00
|
|
|
from typing import Optional
|
2022-08-28 01:31:21 +00:00
|
|
|
import re
|
|
|
|
|
|
|
|
REGISTRIES_DIR = get_dir_location('../azalea-registry/src/lib.rs')
|
|
|
|
|
|
|
|
|
|
|
|
def generate_registries(registries: dict):
|
|
|
|
code = []
|
|
|
|
|
2023-01-30 22:05:18 +00:00
|
|
|
code.append('''#![doc = include_str!("../README.md")]
|
|
|
|
|
|
|
|
// This file is automatically generated in codegen/lib/code/registry.py
|
2022-12-08 03:09:58 +00:00
|
|
|
|
|
|
|
use azalea_buf::{BufReadError, McBufReadable, McBufVarReadable, McBufVarWritable, McBufWritable};
|
|
|
|
use azalea_registry_macros::registry;
|
|
|
|
use std::io::{Cursor, Write};
|
|
|
|
|
|
|
|
pub trait Registry
|
|
|
|
where
|
|
|
|
Self: Sized,
|
|
|
|
{
|
|
|
|
fn from_u32(value: u32) -> Option<Self>;
|
|
|
|
fn to_u32(&self) -> u32;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// A registry that might not be present. This is transmitted as a single
|
|
|
|
/// varint in the protocol.
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
|
|
|
pub struct OptionalRegistry<T: Registry>(Option<T>);
|
|
|
|
|
|
|
|
impl<T: Registry> McBufReadable for OptionalRegistry<T> {
|
|
|
|
fn read_from(buf: &mut Cursor<&[u8]>) -> Result<Self, BufReadError> {
|
|
|
|
Ok(OptionalRegistry(match u32::var_read_from(buf)? {
|
|
|
|
0 => None,
|
|
|
|
value => Some(
|
|
|
|
T::from_u32(value - 1)
|
|
|
|
.ok_or(BufReadError::UnexpectedEnumVariant { id: value as i32 })?,
|
|
|
|
),
|
|
|
|
}))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
impl<T: Registry> McBufWritable for OptionalRegistry<T> {
|
|
|
|
fn write_into(&self, buf: &mut impl Write) -> Result<(), std::io::Error> {
|
|
|
|
match &self.0 {
|
|
|
|
None => 0u32.var_write_into(buf),
|
|
|
|
Some(value) => (value.to_u32() + 1).var_write_into(buf),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
''')
|
2022-08-28 01:31:21 +00:00
|
|
|
|
|
|
|
for registry_name, registry in registries.items():
|
|
|
|
# registry!(Block, {
|
|
|
|
# Air => "minecraft:air",
|
|
|
|
# Stone => "minecraft:stone"
|
|
|
|
# });
|
|
|
|
registry_struct_name = to_camel_case(registry_name.split(':')[1])
|
|
|
|
code.append(f'registry!({registry_struct_name}, {{')
|
|
|
|
registry_entries = sorted(
|
|
|
|
registry['entries'].items(), key=lambda x: x[1]['protocol_id'])
|
|
|
|
for variant_name, _variant in registry_entries:
|
|
|
|
variant_struct_name = to_camel_case(
|
|
|
|
variant_name.split(':')[1])
|
|
|
|
code.append(f'\t{variant_struct_name} => "{variant_name}",')
|
|
|
|
code.append('});')
|
|
|
|
code.append('')
|
|
|
|
|
|
|
|
with open(REGISTRIES_DIR, 'w') as f:
|
|
|
|
f.write('\n'.join(code))
|