Default components (#232)

* add default components

* remove debug prints

* clippy

* use default components

* fix tests

1 files changed, 13 insertions, 12 deletions

diff --git a/codegen/genregistries.py b/codegen/genregistries.py
index 175d1b20..d34cce97 100644
--- a/codegen/genregistries.py
+++ b/codegen/genregistries.py
@@ -1,31 +1,32 @@
 import lib.code.inventory
 import lib.code.registry
 import lib.code.version
-import lib.code.packet
 import lib.code.utils
 import lib.code.tags
-import lib.download
 import lib.extract
-import lib.utils
+
 
 def generate(version_id: str):
     registries = lib.extract.get_registries_report(version_id)
 
     lib.code.registry.generate_registries(registries)
-    lib.code.inventory.update_menus(registries['minecraft:menu']['entries'])
+    lib.code.inventory.update_menus(registries["minecraft:menu"]["entries"])
 
-    block_tags = lib.extract.get_registry_tags(version_id, 'block')
-    item_tags = lib.extract.get_registry_tags(version_id, 'item')
-    fluid_tags = lib.extract.get_registry_tags(version_id, 'fluid')
+    block_tags = lib.extract.get_registry_tags(version_id, "block")
+    item_tags = lib.extract.get_registry_tags(version_id, "item")
+    fluid_tags = lib.extract.get_registry_tags(version_id, "fluid")
+    entity_tags = lib.extract.get_registry_tags(version_id, "entity_type")
 
-    lib.code.tags.generate_tags(block_tags, 'blocks', 'Block')
-    lib.code.tags.generate_tags(item_tags, 'items', 'Item')
-    lib.code.tags.generate_tags(fluid_tags, 'fluids', 'Fluid')
+    lib.code.tags.generate_tags(block_tags, "blocks", "Block")
+    lib.code.tags.generate_tags(item_tags, "items", "Item")
+    lib.code.tags.generate_tags(fluid_tags, "fluids", "Fluid")
+    lib.code.tags.generate_tags(entity_tags, "entities", "EntityKind")
 
     lib.code.utils.fmt()
 
-    print('Done!')
+    print("Done!")
+
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     version_id = lib.code.version.get_version_id()
     generate(version_id)