SoulForge.java

package modtweaker.mods.bloodmagic.handlers;

import WayofTime.bloodmagic.api.recipe.TartaricForgeRecipe;
import WayofTime.bloodmagic.compat.jei.forge.TartaricForgeRecipeJEI;
import com.blamejared.mtlib.helpers.LogHelper;
import com.blamejared.mtlib.utils.BaseListAddition;
import com.blamejared.mtlib.utils.BaseListRemoval;
import minetweaker.MineTweakerAPI;
import minetweaker.api.item.IIngredient;
import minetweaker.api.item.IItemStack;
import modtweaker.mods.bloodmagic.BloodMagicHelper;
import stanhebben.zenscript.annotations.ZenClass;
import stanhebben.zenscript.annotations.ZenMethod;

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

import static com.blamejared.mtlib.helpers.InputHelper.*;
import static com.blamejared.mtlib.helpers.StackHelper.matches;

@ZenClass("mods.bloodmagic.SoulForge")
public class SoulForge
{
    protected static final String name = "Blood Magic Soul Forge";

    @ZenMethod
    public static void addRecipe(IItemStack output, double minimumSouls, double drain, IIngredient[] input)
    {
        MineTweakerAPI.apply(new Add(new TartaricForgeRecipe(toStack(output), minimumSouls, drain, toObjects(input)), BloodMagicHelper.soulForgeList));
    }

    private static class Add extends BaseListAddition<TartaricForgeRecipe> {
        public Add(TartaricForgeRecipe recipe, List<TartaricForgeRecipe> list) {
            super(SoulForge.name, list);
            this.recipes.add(recipe);
        }
    
        @Override
        public void apply() {
            if(recipes.isEmpty()) {
                return;
            }
        
            for(TartaricForgeRecipe recipe : recipes) {
                if(recipe != null) {
                    if(list.add(recipe)) {
                        successful.add(recipe);
                        MineTweakerAPI.getIjeiRecipeRegistry().addRecipe(new TartaricForgeRecipeJEI(recipe));
                    } else {
                        LogHelper.logError(String.format("Error adding %s Recipe for %s", name, getRecipeInfo(recipe)));
                    }
                } else {
                    LogHelper.logError(String.format("Error adding %s Recipe: null object", name));
                }
            }
        }
    
        @Override
        public void undo() {
            if(this.successful.isEmpty()) {
                return;
            }
        
            for(TartaricForgeRecipe recipe : successful) {
                if(recipe != null) {
                    if(!list.remove(recipe)) {
                        LogHelper.logError(String.format("Error removing %s Recipe for %s", name, this.getRecipeInfo(recipe)));
                    }else{
                        MineTweakerAPI.getIjeiRecipeRegistry().removeRecipe(new TartaricForgeRecipeJEI(recipe));
                    }
                } else {
                    LogHelper.logError(String.format("Error removing %s Recipe: null object", name));
                }
            }
        }
        @Override
        public String getRecipeInfo(TartaricForgeRecipe recipe)
        {
            return LogHelper.getStackDescription(recipe.getRecipeOutput());
        }
    }

    @ZenMethod
    public static void removeRecipe(IIngredient output)
    {
        remove(output, BloodMagicHelper.soulForgeList);
    }

    public static void remove(IIngredient output, List<TartaricForgeRecipe> list) {
        if (output == null) {
            LogHelper.logError(String.format("Required parameters missing for %s Recipe.", name));
            return;
        }

        List<TartaricForgeRecipe> recipes = new LinkedList<>();

        for(TartaricForgeRecipe recipe : list)
        {
            if(matches(output, toIItemStack(recipe.getRecipeOutput())))
                recipes.add(recipe);
        }

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

    }

    private static class Remove extends BaseListRemoval<TartaricForgeRecipe> {
        public Remove(List<TartaricForgeRecipe> list, List<TartaricForgeRecipe> recipes) {
            super(SoulForge.name, list, recipes);
        }
        @Override
        public void apply() {
            if (recipes.isEmpty()) {
                return;
            }
            for (TartaricForgeRecipe recipe : this.recipes) {
                if (recipe != null) {
                    if (this.list.remove(recipe)) {
                    
                        successful.add(recipe);
                        MineTweakerAPI.getIjeiRecipeRegistry().removeRecipe(new TartaricForgeRecipeJEI(recipe));
                    } else {
                        LogHelper.logError(String.format("Error removing %s Recipe for %s", name, getRecipeInfo(recipe)));
                    }
                } else {
                    LogHelper.logError(String.format("Error removing %s Recipe: null object", name));
                }
            }
        }
    
        @Override
        public void undo() {
            if (successful.isEmpty()) {
                return;
            }
            for (TartaricForgeRecipe recipe : successful) {
                if (recipe != null) {
                    if (!list.add(recipe)) {
                        LogHelper.logError(String.format("Error restoring %s Recipe for %s", name, getRecipeInfo(recipe)));
                    }else{
                        MineTweakerAPI.getIjeiRecipeRegistry().addRecipe(new TartaricForgeRecipeJEI(recipe));
                    }
                } else {
                    LogHelper.logError(String.format("Error restoring %s Recipe: null object", name));
                }
            }
        }
        @Override
        protected String getRecipeInfo(TartaricForgeRecipe recipe) {
            return LogHelper.getStackDescription(recipe.getRecipeOutput());
        }
    }
}