ModTweaker/src/main/java/modtweaker2/mods/forestry/handlers/Squeezer.java

package modtweaker2.mods.forestry.handlers;

import static modtweaker2.helpers.InputHelper.toFluid;
import static modtweaker2.helpers.InputHelper.toILiquidStack;
import static modtweaker2.helpers.InputHelper.toIItemStack;
import static modtweaker2.helpers.InputHelper.toStack;
import static modtweaker2.helpers.InputHelper.toStacks;
import static modtweaker2.helpers.StackHelper.matches;

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

import minetweaker.MineTweakerAPI;
import minetweaker.api.item.IIngredient;
import minetweaker.api.item.IItemStack;
import minetweaker.api.item.WeightedItemStack;
import minetweaker.api.liquid.ILiquidStack;
import modtweaker2.helpers.LogHelper;
import modtweaker2.utils.BaseListAddition;
import modtweaker2.utils.BaseListRemoval;
import stanhebben.zenscript.annotations.Optional;
import stanhebben.zenscript.annotations.ZenClass;
import stanhebben.zenscript.annotations.ZenMethod;
import forestry.factory.gadgets.MachineSqueezer;
import forestry.factory.gadgets.MachineSqueezer.Recipe;
import forestry.factory.gadgets.MachineSqueezer.RecipeManager;

@ZenClass("mods.forestry.Squeezer")
public class Squeezer {

	public static final String name = "Forestry Squeezer";

	/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

	/*
	 * Adds a recipe without additional item output
	 *
	 * @param fluidOutput recipe fluid amount
	 * @param ingredients recipe ingredients
	 * @param timePerItem time per crafting operation
	 */
	@ZenMethod
	public static void addRecipe(ILiquidStack fluidOutput, int timePerItem, IItemStack[] ingredients) {
		MineTweakerAPI.apply(new Add( new Recipe(timePerItem, toStacks(ingredients), toFluid(fluidOutput), null, 0) ));
	}

	/**
	 * Adds a recipe with additional item output
	 * 
	 * @param fluidOutput recipe fluid amount
	 * @param itemOutput recipe output
	 * @param ingredients recipe ingredients
	 * @param timePerItem time per crafting operation
	 */
	@ZenMethod
	public static void addRecipe(ILiquidStack fluidOutput, WeightedItemStack itemOutput, IItemStack[] ingredients, int timePerItem) {
		MineTweakerAPI.apply(new Add(new Recipe(timePerItem, toStacks(ingredients), toFluid(fluidOutput), toStack(itemOutput.getStack()), (int) itemOutput.getPercent())));
	}
	
	@ZenMethod
	@Deprecated
	public static void addRecipe(int timePerItem, IItemStack[] resources, ILiquidStack liquid, IItemStack remnants, int chance) {
		MineTweakerAPI.apply(new Add(new Recipe(timePerItem, toStacks(resources), toFluid(liquid), toStack(remnants), chance)));
	}

	private static class Add extends BaseListAddition<Recipe> {
		public Add(Recipe recipe) {
			super(Squeezer.name, MachineSqueezer.RecipeManager.recipes);
			recipes.add(recipe);
		}

		@Override
		public void apply() {
			super.apply();
			for (Recipe recipe : recipes) {
				RecipeManager.recipeInputs.addAll(Arrays.asList(recipe.resources));
			}
		}

		@Override
		public void undo() {
			super.undo();
			for (Recipe recipe : recipes) {
				RecipeManager.recipeInputs.removeAll(Arrays.asList(recipe.resources));
			}
		}

		@Override
		public String getRecipeInfo(Recipe recipe) {
			return LogHelper.getStackDescription(recipe.liquid);
		}
	}
	
	/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

	/**
	 * Removes a recipe for the Centrifuge
	 * 
	 * @param liquid liquid output
	 * @param ingredients list of ingredients
	 */
	@ZenMethod
	public static void removeRecipe(IIngredient liquid, @Optional IIngredient[] ingredients) {
	    List<Recipe> recipes = new LinkedList<Recipe>();

		for (Recipe r : RecipeManager.recipes) {
			if (r != null && r.liquid != null && matches(liquid, toILiquidStack(r.liquid))) {
				// optional check for ingredients
				if (ingredients != null) {
					boolean matched = false;
					for (int i = 0; i < ingredients.length; i++) {
						if ( matches(ingredients[i], toIItemStack(r.resources[i])) )
							matched = true;
						else {
							matched = false;
							// if one ingredients doesn't match abort all further checks
							break;
						}
					}
					// if some ingredient doesn't match, the last one is false
					if (matched)
						recipes.add(r);
				} else {
					recipes.add(r);
				}
			}
		}

		if(!recipes.isEmpty()) {
			MineTweakerAPI.apply(new Remove(recipes));
		} else {
			LogHelper.logWarning(String.format("No %s Recipe found for %s. Command ignored!", Squeezer.name, liquid.toString()));
		}
	}

	private static class Remove extends BaseListRemoval<Recipe> {
		public Remove(List<Recipe> recipes) {
			super(Squeezer.name, MachineSqueezer.RecipeManager.recipes, recipes);

		}

		@Override
		public void apply() {
			super.apply();
			for (Recipe recipe : recipes) {
				RecipeManager.recipeInputs.removeAll(Arrays.asList(recipe.resources));
			}
		}

		@Override
		public void undo() {
			super.undo();
			for (Recipe recipe : recipes) {
				RecipeManager.recipeInputs.addAll(Arrays.asList(recipe.resources));
			}
		}
		
		@Override
		public String getRecipeInfo(Recipe recipe) {
			return LogHelper.getStackDescription(recipe.liquid);
		}
	}
}