package net.minecraft.world.level.block;

import com.google.common.annotations.VisibleForTesting;
import com.mojang.serialization.MapCodec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.animal.frog.Tadpole;
import net.minecraft.world.level.BlockGetter;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.LevelAccessor;
import net.minecraft.world.level.LevelReader;
import net.minecraft.world.level.block.state.BlockBehaviour;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.material.FluidState;
import net.minecraft.world.level.material.Fluids;
import net.minecraft.world.phys.shapes.CollisionContext;
import net.minecraft.world.phys.shapes.VoxelShape;

public class FrogspawnBlock extends Block {
	public static final MapCodec<FrogspawnBlock> CODEC = simpleCodec(FrogspawnBlock::new);
	private static final int MIN_TADPOLES_SPAWN = 2;
	private static final int MAX_TADPOLES_SPAWN = 5;
	private static final int DEFAULT_MIN_HATCH_TICK_DELAY = 3600;
	private static final int DEFAULT_MAX_HATCH_TICK_DELAY = 12000;
	protected static final VoxelShape SHAPE = Block.box(0.0, 0.0, 0.0, 16.0, 1.5, 16.0);
	private static int minHatchTickDelay = 3600;
	private static int maxHatchTickDelay = 12000;

	@Override
	public MapCodec<FrogspawnBlock> codec() {
		return CODEC;
	}

	public FrogspawnBlock(BlockBehaviour.Properties properties) {
		super(properties);
	}

	@Override
	protected VoxelShape getShape(BlockState state, BlockGetter level, BlockPos pos, CollisionContext context) {
		return SHAPE;
	}

	@Override
	protected boolean canSurvive(BlockState state, LevelReader level, BlockPos pos) {
		return mayPlaceOn(level, pos.below());
	}

	@Override
	protected void onPlace(BlockState state, Level level, BlockPos pos, BlockState oldState, boolean movedByPiston) {
		level.scheduleTick(pos, this, getFrogspawnHatchDelay(level.getRandom()));
	}

	private static int getFrogspawnHatchDelay(RandomSource random) {
		return random.nextInt(minHatchTickDelay, maxHatchTickDelay);
	}

	@Override
	protected BlockState updateShape(BlockState state, Direction direction, BlockState neighborState, LevelAccessor level, BlockPos pos, BlockPos neighborPos) {
		return !this.canSurvive(state, level, pos) ? Blocks.AIR.defaultBlockState() : super.updateShape(state, direction, neighborState, level, pos, neighborPos);
	}

	@Override
	protected void tick(BlockState state, ServerLevel level, BlockPos pos, RandomSource random) {
		if (!this.canSurvive(state, level, pos)) {
			this.destroyBlock(level, pos);
		} else {
			this.hatchFrogspawn(level, pos, random);
		}
	}

	@Override
	protected void entityInside(BlockState state, Level level, BlockPos pos, Entity entity) {
		if (entity.getType().equals(EntityType.FALLING_BLOCK)) {
			this.destroyBlock(level, pos);
		}
	}

	private static boolean mayPlaceOn(BlockGetter level, BlockPos pos) {
		FluidState fluidState = level.getFluidState(pos);
		FluidState fluidState2 = level.getFluidState(pos.above());
		return fluidState.getType() == Fluids.WATER && fluidState2.getType() == Fluids.EMPTY;
	}

	private void hatchFrogspawn(ServerLevel level, BlockPos pos, RandomSource random) {
		this.destroyBlock(level, pos);
		level.playSound(null, pos, SoundEvents.FROGSPAWN_HATCH, SoundSource.BLOCKS, 1.0F, 1.0F);
		this.spawnTadpoles(level, pos, random);
	}

	private void destroyBlock(Level level, BlockPos pos) {
		level.destroyBlock(pos, false);
	}

	private void spawnTadpoles(ServerLevel level, BlockPos pos, RandomSource random) {
		int i = random.nextInt(2, 6);

		for (int j = 1; j <= i; j++) {
			Tadpole tadpole = EntityType.TADPOLE.create(level);
			if (tadpole != null) {
				double d = pos.getX() + this.getRandomTadpolePositionOffset(random);
				double e = pos.getZ() + this.getRandomTadpolePositionOffset(random);
				int k = random.nextInt(1, 361);
				tadpole.moveTo(d, pos.getY() - 0.5, e, k, 0.0F);
				tadpole.setPersistenceRequired();
				level.addFreshEntity(tadpole);
			}
		}
	}

	private double getRandomTadpolePositionOffset(RandomSource random) {
		double d = 0.2F;
		return Mth.clamp(random.nextDouble(), 0.2F, 0.7999999970197678);
	}

	@VisibleForTesting
	public static void setHatchDelay(int minHatchDelay, int maxHatchDelay) {
		minHatchTickDelay = minHatchDelay;
		maxHatchTickDelay = maxHatchDelay;
	}

	@VisibleForTesting
	public static void setDefaultHatchDelay() {
		minHatchTickDelay = 3600;
		maxHatchTickDelay = 12000;
	}
}