IsInStateCondition.java

package com.hypixel.hytale.server.npc.decisionmaker.core.conditions;

import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.codec.validation.Validators;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.npc.decisionmaker.core.EvaluationContext;
import com.hypixel.hytale.server.npc.decisionmaker.core.conditions.base.SimpleCondition;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import com.hypixel.hytale.server.npc.role.support.StateSupport;
import javax.annotation.Nonnull;

public class IsInStateCondition extends SimpleCondition {
   public static final BuilderCodec<IsInStateCondition> CODEC;
   protected String state;
   protected String subState;

   protected IsInStateCondition() {
   }

   public String getState() {
      return this.state;
   }

   protected boolean evaluate(int selfIndex, @Nonnull ArchetypeChunk<EntityStore> archetypeChunk, Ref<EntityStore> target, CommandBuffer<EntityStore> commandBuffer, EvaluationContext context) {
      NPCEntity npcComponent = (NPCEntity)archetypeChunk.getComponent(selfIndex, NPCEntity.getComponentType());

      assert npcComponent != null;

      StateSupport stateSupport = npcComponent.getRole().getStateSupport();
      return stateSupport.inState(this.state, this.subState);
   }

   @Nonnull
   public String toString() {
      String var10000 = this.state;
      return "IsInStateCondition{state=" + var10000 + ", subState=" + this.subState + "} " + super.toString();
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(IsInStateCondition.class, IsInStateCondition::new, ABSTRACT_CODEC).documentation("A simple boolean condition that returns whether the NPC is in a given state.")).appendInherited(new KeyedCodec("State", Codec.STRING), (condition, s) -> condition.state = s, (condition) -> condition.state, (condition, parent) -> condition.state = parent.state).documentation("The main state to evaluate.").addValidator(Validators.nonNull()).addValidator(Validators.nonEmptyString()).add()).appendInherited(new KeyedCodec("SubState", Codec.STRING), (condition, s) -> condition.subState = s, (condition) -> condition.subState, (condition, parent) -> condition.subState = parent.subState).documentation("The optional substate to evaluate.").add()).build();
   }
}