package net.minecraft.world.level.block;

import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Holder;
import net.minecraft.core.registries.Registries;
import net.minecraft.resources.ResourceKey;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.tags.BlockTags;
import net.minecraft.util.RandomSource;
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.levelgen.feature.AbstractHugeMushroomFeature;
import net.minecraft.world.level.levelgen.feature.ConfiguredFeature;
import net.minecraft.world.level.levelgen.feature.configurations.HugeMushroomFeatureConfiguration;
import net.minecraft.world.phys.shapes.CollisionContext;
import net.minecraft.world.phys.shapes.VoxelShape;

public class MushroomBlock extends VegetationBlock implements BonemealableBlock {
	public static final MapCodec<MushroomBlock> CODEC = RecordCodecBuilder.mapCodec(
		i -> i.group(ResourceKey.codec(Registries.CONFIGURED_FEATURE).fieldOf("feature").forGetter(b -> b.feature), propertiesCodec()).apply(i, MushroomBlock::new)
	);
	private static final VoxelShape SHAPE = Block.column(6.0, 0.0, 6.0);
	private final ResourceKey<ConfiguredFeature<?, ?>> feature;

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

	public MushroomBlock(final ResourceKey<ConfiguredFeature<?, ?>> feature, final BlockBehaviour.Properties properties) {
		super(properties);
		this.feature = feature;
	}

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

	@Override
	protected void randomTick(final BlockState state, final ServerLevel level, BlockPos pos, final RandomSource random) {
		if (random.nextInt(25) == 0) {
			int max = 5;
			int r = 4;

			for (BlockPos blockPos : BlockPos.betweenClosed(pos.offset(-4, -1, -4), pos.offset(4, 1, 4))) {
				if (level.getBlockState(blockPos).is(this)) {
					if (--max <= 0) {
						return;
					}
				}
			}

			BlockPos offset = pos.offset(random.nextInt(3) - 1, random.nextInt(2) - random.nextInt(2), random.nextInt(3) - 1);

			for (int i = 0; i < 4; i++) {
				if (level.isEmptyBlock(offset) && state.canSurvive(level, offset)) {
					pos = offset;
				}

				offset = pos.offset(random.nextInt(3) - 1, random.nextInt(2) - random.nextInt(2), random.nextInt(3) - 1);
			}

			if (level.isEmptyBlock(offset) && state.canSurvive(level, offset)) {
				level.setBlock(offset, state, 2);
			}
		}
	}

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

	@Override
	protected boolean canSurvive(final BlockState state, final LevelReader level, final BlockPos pos) {
		BlockPos belowPos = pos.below();
		BlockState below = level.getBlockState(belowPos);
		return below.is(BlockTags.OVERRIDES_MUSHROOM_LIGHT_REQUIREMENT) ? true : level.getRawBrightness(pos, 0) < 13 && this.mayPlaceOn(below, level, belowPos);
	}

	public boolean growMushroom(final ServerLevel level, final BlockPos pos, final BlockState state, final RandomSource random) {
		Optional<? extends Holder<ConfiguredFeature<?, ?>>> feature = level.registryAccess().lookupOrThrow(Registries.CONFIGURED_FEATURE).get(this.feature);
		if (feature.isEmpty()) {
			return false;
		} else {
			level.removeBlock(pos, false);
			if (((ConfiguredFeature)((Holder)feature.get()).value()).place(level, level.getChunkSource().getGenerator(), random, pos)) {
				return true;
			} else {
				level.setBlock(pos, state, 3);
				return false;
			}
		}
	}

	@Override
	public boolean isValidBonemealTarget(final LevelReader level, final BlockPos pos, final BlockState state) {
		if (level instanceof ServerLevel serverLevel) {
			Optional<? extends Holder<ConfiguredFeature<?, ?>>> featureHolder = serverLevel.registryAccess()
				.lookupOrThrow(Registries.CONFIGURED_FEATURE)
				.get(this.feature);
			if (featureHolder.isPresent()) {
				ConfiguredFeature<?, ?> configuredFeature = (ConfiguredFeature<?, ?>)((Holder)featureHolder.get()).value();
				if (configuredFeature.feature() instanceof AbstractHugeMushroomFeature && configuredFeature.config() instanceof HugeMushroomFeatureConfiguration config) {
					int minHeight = 4 + config.foliageRadius();
					return level.isInsideBuildHeight(pos.above(minHeight));
				} else {
					return false;
				}
			} else {
				return false;
			}
		} else {
			return false;
		}
	}

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

	@Override
	public void performBonemeal(final ServerLevel level, final RandomSource random, final BlockPos pos, final BlockState state) {
		this.growMushroom(level, pos, state, random);
	}
}