package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.RandomSource;
import net.minecraft.world.item.context.BlockPlaceContext;
import net.minecraft.world.level.BlockGetter;
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.phys.shapes.CollisionContext;
import net.minecraft.world.phys.shapes.VoxelShape;
import org.jspecify.annotations.Nullable;

public abstract class GrowingPlantBlock extends Block {
	protected final Direction growthDirection;
	protected final boolean scheduleFluidTicks;
	protected final VoxelShape shape;

	protected GrowingPlantBlock(
		final BlockBehaviour.Properties properties, final Direction growthDirection, final VoxelShape shape, final boolean scheduleFluidTicks
	) {
		super(properties);
		this.growthDirection = growthDirection;
		this.shape = shape;
		this.scheduleFluidTicks = scheduleFluidTicks;
	}

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

	@Nullable
	@Override
	public BlockState getStateForPlacement(final BlockPlaceContext context) {
		BlockState growthDirectionState = context.getLevel().getBlockState(context.getClickedPos().relative(this.growthDirection));
		return !growthDirectionState.is(this.getHeadBlock()) && !growthDirectionState.is(this.getBodyBlock())
			? this.getStateForPlacement(context.getLevel().getRandom())
			: this.getBodyBlock().defaultBlockState();
	}

	public BlockState getStateForPlacement(final RandomSource random) {
		return this.defaultBlockState();
	}

	@Override
	protected boolean canSurvive(final BlockState state, final LevelReader level, final BlockPos pos) {
		BlockPos attachedToPos = pos.relative(this.growthDirection.getOpposite());
		BlockState attachedToState = level.getBlockState(attachedToPos);
		return !this.canAttachTo(attachedToState)
			? false
			: attachedToState.is(this.getHeadBlock())
				|| attachedToState.is(this.getBodyBlock())
				|| attachedToState.isFaceSturdy(level, attachedToPos, this.growthDirection);
	}

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

	protected boolean canAttachTo(final BlockState state) {
		return true;
	}

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

	protected abstract GrowingPlantHeadBlock getHeadBlock();

	protected abstract Block getBodyBlock();
}