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

class="kw">import com.hypixel.hytale.builtin.adventure.memories.component.PlayerMemories;
class="kw">import com.hypixel.hytale.codec.Codec;
class="kw">import com.hypixel.hytale.codec.KeyedCodec;
class="kw">import com.hypixel.hytale.codec.builder.BuilderCodec;
class="kw">import com.hypixel.hytale.component.CommandBuffer;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.protocol.InteractionState;
class="kw">import com.hypixel.hytale.protocol.InteractionType;
class="kw">import com.hypixel.hytale.protocol.WaitForDataFrom;
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.entity.InteractionContext;
class="kw">import com.hypixel.hytale.server.core.io.PacketHandler;
class="kw">import com.hypixel.hytale.server.core.modules.interaction.interaction.CooldownHandler;
class="kw">import com.hypixel.hytale.server.core.modules.interaction.interaction.config.SimpleInstantInteraction;
class="kw">import com.hypixel.hytale.server.core.universe.PlayerRef;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.core.util.NotificationUtil;
class="kw">import javax.annotation.Nonnull;

class="kw">public class SetMemoriesCapacityInteraction class="kw">extends SimpleInstantInteraction {
   @Nonnull
   class="kw">private class="kw">static class="kw">final String NOTIFICATION_ICON_MEMORIES = "NotificationIcons/MemoriesIcon.png";
   @Nonnull
   class="kw">private class="kw">static class="kw">final Message MESSAGE_SERVER_MEMORIES_GENERAL_FEATURE_UNLOCKED_NOTIFICATION = Message.translation(
      "server.memories.general.featureUnlockedNotification"
   );
   @Nonnull
   class="kw">private class="kw">static class="kw">final Message MESSAGE_SERVER_MEMORIES_GENERAL_FEATURE_UNLOCKED_MESSAGE = Message.translation("server.memories.general.featureUnlockedMessage");
   @Nonnull
   class="kw">public class="kw">static class="kw">final BuilderCodec<SetMemoriesCapacityInteraction> CODEC = BuilderCodec.builder(
         SetMemoriesCapacityInteraction.class, SetMemoriesCapacityInteraction::new, SimpleInstantInteraction.CODEC
      )
      .documentation("Sets how many memories a player can store.")
      .<Integer>appendInherited(
         new KeyedCodec<>("Capacity", Codec.INTEGER),
         (var0, var1) -> var0.capacity = var1,
         var0 -> var0.capacity,
         (var0, var1) -> var0.capacity = var1.capacity
      )
      .documentation("Defines the amount of memories that a player can store.")
      .add()
      .build();
   class="kw">private int capacity;

   class="kw">public SetMemoriesCapacityInteraction() {
   }

   @Override
   class="kw">protected void firstRun(@Nonnull InteractionType var1, @Nonnull InteractionContext var2, @Nonnull CooldownHandler var3) {
      Ref var4 = var2.getEntity();
      CommandBuffer var5 = var2.getCommandBuffer();
      assert var5 != null;
      PlayerMemories var6 = var5.ensureAndGetComponent(var4, PlayerMemories.getComponentType());
      if (this.capacity <= var6.getMemoriesCapacity()) {
         var2.getState().state = InteractionState.Failed;
      } else {
         int var7 = var6.getMemoriesCapacity();
         var6.setMemoriesCapacity(this.capacity);
         if (var7 <= 0) {
            PlayerRef var8 = var5.getComponent(var4, PlayerRef.getComponentType());
            if (var8 != null) {
               PacketHandler var9 = var8.getPacketHandler();
               var9.writeNoCache(new UpdateMemoriesFeatureStatus(true));
               NotificationUtil.sendNotification(
                  var9, MESSAGE_SERVER_MEMORIES_GENERAL_FEATURE_UNLOCKED_NOTIFICATION, null, "NotificationIcons/MemoriesIcon.png"
               );
               var8.sendMessage(MESSAGE_SERVER_MEMORIES_GENERAL_FEATURE_UNLOCKED_MESSAGE);
            }
         }

         var2.getState().state = InteractionState.Finished;
      }
   }

   @Nonnull
   @Override
   class="kw">public WaitForDataFrom getWaitForDataFrom() {
      class="kw">return WaitForDataFrom.Server;
   }
}