Fixed Skylands cavegen

Author: mineLdiver

station-flattening-v0/src/main/java/net/modificationstation/stationapi/mixin/flattening/SkylandsChunkGeneratorMixin.java

@@ -1,8 +1,12 @@
 package net.modificationstation.stationapi.mixin.flattening;
 
+import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
+import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
 import net.minecraft.class_208;
+import net.minecraft.class_415;
 import net.minecraft.world.World;
 import net.minecraft.world.chunk.Chunk;
+import net.modificationstation.stationapi.impl.world.CaveGenBaseImpl;
 import net.modificationstation.stationapi.impl.world.chunk.FlattenedChunk;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
@@ -13,6 +17,19 @@
 
 @Mixin(class_208.class)
 class SkylandsChunkGeneratorMixin {
+    @WrapOperation(
+            method = "<init>",
+            at = @At(
+                    value = "NEW",
+                    target = "()Lnet/minecraft/class_415;"
+            )
+    )
+    private class_415 stationapi_setWorldForCaveGen(Operation<class_415> original, World world, long l) {
+        final class_415 caveGen = original.call();
+        ((CaveGenBaseImpl) caveGen).stationapi_setWorld(world);
+        return caveGen;
+    }
+
     @Redirect(
             method = "method_1806",
             at = @At(