class="kw">package com.hypixel.hytale.server.npc.commands;

class="kw">import com.hypixel.hytale.codec.validation.Validators;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.command.system.CommandContext;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.system.FlagArg;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.system.OptionalArg;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.types.EntityWrappedArg;
class="kw">import com.hypixel.hytale.server.core.command.system.basecommands.AbstractWorldCommand;
class="kw">import com.hypixel.hytale.server.core.entity.Frozen;
class="kw">import com.hypixel.hytale.server.core.universe.world.World;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.npc.components.StepComponent;
class="kw">import com.hypixel.hytale.server.npc.entities.NPCEntity;
class="kw">import it.unimi.dsi.fastutil.Pair;
class="kw">import javax.annotation.Nonnull;

class="kw">public class NPCStepCommand class="kw">extends AbstractWorldCommand {
   @Nonnull
   class="kw">private class="kw">final FlagArg allArg = this.withFlagArg("all", "server.commands.npc.step.all");
   @Nonnull
   class="kw">private class="kw">final EntityWrappedArg entityArg = this.withOptionalArg("entity", "server.commands.entity.entity.desc", ArgTypes.ENTITY_ID);
   @Nonnull
   class="kw">private class="kw">final OptionalArg<Float> dtArg = this.withOptionalArg("dt", "server.commands.npc.step.dt.desc", ArgTypes.FLOAT)
      .addValidator(Validators.greaterThan(0.0F));

   class="kw">public NPCStepCommand() {
      super("step", "server.commands.npc.step.desc");
   }

   @Override
   class="kw">protected void execute(@Nonnull CommandContext var1, @Nonnull World var2, @Nonnull Store<EntityStore> var3) {
      float var4 = this.dtArg.provided(var1) ? this.dtArg.get(var1) : 1.0F / var2.getTps();
      if (this.allArg.get(var1)) {
         var3.forEachEntityParallel(NPCEntity.getComponentType(), (var1x, var2x, var3x) -> {
            var3x.ensureComponent(var2x.getReferenceTo(var1x), Frozen.getComponentType());
            var3x.addComponent(var2x.getReferenceTo(var1x), StepComponent.getComponentType(), new StepComponent(var4));
         });
      } else {
         Pair var5 = NPCCommandUtils.getTargetNpc(var1, this.entityArg, var3);
         if (var5 != null) {
            Ref var6 = (Ref<EntityStore>)var5.first();
            var3.ensureComponent(var6, Frozen.getComponentType());
            var3.addComponent(var6, StepComponent.getComponentType(), new StepComponent(var4));
         }
      }
   }
}