package net.minecraft.world.level.levelgen.structure.structures;

import com.mojang.serialization.MapCodec;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import java.util.Set;
import net.minecraft.Util;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Vec3i;
import net.minecraft.util.RandomSource;
import net.minecraft.util.SortedArraySet;
import net.minecraft.world.level.ChunkPos;
import net.minecraft.world.level.StructureManager;
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.chunk.ChunkGenerator;
import net.minecraft.world.level.levelgen.structure.BoundingBox;
import net.minecraft.world.level.levelgen.structure.SinglePieceStructure;
import net.minecraft.world.level.levelgen.structure.Structure;
import net.minecraft.world.level.levelgen.structure.StructurePiece;
import net.minecraft.world.level.levelgen.structure.StructureType;
import net.minecraft.world.level.levelgen.structure.pieces.PiecesContainer;
import net.minecraft.world.level.storage.loot.BuiltInLootTables;

public class DesertPyramidStructure extends SinglePieceStructure {
	public static final MapCodec<DesertPyramidStructure> CODEC = simpleCodec(DesertPyramidStructure::new);

	public DesertPyramidStructure(Structure.StructureSettings structureSettings) {
		super(DesertPyramidPiece::new, 21, 21, structureSettings);
	}

	@Override
	public void afterPlace(
		WorldGenLevel level,
		StructureManager structureManager,
		ChunkGenerator chunkGenerator,
		RandomSource random,
		BoundingBox boundingBox,
		ChunkPos chunkPos,
		PiecesContainer pieces
	) {
		Set<BlockPos> set = SortedArraySet.<BlockPos>create(Vec3i::compareTo);

		for (StructurePiece structurePiece : pieces.pieces()) {
			if (structurePiece instanceof DesertPyramidPiece desertPyramidPiece) {
				set.addAll(desertPyramidPiece.getPotentialSuspiciousSandWorldPositions());
				placeSuspiciousSand(boundingBox, level, desertPyramidPiece.getRandomCollapsedRoofPos());
			}
		}

		ObjectArrayList<BlockPos> objectArrayList = new ObjectArrayList<>(set.stream().toList());
		RandomSource randomSource = RandomSource.create(level.getSeed()).forkPositional().at(pieces.calculateBoundingBox().getCenter());
		Util.shuffle(objectArrayList, randomSource);
		int i = Math.min(set.size(), randomSource.nextInt(5, 8));

		for (BlockPos blockPos : objectArrayList) {
			if (i > 0) {
				i--;
				placeSuspiciousSand(boundingBox, level, blockPos);
			} else if (boundingBox.isInside(blockPos)) {
				level.setBlock(blockPos, Blocks.SAND.defaultBlockState(), 2);
			}
		}
	}

	private static void placeSuspiciousSand(BoundingBox boundingBox, WorldGenLevel worldGenLevel, BlockPos pos) {
		if (boundingBox.isInside(pos)) {
			worldGenLevel.setBlock(pos, Blocks.SUSPICIOUS_SAND.defaultBlockState(), 2);
			worldGenLevel.getBlockEntity(pos, BlockEntityType.BRUSHABLE_BLOCK)
				.ifPresent(brushableBlockEntity -> brushableBlockEntity.setLootTable(BuiltInLootTables.DESERT_PYRAMID_ARCHAEOLOGY, pos.asLong()));
		}
	}

	@Override
	public StructureType<?> type() {
		return StructureType.DESERT_PYRAMID;
	}
}