class="kw">package com.hypixel.hytale.server.core.command.commands.world.entity.stats;

class="kw">import com.hypixel.hytale.common.util.StringUtil;
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.CommandUtil;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
class="kw">import com.hypixel.hytale.server.core.command.system.basecommands.AbstractTargetEntityCommand;
class="kw">import com.hypixel.hytale.server.core.modules.entitystats.EntityStatMap;
class="kw">import com.hypixel.hytale.server.core.modules.entitystats.EntityStatsModule;
class="kw">import com.hypixel.hytale.server.core.modules.entitystats.asset.EntityStatType;
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 java.util.List;
class="kw">import javax.annotation.Nonnull;

class="kw">public class EntityStatsAddCommand class="kw">extends AbstractTargetEntityCommand {
   @Nonnull
   class="kw">private class="kw">final RequiredArg<String> entityStatNameArg = this.withRequiredArg("statName", "server.commands.entity.stats.add.statName.desc", ArgTypes.STRING);
   @Nonnull
   class="kw">private class="kw">final RequiredArg<Integer> statAmountArg = this.withRequiredArg("statAmount", "server.commands.entity.stats.add.statAmount.desc", ArgTypes.INTEGER);

   class="kw">public EntityStatsAddCommand() {
      super("add", "server.commands.entity.stats.add.desc");
   }

   @Override
   class="kw">protected void execute(@Nonnull CommandContext var1, @Nonnull List<Ref<EntityStore>> var2, @Nonnull World var3, @Nonnull Store<EntityStore> var4) {
      int var5 = this.statAmountArg.get(var1);
      String var6 = this.entityStatNameArg.get(var1);
      addEntityStat(var1, var2, var5, var6, var4);
   }

   class="kw">public class="kw">static void addEntityStat(
      @Nonnull CommandContext var0, @Nonnull List<Ref<EntityStore>> var1, int var2, @Nonnull String var3, @Nonnull Store<EntityStore> var4
   ) {
      int var5 = EntityStatType.getAssetMap().getIndex(var3);
      if (var5 == Integer.MIN_VALUE) {
         var0.sendMessage(Message.translation("server.commands.entityStats.entityStatNotFound").param("name", var3));
         var0.sendMessage(
            Message.translation("server.general.failed.didYouMean")
               .param(
                  "choices", StringUtil.sortByFuzzyDistance(var3, EntityStatType.getAssetMap().getAssetMap().keySet(), CommandUtil.RECOMMEND_COUNT).toString()
               )
         );
      } else {
         for (Ref var7 : var1) {
            EntityStatMap var8 = var4.getComponent(var7, EntityStatsModule.get().getEntityStatMapComponentType());
            if (var8 != null) {
               if (var8.get(var5) == null) {
                  var0.sendMessage(Message.translation("server.commands.entityStats.entityStatNotFound").param("name", var3));
                  var0.sendMessage(
                     Message.translation("server.general.failed.didYouMean")
                        .param(
                           "choices",
                           StringUtil.sortByFuzzyDistance(var3, EntityStatType.getAssetMap().getAssetMap().keySet(), CommandUtil.RECOMMEND_COUNT).toString()
                        )
                  );
               } else {
                  float var9 = var8.addStatValue(var5, var2);
                  var0.sendMessage(Message.translation("server.commands.entityStats.success").param("name", var3).param("value", var9));
               }
            }
         }
      }
   }
}