package net.minecraft.world.entity.npc;

import java.util.List;
import net.minecraft.core.BlockPos;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.tags.StructureTags;
import net.minecraft.util.RandomSource;
import net.minecraft.world.entity.EntitySpawnReason;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.SpawnPlacements;
import net.minecraft.world.entity.ai.village.poi.PoiTypes;
import net.minecraft.world.entity.ai.village.poi.PoiManager.Occupancy;
import net.minecraft.world.entity.animal.Cat;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.level.CustomSpawner;
import net.minecraft.world.level.GameRules;
import net.minecraft.world.phys.AABB;

public class CatSpawner implements CustomSpawner {
	private static final int TICK_DELAY = 1200;
	private int nextTick;

	@Override
	public void tick(ServerLevel level, boolean spawnEnemies, boolean spawnFriendlies) {
		if (spawnFriendlies && level.getGameRules().getBoolean(GameRules.RULE_DOMOBSPAWNING)) {
			this.nextTick--;
			if (this.nextTick <= 0) {
				this.nextTick = 1200;
				Player player = level.getRandomPlayer();
				if (player != null) {
					RandomSource randomSource = level.random;
					int i = (8 + randomSource.nextInt(24)) * (randomSource.nextBoolean() ? -1 : 1);
					int j = (8 + randomSource.nextInt(24)) * (randomSource.nextBoolean() ? -1 : 1);
					BlockPos blockPos = player.blockPosition().offset(i, 0, j);
					int k = 10;
					if (level.hasChunksAt(blockPos.getX() - 10, blockPos.getZ() - 10, blockPos.getX() + 10, blockPos.getZ() + 10)) {
						if (SpawnPlacements.isSpawnPositionOk(EntityType.CAT, level, blockPos)) {
							if (level.isCloseToVillage(blockPos, 2)) {
								this.spawnInVillage(level, blockPos);
							} else if (level.structureManager().getStructureWithPieceAt(blockPos, StructureTags.CATS_SPAWN_IN).isValid()) {
								this.spawnInHut(level, blockPos);
							}
						}
					}
				}
			}
		}
	}

	private void spawnInVillage(ServerLevel level, BlockPos pos) {
		int i = 48;
		if (level.getPoiManager().getCountInRange(holder -> holder.is(PoiTypes.HOME), pos, 48, Occupancy.IS_OCCUPIED) > 4L) {
			List<Cat> list = level.getEntitiesOfClass(Cat.class, new AABB(pos).inflate(48.0, 8.0, 48.0));
			if (list.size() < 5) {
				this.spawnCat(pos, level, false);
			}
		}
	}

	private void spawnInHut(ServerLevel level, BlockPos pos) {
		int i = 16;
		List<Cat> list = level.getEntitiesOfClass(Cat.class, new AABB(pos).inflate(16.0, 8.0, 16.0));
		if (list.isEmpty()) {
			this.spawnCat(pos, level, true);
		}
	}

	private void spawnCat(BlockPos pos, ServerLevel level, boolean persistent) {
		Cat cat = EntityType.CAT.create(level, EntitySpawnReason.NATURAL);
		if (cat != null) {
			cat.finalizeSpawn(level, level.getCurrentDifficultyAt(pos), EntitySpawnReason.NATURAL, null);
			if (persistent) {
				cat.setPersistenceRequired();
			}

			cat.snapTo(pos, 0.0F, 0.0F);
			level.addFreshEntityWithPassengers(cat);
		}
	}
}