ActuallyAdditions/src/main/java/ellpeck/actuallyadditions/gen/JamVillagerTradeHandler.java

package ellpeck.actuallyadditions.gen;

import cpw.mods.fml.common.registry.VillagerRegistry;
import ellpeck.actuallyadditions.items.InitItems;
import ellpeck.actuallyadditions.items.metalists.TheJams;
import net.minecraft.entity.passive.EntityVillager;
import net.minecraft.init.Items;
import net.minecraft.item.ItemStack;
import net.minecraft.village.MerchantRecipe;
import net.minecraft.village.MerchantRecipeList;
import net.minecraftforge.oredict.OreDictionary;

import java.util.ArrayList;
import java.util.Random;

public class JamVillagerTradeHandler implements VillagerRegistry.IVillageTradeHandler{

    private ArrayList<Trade> trades = new ArrayList<Trade>();

    public JamVillagerTradeHandler(){
        this.addWants("ingotGold", 5, 2);
        this.addWants("cropWheat", 15, 10);
        this.addWants("dustRedstone", 25, 15);
        this.addWants(new ItemStack(Items.bucket), 5, 4);
        this.addWants(new ItemStack(Items.glass_bottle), 12, 5);
        this.addWants(new ItemStack(Items.potionitem), 1, 0);
        this.addWants("ingotIron", 10, 5);
        this.addWants("gemDiamond", 1, 2);
        this.addWants("dustGlowstone", 12, 10);
    }

    @Override
    @SuppressWarnings("all")
    public void manipulateTradesForVillager(EntityVillager villager, MerchantRecipeList recipeList, Random rand){
        for(int i = 0; i < trades.size(); i++){
            for(int j = 0; j < trades.get(i).wants.size(); j++){
                ItemStack wantsTwo = null;
                ItemStack wantsOne = trades.get(i).wants.get(j);

                wantsOne.stackSize = rand.nextInt(trades.get(i).extraStackSize) + trades.get(i).baseStackSize;
                if(rand.nextInt(3) == 0){
                    int toGet = rand.nextInt(trades.size());
                    for(int k = 0; k < trades.get(toGet).wants.size(); k++){
                        wantsTwo = trades.get(toGet).wants.get(k);
                        wantsTwo.stackSize = rand.nextInt(trades.get(k).extraStackSize) + trades.get(k).baseStackSize;
                    }
                }
                if(wantsOne == wantsTwo) wantsTwo = null;

                for(int k = 0; k < TheJams.values().length; k++){
                    recipeList.add(new MerchantRecipe(wantsOne, wantsTwo, new ItemStack(InitItems.itemJams, rand.nextInt(3)+1, k)));
                }
            }
        }
    }

    public void addWants(String oredictName, int minSize, int maxSize){
        ArrayList<ItemStack> stacks = (ArrayList<ItemStack>)OreDictionary.getOres(oredictName, false);
        trades.add(new Trade(stacks, minSize, maxSize));
    }

    public void addWants(ItemStack stack, int minSize, int maxSize){
        trades.add(new Trade(stack, minSize, maxSize));
    }

    public static class Trade{

        public final ArrayList<ItemStack> wants = new ArrayList<ItemStack>();
        public final int baseStackSize;
        public final int extraStackSize;

        public Trade(ArrayList<ItemStack> wants, int minStackSize, int maxStackSize){
            this.wants.addAll(wants);
            this.baseStackSize = minStackSize <= 0 ? 1 : minStackSize;
            this.extraStackSize = maxStackSize <= 0 ? 1 : maxStackSize;
        }

        public Trade(ItemStack want, int minStackSize, int maxStackSize){
            this.wants.add(want);
            this.baseStackSize = minStackSize <= 0 ? 1 : minStackSize;
            this.extraStackSize = maxStackSize <= 0 ? 1 : maxStackSize;
        }

    }
}