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

class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.Message;
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.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.modules.entity.component.Interactable;
class="kw">import com.hypixel.hytale.server.core.modules.entity.item.ItemComponent;
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.entities.NPCEntity;
class="kw">import it.unimi.dsi.fastutil.Pair;
class="kw">import javax.annotation.Nonnull;

class="kw">public class NPCFreezeCommand class="kw">extends AbstractWorldCommand {
   @Nonnull
   class="kw">private class="kw">final FlagArg allArg = this.withFlagArg("all", "server.commands.npc.freeze.all");
   @Nonnull
   class="kw">private class="kw">final FlagArg toggleArg = this.withFlagArg("toggle", "server.commands.npc.freeze.toggle");
   @Nonnull
   class="kw">private class="kw">final EntityWrappedArg entityArg = this.withOptionalArg("entity", "server.commands.entity.entity.desc", ArgTypes.ENTITY_ID);

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

   @Override
   class="kw">protected void execute(@Nonnull CommandContext var1, @Nonnull World var2, @Nonnull Store<EntityStore> var3) {
      if (this.allArg.get(var1)) {
         var3.forEachEntityParallel(
            NPCEntity.getComponentType(), (var0, var1x, var2x) -> var2x.ensureComponent(var1x.getReferenceTo(var0), Frozen.getComponentType())
         );
         var3.forEachEntityParallel(ItemComponent.getComponentType(), (var0, var1x, var2x) -> {
            Ref var3 = var1x.getReferenceTo(var0);
            var2x.ensureComponent(var3, Frozen.getComponentType());
            var2x.ensureComponent(var3, Interactable.getComponentType());
         });
      } else {
         Pair var4 = NPCCommandUtils.getTargetNpc(var1, this.entityArg, var3);
         if (var4 != null) {
            Ref var5 = (Ref<EntityStore>)var4.first();
            String var6 = ((NPCEntity)var4.second()).getRoleName();
            if (this.toggleArg.get(var1)) {
               boolean var7 = var3.getArchetype(var5).contains(Frozen.getComponentType());
               if (var7) {
                  var3.tryRemoveComponent(var5, Frozen.getComponentType());
                  var1.sendMessage(Message.translation("server.commands.npc.thaw.npc").param("role", var6));
               } else {
                  var3.ensureComponent(var5, Frozen.getComponentType());
                  var1.sendMessage(Message.translation("server.commands.npc.freeze.npc").param("role", var6));
               }
            } else {
               var3.ensureComponent(var5, Frozen.getComponentType());
               var1.sendMessage(Message.translation("server.commands.npc.freeze.npc").param("role", var6));
            }
         }
      }
   }
}