package net.minecraft.world.level.block;

import com.mojang.serialization.Codec;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Map;
import java.util.function.BiConsumer;
import java.util.function.Function;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.sounds.SoundEvent;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.RandomSource;
import net.minecraft.world.InteractionResult;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.InsideBlockEffectApplier;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.entity.projectile.AbstractArrow;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.BlockGetter;
import net.minecraft.world.level.Explosion;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.LevelAccessor;
import net.minecraft.world.level.block.state.BlockBehaviour;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.StateDefinition.Builder;
import net.minecraft.world.level.block.state.properties.AttachFace;
import net.minecraft.world.level.block.state.properties.BlockSetType;
import net.minecraft.world.level.block.state.properties.BlockStateProperties;
import net.minecraft.world.level.block.state.properties.BooleanProperty;
import net.minecraft.world.level.gameevent.GameEvent;
import net.minecraft.world.level.redstone.ExperimentalRedstoneUtils;
import net.minecraft.world.level.redstone.Orientation;
import net.minecraft.world.phys.BlockHitResult;
import net.minecraft.world.phys.shapes.BooleanOp;
import net.minecraft.world.phys.shapes.CollisionContext;
import net.minecraft.world.phys.shapes.Shapes;
import net.minecraft.world.phys.shapes.VoxelShape;
import org.jetbrains.annotations.Nullable;

public class ButtonBlock extends FaceAttachedHorizontalDirectionalBlock {
	public static final MapCodec<ButtonBlock> CODEC = RecordCodecBuilder.mapCodec(
		instance -> instance.group(
				BlockSetType.CODEC.fieldOf("block_set_type").forGetter(buttonBlock -> buttonBlock.type),
				Codec.intRange(1, 1024).fieldOf("ticks_to_stay_pressed").forGetter(buttonBlock -> buttonBlock.ticksToStayPressed),
				propertiesCodec()
			)
			.apply(instance, ButtonBlock::new)
	);
	public static final BooleanProperty POWERED = BlockStateProperties.POWERED;
	private final BlockSetType type;
	private final int ticksToStayPressed;
	private final Function<BlockState, VoxelShape> shapes;

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

	protected ButtonBlock(BlockSetType type, int ticksToStayPressed, BlockBehaviour.Properties properties) {
		super(properties.sound(type.soundType()));
		this.type = type;
		this.registerDefaultState(this.stateDefinition.any().setValue(FACING, Direction.NORTH).setValue(POWERED, false).setValue(FACE, AttachFace.WALL));
		this.ticksToStayPressed = ticksToStayPressed;
		this.shapes = this.makeShapes();
	}

	private Function<BlockState, VoxelShape> makeShapes() {
		VoxelShape voxelShape = Block.cube(14.0);
		VoxelShape voxelShape2 = Block.cube(12.0);
		Map<AttachFace, Map<Direction, VoxelShape>> map = Shapes.rotateAttachFace(Block.boxZ(6.0, 4.0, 8.0, 16.0));
		return this.getShapeForEachState(
			blockState -> Shapes.join(
				(VoxelShape)((Map)map.get(blockState.getValue(FACE))).get(blockState.getValue(FACING)),
				blockState.getValue(POWERED) ? voxelShape : voxelShape2,
				BooleanOp.ONLY_FIRST
			)
		);
	}

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

	@Override
	protected InteractionResult useWithoutItem(BlockState state, Level level, BlockPos pos, Player player, BlockHitResult hitResult) {
		if ((Boolean)state.getValue(POWERED)) {
			return InteractionResult.CONSUME;
		} else {
			this.press(state, level, pos, player);
			return InteractionResult.SUCCESS;
		}
	}

	@Override
	protected void onExplosionHit(BlockState state, ServerLevel level, BlockPos pos, Explosion explosion, BiConsumer<ItemStack, BlockPos> dropConsumer) {
		if (explosion.canTriggerBlocks() && !(Boolean)state.getValue(POWERED)) {
			this.press(state, level, pos, null);
		}

		super.onExplosionHit(state, level, pos, explosion, dropConsumer);
	}

	public void press(BlockState state, Level level, BlockPos pos, @Nullable Player player) {
		level.setBlock(pos, state.setValue(POWERED, true), 3);
		this.updateNeighbours(state, level, pos);
		level.scheduleTick(pos, this, this.ticksToStayPressed);
		this.playSound(player, level, pos, true);
		level.gameEvent(player, GameEvent.BLOCK_ACTIVATE, pos);
	}

	protected void playSound(@Nullable Player player, LevelAccessor level, BlockPos pos, boolean hitByArrow) {
		level.playSound(hitByArrow ? player : null, pos, this.getSound(hitByArrow), SoundSource.BLOCKS);
	}

	protected SoundEvent getSound(boolean isOn) {
		return isOn ? this.type.buttonClickOn() : this.type.buttonClickOff();
	}

	@Override
	protected void affectNeighborsAfterRemoval(BlockState state, ServerLevel level, BlockPos pos, boolean movedByPiston) {
		if (!movedByPiston && (Boolean)state.getValue(POWERED)) {
			this.updateNeighbours(state, level, pos);
		}
	}

	@Override
	protected int getSignal(BlockState state, BlockGetter level, BlockPos pos, Direction direction) {
		return state.getValue(POWERED) ? 15 : 0;
	}

	@Override
	protected int getDirectSignal(BlockState state, BlockGetter level, BlockPos pos, Direction direction) {
		return state.getValue(POWERED) && getConnectedDirection(state) == direction ? 15 : 0;
	}

	@Override
	protected boolean isSignalSource(BlockState state) {
		return true;
	}

	@Override
	protected void tick(BlockState state, ServerLevel level, BlockPos pos, RandomSource random) {
		if ((Boolean)state.getValue(POWERED)) {
			this.checkPressed(state, level, pos);
		}
	}

	@Override
	protected void entityInside(BlockState state, Level level, BlockPos pos, Entity entity, InsideBlockEffectApplier effectApplier) {
		if (!level.isClientSide && this.type.canButtonBeActivatedByArrows() && !(Boolean)state.getValue(POWERED)) {
			this.checkPressed(state, level, pos);
		}
	}

	protected void checkPressed(BlockState state, Level level, BlockPos pos) {
		AbstractArrow abstractArrow = this.type.canButtonBeActivatedByArrows()
			? (AbstractArrow)level.getEntitiesOfClass(AbstractArrow.class, state.getShape(level, pos).bounds().move(pos)).stream().findFirst().orElse(null)
			: null;
		boolean bl = abstractArrow != null;
		boolean bl2 = (Boolean)state.getValue(POWERED);
		if (bl != bl2) {
			level.setBlock(pos, state.setValue(POWERED, bl), 3);
			this.updateNeighbours(state, level, pos);
			this.playSound(null, level, pos, bl);
			level.gameEvent(abstractArrow, bl ? GameEvent.BLOCK_ACTIVATE : GameEvent.BLOCK_DEACTIVATE, pos);
		}

		if (bl) {
			level.scheduleTick(new BlockPos(pos), this, this.ticksToStayPressed);
		}
	}

	private void updateNeighbours(BlockState state, Level level, BlockPos pos) {
		Direction direction = getConnectedDirection(state).getOpposite();
		Orientation orientation = ExperimentalRedstoneUtils.initialOrientation(
			level, direction, direction.getAxis().isHorizontal() ? Direction.UP : state.getValue(FACING)
		);
		level.updateNeighborsAt(pos, this, orientation);
		level.updateNeighborsAt(pos.relative(direction), this, orientation);
	}

	@Override
	protected void createBlockStateDefinition(Builder<Block, BlockState> builder) {
		builder.add(FACING, POWERED, FACE);
	}
}