class="kw">package com.hypixel.hytale.builtin.adventure.memories.commands;

class="kw">import com.hypixel.hytale.builtin.adventure.memories.component.PlayerMemories;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.protocol.packets.player.UpdateMemoriesFeatureStatus;
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.command.system.basecommands.AbstractPlayerCommand;
class="kw">import com.hypixel.hytale.server.core.io.PacketHandler;
class="kw">import com.hypixel.hytale.server.core.universe.PlayerRef;
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 javax.annotation.Nonnull;

class="kw">public class MemoriesCapacityCommand class="kw">extends AbstractPlayerCommand {
   @Nonnull
   class="kw">private class="kw">final RequiredArg<Integer> capacityArg = this.withRequiredArg("capacity", "server.commands.memories.capacity.capacity.desc", ArgTypes.INTEGER);

   class="kw">public MemoriesCapacityCommand() {
      super("capacity", "server.commands.memories.capacity.desc");
   }

   @Override
   class="kw">protected void execute(
      @Nonnull CommandContext var1, @Nonnull Store<EntityStore> var2, @Nonnull Ref<EntityStore> var3, @Nonnull PlayerRef var4, @Nonnull World var5
   ) {
      Integer var6 = this.capacityArg.get(var1);
      PacketHandler var7 = var4.getPacketHandler();
      if (var6 <= 0) {
         var2.tryRemoveComponent(var3, PlayerMemories.getComponentType());
         var1.sendMessage(Message.translation("server.commands.memories.capacity.success").param("capacity", 0));
         var7.writeNoCache(new UpdateMemoriesFeatureStatus(false));
      } else {
         PlayerMemories var8 = var2.ensureAndGetComponent(var3, PlayerMemories.getComponentType());
         var8.setMemoriesCapacity(var6);
         var1.sendMessage(Message.translation("server.commands.memories.capacity.success").param("capacity", var6));
         var7.writeNoCache(new UpdateMemoriesFeatureStatus(true));
      }
   }
}