RepairItemInteraction.java

package com.hypixel.hytale.server.core.entity.entities.player.pages.itemrepair;

import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.math.util.MathUtil;
import com.hypixel.hytale.protocol.SoundCategory;
import com.hypixel.hytale.protocol.packets.interface_.Page;
import com.hypixel.hytale.server.core.Message;
import com.hypixel.hytale.server.core.entity.entities.Player;
import com.hypixel.hytale.server.core.entity.entities.player.pages.PageManager;
import com.hypixel.hytale.server.core.entity.entities.player.pages.choices.ChoiceInteraction;
import com.hypixel.hytale.server.core.inventory.ItemContext;
import com.hypixel.hytale.server.core.inventory.ItemStack;
import com.hypixel.hytale.server.core.inventory.container.ItemContainer;
import com.hypixel.hytale.server.core.inventory.container.SimpleItemContainer;
import com.hypixel.hytale.server.core.inventory.transaction.ItemStackSlotTransaction;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.SoundUtil;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.core.util.TempAssetIdUtil;
import javax.annotation.Nonnull;

public class RepairItemInteraction extends ChoiceInteraction {
   protected final ItemContext itemContext;
   protected final double repairPenalty;
   protected final ItemContext heldItemContext;

   public RepairItemInteraction(ItemContext itemContext, double repairPenalty, ItemContext heldItemContext) {
      this.itemContext = itemContext;
      this.repairPenalty = repairPenalty;
      this.heldItemContext = heldItemContext;
   }

   public void run(@Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref, @Nonnull PlayerRef playerRef) {
      Player playerComponent = (Player)store.getComponent(ref, Player.getComponentType());
      PageManager pageManager = playerComponent.getPageManager();
      ItemStack itemStack = this.itemContext.getItemStack();
      double itemStackDurability = itemStack.getDurability();
      double itemStackMaxDurability = itemStack.getMaxDurability();
      double ratioAmountRepaired = 1.0 - itemStackDurability / itemStackMaxDurability;
      double newMaxDurability = (double)MathUtil.floor(itemStackMaxDurability - itemStack.getItem().getMaxDurability() * this.repairPenalty * ratioAmountRepaired);
      if (itemStackDurability >= newMaxDurability) {
         playerRef.sendMessage(Message.translation("server.general.repair.penaltyTooBig").color("#ff5555"));
         pageManager.setPage(ref, store, Page.None);
      } else {
         if (newMaxDurability <= 10.0) {
            newMaxDurability = 10.0;
            playerRef.sendMessage(Message.translation("server.general.repair.tooLowDurability").color("#ff5555"));
         }

         ItemContainer heldItemContainer = this.heldItemContext.getContainer();
         short heldItemSlot = this.heldItemContext.getSlot();
         ItemStack heldItemStack = this.heldItemContext.getItemStack();
         ItemStackSlotTransaction slotTransaction = heldItemContainer.removeItemStackFromSlot(heldItemSlot, heldItemStack, 1);
         if (!slotTransaction.succeeded()) {
            pageManager.setPage(ref, store, Page.None);
         } else {
            ItemStack newItemStack = itemStack.withRestoredDurability(newMaxDurability);
            ItemStackSlotTransaction replaceTransaction = this.itemContext.getContainer().replaceItemStackInSlot(this.itemContext.getSlot(), itemStack, newItemStack);
            if (!replaceTransaction.succeeded()) {
               SimpleItemContainer.addOrDropItemStack(store, ref, heldItemContainer, heldItemSlot, heldItemStack.withQuantity(1));
               pageManager.setPage(ref, store, Page.None);
            } else {
               Message newItemStackMessage = Message.translation(newItemStack.getItem().getTranslationKey());
               playerRef.sendMessage(Message.translation("server.general.repair.successful").param("itemName", newItemStackMessage));
               pageManager.setPage(ref, store, Page.None);
               SoundUtil.playSoundEvent2d(ref, TempAssetIdUtil.getSoundEventIndex("SFX_Item_Repair"), SoundCategory.UI, store);
            }
         }
      }
   }
}