package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.sounds.SoundSource;
import net.minecraft.tags.FluidTags;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.entity.BlockEntity;
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.redstone.Orientation;
import org.jspecify.annotations.Nullable;

/**
 * Access widened by fabric-transitive-access-wideners-v1 to accessible
 */
public class SpongeBlock extends Block {
	public static final MapCodec<SpongeBlock> CODEC = simpleCodec(SpongeBlock::new);
	public static final int MAX_DEPTH = 6;
	public static final int MAX_COUNT = 64;
	private static final Direction[] ALL_DIRECTIONS = Direction.values();

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

	/**
	 * Access widened by fabric-transitive-access-wideners-v1 to accessible
	 */
	public SpongeBlock(final BlockBehaviour.Properties properties) {
		super(properties);
	}

	@Override
	protected void onPlace(final BlockState state, final Level level, final BlockPos pos, final BlockState oldState, final boolean movedByPiston) {
		if (!oldState.is(state.getBlock())) {
			this.tryAbsorbWater(level, pos);
		}
	}

	@Override
	protected void neighborChanged(
		final BlockState state, final Level level, final BlockPos pos, final Block block, @Nullable final Orientation orientation, final boolean movedByPiston
	) {
		this.tryAbsorbWater(level, pos);
		super.neighborChanged(state, level, pos, block, orientation, movedByPiston);
	}

	protected void tryAbsorbWater(final Level level, final BlockPos pos) {
		if (this.removeWaterBreadthFirstSearch(level, pos)) {
			level.setBlock(pos, Blocks.WET_SPONGE.defaultBlockState(), 2);
			level.playSound(null, pos, SoundEvents.SPONGE_ABSORB, SoundSource.BLOCKS, 1.0F, 1.0F);
		}
	}

	private boolean removeWaterBreadthFirstSearch(final Level level, final BlockPos startPos) {
		return BlockPos.breadthFirstTraversal(startPos, 6, 65, (pos, consumer) -> {
			for (Direction direction : ALL_DIRECTIONS) {
				consumer.accept(pos.relative(direction));
			}
		}, pos -> {
			if (pos.equals(startPos)) {
				return BlockPos.TraversalNodeStatus.ACCEPT;
			} else {
				BlockState state = level.getBlockState(pos);
				FluidState fluidState = level.getFluidState(pos);
				if (!fluidState.is(FluidTags.WATER)) {
					return BlockPos.TraversalNodeStatus.SKIP;
				} else if (state.getBlock() instanceof BucketPickup bucketPickup && !bucketPickup.pickupBlock(null, level, pos, state).isEmpty()) {
					return BlockPos.TraversalNodeStatus.ACCEPT;
				} else {
					if (state.getBlock() instanceof LiquidBlock) {
						level.setBlock(pos, Blocks.AIR.defaultBlockState(), 3);
					} else {
						if (!state.is(Blocks.KELP) && !state.is(Blocks.KELP_PLANT) && !state.is(Blocks.SEAGRASS) && !state.is(Blocks.TALL_SEAGRASS)) {
							return BlockPos.TraversalNodeStatus.SKIP;
						}

						BlockEntity blockEntity = state.hasBlockEntity() ? level.getBlockEntity(pos) : null;
						dropResources(state, level, pos, blockEntity);
						level.setBlock(pos, Blocks.AIR.defaultBlockState(), 3);
					}

					return BlockPos.TraversalNodeStatus.ACCEPT;
				}
			}
		}) > 1;
	}
}