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

import java.util.Optional;
import java.util.function.Predicate;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.RandomSource;
import net.minecraft.util.valueproviders.UniformInt;
import net.minecraft.world.entity.EntityType;
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.memory.NearestVisibleLivingEntities;

@Deprecated
public class SetEntityLookTargetSometimes {
    public static BehaviorControl<LivingEntity> create(float p_259047_, UniformInt p_260065_) {
        return create(p_259047_, p_260065_, p_259715_ -> true);
    }

    public static BehaviorControl<LivingEntity> create(EntityType<?> p_259588_, float p_259105_, UniformInt p_259422_) {
        return create(p_259105_, p_259422_, p_390583_ -> p_259588_.equals(p_390583_.getType()));
    }

    private static BehaviorControl<LivingEntity> create(float p_260336_, UniformInt p_259266_, Predicate<LivingEntity> p_260173_) {
        float f = p_260336_ * p_260336_;
        SetEntityLookTargetSometimes.Ticker setentitylooktargetsometimes$ticker = new SetEntityLookTargetSometimes.Ticker(p_259266_);
        return BehaviorBuilder.create(
            p_259288_ -> p_259288_.group(p_259288_.absent(MemoryModuleType.LOOK_TARGET), p_259288_.present(MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES))
                .apply(
                    p_259288_,
                    (p_259350_, p_260134_) -> (p_264952_, p_264953_, p_264954_) -> {
                        Optional<LivingEntity> optional = p_259288_.<NearestVisibleLivingEntities>get(p_260134_)
                            .findClosest(p_260173_.and(p_326890_ -> p_326890_.distanceToSqr(p_264953_) <= f));
                        if (optional.isEmpty()) {
                            return false;
                        } else if (!setentitylooktargetsometimes$ticker.tickDownAndCheck(p_264952_.random)) {
                            return false;
                        } else {
                            p_259350_.set(new EntityTracker(optional.get(), true));
                            return true;
                        }
                    }
                )
        );
    }

    public static final class Ticker {
        private final UniformInt interval;
        private int ticksUntilNextStart;

        public Ticker(UniformInt p_260279_) {
            if (p_260279_.getMinValue() <= 1) {
                throw new IllegalArgumentException();
            } else {
                this.interval = p_260279_;
            }
        }

        public boolean tickDownAndCheck(RandomSource p_260276_) {
            if (this.ticksUntilNextStart == 0) {
                this.ticksUntilNextStart = this.interval.sample(p_260276_) - 1;
                return false;
            } else {
                return --this.ticksUntilNextStart == 0;
            }
        }
    }
}