feat: Condition datagen

Zepalesque · May 24, 2024 · 6186f21 · 6186f21
1 parent 4c55c7c
commit 6186f21
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/main/java/net/zepalesque/zenith/Zenith.java b/src/main/java/net/zepalesque/zenith/Zenith.java
@@ -17,6 +17,7 @@
 import net.neoforged.fml.event.lifecycle.FMLClientSetupEvent;
 import net.neoforged.fml.event.lifecycle.FMLCommonSetupEvent;
 import net.neoforged.fml.loading.FMLPaths;
+import net.neoforged.neoforge.common.data.DatapackBuiltinEntriesProvider;
 import net.neoforged.neoforge.common.data.ExistingFileHelper;
 import net.neoforged.neoforge.data.event.GatherDataEvent;
 import net.neoforged.neoforge.registries.DataPackRegistryEvent;
@@ -32,6 +33,7 @@
 import net.zepalesque.zenith.api.condition.config.ConfigSerializer;
 import net.zepalesque.zenith.config.ZConfig;
 import net.zepalesque.zenith.data.generator.ZenithDataMapGen;
+import net.zepalesque.zenith.data.generator.ZenithRegistrySets;
 import net.zepalesque.zenith.loot.condition.ZenithLootConditions;
 import net.zepalesque.zenith.recipe.condition.ZenithRecipeConditions;
 import net.zepalesque.zenith.world.placement.ZenithPlacementModifiers;
@@ -86,6 +88,10 @@ private void dataSetup(GatherDataEvent event) {
 
 
         generator.addProvider(event.includeServer(), new ZenithDataMapGen(packOutput, lookupProvider));
+
+        ZenithRegistrySets registrySets = new ZenithRegistrySets(packOutput, lookupProvider);
+//        CompletableFuture<HolderLookup.Provider> registryProvider = registrySets.getRegistryProvider();
+        generator.addProvider(event.includeServer(), registrySets);
     }
 
     private void registerDataMaps(RegisterDataMapTypesEvent event) {

diff --git a/src/main/java/net/zepalesque/zenith/data/generator/ZenithRegistrySets.java b/src/main/java/net/zepalesque/zenith/data/generator/ZenithRegistrySets.java
@@ -0,0 +1,21 @@
+package net.zepalesque.zenith.data.generator;
+
+import net.minecraft.core.HolderLookup;
+import net.minecraft.core.RegistrySetBuilder;
+import net.minecraft.data.PackOutput;
+import net.neoforged.neoforge.common.data.DatapackBuiltinEntriesProvider;
+import net.zepalesque.zenith.Zenith;
+import net.zepalesque.zenith.data.resource.Conditions;
+
+import java.util.Collections;
+import java.util.concurrent.CompletableFuture;
+
+public class ZenithRegistrySets extends DatapackBuiltinEntriesProvider {
+    public static final RegistrySetBuilder BUILDER = new RegistrySetBuilder()
+            .add(Zenith.Keys.CONDITION, Conditions::bootstrap);
+
+
+    public ZenithRegistrySets(PackOutput output, CompletableFuture<HolderLookup.Provider> registries) {
+        super(output, registries, BUILDER, Collections.singleton(Zenith.MODID));
+    }
+}