ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/booklet/page/PageCrafting.java

package de.ellpeck.actuallyadditions.booklet.page;

import java.util.Arrays;
import java.util.List;

import de.ellpeck.actuallyadditions.api.booklet.internal.GuiBookletBase;
import de.ellpeck.actuallyadditions.common.ActuallyAdditions;
import de.ellpeck.actuallyadditions.booklet.gui.GuiBooklet;
import de.ellpeck.actuallyadditions.common.util.RefHelp;
import de.ellpeck.actuallyadditions.common.util.StackUtil;
import de.ellpeck.actuallyadditions.common.util.StringUtil;
import de.ellpeck.actuallyadditions.common.util.Util;
import de.ellpeck.actuallyadditions.common.util.crafting.BlankRecipe;
import net.minecraft.item.ItemStack;
import net.minecraft.item.crafting.IRecipe;
import net.minecraft.item.crafting.Ingredient;
import net.minecraft.item.crafting.ShapedRecipes;
import net.minecraft.item.crafting.ShapelessRecipes;
import net.minecraftforge.fml.client.config.GuiUtils;
import net.minecraftforge.fml.relauncher.Side;
import net.minecraftforge.fml.relauncher.SideOnly;
import net.minecraftforge.oredict.ShapedOreRecipe;
import net.minecraftforge.oredict.ShapelessOreRecipe;

public class PageCrafting extends BookletPage {

    private final List<IRecipe> recipes;
    private int recipeAt;
    private String recipeTypeLocKey;
    private boolean isWildcard;

    public PageCrafting(int localizationKey, int priority, List<IRecipe> recipes) {
        super(localizationKey, priority);
        this.recipes = recipes;
    }

    public PageCrafting(int localizationKey, List<IRecipe> recipes) {
        this(localizationKey, 0, recipes);
    }

    public PageCrafting(int localizationKey, IRecipe... recipes) {
        this(localizationKey, 0, recipes);
    }

    public PageCrafting(int localizationKey, int priority, IRecipe... recipes) {
        this(localizationKey, priority, Arrays.asList(recipes));
    }

    public BookletPage setWildcard() {
        this.isWildcard = true;
        return this;
    }

    @Override
    @SideOnly(Side.CLIENT)
    public void drawScreenPre(GuiBookletBase gui, int startX, int startY, int mouseX, int mouseY, float partialTicks) {
        super.drawScreenPre(gui, startX, startY, mouseX, mouseY, partialTicks);

        gui.mc.getTextureManager().bindTexture(GuiBooklet.RES_LOC_GADGETS);
        GuiUtils.drawTexturedModalRect(startX + 5, startY + 6, 20, 0, 116, 54, 0);

        if (recipeTypeLocKey != null) gui.renderScaledAsciiString("(" + StringUtil.localize(this.recipeTypeLocKey) + ")", startX + 6, startY + 65, 0, false, gui.getMediumFontSize());

        PageTextOnly.renderTextToPage(gui, this, startX + 6, startY + 80);
    }

    @Override
    @SideOnly(Side.CLIENT)
    public void updateScreen(GuiBookletBase gui, int startX, int startY, int pageTimer) {
        super.updateScreen(gui, startX, startY, pageTimer);

        if (pageTimer % 20 == 0) {
            this.findRecipe(gui, startX, startY);
        }
    }

    private void findRecipe(GuiBookletBase gui, int startX, int startY) {
        if (!this.recipes.isEmpty()) {
            IRecipe recipe = this.recipes.get(this.recipeAt);
            if (recipe != null) {
                this.setupRecipe(gui, recipe, startX, startY);
            }

            this.recipeAt++;
            if (this.recipeAt >= this.recipes.size()) {
                this.recipeAt = 0;
            }
        }
    }

    @Override
    public void initGui(GuiBookletBase gui, int startX, int startY) {
        super.initGui(gui, startX, startY);
        this.findRecipe(gui, startX, startY);
    }

    @Override
    public void getItemStacksForPage(List<ItemStack> list) {
        super.getItemStacksForPage(list);

        if (!this.recipes.isEmpty()) {
            for (IRecipe recipe : this.recipes) {
                if (recipe != null) {
                    ItemStack output = recipe.getRecipeOutput();
                    if (StackUtil.isValid(output)) {
                        ItemStack copy = output.copy();
                        if (this.isWildcard) {
                            copy.setItemDamage(Util.WILDCARD);
                        }
                        list.add(copy);
                    }
                }
            }
        }
    }

    private void setupRecipe(GuiBookletBase gui, IRecipe recipe, int startX, int startY) {
        Ingredient[] ings = new Ingredient[9];
        int width = 3;
        int height = 3;

        if (recipe instanceof BlankRecipe) {
            this.recipeTypeLocKey = "tooltip." + ActuallyAdditions.MODID + ".disabled";
            gui.addOrModifyItemRenderer(recipe.getRecipeOutput(), startX + 100, startY + 25, 1F, false);
            return;
        } else if (recipe instanceof ShapedRecipes) {
            ShapedRecipes shaped = (ShapedRecipes) recipe;
            width = shaped.recipeWidth;
            height = shaped.recipeHeight;
            ings = shaped.recipeItems.toArray(new Ingredient[shaped.recipeItems.size()]);
            this.recipeTypeLocKey = "booklet." + ActuallyAdditions.MODID + ".shapedRecipe";
        } else if (recipe instanceof ShapelessRecipes) {
            ShapelessRecipes shapeless = (ShapelessRecipes) recipe;
            for (int i = 0; i < shapeless.recipeItems.size(); i++) {
                ings[i] = shapeless.recipeItems.get(i);
            }
            this.recipeTypeLocKey = "booklet." + ActuallyAdditions.MODID + ".shapelessRecipe";
        } else if (recipe instanceof ShapedOreRecipe) {
            ShapedOreRecipe shaped = (ShapedOreRecipe) recipe;
            try {
                width = RefHelp.getPrivateValue(ShapedOreRecipe.class, shaped, 4);
                height = RefHelp.getPrivateValue(ShapedOreRecipe.class, shaped, 5);
            } catch (Exception e) {
                ActuallyAdditions.LOGGER.error("Something went wrong trying to get the Crafting Recipe in the booklet to display!", e);
            }
            for (int i = 0; i < shaped.getIngredients().size(); i++) {
                ings[i] = shaped.getIngredients().get(i);
            }
            this.recipeTypeLocKey = "booklet." + ActuallyAdditions.MODID + ".shapedOreRecipe";
        } else if (recipe instanceof ShapelessOreRecipe) {
            ShapelessOreRecipe shapeless = (ShapelessOreRecipe) recipe;
            for (int i = 0; i < shapeless.getIngredients().size(); i++) {
                ings[i] = shapeless.getIngredients().get(i);
            }
            this.recipeTypeLocKey = "booklet." + ActuallyAdditions.MODID + ".shapelessOreRecipe";
        }

        for (int x = 0; x < width; x++) {
            for (int y = 0; y < height; y++) {
                Ingredient ing = ings[y * width + x];
                if (ing != null) {
                    ItemStack[] stacks = ing.getMatchingStacks();
                    if (stacks != null && stacks.length > 0) {
                        ItemStack stack = stacks[0];
                        if (StackUtil.isValid(stack)) {
                            ItemStack copy = stack.copy();
                            copy.setCount(1);
                            if (copy.getItemDamage() == Util.WILDCARD) {
                                copy.setItemDamage(0);
                            }

                            gui.addOrModifyItemRenderer(copy, startX + 6 + x * 18, startY + 7 + y * 18, 1F, true);
                        }
                    }
                }
            }
        }

        gui.addOrModifyItemRenderer(recipe.getRecipeOutput(), startX + 100, startY + 25, 1F, false);
    }
}