class="kw">package com.hypixel.hytale.server.flock.corecomponents.builders;

class="kw">import com.google.gson.JsonElement;
class="kw">import com.hypixel.hytale.server.flock.corecomponents.EntityFilterFlock;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.Builder;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderDescriptorState;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.InstructionType;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.IEntityFilter;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.builders.BuilderEntityFilterBase;
class="kw">import com.hypixel.hytale.server.npc.movement.FlockMembershipType;
class="kw">import com.hypixel.hytale.server.npc.movement.FlockPlayerMembership;
class="kw">import javax.annotation.Nonnull;

class="kw">public class BuilderEntityFilterFlock class="kw">extends BuilderEntityFilterBase {
   class="kw">protected FlockMembershipType flockMembership;
   class="kw">protected FlockPlayerMembership flockPlayerMembership;
   class="kw">protected int[] size;
   class="kw">protected boolean checkCanJoin;

   class="kw">public BuilderEntityFilterFlock() {
   }

   @Nonnull
   class="kw">public IEntityFilter build(BuilderSupport var1) {
      class="kw">return new EntityFilterFlock(this);
   }

   @Nonnull
   @Override
   class="kw">public String getShortDescription() {
      class="kw">return "Test for flock membership and related properties";
   }

   @Nonnull
   @Override
   class="kw">public String getLongDescription() {
      class="kw">return this.getShortDescription();
   }

   @Nonnull
   @Override
   class="kw">public BuilderDescriptorState getBuilderDescriptorState() {
      class="kw">return BuilderDescriptorState.Stable;
   }

   @Nonnull
   @Override
   class="kw">public Builder<IEntityFilter> readConfig(@Nonnull JsonElement var1) {
      this.getEnum(
         var1,
         "FlockStatus",
         var1x -> this.flockMembership = var1x,
         FlockMembershipType.class,
         FlockMembershipType.Any,
         BuilderDescriptorState.Stable,
         "Test for NPC status in relation to flock",
         null
      );
      this.getEnum(
         var1,
         "FlockPlayerStatus",
         var1x -> this.flockPlayerMembership = var1x,
         FlockPlayerMembership.class,
         FlockPlayerMembership.Any,
         BuilderDescriptorState.Stable,
         "Test for Player status for flock NPC is member",
         null
      );
      this.getIntRange(
         var1, "Size", var1x -> this.size = var1x, null, null, BuilderDescriptorState.Stable, "Check for a certain range of NPCs in the flock", null
      );
      this.getBoolean(
         var1,
         "CheckCanJoin",
         var1x -> this.checkCanJoin = var1x,
         false,
         BuilderDescriptorState.Stable,
         "If true, will filter entities in a flock the executor can join",
         null
      );
      this.requireInstructionType(InstructionType.NPCOnlyInstructions);
      class="kw">return this;
   }

   class="kw">public int[] getSize() {
      class="kw">return this.size;
   }

   class="kw">public FlockMembershipType getFlockMembership() {
      class="kw">return this.flockMembership;
   }

   class="kw">public FlockPlayerMembership getFlockPlayerMembership() {
      class="kw">return this.flockPlayerMembership;
   }

   class="kw">public boolean isCheckCanJoin() {
      class="kw">return this.checkCanJoin;
   }
}