package net.devtech.orecentration.mixin;

import net.devtech.orecentration.OreCentration;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Redirect;

import net.minecraft.block.Block;
import net.minecraft.world.gen.carver.Carver;
import net.minecraft.world.gen.carver.NetherCaveCarver;
import net.minecraft.world.gen.carver.UnderwaterCaveCarver;
import net.minecraft.world.gen.carver.UnderwaterRavineCarver;
import net.minecraft.world.gen.chunk.ChunkGeneratorSettings;
import net.minecraft.world.gen.feature.ConfiguredFeatures;
import net.minecraft.world.gen.feature.Feature;
import net.minecraft.world.gen.feature.LakeFeature;
import net.minecraft.world.gen.feature.VoidStartPlatformFeature;
import net.minecraft.world.gen.surfacebuilder.SurfaceBuilder;

@Mixin({
        Feature.class,
        SurfaceBuilder.class,
        ChunkGeneratorSettings.class,
        UnderwaterRavineCarver.class,
        UnderwaterCaveCarver.class,
        NetherCaveCarver.class,
        Carver.class,
        ConfiguredFeatures.class,
        ConfiguredFeatures.Configs.class,
        ConfiguredFeatures.States.class,
        LakeFeature.class,
        VoidStartPlatformFeature.class
})
public class StoneReplaceMixin {
    @Redirect(method = {
            "<init>(Lcom/mojang/serialization/Codec;I)V",
            "<init>(Lcom/mojang/serialization/Codec;)V",
            "generate(Lnet/minecraft/world/StructureWorldAccess;Lnet/minecraft/world/gen/chunk/ChunkGenerator;Ljava/util/Random;Lnet/minecraft/util/math/BlockPos;Lnet/minecraft/world/gen/feature/DefaultFeatureConfig;)Z"
    }, at = @At(value = "FIELD", target = "Lnet/minecraft/block/Blocks;STONE:Lnet/minecraft/block/Block;"), require = 0)
    public Block getStone() {
        return OreCentration.NATURAL_STONE;
    }

    @Redirect(method = {
            "isStone(Lnet/minecraft/block/Block;)Z",
            "<clinit>()V",
            "createSurfaceSettings(Lnet/minecraft/world/gen/chunk/StructuresConfig;ZLnet/minecraft/util/Identifier;)Lnet/minecraft/world/gen/chunk/ChunkGeneratorSettings;",
    }, at = @At(value = "FIELD", target = "Lnet/minecraft/block/Blocks;STONE:Lnet/minecraft/block/Block;"), require = 0)
    private static Block getStoneStatic() {
        return OreCentration.NATURAL_STONE;
    }
}