ActionFlockSetTarget.java

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

import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.flock.FlockPlugin;
import com.hypixel.hytale.server.flock.corecomponents.builders.BuilderActionFlockSetTarget;
import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
import com.hypixel.hytale.server.npc.corecomponents.ActionBase;
import com.hypixel.hytale.server.npc.role.Role;
import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class ActionFlockSetTarget extends ActionBase {
   protected final boolean clear;
   protected final String targetSlot;

   public ActionFlockSetTarget(@Nonnull BuilderActionFlockSetTarget builderActionFlockSetTarget, @Nonnull BuilderSupport support) {
      super(builderActionFlockSetTarget);
      this.clear = builderActionFlockSetTarget.isClear();
      this.targetSlot = builderActionFlockSetTarget.getTargetSlot(support);
   }

   public boolean canExecute(@Nonnull Ref<EntityStore> ref, @Nonnull Role role, @Nullable InfoProvider sensorInfo, double dt, @Nonnull Store<EntityStore> store) {
      if (super.canExecute(ref, role, sensorInfo, dt, store) && FlockPlugin.isFlockMember(ref, store)) {
         if (this.clear) {
            return true;
         } else {
            Ref<EntityStore> target = sensorInfo != null && sensorInfo.hasPosition() ? sensorInfo.getPositionProvider().getTarget() : null;
            return target != null;
         }
      } else {
         return false;
      }
   }

   public boolean execute(@Nonnull Ref<EntityStore> ref, @Nonnull Role role, @Nonnull InfoProvider sensorInfo, double dt, @Nonnull Store<EntityStore> store) {
      super.execute(ref, role, sensorInfo, dt, store);
      if (this.clear) {
         role.getMarkedEntitySupport().flockSetTarget(this.targetSlot, (Ref)null, store);
         return true;
      } else {
         Ref<EntityStore> targetRef = sensorInfo.getPositionProvider().getTarget();
         role.getMarkedEntitySupport().flockSetTarget(this.targetSlot, targetRef, store);
         return true;
      }
   }
}