package net.minecraft.world.level.levelgen.feature;

import com.mojang.serialization.Codec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.LevelAccessor;
import net.minecraft.world.level.WorldGenLevel;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.levelgen.feature.configurations.ReplaceSphereConfiguration;
import org.jspecify.annotations.Nullable;

public class ReplaceBlobsFeature extends Feature<ReplaceSphereConfiguration> {
	public ReplaceBlobsFeature(final Codec<ReplaceSphereConfiguration> codec) {
		super(codec);
	}

	@Override
	public boolean place(final FeaturePlaceContext<ReplaceSphereConfiguration> context) {
		ReplaceSphereConfiguration config = context.config();
		WorldGenLevel level = context.level();
		RandomSource random = context.random();
		Block targetBlock = config.targetState.getBlock();
		BlockPos centerPos = findTarget(level, context.origin().mutable().clamp(Direction.Axis.Y, level.getMinY() + 1, level.getMaxY()), targetBlock);
		if (centerPos == null) {
			return false;
		} else {
			int radiusX = config.radius().sample(random);
			int radiusY = config.radius().sample(random);
			int radiusZ = config.radius().sample(random);
			int maximumRadius = Math.max(radiusX, Math.max(radiusY, radiusZ));
			boolean replacedAny = false;

			for (BlockPos pos : BlockPos.withinManhattan(centerPos, radiusX, radiusY, radiusZ)) {
				if (pos.distManhattan(centerPos) > maximumRadius) {
					break;
				}

				BlockState blockState = level.getBlockState(pos);
				if (blockState.is(targetBlock)) {
					this.setBlock(level, pos, config.replaceState);
					replacedAny = true;
				}
			}

			return replacedAny;
		}
	}

	@Nullable
	private static BlockPos findTarget(final LevelAccessor level, final BlockPos.MutableBlockPos cursor, final Block target) {
		while (cursor.getY() > level.getMinY() + 1) {
			BlockState blockState = level.getBlockState(cursor);
			if (blockState.is(target)) {
				return cursor;
			}

			cursor.move(Direction.DOWN);
		}

		return null;
	}
}