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

import java.util.Optional;
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.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.memory.NearestVisibleLivingEntities;

public class SetLookAndInteract {
	public static BehaviorControl<LivingEntity> create(EntityType<?> entityType, int maxDist) {
		int i = maxDist * maxDist;
		return BehaviorBuilder.create(
			instance -> instance.group(
					instance.registered(MemoryModuleType.LOOK_TARGET),
					instance.absent(MemoryModuleType.INTERACTION_TARGET),
					instance.present(MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES)
				)
				.apply(
					instance,
					(memoryAccessor, memoryAccessor2, memoryAccessor3) -> (serverLevel, livingEntity, l) -> {
						Optional<LivingEntity> optional = instance.<NearestVisibleLivingEntities>get(memoryAccessor3)
							.findClosest(livingEntity2x -> livingEntity2x.distanceToSqr(livingEntity) <= i && entityType.equals(livingEntity2x.getType()));
						if (optional.isEmpty()) {
							return false;
						} else {
							LivingEntity livingEntity2 = (LivingEntity)optional.get();
							memoryAccessor2.set(livingEntity2);
							memoryAccessor.set(new EntityTracker(livingEntity2, true));
							return true;
						}
					}
				)
		);
	}
}