NotEnoughItems/src/main/java/codechicken/nei/recipe/AutoCraftingManager.java

package codechicken.nei.recipe;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

import net.minecraft.client.gui.inventory.GuiContainer;
import net.minecraft.entity.player.InventoryPlayer;

import codechicken.nei.ItemStackAmount;
import codechicken.nei.NEIClientUtils;
import codechicken.nei.RestartableTask;
import codechicken.nei.bookmark.BookmarkItem;
import codechicken.nei.recipe.Recipe.RecipeId;
import codechicken.nei.recipe.chain.RecipeChainIterator;
import codechicken.nei.recipe.chain.RecipeChainMath;

public class AutoCraftingManager {

    private static RecipeChainMath math;

    private static final RestartableTask task = new RestartableTask("NEI Bookmark AutoCraft Processing") {

        @Override
        public void execute() {
            final GuiContainer guiContainer = NEIClientUtils.getGuiContainer();
            final InventoryPlayer playerInventory = guiContainer.mc.thePlayer.inventory;
            final ItemStackAmount inventory = ItemStackAmount.of(Arrays.asList(playerInventory.mainInventory));
            final List<BookmarkItem> initialItems = prepareInitialItems(math, inventory);
            boolean processed = false;
            boolean changed = false;

            do {
                changed = false;

                RecipeChainIterator iterator = new RecipeChainIterator(math, initialItems);
                iterator.updateInventory(playerInventory.mainInventory);

                while (iterator.hasNext() && !interrupted(guiContainer)) {
                    final Map<RecipeId, Long> recipes = iterator.next();
                    boolean craft = false;

                    for (Map.Entry<RecipeId, Long> entry : recipes.entrySet()) {
                        final RecipeHandlerRef handler = RecipeHandlerRef.of(entry.getKey());

                        if (handler != null && handler.canCraft(guiContainer)) {
                            long multiplier = entry.getValue();

                            while (multiplier > 0 && !interrupted(guiContainer)
                                    && handler.craft(guiContainer, (int) Math.min(64, multiplier))) {
                                multiplier -= 64;
                            }

                            craft = multiplier != entry.getValue();
                        }

                        if (interrupted(guiContainer)) break;
                    }

                    if (craft) {
                        changed = true;
                        processed = true;
                        iterator.updateInventory(playerInventory.mainInventory);
                    }
                }

            } while (changed && !interrupted(guiContainer));

            if (processed && !changed && !interrupted(guiContainer)) {
                NEIClientUtils.playClickSound();
            }

        }

        @Override
        public void clearTasks() {
            super.clearTasks();
            AutoCraftingManager.math = null;
        }

        private boolean interrupted(GuiContainer guiContainer) {
            return interrupted() || guiContainer != NEIClientUtils.getGuiContainer();
        }

        private List<BookmarkItem> prepareInitialItems(RecipeChainMath math, ItemStackAmount inventory) {
            final List<BookmarkItem> initialItems = new ArrayList<>();

            for (BookmarkItem item : math.initialItems) {
                final long invStackSize = inventory.getOrDefault(item.itemStack, 0L);
                final long amount = Math.max(0, item.amount - invStackSize * item.fluidCellAmount);
                if (amount > 0) {
                    initialItems.add(item.copyWithAmount(amount));
                }
            }

            return initialItems;
        }

    };

    private AutoCraftingManager() {}

    public static void runProcessing(RecipeChainMath math) {
        task.stop();
        AutoCraftingManager.math = math;

        if (AutoCraftingManager.math != null) {
            math.createMasterRoot();
            task.restart();
        }
    }

    public static boolean processing() {
        return AutoCraftingManager.math != null && !task.interrupted();
    }

}