/*
 * This file is licensed under the MIT License, part of Roughly Enough Items.
 * Copyright (c) 2018, 2019, 2020 shedaniel
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package me.shedaniel.rei.server;

import it.unimi.dsi.fastutil.ints.IntArrayList;
import it.unimi.dsi.fastutil.ints.IntList;
import net.minecraft.container.Container;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.inventory.Inventory;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.recipe.Ingredient;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.util.Identifier;
import net.minecraft.util.collection.DefaultedList;

import java.util.*;

public class InputSlotCrafter<C extends Inventory> implements RecipeGridAligner<Integer>, ContainerContext {
    
    protected Container container;
    protected ContainerInfo containerInfo;
    private List<StackAccessor> gridStacks;
    private List<StackAccessor> inventoryStacks;
    private ServerPlayerEntity player;
    
    private InputSlotCrafter(Container container, ContainerInfo<? extends Container> containerInfo) {
        this.container = container;
        this.containerInfo = containerInfo;
    }
    
    public static <C extends Inventory> void start(Identifier category, Container craftingContainer_1, ServerPlayerEntity player, Map<Integer, List<ItemStack>> map, boolean hasShift) {
        ContainerInfo<? extends Container> containerInfo = Objects.requireNonNull(ContainerInfoHandler.getContainerInfo(category, craftingContainer_1.getClass()), "Container Info does not exist on the server!");
        new InputSlotCrafter<C>(craftingContainer_1, containerInfo).fillInputSlots(player, map, hasShift);
    }
    
    private void fillInputSlots(ServerPlayerEntity player, Map<Integer, List<ItemStack>> map, boolean hasShift) {
        this.player = player;
        this.inventoryStacks = this.containerInfo.getInventoryStacks(this);
        this.gridStacks = this.containerInfo.getGridStacks(this);
        
        player.skipPacketSlotUpdates = true;
        // Return the already placed items on the grid
        this.returnInputs();
        
        RecipeFinder recipeFinder = new RecipeFinder();
        this.containerInfo.getRecipeFinderPopulator().populate(this).accept(recipeFinder);
        DefaultedList<Ingredient> ingredients = DefaultedList.of();
        map.entrySet().stream().sorted(Comparator.comparingInt(Map.Entry::getKey)).forEach(entry -> {
            ingredients.add(Ingredient.ofItems(entry.getValue().stream().map(ItemStack::getItem).toArray(Item[]::new)));
        });
        
        if (recipeFinder.findRecipe(ingredients, null)) {
            this.fillInputSlots(recipeFinder, ingredients, hasShift);
        } else {
            this.returnInputs();
            player.skipPacketSlotUpdates = false;
            this.containerInfo.markDirty(this);
            throw new NotEnoughMaterialsException();
        }
        
        player.skipPacketSlotUpdates = false;
        this.containerInfo.markDirty(this);
    }
    
    @Override
    public void acceptAlignedInput(Iterator<Integer> iterator_1, StackAccessor gridSlot, int craftsAmount) {
        ItemStack toBeTakenStack = RecipeFinder.getStackFromId(iterator_1.next());
        if (!toBeTakenStack.isEmpty()) {
            for (int i = 0; i < craftsAmount; ++i) {
                this.fillInputSlot(gridSlot, toBeTakenStack);
            }
        }
    }
    
    protected void fillInputSlot(StackAccessor slot, ItemStack toBeTakenStack) {
        int takenSlotIndex = this.method_7371(toBeTakenStack);
        if (takenSlotIndex != -1) {
            ItemStack takenStack = this.inventoryStacks.get(takenSlotIndex).getItemStack().copy();
            if (!takenStack.isEmpty()) {
                if (takenStack.getCount() > 1) {
                    this.inventoryStacks.get(takenSlotIndex).takeStack(1);
                } else {
                    this.inventoryStacks.get(takenSlotIndex).setItemStack(ItemStack.EMPTY);
                }
                
                takenStack.setCount(1);
                if (!slot.canPlace(takenStack)) {
                    return;
                }
                
                if (slot.getItemStack().isEmpty()) {
                    slot.setItemStack(takenStack);
                } else {
                    slot.getItemStack().increment(1);
                }
            }
        }
    }
    
    protected void fillInputSlots(RecipeFinder recipeFinder, DefaultedList<Ingredient> ingredients, boolean hasShift) {
        int recipeCrafts = recipeFinder.countRecipeCrafts(ingredients, null);
        int amountToFill = this.getAmountToFill(hasShift, recipeCrafts, false);
        IntList intList_1 = new IntArrayList();
        if (recipeFinder.findRecipe(ingredients, intList_1, amountToFill)) {
            int finalCraftsAmount = amountToFill;
            
            for (int itemId : intList_1) {
                finalCraftsAmount = Math.min(finalCraftsAmount, RecipeFinder.getStackFromId(itemId).getMaxCount());
            }
            
            if (recipeFinder.findRecipe(ingredients, intList_1, finalCraftsAmount)) {
                this.returnInputs();
                this.alignRecipeToGrid(gridStacks, intList_1.iterator(), finalCraftsAmount);
            }
        }
        
    }
    
    protected int getAmountToFill(boolean hasShift, int recipeCrafts, boolean boolean_2) {
        int amountToFill = 1;
        if (hasShift) {
            amountToFill = recipeCrafts;
        } else if (boolean_2) {
            amountToFill = 64;
            for (StackAccessor stackAccessor : gridStacks) {
                ItemStack itemStack = stackAccessor.getItemStack();
                if (!itemStack.isEmpty() && amountToFill > itemStack.getCount()) {
                    amountToFill = itemStack.getCount();
                }
            }
            if (amountToFill < 64) {
                ++amountToFill;
            }
        }
        return amountToFill;
    }
    
    protected void returnInputs() {
        this.containerInfo.getGridCleanHandler().clean(this);
    }
    
    public int method_7371(ItemStack itemStack) {
        boolean rejectedModification = false;
        for (int i = 0; i < inventoryStacks.size(); i++) {
            ItemStack itemStack1 = this.inventoryStacks.get(i).getItemStack();
            if (!itemStack1.isEmpty() && areItemsEqual(itemStack, itemStack1) && !itemStack1.isDamaged() && !itemStack1.hasEnchantments() && !itemStack1.hasCustomName()) {
                if (!this.inventoryStacks.get(i).allowModification(player)) {
                    rejectedModification = true;
                } else {
                    return i;
                }
            }
        }
        
        if (rejectedModification) {
            throw new IllegalStateException("Unable to take item from inventory due to slot not allowing modification! Item requested: " + itemStack);
        }
        
        return -1;
    }
    
    private static boolean areItemsEqual(ItemStack stack1, ItemStack stack2) {
        return stack1.getItem() == stack2.getItem() && ItemStack.areTagsEqual(stack1, stack2);
    }
    
    private int getFreeInventorySlots() {
        int int_1 = 0;
        for (StackAccessor inventoryStack : inventoryStacks) {
            if (inventoryStack.getItemStack().isEmpty()) {
                ++int_1;
            }
        }
        return int_1;
    }
    
    @Override
    public Container getContainer() {
        return container;
    }
    
    @Override
    public PlayerEntity getPlayerEntity() {
        return player;
    }
    
    @Override
    public ContainerInfo getContainerInfo() {
        return containerInfo;
    }
    
    public static class NotEnoughMaterialsException extends RuntimeException {}
    
}