package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import java.util.List;
import java.util.Optional;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Holder;
import net.minecraft.core.registries.Registries;
import net.minecraft.data.worldgen.placement.VegetationPlacements;
import net.minecraft.references.BlockItemIds;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.RandomSource;
import net.minecraft.util.Util;
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.levelgen.feature.ConfiguredFeature;
import net.minecraft.world.level.levelgen.placement.PlacedFeature;

public class GrassBlock extends SpreadingSnowyBlock implements BonemealableBlock {
	public static final MapCodec<GrassBlock> CODEC = simpleCodec(GrassBlock::new);

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

	public GrassBlock(final BlockBehaviour.Properties properties) {
		super(properties, BlockItemIds.DIRT.block());
	}

	@Override
	public boolean isValidBonemealTarget(final LevelReader level, final BlockPos pos, final BlockState state) {
		return level.getBlockState(pos.above()).isAir() && level.isInsideBuildHeight(pos.above());
	}

	@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) {
		BlockPos above = pos.above();
		BlockState grass = Blocks.SHORT_GRASS.defaultBlockState();
		Optional<Holder.Reference<PlacedFeature>> grassFeature = level.registryAccess()
			.lookupOrThrow(Registries.PLACED_FEATURE)
			.get(VegetationPlacements.GRASS_BONEMEAL);

		label48:
		for (int j = 0; j < 128; j++) {
			BlockPos testPos = above;

			for (int i = 0; i < j / 16; i++) {
				testPos = testPos.offset(random.nextInt(3) - 1, (random.nextInt(3) - 1) * random.nextInt(3) / 2, random.nextInt(3) - 1);
				if (!level.getBlockState(testPos.below()).is(this) || level.getBlockState(testPos).isCollisionShapeFullBlock(level, testPos)) {
					continue label48;
				}
			}

			BlockState testState = level.getBlockState(testPos);
			if (testState.is(grass.getBlock()) && random.nextInt(10) == 0) {
				BonemealableBlock bonemealableBlock = (BonemealableBlock)grass.getBlock();
				if (bonemealableBlock.isValidBonemealTarget(level, testPos, testState)) {
					bonemealableBlock.performBonemeal(level, random, testPos, testState);
				}
			}

			if (testState.isAir() && !level.isOutsideBuildHeight(testPos)) {
				if (random.nextInt(8) == 0) {
					List<ConfiguredFeature<?, ?>> features = level.getBiome(testPos).value().getGenerationSettings().getBoneMealFeatures();
					if (!features.isEmpty()) {
						ConfiguredFeature<?, ?> placementFeature = Util.getRandom(features, random);
						placementFeature.place(level, level.getChunkSource().getGenerator(), random, testPos);
					}
				} else if (grassFeature.isPresent()) {
					((PlacedFeature)((Holder.Reference)grassFeature.get()).value()).place(level, level.getChunkSource().getGenerator(), random, testPos);
				}
			}
		}
	}

	@Override
	public BonemealableBlock.Type getType() {
		return BonemealableBlock.Type.NEIGHBOR_SPREADER;
	}
}