package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.RandomSource;
import net.minecraft.world.item.context.BlockPlaceContext;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.LevelReader;
import net.minecraft.world.level.ScheduledTickAccess;
import net.minecraft.world.level.block.entity.BlockEntity;
import net.minecraft.world.level.block.entity.BlockEntityTicker;
import net.minecraft.world.level.block.entity.BlockEntityType;
import net.minecraft.world.level.block.entity.BlockEntityTypes;
import net.minecraft.world.level.block.entity.PotentSulfurBlockEntity;
import net.minecraft.world.level.block.state.BlockBehaviour;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.StateDefinition;
import net.minecraft.world.level.block.state.properties.BlockStateProperties;
import net.minecraft.world.level.block.state.properties.EnumProperty;
import net.minecraft.world.level.block.state.properties.PotentSulfurState;
import net.minecraft.world.level.block.state.properties.Property;
import net.minecraft.world.level.gameevent.GameEvent;
import net.minecraft.world.level.material.Fluids;
import org.jspecify.annotations.Nullable;

public class PotentSulfurBlock extends BaseEntityBlock {
	public static final int ALLOWED_WATER_BLOCKS_ABOVE = 4;
	public static final MapCodec<PotentSulfurBlock> CODEC = simpleCodec(PotentSulfurBlock::new);
	public static final EnumProperty<PotentSulfurState> STATE = BlockStateProperties.POTENT_SULFUR_STATE;

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

	public PotentSulfurBlock(final BlockBehaviour.Properties properties) {
		super(properties);
		this.registerDefaultState(this.defaultBlockState().setValue(STATE, PotentSulfurState.DRY));
	}

	@Override
	protected void createBlockStateDefinition(final StateDefinition.Builder<Block, BlockState> builder) {
		builder.add(STATE);
	}

	@Nullable
	@Override
	public BlockEntity newBlockEntity(final BlockPos worldPosition, final BlockState blockState) {
		return new PotentSulfurBlockEntity(worldPosition, blockState);
	}

	@Override
	protected BlockState updateShape(
		final BlockState state,
		final LevelReader level,
		final ScheduledTickAccess ticks,
		final BlockPos pos,
		final Direction directionToNeighbour,
		final BlockPos neighbourPos,
		final BlockState neighbourState,
		final RandomSource random
	) {
		return validBlockState(state, level, pos);
	}

	@Nullable
	@Override
	public BlockState getStateForPlacement(final BlockPlaceContext context) {
		return validBlockState(this.defaultBlockState(), context.getLevel(), context.getClickedPos());
	}

	private static BlockState validBlockState(final BlockState state, final LevelReader level, final BlockPos pos) {
		if (!level.getFluidState(pos.above()).isSourceOfType(Fluids.WATER)) {
			return state.setValue(STATE, PotentSulfurState.DRY);
		} else {
			BlockState belowState = level.getBlockState(pos.below());
			if (!belowState.is(Blocks.MAGMA_BLOCK)) {
				return state.setValue(STATE, PotentSulfurState.WET);
			} else {
				boolean isGeyser = state.getValue(STATE) == PotentSulfurState.ERUPTING || state.getValue(STATE) == PotentSulfurState.DORMANT;
				if (!isGeyser && level.getBlockEntity(pos) instanceof PotentSulfurBlockEntity potentSulfurEntity) {
					potentSulfurEntity.resetCountdown();
				}

				return state.getValue(STATE) == PotentSulfurState.ERUPTING ? state : state.setValue(STATE, PotentSulfurState.DORMANT);
			}
		}
	}

	@Override
	protected void onPlace(final BlockState state, final Level level, final BlockPos pos, final BlockState oldState, final boolean movedByPiston) {
		super.onPlace(state, level, pos, oldState, movedByPiston);
		if (state.getValue(STATE) == PotentSulfurState.ERUPTING) {
			level.blockEvent(pos, this, 0, 0);
			level.playSound(null, pos, SoundEvents.GEYSER_ERUPTION_START, SoundSource.BLOCKS, 1.0F, 1.0F);
			level.gameEvent(GameEvent.BLOCK_ACTIVATE, pos, GameEvent.Context.of(state));
		}
	}

	@Override
	public void animateTick(final BlockState state, final Level level, final BlockPos pos, final RandomSource random) {
		if (state.getValue(STATE) != PotentSulfurState.DRY) {
			if (level.getFluidState(pos.above()).isSourceOfType(Fluids.WATER)) {
				spawnBubbleParticlesAt(level, random, pos.getX(), pos.getY() + 1, pos.getZ());
				spawnBubbleParticlesAt(level, random, pos.getX(), pos.getY() + 1, pos.getZ());
				if (random.nextInt(10) == 0) {
					level.playLocalSound(pos.getX(), pos.getY(), pos.getZ(), SoundEvents.NOXIOUS_GAS, SoundSource.AMBIENT, 1.0F, 1.0F, false);
				}
			}
		}
	}

	private static void spawnBubbleParticlesAt(final Level level, final RandomSource random, final double x, final double y, final double z) {
		level.addAlwaysVisibleParticle(ParticleTypes.SULFUR_BUBBLES, x + random.nextFloat(), y + random.nextFloat(), z + random.nextFloat(), 0.0, 0.0, 0.0);
	}

	@Override
	protected boolean triggerEvent(final BlockState state, final Level level, final BlockPos pos, final int b0, final int b1) {
		if (level.getBlockEntity(pos) instanceof PotentSulfurBlockEntity entity) {
			entity.eruptionTick = level.getGameTime();
		}

		return true;
	}

	@Nullable
	@Override
	public <T extends BlockEntity> BlockEntityTicker<T> getTicker(final Level level, final BlockState blockState, final BlockEntityType<T> type) {
		boolean client = level.isClientSide();

		return createTickerHelper(
			type,
			BlockEntityTypes.POTENT_SULFUR,
			switch ((PotentSulfurState)blockState.getValue((Property<T>)STATE)) {
				case DRY -> null;
				case WET -> client ? PotentSulfurBlockEntity.CLIENT_NOXIOUS_GAS_TICKER : PotentSulfurBlockEntity.SERVER_NAUSEA_EFFECT_TICKER;
				case DORMANT -> client
					? PotentSulfurBlockEntity.CLIENT_NOXIOUS_GAS_TICKER
					: PotentSulfurBlockEntity.SERVER_WAITING_COUNTDOWN_TICKER.andThen(PotentSulfurBlockEntity.SERVER_NAUSEA_EFFECT_TICKER);
				case ERUPTING -> client
					? PotentSulfurBlockEntity.CLIENT_GEYSER_PLUME_TICKER
					: PotentSulfurBlockEntity.SERVER_LAUNCH_ENTITY_TICKER.andThen(PotentSulfurBlockEntity.SERVER_WAITING_COUNTDOWN_TICKER);
			}
		);
	}
}