basic code generator

1 files changed, 100 insertions, 0 deletions

diff --git a/data-code-generator/packetcodegen.py b/data-code-generator/packetcodegen.py
new file mode 100644
index 00000000..674f97c2
--- /dev/null
+++ b/data-code-generator/packetcodegen.py
@@ -0,0 +1,100 @@
+from utils import to_snake_case, to_camel_case
+from mappings import Mappings
+
+
+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'
+    elif burger_type == 'identifier':
+        field_type_rs = 'ResourceLocation'
+    elif burger_type == 'uuid':
+        field_type_rs = 'Uuid'
+    elif burger_type == 'position':
+        field_type_rs = 'BlockPos'
+    elif burger_type == 'nbtcompound':
+        field_type_rs = 'azalea_nbt::Tag'
+    elif burger_type == 'itemstack':
+        field_type_rs = 'Slot'
+    elif burger_type == 'metadata':
+        field_type_rs = '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 generate(burger_packets, mappings: Mappings):
+    for packet in burger_packets.values():
+        direction = packet['direction'].lower()  # serverbound or clientbound
+        state = packet['state'].lower()
+
+        generated_packet_code = []
+        generated_packet_code.append(
+            f'#[derive(Clone, Debug, {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, 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},')
+            else:
+                generated_packet_code.append(f'// TODO: {instruction}')
+                continue
+
+        generated_packet_code.append('}')
+        print(generated_packet_code)
+        print()