package net.minecraft.world.entity.ai.behavior;

import java.util.List;
import java.util.function.Predicate;
import net.minecraft.core.BlockPos;
import net.minecraft.core.GlobalPos;
import net.minecraft.core.Holder;
import net.minecraft.network.protocol.game.DebugPackets;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.tags.BlockTags;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.ai.behavior.declarative.BehaviorBuilder;
import net.minecraft.world.entity.ai.behavior.declarative.MemoryAccessor;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.village.poi.PoiType;
import net.minecraft.world.entity.npc.Villager;
import net.minecraft.world.level.block.BedBlock;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.phys.AABB;

public class ValidateNearbyPoi {
    private static final int MAX_DISTANCE = 16;

    public static BehaviorControl<LivingEntity> create(Predicate<Holder<PoiType>> p_259460_, MemoryModuleType<GlobalPos> p_259635_) {
        return BehaviorBuilder.create(
            p_259215_ -> p_259215_.group(p_259215_.present(p_259635_)).apply(p_259215_, p_259498_ -> (p_375053_, p_375054_, p_375055_) -> {
                GlobalPos globalpos = p_259215_.get(p_259498_);
                BlockPos blockpos = globalpos.pos();
                if (p_375053_.dimension() == globalpos.dimension() && blockpos.closerToCenterThan(p_375054_.position(), 16.0)) {
                    ServerLevel serverlevel = p_375053_.getServer().getLevel(globalpos.dimension());
                    if (serverlevel == null || !serverlevel.getPoiManager().exists(blockpos, p_259460_)) {
                        p_259498_.erase();
                    } else if (bedIsOccupied(serverlevel, blockpos, p_375054_)) {
                        p_259498_.erase();
                        if (!bedIsOccupiedByVillager(serverlevel, blockpos)) {
                            p_375053_.getPoiManager().release(blockpos);
                            DebugPackets.sendPoiTicketCountPacket(p_375053_, blockpos);
                        }
                    }

                    return true;
                } else {
                    return false;
                }
            })
        );
    }

    private static boolean bedIsOccupied(ServerLevel p_24531_, BlockPos p_24532_, LivingEntity p_24533_) {
        BlockState blockstate = p_24531_.getBlockState(p_24532_);
        return blockstate.is(BlockTags.BEDS) && blockstate.getValue(BedBlock.OCCUPIED) && !p_24533_.isSleeping();
    }

    private static boolean bedIsOccupiedByVillager(ServerLevel p_377892_, BlockPos p_376377_) {
        List<Villager> list = p_377892_.getEntitiesOfClass(Villager.class, new AABB(p_376377_), LivingEntity::isSleeping);
        return !list.isEmpty();
    }
}