From f27c87b291a920376e006a470efb353dee46ed17 Mon Sep 17 00:00:00 2001
From: mat <git@matdoes.dev>
Date: Sat, 31 May 2025 08:56:17 +0500
Subject: more formatting fixes

---
 codegen/lib/code/registry.py | 58 ++++++++++++++++++++++----------------------
 1 file changed, 29 insertions(+), 29 deletions(-)

(limited to 'codegen/lib/code/registry.py')

diff --git a/codegen/lib/code/registry.py b/codegen/lib/code/registry.py
index d1399da5..84d613ba 100644
--- a/codegen/lib/code/registry.py
+++ b/codegen/lib/code/registry.py
@@ -1,13 +1,11 @@
-from lib.utils import to_snake_case, upper_first_letter, get_dir_location, to_camel_case
-from ..mappings import Mappings
-from typing import Optional
-import re
+from lib.utils import get_dir_location, to_camel_case
+
+REGISTRIES_DIR = get_dir_location("../azalea-registry/src/lib.rs")
 
-REGISTRIES_DIR = get_dir_location('../azalea-registry/src/lib.rs')
 
 def generate_registries(registries: dict):
-    with open(REGISTRIES_DIR, 'r') as f:
-        code = f.read().split('\n')
+    with open(REGISTRIES_DIR, "r") as f:
+        code = f.read().split("\n")
 
     existing_registry_enum_names = set()
 
@@ -17,19 +15,20 @@ def generate_registries(registries: dict):
         #     Stone => "minecraft:stone"
         # });
 
-        registry_name = registry_name.split(':')[1]
+        registry_name = registry_name.split(":")[1]
         registry_enum_name = registry_name_to_enum_name(registry_name)
 
         existing_registry_enum_names.add(registry_enum_name)
 
         registry_code = []
-        registry_code.append(f'enum {registry_enum_name} {{')
+        registry_code.append(f"enum {registry_enum_name} {{")
         registry_entries = sorted(
-            registry['entries'].items(), key=lambda x: x[1]['protocol_id'])
+            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])
+            variant_struct_name = to_camel_case(variant_name.split(":")[-1])
             registry_code.append(f'\t{variant_struct_name} => "{variant_name}",')
-        registry_code.append('}')
+        registry_code.append("}")
 
         # when we find a "registry! {" line, find the next line that starts
         # with "enum <name>" and replace that until we find a line that's "}"
@@ -40,28 +39,28 @@ def generate_registries(registries: dict):
                 in_registry_macro = True
             elif in_registry_macro and line == registry_code[0]:
                 # found it, now delete until we get to "}"
-                while code[i] != '}':
+                while code[i] != "}":
                     code.pop(i)
-                code[i] = '\n'.join(registry_code)
+                code[i] = "\n".join(registry_code)
                 found = True
                 break
         if not found:
-            code.append('registry! {')
-            code.append('\n'.join(registry_code))
-            code.append('}')
-            code.append('')
+            code.append("registry! {")
+            code.append("\n".join(registry_code))
+            code.append("}")
+            code.append("")
 
     # delete the unused registries
     i = 0
     while i < len(code):
-        if code[i] == 'registry! {':
+        if code[i] == "registry! {":
             # skip until we get to the enum line
-            while not code[i].startswith('enum '):
+            while not code[i].startswith("enum "):
                 i += 1
-            enum_name = code[i].split(' ')[1]
+            enum_name = code[i].split(" ")[1]
             if enum_name not in existing_registry_enum_names:
                 i -= 1
-                while code[i] != '}':
+                while code[i] != "}":
                     code.pop(i)
                 code.pop(i)
                 # close the registry! block
@@ -69,17 +68,18 @@ def generate_registries(registries: dict):
         else:
             i += 1
 
-    with open(REGISTRIES_DIR, 'w') as f:
-        f.write('\n'.join(code))
+    with open(REGISTRIES_DIR, "w") as f:
+        f.write("\n".join(code))
+
 
 def registry_name_to_enum_name(registry_name: str) -> str:
-    registry_name = registry_name.split(':')[-1]
+    registry_name = registry_name.split(":")[-1]
 
-    if registry_name.endswith('_type'):
+    if registry_name.endswith("_type"):
         # change _type to _kind because that's Rustier (and because _type
         # is a reserved keyword)
-        registry_name = registry_name[:-5] + '_kind'
-    elif registry_name in {'menu'}:
-        registry_name += '_kind'
+        registry_name = registry_name[:-5] + "_kind"
+    elif registry_name in {"menu"}:
+        registry_name += "_kind"
 
     return to_camel_case(registry_name)
-- 
cgit v1.2.3