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

package com.hypixel.hytale.server.flock.corecomponents;

import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.server.core.entity.group.EntityGroup;
import com.hypixel.hytale.server.flock.FlockMembership;
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 javax.annotation.Nonnull;
import com.hypixel.hytale.server.flock.corecomponents.builders.BuilderSensorFlockLeader;
import com.hypixel.hytale.server.npc.sensorinfo.EntityPositionProvider;
import com.hypixel.hytale.server.npc.corecomponents.SensorBase;

public class SensorFlockLeader extends SensorBase
{
    protected final EntityPositionProvider positionProvider;
    
    public SensorFlockLeader(@Nonnull final BuilderSensorFlockLeader builder) {
        super(builder);
        this.positionProvider = new EntityPositionProvider();
    }
    
    @Override
    public boolean matches(@Nonnull final Ref<EntityStore> ref, @Nonnull final Role role, final double dt, @Nonnull final Store<EntityStore> store) {
        if (!super.matches(ref, role, dt, store)) {
            this.positionProvider.clear();
            return false;
        }
        final FlockMembership membership = store.getComponent(ref, FlockMembership.getComponentType());
        if (membership == null) {
            this.positionProvider.clear();
            return false;
        }
        EntityGroup group = null;
        final Ref<EntityStore> flockReference = membership.getFlockRef();
        if (flockReference != null && flockReference.isValid()) {
            group = store.getComponent(flockReference, EntityGroup.getComponentType());
        }
        if (group == null) {
            this.positionProvider.clear();
            return false;
        }
        return this.positionProvider.setTarget(group.getLeaderRef(), store) != null;
    }
    
    @Override
    public InfoProvider getSensorInfo() {
        return this.positionProvider;
    }
}