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

import java.util.Optional;
import java.util.function.Function;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.Mob;
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.memory.NearestVisibleLivingEntities;
import net.minecraft.world.entity.ai.memory.WalkTarget;

public class SetWalkTargetFromAttackTargetIfTargetOutOfReach {
    private static final int PROJECTILE_ATTACK_RANGE_BUFFER = 1;

    public static BehaviorControl<Mob> create(float p_259228_) {
        return create(p_147908_ -> p_259228_);
    }

    public static BehaviorControl<Mob> create(Function<LivingEntity, Float> p_259507_) {
        return BehaviorBuilder.create(
            p_258687_ -> p_258687_.group(
                    p_258687_.registered(MemoryModuleType.WALK_TARGET),
                    p_258687_.registered(MemoryModuleType.LOOK_TARGET),
                    p_258687_.present(MemoryModuleType.ATTACK_TARGET),
                    p_258687_.registered(MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES)
                )
                .apply(p_258687_, (p_258699_, p_258700_, p_258701_, p_258702_) -> (p_258694_, p_258695_, p_258696_) -> {
                    LivingEntity livingentity = p_258687_.get(p_258701_);
                    Optional<NearestVisibleLivingEntities> optional = p_258687_.tryGet(p_258702_);
                    if (optional.isPresent() && optional.get().contains(livingentity) && BehaviorUtils.isWithinAttackRange(p_258695_, livingentity, 1)) {
                        p_258699_.erase();
                    } else {
                        p_258700_.set(new EntityTracker(livingentity, true));
                        p_258699_.set(new WalkTarget(new EntityTracker(livingentity, false), p_259507_.apply(p_258695_), 0));
                    }

                    return true;
                })
        );
    }
}