Change Villager Type generator to a generic registry generator (#33)

Minestom · Nov 12, 2024 · 726130e · 726130e
1 parent c63a8fb
commit 726130e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 21 deletions.
diff --git a/DataGenerator/src/main/java/net/minestom/datagen/DataGenType.java b/DataGenerator/src/main/java/net/minestom/datagen/DataGenType.java
@@ -31,7 +31,7 @@ public enum DataGenType {
     SOUNDS("sounds", new SoundGenerator()),
     SOUND_SOURCES("sound_sources", new SoundSourceGenerator()),
     VILLAGER_PROFESSIONS("villager_professions", new VillagerProfessionGenerator()),
-    VILLAGER_TYPES("villager_types", new VillagerTypeGenerator()),
+    VILLAGER_TYPES("villager_types", new GenericRegistryGenerator<>(BuiltInRegistries.VILLAGER_TYPE)),
     RECIPE_TYPE("recipe_types", new GenericRegistryGenerator<>(BuiltInRegistries.RECIPE_TYPE)),
     RECIPE_DISPLAY_TYPE("recipe_display_types", new GenericRegistryGenerator<>(BuiltInRegistries.RECIPE_DISPLAY)),
     SLOT_DISPLAY_TYPE("slot_display_types", new GenericRegistryGenerator<>(BuiltInRegistries.SLOT_DISPLAY)),

diff --git a/DataGenerator/src/main/java/net/minestom/generators/VillagerTypeGenerator.java b/DataGenerator/src/main/java/net/minestom/generators/VillagerTypeGenerator.java