class="kw">package com.hypixel.hytale.server.core.modules.prefabspawner.commands;

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.RequiredArg;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
class="kw">import com.hypixel.hytale.server.core.modules.block.BlockModule;
class="kw">import com.hypixel.hytale.server.core.modules.prefabspawner.PrefabSpawnerBlock;
class="kw">import com.hypixel.hytale.server.core.prefab.PrefabWeights;
class="kw">import javax.annotation.Nonnull;

class="kw">public class PrefabSpawnerWeightCommand class="kw">extends TargetPrefabSpawnerCommand {
   @Nonnull
   class="kw">private class="kw">final RequiredArg<String> prefabArg = this.withRequiredArg("prefab", "server.commands.prefabspawner.weight.prefab.desc", ArgTypes.STRING);
   @Nonnull
   class="kw">private class="kw">final RequiredArg<Float> weightArg = this.withRequiredArg("weight", "server.commands.prefabspawner.weight.weight.desc", ArgTypes.FLOAT);

   class="kw">public PrefabSpawnerWeightCommand() {
      super("weight", "server.commands.prefabspawner.weight.desc");
   }

   @Override
   class="kw">protected void execute(@Nonnull CommandContext var1, @Nonnull PrefabSpawnerBlock var2, @Nonnull BlockModule.BlockStateInfo var3) {
      String var4 = this.prefabArg.get(var1);
      Float var5 = this.weightArg.get(var1);
      PrefabWeights var6 = var2.getPrefabWeights();
      if (var6 == PrefabWeights.NONE) {
         var6 = new PrefabWeights();
      }

      if (var5 < 0.0F) {
         var6.removeWeight(var4);
         var1.sendMessage(Message.translation("server.commands.prefabspawner.weight.remove").param("prefab", var4));
      } else {
         var6.setWeight(var4, var5.floatValue());
         var1.sendMessage(Message.translation("server.commands.prefabspawner.weight.set").param("prefab", var4).param("weight", var5));
      }

      var2.setPrefabWeights(var6);
      var3.markNeedsSaving();
   }
}