AssemblerRecipeTransferHandler.java

package blusunrize.immersiveengineering.common.util.compat.jei;

import blusunrize.immersiveengineering.ImmersiveEngineering;
import blusunrize.immersiveengineering.api.Lib;
import blusunrize.immersiveengineering.common.gui.ContainerAssembler;
import blusunrize.immersiveengineering.common.util.network.MessageTileSync;
import mezz.jei.api.gui.IGuiIngredient;
import mezz.jei.api.gui.IGuiItemStackGroup;
import mezz.jei.api.gui.IRecipeLayout;
import mezz.jei.api.recipe.VanillaRecipeCategoryUid;
import mezz.jei.api.recipe.transfer.IRecipeTransferError;
import mezz.jei.api.recipe.transfer.IRecipeTransferHandler;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.util.text.translation.I18n;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.List;

/**
 * @author BluSunrize - 08.09.2016
 */
public class AssemblerRecipeTransferHandler implements IRecipeTransferHandler<ContainerAssembler>
{
	@Override
	public Class getContainerClass()
	{
		return ContainerAssembler.class;
	}

	@Override
	public String getRecipeCategoryUid()
	{
		return VanillaRecipeCategoryUid.CRAFTING;
	}

	@Nullable
	@Override
	public IRecipeTransferError transferRecipe(@Nonnull ContainerAssembler container, @Nonnull IRecipeLayout recipeLayout, @Nonnull EntityPlayer player, boolean maxTransfer, boolean doTransfer)
	{
		for(int i = 0; i < 3; i++)
			if(container.tile.patterns[i].recipe == null)
			{
				if(doTransfer)
				{
					IGuiItemStackGroup stacks = recipeLayout.getItemStacks();
					NBTTagList tagList = new NBTTagList();
					ItemStack[] convertedInput = new ItemStack[stacks.getGuiIngredients().size() - 1];
					int j = 0;
					for(IGuiIngredient<ItemStack> ingr : stacks.getGuiIngredients().values())
					{
						if(j > 0)
						{
							List<ItemStack> list = ingr.getAllIngredients();
							if(list != null && list.size() > 0)
								convertedInput[j - 1] = list.get(0);
						}
						j++;
					}
					for(int slot = 0; slot < Math.min(convertedInput.length, 9); slot++)
					{
						container.putStackInSlot(i * 10 + slot, convertedInput[slot]);
						NBTTagCompound itemTag = new NBTTagCompound();
						if(convertedInput[slot] != null)
							convertedInput[slot].writeToNBT(itemTag);
						itemTag.setInteger("slot", slot);
						tagList.appendTag(itemTag);
					}
					NBTTagCompound tag = new NBTTagCompound();
					tag.setTag("patternSync", tagList);
					tag.setInteger("recipe", i);
					ImmersiveEngineering.packetHandler.sendToServer(new MessageTileSync(container.tile, tag));
				}
				return null;
			}
		return JEIHelper.jeiHelpers.recipeTransferHandlerHelper().createUserErrorWithTooltip(I18n.translateToLocal(Lib.GUI + "assembler.nospace"));
	}
}