package net.minecraft.world.level.levelgen.structure.pieces;

import java.util.Optional;
import java.util.function.Predicate;
import net.minecraft.core.Holder;
import net.minecraft.core.QuartPos;
import net.minecraft.core.RegistryAccess;
import net.minecraft.world.level.ChunkPos;
import net.minecraft.world.level.LevelHeightAccessor;
import net.minecraft.world.level.biome.Biome;
import net.minecraft.world.level.biome.BiomeSource;
import net.minecraft.world.level.chunk.ChunkGenerator;
import net.minecraft.world.level.levelgen.Heightmap;
import net.minecraft.world.level.levelgen.RandomState;
import net.minecraft.world.level.levelgen.feature.configurations.FeatureConfiguration;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureTemplateManager;

@FunctionalInterface
public interface PieceGeneratorSupplier<C extends FeatureConfiguration> {
	Optional<PieceGenerator<C>> createGenerator(PieceGeneratorSupplier.Context<C> context);

	static <C extends FeatureConfiguration> PieceGeneratorSupplier<C> simple(
		final Predicate<PieceGeneratorSupplier.Context<C>> check, final PieceGenerator<C> generator
	) {
		Optional<PieceGenerator<C>> result = Optional.of(generator);
		return context -> check.test(context) ? result : Optional.empty();
	}

	static <C extends FeatureConfiguration> Predicate<PieceGeneratorSupplier.Context<C>> checkForBiomeOnTop(final Heightmap.Types type) {
		return context -> context.validBiomeOnTop(type);
	}

	public record Context<C extends FeatureConfiguration>(
		ChunkGenerator chunkGenerator,
		BiomeSource biomeSource,
		RandomState randomState,
		long seed,
		ChunkPos chunkPos,
		C config,
		LevelHeightAccessor heightAccessor,
		Predicate<Holder<Biome>> validBiome,
		StructureTemplateManager structureTemplateManager,
		RegistryAccess registryAccess
	) {
		public boolean validBiomeOnTop(final Heightmap.Types type) {
			int blockX = this.chunkPos.getMiddleBlockX();
			int blockZ = this.chunkPos.getMiddleBlockZ();
			int blockY = this.chunkGenerator.getFirstOccupiedHeight(blockX, blockZ, type, this.heightAccessor, this.randomState);
			Holder<Biome> biome = this.chunkGenerator
				.getBiomeSource()
				.getNoiseBiome(QuartPos.fromBlock(blockX), QuartPos.fromBlock(blockY), QuartPos.fromBlock(blockZ), this.randomState.sampler());
			return this.validBiome.test(biome);
		}
	}
}