net.minestom.server.entity.ai.target.LastEntityDamagerTarget Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of minestom-snapshots Show documentation

1.20.4 Lightweight Minecraft server

There is a newer version: 7320437640

package net.minestom.server.entity.ai.target;

import net.minestom.server.entity.Entity;
import net.minestom.server.entity.EntityCreature;
import net.minestom.server.entity.ai.TargetSelector;
import net.minestom.server.entity.damage.Damage;
import net.minestom.server.entity.damage.DamageType;
import net.minestom.server.entity.damage.EntityDamage;
import org.jetbrains.annotations.NotNull;

/**
 * Targets the last damager of this entity.
 */
public class LastEntityDamagerTarget extends TargetSelector {

    private final float range;

    public LastEntityDamagerTarget(@NotNull EntityCreature entityCreature, float range) {
        super(entityCreature);
        this.range = range;
    }

    @Override
    public Entity findTarget() {
        final Damage damage = entityCreature.getLastDamageSource();
        if (!(damage instanceof EntityDamage entityDamage)) {
            // No damager recorded, return null
            return null;
        }
        final Entity entity = entityDamage.getSource();
        if (entity.isRemoved()) {
            // Entity not valid
            return null;
        }
        // Check range
        return entityCreature.getDistanceSquared(entity) < range * range ? entity : null;
    }
}