package ru.bclib.mixin.common;

import com.google.common.collect.Lists;
import net.fabricmc.fabric.impl.biome.InternalBiomeData;
import net.fabricmc.fabric.impl.biome.WeightedBiomePicker;
import net.minecraft.resources.ResourceKey;
import net.minecraft.world.level.biome.Biome;
import net.minecraft.world.level.biome.Biomes;
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;
import ru.bclib.interfaces.BiomeListProvider;

import java.util.List;

@Mixin(value = WeightedBiomePicker.class, remap = false)
public class WeightedBiomePickerMixin implements BiomeListProvider {
	private final List<ResourceKey<Biome>> biomes = Lists.newArrayList();
	
	@Inject(method = "addBiome", at = @At("TAIL"))
	private void bclib_addBiome(final ResourceKey<Biome> biome, final double weight, CallbackInfo info) {
		if (be_isCorrectPicker(WeightedBiomePicker.class.cast(this))) {
			biomes.add(biome);
		}
	}
	
	private boolean be_isCorrectPicker(WeightedBiomePicker picker) {
		return picker == InternalBiomeData.getEndBiomesMap().get(Biomes.SMALL_END_ISLANDS) ||
			picker == InternalBiomeData.getEndBarrensMap().get(Biomes.END_BARRENS);
	}
	
	@Override
	public List<ResourceKey<Biome>> getBiomes() {
		return biomes;
	}
}