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 EntityStatsResetCommand class="kw">extends AbstractTargetEntityCommand {
   @Nonnull
   class="kw">private class="kw">final RequiredArg<String> entityStatNameArg = this.withRequiredArg("statName", "server.commands.entity.stats.reset.statName.desc", ArgTypes.STRING);

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

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

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