package net.minecraft.world.level.biome;

import com.google.common.collect.Sets;
import com.mojang.datafixers.util.Pair;
import com.mojang.serialization.MapCodec;
import java.util.Set;
import java.util.function.Predicate;
import java.util.stream.Stream;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Holder;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.LevelReader;
import org.jspecify.annotations.Nullable;

public class FixedBiomeSource extends BiomeSource implements BiomeManager.NoiseBiomeSource {
	public static final MapCodec<FixedBiomeSource> CODEC = Biome.CODEC.fieldOf("biome").<FixedBiomeSource>xmap(FixedBiomeSource::new, s -> s.biome).stable();
	private final Holder<Biome> biome;

	public FixedBiomeSource(final Holder<Biome> biome) {
		this.biome = biome;
	}

	@Override
	protected Stream<Holder<Biome>> collectPossibleBiomes() {
		return Stream.of(this.biome);
	}

	@Override
	protected MapCodec<? extends BiomeSource> codec() {
		return CODEC;
	}

	@Override
	public Holder<Biome> getNoiseBiome(final int quartX, final int quartY, final int quartZ, final Climate.Sampler sampler) {
		return this.biome;
	}

	@Override
	public Holder<Biome> getNoiseBiome(final int quartX, final int quartY, final int quartZ) {
		return this.biome;
	}

	@Nullable
	@Override
	public Pair<BlockPos, Holder<Biome>> findBiomeHorizontal(
		final int originX,
		final int originY,
		final int originZ,
		final int r,
		final int skipStep,
		final Predicate<Holder<Biome>> allowed,
		final RandomSource random,
		final boolean findClosest,
		final Climate.Sampler sampler
	) {
		if (allowed.test(this.biome)) {
			return findClosest
				? Pair.of(new BlockPos(originX, originY, originZ), this.biome)
				: Pair.of(new BlockPos(originX - r + random.nextInt(r * 2 + 1), originY, originZ - r + random.nextInt(r * 2 + 1)), this.biome);
		} else {
			return null;
		}
	}

	@Nullable
	@Override
	public Pair<BlockPos, Holder<Biome>> findClosestBiome3d(
		final BlockPos origin,
		final int searchRadius,
		final int sampleResolutionHorizontal,
		final int sampleResolutionVertical,
		final Predicate<Holder<Biome>> allowed,
		final Climate.Sampler sampler,
		final LevelReader level
	) {
		return allowed.test(this.biome) ? Pair.of(origin.atY(Mth.clamp(origin.getY(), level.getMinY() + 1, level.getMaxY() + 1)), this.biome) : null;
	}

	@Override
	public Set<Holder<Biome>> getBiomesWithin(final int x, final int y, final int z, final int r, final Climate.Sampler sampler) {
		return Sets.<Holder<Biome>>newHashSet(Set.of(this.biome));
	}
}