package net.minecraft.world.level.levelgen.feature;

import com.mojang.serialization.Codec;
import java.util.List;
import net.minecraft.Util;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.WorldGenLevel;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.entity.BlockEntityType;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.predicate.BlockStatePredicate;
import net.minecraft.world.level.levelgen.feature.configurations.NoneFeatureConfiguration;
import net.minecraft.world.level.storage.loot.BuiltInLootTables;

public class DesertWellFeature extends Feature<NoneFeatureConfiguration> {
	private static final BlockStatePredicate IS_SAND = BlockStatePredicate.forBlock(Blocks.SAND);
	private final BlockState sand = Blocks.SAND.defaultBlockState();
	private final BlockState sandSlab = Blocks.SANDSTONE_SLAB.defaultBlockState();
	private final BlockState sandstone = Blocks.SANDSTONE.defaultBlockState();
	private final BlockState water = Blocks.WATER.defaultBlockState();

	public DesertWellFeature(Codec<NoneFeatureConfiguration> codec) {
		super(codec);
	}

	@Override
	public boolean place(FeaturePlaceContext<NoneFeatureConfiguration> context) {
		WorldGenLevel worldGenLevel = context.level();
		BlockPos blockPos = context.origin();
		blockPos = blockPos.above();

		while (worldGenLevel.isEmptyBlock(blockPos) && blockPos.getY() > worldGenLevel.getMinY() + 2) {
			blockPos = blockPos.below();
		}

		if (!IS_SAND.test(worldGenLevel.getBlockState(blockPos))) {
			return false;
		} else {
			for (int i = -2; i <= 2; i++) {
				for (int j = -2; j <= 2; j++) {
					if (worldGenLevel.isEmptyBlock(blockPos.offset(i, -1, j)) && worldGenLevel.isEmptyBlock(blockPos.offset(i, -2, j))) {
						return false;
					}
				}
			}

			for (int i = -2; i <= 0; i++) {
				for (int jx = -2; jx <= 2; jx++) {
					for (int k = -2; k <= 2; k++) {
						worldGenLevel.setBlock(blockPos.offset(jx, i, k), this.sandstone, 2);
					}
				}
			}

			worldGenLevel.setBlock(blockPos, this.water, 2);

			for (Direction direction : Direction.Plane.HORIZONTAL) {
				worldGenLevel.setBlock(blockPos.relative(direction), this.water, 2);
			}

			BlockPos blockPos2 = blockPos.below();
			worldGenLevel.setBlock(blockPos2, this.sand, 2);

			for (Direction direction2 : Direction.Plane.HORIZONTAL) {
				worldGenLevel.setBlock(blockPos2.relative(direction2), this.sand, 2);
			}

			for (int jx = -2; jx <= 2; jx++) {
				for (int k = -2; k <= 2; k++) {
					if (jx == -2 || jx == 2 || k == -2 || k == 2) {
						worldGenLevel.setBlock(blockPos.offset(jx, 1, k), this.sandstone, 2);
					}
				}
			}

			worldGenLevel.setBlock(blockPos.offset(2, 1, 0), this.sandSlab, 2);
			worldGenLevel.setBlock(blockPos.offset(-2, 1, 0), this.sandSlab, 2);
			worldGenLevel.setBlock(blockPos.offset(0, 1, 2), this.sandSlab, 2);
			worldGenLevel.setBlock(blockPos.offset(0, 1, -2), this.sandSlab, 2);

			for (int jx = -1; jx <= 1; jx++) {
				for (int kx = -1; kx <= 1; kx++) {
					if (jx == 0 && kx == 0) {
						worldGenLevel.setBlock(blockPos.offset(jx, 4, kx), this.sandstone, 2);
					} else {
						worldGenLevel.setBlock(blockPos.offset(jx, 4, kx), this.sandSlab, 2);
					}
				}
			}

			for (int jx = 1; jx <= 3; jx++) {
				worldGenLevel.setBlock(blockPos.offset(-1, jx, -1), this.sandstone, 2);
				worldGenLevel.setBlock(blockPos.offset(-1, jx, 1), this.sandstone, 2);
				worldGenLevel.setBlock(blockPos.offset(1, jx, -1), this.sandstone, 2);
				worldGenLevel.setBlock(blockPos.offset(1, jx, 1), this.sandstone, 2);
			}

			List<BlockPos> list = List.of(blockPos, blockPos.east(), blockPos.south(), blockPos.west(), blockPos.north());
			RandomSource randomSource = context.random();
			placeSusSand(worldGenLevel, Util.<BlockPos>getRandom(list, randomSource).below(1));
			placeSusSand(worldGenLevel, Util.<BlockPos>getRandom(list, randomSource).below(2));
			return true;
		}
	}

	private static void placeSusSand(WorldGenLevel level, BlockPos pos) {
		level.setBlock(pos, Blocks.SUSPICIOUS_SAND.defaultBlockState(), 3);
		level.getBlockEntity(pos, BlockEntityType.BRUSHABLE_BLOCK)
			.ifPresent(brushableBlockEntity -> brushableBlockEntity.setLootTable(BuiltInLootTables.DESERT_WELL_ARCHAEOLOGY, pos.asLong()));
	}
}