evil registry hacks

platforms/mixin-lifecycle/src/main/java/com/dfsek/terra/lifecycle/mixin/RegistryEntryReferenceInvoker.java

@@ -0,0 +1,13 @@
+package com.dfsek.terra.lifecycle.mixin;
+
+
+import net.minecraft.registry.entry.RegistryEntry.Reference;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.gen.Invoker;
+
+
+@Mixin(Reference.class)
+public interface RegistryEntryReferenceInvoker<T> {
+    @Invoker("setValue")
+    void invokeSetValue(T value);
+}

platforms/mixin-lifecycle/src/main/java/com/dfsek/terra/lifecycle/mixin/SimpleRegistryMixin.java

@@ -0,0 +1,28 @@
+package com.dfsek.terra.lifecycle.mixin;
+
+import com.dfsek.terra.lifecycle.util.RegistryHack;
+
+import net.minecraft.registry.SimpleRegistry;
+import net.minecraft.registry.entry.RegistryEntry;
+import net.minecraft.registry.entry.RegistryEntry.Reference;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+
+import java.util.Map;
+
+
+@Mixin(SimpleRegistry.class)
+public class SimpleRegistryMixin<T> implements RegistryHack {
+    @Shadow
+    @Final
+    private Map<T, Reference<T>> valueToEntry;
+    
+    @Override
+    public void terra_bind() {
+        valueToEntry.forEach((value, entry) -> {
+            //noinspection unchecked
+            ((RegistryEntryReferenceInvoker<T>) entry).invokeSetValue(value);
+        });
+    }
+}

platforms/mixin-lifecycle/src/main/java/com/dfsek/terra/lifecycle/mixin/lifecycle/RegistryLoaderMixin.java

@@ -2,6 +2,8 @@ package com.dfsek.terra.lifecycle.mixin.lifecycle;
 
 import com.dfsek.terra.lifecycle.LifecyclePlatform;
 
+import com.dfsek.terra.lifecycle.util.RegistryHack;
+
 import com.mojang.datafixers.util.Pair;
 import net.minecraft.registry.MutableRegistry;
 import net.minecraft.registry.RegistryKeys;
@@ -34,32 +36,18 @@ public class RegistryLoaderMixin {
                     )
     )
     private static void grabManager(List<Pair<MutableRegistry<?>, Object>> instance, Consumer<Pair<MutableRegistry<?>, Object>> consumer) {
-        MutableRegistry<Biome> biomeMutableRegistry = (MutableRegistry<Biome>)
+        
-                instance.stream()
+        //LifecyclePlatform.setRegistries(biomeMutableRegistry, dimensionTypeMutableRegistry, chunkGeneratorSettingsMutableRegistry);
-                        .map(Pair::getFirst)
+        //LifecycleUtil.initialize(biomeMutableRegistry, worldPresetMutableRegistry);
-                        .filter(r -> r.getKey().equals(RegistryKeys.BIOME))
+        instance.forEach(mutableRegistryObjectPair -> {
-                        .findFirst()
+            System.out.println(mutableRegistryObjectPair.getFirst());
-                        .orElseThrow();
+            System.out.println(mutableRegistryObjectPair.getFirst().size());
-        MutableRegistry<WorldPreset> worldPresetMutableRegistry = (MutableRegistry<WorldPreset>)
+            if(mutableRegistryObjectPair.getFirst().getKey().equals(RegistryKeys.BIOME)) {
-                instance.stream()
+                ((RegistryHack) mutableRegistryObjectPair.getFirst()).terra_bind();
-                        .map(Pair::getFirst)
+                System.out.println("BIOMES: " + mutableRegistryObjectPair.getFirst().stream().toList());
-                        .filter(r -> r.getKey().equals(RegistryKeys.WORLD_PRESET))
+            }
-                        .findFirst()
+            //System.out.println(mutableRegistryObjectPair.getFirst().stream().toList());
-                        .orElseThrow();
+        });
-        MutableRegistry<DimensionType> dimensionTypeMutableRegistry = (MutableRegistry<DimensionType>)
-                instance.stream()
-                        .map(Pair::getFirst)
-                        .filter(r -> r.getKey().equals(RegistryKeys.DIMENSION_TYPE))
-                        .findFirst()
-                        .orElseThrow();
-        MutableRegistry<ChunkGeneratorSettings> chunkGeneratorSettingsMutableRegistry = (MutableRegistry<ChunkGeneratorSettings>)
-                instance.stream()
-                        .map(Pair::getFirst)
-                        .filter(r -> r.getKey().equals(RegistryKeys.CHUNK_GENERATOR_SETTINGS))
-                        .findFirst()
-                        .orElseThrow();
-        LifecyclePlatform.setRegistries(biomeMutableRegistry, dimensionTypeMutableRegistry, chunkGeneratorSettingsMutableRegistry);
-        LifecycleUtil.initialize(biomeMutableRegistry, worldPresetMutableRegistry);
         instance.forEach(consumer);
     }
 }

platforms/mixin-lifecycle/src/main/java/com/dfsek/terra/lifecycle/util/RegistryHack.java

@@ -0,0 +1,5 @@
+package com.dfsek.terra.lifecycle.util;
+
+public interface RegistryHack {
+    void terra_bind();
+}

platforms/mixin-lifecycle/src/main/resources/terra.lifecycle.mixins.json

@@ -5,7 +5,9 @@
     "compatibilityLevel" : "JAVA_17",
     "mixins" : [
         "NoiseConfigMixin",
+        "RegistryEntryReferenceInvoker",
         "RegistryMixin",
+        "SimpleRegistryMixin",
         "lifecycle.MinecraftServerMixin",
         "lifecycle.RegistryLoaderMixin"
     ],