EnergonRelics/src/main/java/com/thebrokenrail/energonrelics/mixin/MixinDefaultBiomeFeatures.java

package com.thebrokenrail.energonrelics.mixin;

import com.thebrokenrail.energonrelics.EnergonRelics;
import com.thebrokenrail.energonrelics.block.structure.StructureGeneratorBlock;
import net.minecraft.world.biome.Biome;
import net.minecraft.world.gen.GenerationStep;
import net.minecraft.world.gen.feature.DefaultBiomeFeatures;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

@Mixin(DefaultBiomeFeatures.class)
public class MixinDefaultBiomeFeatures {
    @Inject(at = @At("RETURN"), method = "addDefaultOres")
    private static void addDefaultOres(Biome biome, CallbackInfo info) {
        biome.addFeature(GenerationStep.Feature.UNDERGROUND_ORES, EnergonRelics.VERIDIUM_ORE_FEATURE);
    }

    @Inject(at = @At("RETURN"), method = "addDefaultUndergroundStructures")
    private static void addDefaultUndergroundStructures(Biome biome, CallbackInfo info) {
        StructureGeneratorBlock.addToBiome(biome);
    }
}