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

import com.google.common.collect.ImmutableMap;
import java.util.List;
import net.minecraft.server.level.WorldServer;
import net.minecraft.util.MathHelper;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.EntityInsentient;
import net.minecraft.world.entity.EntityLiving;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.memory.MemoryStatus;

public class BehaviorRetreat<E extends EntityInsentient> extends Behavior<E> {

    private final int tooCloseDistance;
    private final float strafeSpeed;

    public BehaviorRetreat(int i, float f) {
        super(ImmutableMap.of(MemoryModuleType.WALK_TARGET, MemoryStatus.VALUE_ABSENT, MemoryModuleType.LOOK_TARGET, MemoryStatus.REGISTERED, MemoryModuleType.ATTACK_TARGET, MemoryStatus.VALUE_PRESENT, MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES, MemoryStatus.VALUE_PRESENT));
        this.tooCloseDistance = i;
        this.strafeSpeed = f;
    }

    protected boolean a(WorldServer worldserver, E e0) {
        return this.a(e0) && this.b(e0);
    }

    protected void a(WorldServer worldserver, E e0, long i) {
        e0.getBehaviorController().setMemory(MemoryModuleType.LOOK_TARGET, (Object) (new BehaviorPositionEntity(this.c(e0), true)));
        e0.getControllerMove().a(-this.strafeSpeed, 0.0F);
        e0.setYRot(MathHelper.c(e0.getYRot(), e0.yHeadRot, 0.0F));
    }

    private boolean a(E e0) {
        return ((List) e0.getBehaviorController().getMemory(MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES).get()).contains(this.c(e0));
    }

    private boolean b(E e0) {
        return this.c(e0).a((Entity) e0, (double) this.tooCloseDistance);
    }

    private EntityLiving c(E e0) {
        return (EntityLiving) e0.getBehaviorController().getMemory(MemoryModuleType.ATTACK_TARGET).get();
    }
}