mirror of
https://github.com/mat-1/azalea.git
synced 2025-08-02 14:26:04 +00:00
130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
from utils import to_snake_case, to_camel_case
|
|
from mappings import Mappings
|
|
import os
|
|
|
|
|
|
def burger_type_to_rust_type(burger_type):
|
|
is_var = False
|
|
uses = set()
|
|
|
|
if burger_type == 'byte':
|
|
field_type_rs = 'i8'
|
|
elif burger_type == 'short':
|
|
field_type_rs = 'i16'
|
|
elif burger_type == 'int':
|
|
field_type_rs = 'i32'
|
|
elif burger_type == 'long':
|
|
field_type_rs = 'i64'
|
|
elif burger_type == 'float':
|
|
field_type_rs = 'f32'
|
|
elif burger_type == 'double':
|
|
field_type_rs = 'f64'
|
|
|
|
elif burger_type == 'varint':
|
|
is_var = True
|
|
field_type_rs = 'i32'
|
|
elif burger_type == 'varlong':
|
|
is_var = True
|
|
field_type_rs = 'i64'
|
|
|
|
elif burger_type == 'boolean':
|
|
field_type_rs = 'bool'
|
|
elif burger_type == 'string':
|
|
field_type_rs = 'String'
|
|
|
|
elif burger_type == 'chatcomponent':
|
|
field_type_rs = 'Component'
|
|
uses.add('azalea_chat::component::Component')
|
|
elif burger_type == 'identifier':
|
|
field_type_rs = 'ResourceLocation'
|
|
uses.add('azalea_core::resource_location::ResourceLocation')
|
|
elif burger_type == 'uuid':
|
|
field_type_rs = 'Uuid'
|
|
uses.add('uuid::Uuid')
|
|
elif burger_type == 'position':
|
|
field_type_rs = 'BlockPos'
|
|
uses.add('azalea_core::BlockPos')
|
|
elif burger_type == 'nbtcompound':
|
|
field_type_rs = 'azalea_nbt::Tag'
|
|
elif burger_type == 'itemstack':
|
|
field_type_rs = 'Slot'
|
|
uses.add('azalea_core::Slot')
|
|
elif burger_type == 'metadata':
|
|
field_type_rs = 'EntityMetadata'
|
|
uses.add('crate::mc_buf::EntityMetadata')
|
|
elif burger_type == 'enum':
|
|
# enums are too complicated, leave those to the user
|
|
field_type_rs = 'todo!()'
|
|
elif burger_type.endswith('[]'):
|
|
field_type_rs, is_var, uses = burger_type_to_rust_type(
|
|
burger_type[:-2])
|
|
field_type_rs = f'Vec<{field_type_rs}>'
|
|
else:
|
|
print('Unknown field type:', burger_type)
|
|
exit()
|
|
return field_type_rs, is_var, uses
|
|
|
|
|
|
def write_packet_file(state, packet_name_snake_case, code):
|
|
path = os.path.join(
|
|
'..', f'azalea-protocol/src/packets/{state}/{packet_name_snake_case}.rs')
|
|
with open(path, 'w') as f:
|
|
f.write(code)
|
|
|
|
|
|
def generate(burger_packets, mappings: Mappings):
|
|
packet_ids = [75]
|
|
for packet in burger_packets.values():
|
|
if packet['id'] not in packet_ids:
|
|
continue
|
|
|
|
direction = packet['direction'].lower() # serverbound or clientbound
|
|
state = {'PLAY': 'game'}.get(packet['state'], packet['state'].lower())
|
|
|
|
generated_packet_code = []
|
|
uses = set()
|
|
generated_packet_code.append(
|
|
f'#[derive(Clone, Debug, {to_camel_case(state)}Packet)]')
|
|
uses.add(f'packet_macros::{to_camel_case(state)}Packet')
|
|
|
|
obfuscated_class_name = packet['class'].split('.')[0]
|
|
class_name = mappings.get_class(obfuscated_class_name).split('.')[-1]
|
|
|
|
generated_packet_code.append(
|
|
f'pub struct {to_camel_case(class_name)} {{')
|
|
|
|
for instruction in packet.get('instructions', []):
|
|
if instruction['operation'] == 'write':
|
|
obfuscated_field_name = instruction['field']
|
|
if '.' in obfuscated_field_name or ' ' in obfuscated_field_name or '(' in obfuscated_field_name:
|
|
continue
|
|
field_name = mappings.get_field(
|
|
obfuscated_class_name, obfuscated_field_name)
|
|
if not field_name:
|
|
generated_packet_code.append(f'// TODO: {instruction}')
|
|
continue
|
|
|
|
field_type = instruction['type']
|
|
field_type_rs, is_var, instruction_uses = burger_type_to_rust_type(
|
|
field_type)
|
|
if is_var:
|
|
generated_packet_code.append('#[var]')
|
|
generated_packet_code.append(
|
|
f'pub {to_snake_case(field_name)}: {field_type_rs},')
|
|
uses.update(instruction_uses)
|
|
else:
|
|
generated_packet_code.append(f'// TODO: {instruction}')
|
|
continue
|
|
|
|
generated_packet_code.append('}')
|
|
|
|
if uses:
|
|
# empty line before the `use` statements
|
|
generated_packet_code.insert(0, '')
|
|
for use in uses:
|
|
generated_packet_code.insert(0, f'use {use};')
|
|
|
|
print(generated_packet_code)
|
|
write_packet_file(state, to_snake_case(class_name),
|
|
'\n'.join(generated_packet_code))
|
|
print()
|