package net.minecraft.world.level.block;

import com.mojang.datafixers.DataFixUtils;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.core.Registry;
import net.minecraft.core.registries.Registries;
import net.minecraft.resources.ResourceKey;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.tags.TagKey;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.BlockGetter;
import net.minecraft.world.level.Level;
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.level.block.state.StateDefinition;
import net.minecraft.world.level.block.state.properties.BlockStateProperties;
import net.minecraft.world.level.block.state.properties.IntegerProperty;
import net.minecraft.world.phys.shapes.CollisionContext;
import net.minecraft.world.phys.shapes.VoxelShape;

/**
 * Access widened by fabric-transitive-access-wideners-v1 to accessible
 */
public class StemBlock extends VegetationBlock implements BonemealableBlock {
	public static final MapCodec<StemBlock> CODEC = RecordCodecBuilder.mapCodec(
		i -> i.group(
				ResourceKey.codec(Registries.BLOCK).fieldOf("fruit").forGetter(b -> b.fruit),
				ResourceKey.codec(Registries.BLOCK).fieldOf("attached_stem").forGetter(b -> b.attachedStem),
				ResourceKey.codec(Registries.ITEM).fieldOf("seed").forGetter(b -> b.seed),
				TagKey.codec(Registries.BLOCK).fieldOf("stem_support_blocks").forGetter(b -> b.stemSupportBlocks),
				TagKey.codec(Registries.BLOCK).fieldOf("fruit_support_blocks").forGetter(b -> b.fruitSupportBlocks),
				propertiesCodec()
			)
			.apply(i, StemBlock::new)
	);
	public static final int MAX_AGE = 7;
	public static final IntegerProperty AGE = BlockStateProperties.AGE_7;
	private static final VoxelShape[] SHAPES = Block.boxes(7, age -> Block.column(2.0, 0.0, 2 + age * 2));
	private final ResourceKey<Block> fruit;
	private final ResourceKey<Block> attachedStem;
	private final ResourceKey<Item> seed;
	private final TagKey<Block> stemSupportBlocks;
	private final TagKey<Block> fruitSupportBlocks;

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

	/**
	 * Access widened by fabric-transitive-access-wideners-v1 to accessible
	 */
	public StemBlock(
		final ResourceKey<Block> fruit,
		final ResourceKey<Block> attachedStem,
		final ResourceKey<Item> seed,
		final TagKey<Block> stemSupportBlocks,
		final TagKey<Block> fruitSupportBlocks,
		final BlockBehaviour.Properties properties
	) {
		super(properties);
		this.fruit = fruit;
		this.attachedStem = attachedStem;
		this.seed = seed;
		this.stemSupportBlocks = stemSupportBlocks;
		this.fruitSupportBlocks = fruitSupportBlocks;
		this.registerDefaultState(this.stateDefinition.any().setValue(AGE, 0));
	}

	@Override
	protected VoxelShape getShape(final BlockState state, final BlockGetter level, final BlockPos pos, final CollisionContext context) {
		return SHAPES[state.getValue(AGE)];
	}

	@Override
	protected boolean mayPlaceOn(final BlockState state, final BlockGetter level, final BlockPos pos) {
		return state.is(this.stemSupportBlocks);
	}

	@Override
	protected void randomTick(BlockState state, final ServerLevel level, final BlockPos pos, final RandomSource random) {
		if (level.getRawBrightness(pos, 0) >= 9) {
			float growthSpeed = CropBlock.getGrowthSpeed(this, level, pos);
			if (random.nextInt((int)(25.0F / growthSpeed) + 1) == 0) {
				int age = (Integer)state.getValue(AGE);
				if (age < 7) {
					state = state.setValue(AGE, age + 1);
					level.setBlock(pos, state, 2);
				} else {
					Direction direction = Direction.Plane.HORIZONTAL.getRandomDirection(random);
					BlockPos relative = pos.relative(direction);
					BlockState stateBelow = level.getBlockState(relative.below());
					if (level.getBlockState(relative).isAir() && stateBelow.is(this.fruitSupportBlocks)) {
						Registry<Block> blocks = level.registryAccess().lookupOrThrow(Registries.BLOCK);
						Optional<Block> fruit = blocks.getOptional(this.fruit);
						Optional<Block> stem = blocks.getOptional(this.attachedStem);
						if (fruit.isPresent() && stem.isPresent()) {
							level.setBlockAndUpdate(relative, ((Block)fruit.get()).defaultBlockState());
							level.setBlockAndUpdate(pos, ((Block)stem.get()).defaultBlockState().setValue(HorizontalDirectionalBlock.FACING, direction));
						}
					}
				}
			}
		}
	}

	@Override
	protected ItemStack getCloneItemStack(final LevelReader level, final BlockPos pos, final BlockState state, final boolean includeData) {
		return new ItemStack(DataFixUtils.orElse(level.registryAccess().lookupOrThrow(Registries.ITEM).getOptional(this.seed), this));
	}

	@Override
	public boolean isValidBonemealTarget(final LevelReader level, final BlockPos pos, final BlockState state) {
		return (Integer)state.getValue(AGE) != 7;
	}

	@Override
	public boolean isBonemealSuccess(final Level level, final RandomSource random, final BlockPos pos, final BlockState state) {
		return true;
	}

	@Override
	public void performBonemeal(final ServerLevel level, final RandomSource random, final BlockPos pos, final BlockState state) {
		int age = Math.min(7, (Integer)state.getValue(AGE) + Mth.nextInt(random, 2, 5));
		BlockState newState = state.setValue(AGE, age);
		level.setBlock(pos, newState, 2);
		if (age == 7) {
			newState.randomTick(level, pos, random);
		}
	}

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