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

import com.google.common.collect.ImmutableMap;
import net.minecraft.server.level.WorldServer;
import net.minecraft.world.EnumHand;
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;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemProjectileWeapon;

public class BehaviorAttack extends Behavior<EntityInsentient> {

    private final int cooldownBetweenAttacks;

    public BehaviorAttack(int i) {
        super(ImmutableMap.of(MemoryModuleType.LOOK_TARGET, MemoryStatus.REGISTERED, MemoryModuleType.ATTACK_TARGET, MemoryStatus.VALUE_PRESENT, MemoryModuleType.ATTACK_COOLING_DOWN, MemoryStatus.VALUE_ABSENT));
        this.cooldownBetweenAttacks = i;
    }

    protected boolean a(WorldServer worldserver, EntityInsentient entityinsentient) {
        EntityLiving entityliving = this.b(entityinsentient);

        return !this.a(entityinsentient) && BehaviorUtil.b((EntityLiving) entityinsentient, entityliving) && BehaviorUtil.a(entityinsentient, entityliving);
    }

    private boolean a(EntityInsentient entityinsentient) {
        return entityinsentient.b((itemstack) -> {
            Item item = itemstack.getItem();

            return item instanceof ItemProjectileWeapon && entityinsentient.a((ItemProjectileWeapon) item);
        });
    }

    protected void a(WorldServer worldserver, EntityInsentient entityinsentient, long i) {
        EntityLiving entityliving = this.b(entityinsentient);

        BehaviorUtil.a((EntityLiving) entityinsentient, entityliving);
        entityinsentient.swingHand(EnumHand.MAIN_HAND);
        entityinsentient.attackEntity(entityliving);
        entityinsentient.getBehaviorController().a(MemoryModuleType.ATTACK_COOLING_DOWN, true, (long) this.cooldownBetweenAttacks);
    }

    private EntityLiving b(EntityInsentient entityinsentient) {
        return (EntityLiving) entityinsentient.getBehaviorController().getMemory(MemoryModuleType.ATTACK_TARGET).get();
    }
}