package techreborn.compat.minetweaker; import minetweaker.IUndoableAction; import minetweaker.MineTweakerAPI; import minetweaker.api.item.IIngredient; import minetweaker.api.item.IItemStack; import net.minecraft.item.ItemStack; import stanhebben.zenscript.annotations.ZenClass; import stanhebben.zenscript.annotations.ZenMethod; import techreborn.api.recipe.IBaseRecipeType; import techreborn.api.recipe.RecipeHandler; import techreborn.api.recipe.machines.AssemblingMachineRecipe; import techreborn.lib.Reference; import java.util.ArrayList; import java.util.List; @ZenClass("mods.techreborn.assemblingMachine") public class MTAssemblingMachine { @ZenMethod public static void addRecipe(IItemStack output, IIngredient input1, IIngredient input2, int ticktime, int euTick) { ItemStack oInput1 = (ItemStack) MinetweakerCompat.toObject(input1); ItemStack oInput2 = (ItemStack) MinetweakerCompat.toObject(input2); AssemblingMachineRecipe r = new AssemblingMachineRecipe(oInput1, oInput2, MinetweakerCompat.toStack(output), ticktime, euTick); MineTweakerAPI.apply(new Add(r)); } private static class Add implements IUndoableAction { private final AssemblingMachineRecipe recipe; public Add(AssemblingMachineRecipe recipe) { this.recipe = recipe; } @Override public void apply() { RecipeHandler.addRecipe(recipe); } @Override public boolean canUndo() { return true; } @Override public void undo() { RecipeHandler.recipeList.remove(recipe); } @Override public String describe() { return "Adding Assembling Machine Recipe for " + recipe.getOutput(0).getDisplayName(); } @Override public String describeUndo() { return "Removing Assembling Machine Recipe for " + recipe.getOutput(0).getDisplayName(); } @Override public Object getOverrideKey() { return null; } } @ZenMethod public static void removeRecipe(IIngredient output) { for(IItemStack itemStack : output.getItems()){ MineTweakerAPI.apply(new Remove(MinetweakerCompat.toStack(itemStack))); } } private static class Remove implements IUndoableAction { private final ItemStack output; List removedRecipes = new ArrayList(); public Remove(ItemStack output) { this.output = output; } @Override public void apply() { for (IBaseRecipeType recipeType : RecipeHandler.getRecipeClassFromName(Reference.assemblingMachineRecipe)) { for (ItemStack stack : recipeType.getOutputs()) { if (ItemUtils.isItemEqual(stack, output, true, false)) { removedRecipes.add((AssemblingMachineRecipe) recipeType); RecipeHandler.recipeList.remove(recipeType); break; } } } } @Override public void undo() { if (removedRecipes != null) { for (AssemblingMachineRecipe recipe : removedRecipes) { if (recipe != null) { RecipeHandler.addRecipe(recipe); } } } } @Override public String describe() { return "Removing Assembling Machine Recipe for " + output.getDisplayName(); } @Override public String describeUndo() { return "Re-Adding Assembling Machine Recipe for " + output.getDisplayName(); } @Override public Object getOverrideKey() { return null; } @Override public boolean canUndo() { return true; } } }