// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.npc.corecomponents.entity;

import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.server.npc.util.DamageData;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.npc.role.Role;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.npc.corecomponents.builders.BuilderSensorBase;
import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.npc.corecomponents.entity.builders.BuilderSensorKill;
import com.hypixel.hytale.server.npc.sensorinfo.PositionProvider;
import com.hypixel.hytale.server.npc.corecomponents.SensorBase;

public class SensorKill extends SensorBase
{
    protected final int targetSlot;
    protected final PositionProvider positionProvider;
    
    public SensorKill(@Nonnull final BuilderSensorKill builder, @Nonnull final BuilderSupport support) {
        super(builder);
        this.positionProvider = new PositionProvider();
        this.targetSlot = builder.getTargetSlot(support);
    }
    
    @Override
    public boolean matches(@Nonnull final Ref<EntityStore> ref, @Nonnull final Role role, final double dt, @Nonnull final Store<EntityStore> store) {
        final NPCEntity npcComponent = store.getComponent(ref, NPCEntity.getComponentType());
        assert npcComponent != null;
        final DamageData damageData = npcComponent.getDamageData();
        if (!super.matches(ref, role, dt, store) || !damageData.haveKill()) {
            this.positionProvider.clear();
            return false;
        }
        Ref<EntityStore> targetRef;
        if (this.targetSlot >= 0) {
            targetRef = role.getMarkedEntitySupport().getMarkedEntityRef(this.targetSlot);
            if (targetRef == null || !damageData.haveKilled(targetRef)) {
                this.positionProvider.clear();
                return false;
            }
        }
        else {
            targetRef = damageData.getAnyKilled();
        }
        if (targetRef == null) {
            this.positionProvider.clear();
            return false;
        }
        final Vector3d killPosition = damageData.getKillPosition(targetRef);
        if (killPosition == null) {
            this.positionProvider.clear();
            return false;
        }
        this.positionProvider.setTarget(killPosition);
        return true;
    }
    
    @Override
    public InfoProvider getSensorInfo() {
        return this.positionProvider;
    }
}