From 375241b4da0102b7b57d282fc64266b2eb04a7cf Mon Sep 17 00:00:00 2001
From: Vos <isedenwildfire@proton.com>
Date: Wed, 12 Nov 2025 21:09:17 -0600
Subject: [PATCH] More World Gen Prep

---
 .../java/com/acethewildfire/acesbs/AcesBSDataGenerator.java   | 4 ++++
 .../acesbs/datagen/ModRegistryDataGenerator.java              | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/src/main/java/com/acethewildfire/acesbs/AcesBSDataGenerator.java b/src/main/java/com/acethewildfire/acesbs/AcesBSDataGenerator.java
index 2773441..a080281 100644
--- a/src/main/java/com/acethewildfire/acesbs/AcesBSDataGenerator.java
+++ b/src/main/java/com/acethewildfire/acesbs/AcesBSDataGenerator.java
@@ -4,6 +4,8 @@ import com.acethewildfire.acesbs.datagen.*;
 import com.acethewildfire.acesbs.enchantment.ModEnchantments;
 import com.acethewildfire.acesbs.trim.ModTrimMaterials;
 import com.acethewildfire.acesbs.trim.ModTrimPatterns;
+import com.acethewildfire.acesbs.world.ModConfiguredFeatures;
+import com.acethewildfire.acesbs.world.ModPlacedFeatures;
 import net.fabricmc.fabric.api.datagen.v1.DataGeneratorEntrypoint;
 import net.fabricmc.fabric.api.datagen.v1.FabricDataGenerator;
 import net.minecraft.registry.RegistryBuilder;
@@ -27,5 +29,7 @@ public class AcesBSDataGenerator implements DataGeneratorEntrypoint {
 		registryBuilder.addRegistry(RegistryKeys.TRIM_MATERIAL, ModTrimMaterials::bootstrap);
 		registryBuilder.addRegistry(RegistryKeys.TRIM_PATTERN, ModTrimPatterns::bootstrap);
 		registryBuilder.addRegistry(RegistryKeys.ENCHANTMENT, ModEnchantments::bootstrap);
+		registryBuilder.addRegistry(RegistryKeys.CONFIGURED_FEATURE, ModConfiguredFeatures::bootstrap);
+		registryBuilder.addRegistry(RegistryKeys.PLACED_FEATURE, ModPlacedFeatures::bootstrap);
 	}
 }
diff --git a/src/main/java/com/acethewildfire/acesbs/datagen/ModRegistryDataGenerator.java b/src/main/java/com/acethewildfire/acesbs/datagen/ModRegistryDataGenerator.java
index f483227..274b527 100644
--- a/src/main/java/com/acethewildfire/acesbs/datagen/ModRegistryDataGenerator.java
+++ b/src/main/java/com/acethewildfire/acesbs/datagen/ModRegistryDataGenerator.java
@@ -17,6 +17,8 @@ public class ModRegistryDataGenerator extends FabricDynamicRegistryProvider {
         entries.addAll(registries.getWrapperOrThrow(RegistryKeys.TRIM_MATERIAL));
         entries.addAll(registries.getWrapperOrThrow(RegistryKeys.TRIM_PATTERN));
         entries.addAll(registries.getWrapperOrThrow(RegistryKeys.ENCHANTMENT));
+        entries.addAll(registries.getWrapperOrThrow(RegistryKeys.CONFIGURED_FEATURE));
+        entries.addAll(registries.getWrapperOrThrow(RegistryKeys.PLACED_FEATURE));
     }
 
     @Override