package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import java.util.Map;
import java.util.Map.Entry;
import java.util.function.Function;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.util.Util;
import net.minecraft.world.level.BlockGetter;
import net.minecraft.world.level.block.state.BlockBehaviour;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.properties.BlockStateProperties;
import net.minecraft.world.level.block.state.properties.BooleanProperty;
import net.minecraft.world.level.block.state.properties.Property;
import net.minecraft.world.level.material.FluidState;
import net.minecraft.world.level.material.Fluids;
import net.minecraft.world.level.pathfinder.PathComputationType;
import net.minecraft.world.phys.shapes.CollisionContext;
import net.minecraft.world.phys.shapes.Shapes;
import net.minecraft.world.phys.shapes.VoxelShape;

public abstract class CrossCollisionBlock extends Block implements SimpleWaterloggedBlock {
	public static final BooleanProperty NORTH = PipeBlock.NORTH;
	public static final BooleanProperty EAST = PipeBlock.EAST;
	public static final BooleanProperty SOUTH = PipeBlock.SOUTH;
	public static final BooleanProperty WEST = PipeBlock.WEST;
	public static final BooleanProperty WATERLOGGED = BlockStateProperties.WATERLOGGED;
	public static final Map<Direction, BooleanProperty> PROPERTY_BY_DIRECTION = (Map<Direction, BooleanProperty>)PipeBlock.PROPERTY_BY_DIRECTION
		.entrySet()
		.stream()
		.filter(e -> ((Direction)e.getKey()).getAxis().isHorizontal())
		.collect(Util.toMap());
	private final Function<BlockState, VoxelShape> collisionShapes;
	private final Function<BlockState, VoxelShape> shapes;

	protected CrossCollisionBlock(
		final float postWidth,
		final float postHeight,
		final float wallWidth,
		final float wallHeight,
		final float collisionHeight,
		final BlockBehaviour.Properties properties
	) {
		super(properties);
		this.collisionShapes = this.makeShapes(postWidth, collisionHeight, wallWidth, 0.0F, collisionHeight);
		this.shapes = this.makeShapes(postWidth, postHeight, wallWidth, 0.0F, wallHeight);
	}

	@Override
	protected abstract MapCodec<? extends CrossCollisionBlock> codec();

	protected Function<BlockState, VoxelShape> makeShapes(
		final float postWidth, final float postHeight, final float wallWidth, final float wallBottom, final float wallTop
	) {
		VoxelShape post = Block.column(postWidth, 0.0, postHeight);
		Map<Direction, VoxelShape> arms = Shapes.rotateHorizontal(Block.boxZ(wallWidth, wallBottom, wallTop, 0.0, 8.0));
		return this.getShapeForEachState(state -> {
			VoxelShape shape = post;

			for (Entry<Direction, BooleanProperty> entry : PROPERTY_BY_DIRECTION.entrySet()) {
				if ((Boolean)state.getValue((Property)entry.getValue())) {
					shape = Shapes.or(shape, (VoxelShape)arms.get(entry.getKey()));
				}
			}

			return shape;
		}, new Property[]{WATERLOGGED});
	}

	@Override
	protected boolean propagatesSkylightDown(final BlockState state) {
		return !(Boolean)state.getValue(WATERLOGGED);
	}

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

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

	@Override
	protected FluidState getFluidState(final BlockState state) {
		return state.getValue(WATERLOGGED) ? Fluids.WATER.getSource(false) : super.getFluidState(state);
	}

	@Override
	protected boolean isPathfindable(final BlockState state, final PathComputationType type) {
		return false;
	}

	@Override
	protected BlockState rotate(final BlockState state, final Rotation rotation) {
		return switch (rotation) {
			case CLOCKWISE_180 -> (BlockState)state.setValue(NORTH, (Boolean)state.getValue(SOUTH))
				.setValue(EAST, (Boolean)state.getValue(WEST))
				.setValue(SOUTH, (Boolean)state.getValue(NORTH))
				.setValue(WEST, (Boolean)state.getValue(EAST));
			case COUNTERCLOCKWISE_90 -> (BlockState)state.setValue(NORTH, (Boolean)state.getValue(EAST))
				.setValue(EAST, (Boolean)state.getValue(SOUTH))
				.setValue(SOUTH, (Boolean)state.getValue(WEST))
				.setValue(WEST, (Boolean)state.getValue(NORTH));
			case CLOCKWISE_90 -> (BlockState)state.setValue(NORTH, (Boolean)state.getValue(WEST))
				.setValue(EAST, (Boolean)state.getValue(NORTH))
				.setValue(SOUTH, (Boolean)state.getValue(EAST))
				.setValue(WEST, (Boolean)state.getValue(SOUTH));
			default -> state;
		};
	}

	@Override
	protected BlockState mirror(final BlockState state, final Mirror mirror) {
		switch (mirror) {
			case LEFT_RIGHT:
				return state.setValue(NORTH, (Boolean)state.getValue(SOUTH)).setValue(SOUTH, (Boolean)state.getValue(NORTH));
			case FRONT_BACK:
				return state.setValue(EAST, (Boolean)state.getValue(WEST)).setValue(WEST, (Boolean)state.getValue(EAST));
			default:
				return super.mirror(state, mirror);
		}
	}
}