// 
// Decompiled by Procyon v0.6.0
// 

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

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

public class RepairItemInteraction extends ChoiceInteraction
{
    protected final ItemContext itemContext;
    protected final double repairPenalty;
    protected final ItemContext heldItemContext;
    
    public RepairItemInteraction(final ItemContext itemContext, final double repairPenalty, final ItemContext heldItemContext) {
        this.itemContext = itemContext;
        this.repairPenalty = repairPenalty;
        this.heldItemContext = heldItemContext;
    }
    
    @Override
    public void run(@Nonnull final Store<EntityStore> store, @Nonnull final Ref<EntityStore> ref, @Nonnull final PlayerRef playerRef) {
        final Player playerComponent = store.getComponent(ref, Player.getComponentType());
        final PageManager pageManager = playerComponent.getPageManager();
        final ItemStack itemStack = this.itemContext.getItemStack();
        final double itemStackDurability = itemStack.getDurability();
        final double itemStackMaxDurability = itemStack.getMaxDurability();
        final double ratioAmountRepaired = 1.0 - itemStackDurability / itemStackMaxDurability;
        double newMaxDurability = 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);
            return;
        }
        if (newMaxDurability <= 10.0) {
            newMaxDurability = 10.0;
            playerRef.sendMessage(Message.translation("server.general.repair.tooLowDurability").color("#ff5555"));
        }
        final ItemContainer heldItemContainer = this.heldItemContext.getContainer();
        final short heldItemSlot = this.heldItemContext.getSlot();
        final ItemStack heldItemStack = this.heldItemContext.getItemStack();
        final ItemStackSlotTransaction slotTransaction = heldItemContainer.removeItemStackFromSlot(heldItemSlot, heldItemStack, 1);
        if (!slotTransaction.succeeded()) {
            pageManager.setPage(ref, store, Page.None);
            return;
        }
        final ItemStack newItemStack = itemStack.withRestoredDurability(newMaxDurability);
        final 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);
            return;
        }
        final 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);
    }
}