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

import java.util.List;
import java.util.Optional;
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.world.entity.LivingEntity;
import net.minecraft.world.entity.PathfinderMob;
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.entity.npc.VillagerProfession;
import net.minecraft.world.level.pathfinder.Path;

public class YieldJobSite {
    public static BehaviorControl<Villager> create(float p_259768_) {
        return BehaviorBuilder.create(
            p_258916_ -> p_258916_.group(
                    p_258916_.present(MemoryModuleType.POTENTIAL_JOB_SITE),
                    p_258916_.absent(MemoryModuleType.JOB_SITE),
                    p_258916_.present(MemoryModuleType.NEAREST_LIVING_ENTITIES),
                    p_258916_.registered(MemoryModuleType.WALK_TARGET),
                    p_258916_.registered(MemoryModuleType.LOOK_TARGET)
                )
                .apply(
                    p_258916_,
                    (p_258901_, p_258902_, p_258903_, p_258904_, p_258905_) -> (p_258912_, p_258913_, p_258914_) -> {
                        if (p_258913_.isBaby()) {
                            return false;
                        } else if (!p_258913_.getVillagerData().profession().is(VillagerProfession.NONE)) {
                            return false;
                        } else {
                            BlockPos blockpos = p_258916_.<GlobalPos>get(p_258901_).pos();
                            Optional<Holder<PoiType>> optional = p_258912_.getPoiManager().getType(blockpos);
                            if (optional.isEmpty()) {
                                return true;
                            } else {
                                p_258916_.<List<LivingEntity>>get(p_258903_)
                                    .stream()
                                    .filter(p_258898_ -> p_258898_ instanceof Villager && p_258898_ != p_258913_)
                                    .map(p_258896_ -> (Villager)p_258896_)
                                    .filter(LivingEntity::isAlive)
                                    .filter(p_258919_ -> nearbyWantsJobsite(optional.get(), p_258919_, blockpos))
                                    .findFirst()
                                    .ifPresent(p_390624_ -> {
                                        p_258904_.erase();
                                        p_258905_.erase();
                                        p_258901_.erase();
                                        if (p_390624_.getBrain().getMemory(MemoryModuleType.JOB_SITE).isEmpty()) {
                                            BehaviorUtils.setWalkAndLookTargetMemories(p_390624_, blockpos, p_259768_, 1);
                                            p_390624_.getBrain().setMemory(MemoryModuleType.POTENTIAL_JOB_SITE, GlobalPos.of(p_258912_.dimension(), blockpos));
                                            DebugPackets.sendPoiTicketCountPacket(p_258912_, blockpos);
                                        }
                                    });
                                return true;
                            }
                        }
                    }
                )
        );
    }

    private static boolean nearbyWantsJobsite(Holder<PoiType> p_217511_, Villager p_217512_, BlockPos p_217513_) {
        boolean flag = p_217512_.getBrain().getMemory(MemoryModuleType.POTENTIAL_JOB_SITE).isPresent();
        if (flag) {
            return false;
        } else {
            Optional<GlobalPos> optional = p_217512_.getBrain().getMemory(MemoryModuleType.JOB_SITE);
            Holder<VillagerProfession> holder = p_217512_.getVillagerData().profession();
            if (holder.value().heldJobSite().test(p_217511_)) {
                return optional.isEmpty() ? canReachPos(p_217512_, p_217513_, p_217511_.value()) : optional.get().pos().equals(p_217513_);
            } else {
                return false;
            }
        }
    }

    private static boolean canReachPos(PathfinderMob p_260080_, BlockPos p_259875_, PoiType p_259606_) {
        Path path = p_260080_.getNavigation().createPath(p_259875_, p_259606_.validRange());
        return path != null && path.canReach();
    }
}