Implemented groundwork for fixing Data Stick exploiting.

1 files changed, 374 insertions, 0 deletions

diff --git a/src/main/java/gregtech/api/util/GT_AssemblyLineUtils.java b/src/main/java/gregtech/api/util/GT_AssemblyLineUtils.java
new file mode 100644
index 0000000000..17a7583ebe
--- /dev/null
+++ b/src/main/java/gregtech/api/util/GT_AssemblyLineUtils.java
@@ -0,0 +1,374 @@
+package gregtech.api.util;
+
+import static gregtech.GT_Mod.GT_FML_LOGGER;
+
+import cpw.mods.fml.common.FMLCommonHandler;
+import gregtech.api.enums.GT_Values;
+import gregtech.api.enums.ItemList;
+import gregtech.api.util.GT_Recipe.GT_Recipe_AssemblyLine;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.nbt.NBTTagList;
+import net.minecraft.nbt.NBTTagString;
+import net.minecraftforge.fluids.FluidStack;
+import scala.actors.threadpool.Arrays;
+
+public class GT_AssemblyLineUtils {
+
+	/**
+	 * Checks the DataStick for deprecated/invalid recipes, updating them as required.
+	 * @param aDataStick - The DataStick to process 
+	 * @return Is this DataStick now valid with a current recipe?
+	 */
+	public static boolean processDataStick(ItemStack aDataStick) {
+		if (!isItemDataStick(aDataStick)) {
+			return false;
+		}
+		if (doesDataStickNeedUpdate(aDataStick)) {
+			ItemStack aStickOutput = getDataStickOutput(aDataStick);
+			if (aStickOutput != null) {
+				GT_Recipe_AssemblyLine aIntendedRecipe = findAssemblyLineRecipeByOutput(aStickOutput);
+				if (aIntendedRecipe != null) {
+					return setAssemblyLineRecipeOnDataStick(aDataStick, aIntendedRecipe);
+				}
+			}
+			return false;			
+		}
+		return true;
+	}
+
+	/**
+	 * Finds an Assembly Line recipe from a DataStick.
+	 * @param aDataStick - The DataStick to check.
+	 * @return The GT_Recipe_AssemblyLine recipe contained on the DataStick, if any.
+	 */
+	public static GT_Recipe_AssemblyLine findAssemblyLineRecipeFromDataStick(ItemStack aDataStick) {	
+		if (!isItemDataStick(aDataStick)) {
+			return null;
+		}
+		ItemStack[] aInputs = new ItemStack[15];
+		ItemStack[] aOutputs = new ItemStack[1];
+		FluidStack[] aFluidInputs = new FluidStack[4];
+		boolean aFoundRecipe = false;
+
+		NBTTagCompound aTag = aDataStick.getTagCompound();
+		if (aTag == null) {
+			return null;
+		}
+
+		for (int i = 0; i < 15; i++) {
+			int count = aTag.getInteger("a" + i);
+			if (!aTag.hasKey("" + i) && count <= 0) {
+				continue;
+			}
+
+			boolean flag = true;
+			if (count > 0) {
+				for (int j = 0; j < count; j++) {
+					aInputs[i] = GT_Utility.loadItem(aTag, "a" + i + ":" + j);
+					if (aInputs[i] == null) {
+						continue;
+					}
+					if (GT_Values.D1) {
+						GT_FML_LOGGER.info("Item " + i + " : " + aInputs[i].getUnlocalizedName());
+					}
+					flag = false;
+					break;
+				}
+			}
+			if (flag) {
+				aInputs[i] = GT_Utility.loadItem(aTag, "" + i);
+				if (aInputs[i] == null) {
+					flag = false;
+					continue;
+				}
+				if (GT_Values.D1) {
+					GT_FML_LOGGER.info("Item " + i + " : " + aInputs[i].getUnlocalizedName());
+				}
+				flag = false;
+			}
+			if (GT_Values.D1) {
+				GT_FML_LOGGER.info(i + (flag ? " not accepted" : " accepted"));
+			}
+		}
+
+		if (GT_Values.D1) GT_FML_LOGGER.info("All Items done, start fluid check");
+		for (int i = 0; i < 4; i++) {
+			if (!aTag.hasKey("f" + i)) continue;
+			aFluidInputs[i] = GT_Utility.loadFluid(aTag, "f" + i);
+			if (aFluidInputs[i] == null) continue;
+			if (GT_Values.D1) {
+				GT_FML_LOGGER.info("Fluid " + i + " " + aFluidInputs[i].getUnlocalizedName());
+			}
+			if (GT_Values.D1) {
+				GT_FML_LOGGER.info(i + " accepted");
+			}
+		}
+
+		if (GT_Values.D1) {
+			GT_FML_LOGGER.info("Input accepted, check other values");
+		}
+		aOutputs = new ItemStack[]{GT_Utility.loadItem(aTag, "output")};
+		if (!aTag.hasKey("output") || !aTag.hasKey("time") || aTag.getInteger("time") <= 0 || !aTag.hasKey("eu") || aOutputs[0] == null || !GT_Utility.isStackValid(aOutputs[0])) {
+			return null;
+		}
+		if (GT_Values.D1) {
+			GT_FML_LOGGER.info("Find Data Stick recipe");
+		}
+		aFoundRecipe = true;
+
+		if (aFoundRecipe) {
+			int aTime = aTag.getInteger("time");
+			int aEU = aTag.getInteger("eu");			
+			for (GT_Recipe_AssemblyLine aRecipe : GT_Recipe.GT_Recipe_AssemblyLine.sAssemblylineRecipes) {
+				if (aRecipe.mEUt == aEU && aRecipe.mDuration == aTime) {
+					if (GT_Utility.areStacksEqual(aOutputs[0], aRecipe.mOutput, true)) {
+						if (Arrays.equals(aRecipe.mInputs, aInputs) && Arrays.equals(aRecipe.mFluidInputs, aFluidInputs)) {
+							return aRecipe;
+						}
+					}
+				}
+			}			
+		}
+		return null;
+	}
+
+
+	/**
+	 * Finds a GT_Recipe_AssemblyLine based on the expected output ItemStack.
+	 * @param aOutput - The Output of a GT_Recipe_AssemblyLine.
+	 * @return First found GT_Recipe_AssemblyLine with matching output.
+	 */
+	public static GT_Recipe_AssemblyLine findAssemblyLineRecipeByOutput(ItemStack aOutput) {
+		for (GT_Recipe_AssemblyLine aRecipe : GT_Recipe.GT_Recipe_AssemblyLine.sAssemblylineRecipes) {
+			ItemStack aRecipeOutput = aRecipe.mOutput;
+			if (GT_Utility.areStacksEqual(aRecipeOutput, aOutput)) {
+				return aRecipe;
+			}
+		}
+		return null;
+	}
+
+
+	/**
+	 * Get the Output ItemStack from a Data Stick.
+	 * @param aDataStick - The Data Stick to check.
+	 * @return Output ItemStack contained on the Data Stick.
+	 */
+	public static ItemStack getDataStickOutput(ItemStack aDataStick) {
+		if (doesDataStickHaveOutput(aDataStick)) {
+			ItemStack aOutput = GT_Utility.loadItem(aDataStick.getTagCompound(), "output");
+			return aOutput;
+		}
+		return null;
+	}
+
+	/**
+	 * @param aDataStick - The Data Stick to check.
+	 * @return Does this Data Stick have a valid output ItemStack?
+	 */
+	public static boolean doesDataStickHaveOutput(ItemStack aDataStick) {
+		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound() && aDataStick.getTagCompound().hasKey("output")) {
+			return true;
+		}
+		return false;
+	}
+
+	/**
+	 * @param aDataStick - The Data Stick to check.
+	 * @return Does this Data Stick need recipe data updated.
+	 */
+	public static boolean doesDataStickNeedUpdate(ItemStack aDataStick) {
+		if (isItemDataStick(aDataStick) && doesDataStickHaveRecipeHash(aDataStick)) {
+			String aStickHash = getHashFromDataStack(aDataStick);
+			if (isValidHash(aStickHash) && doesDataStickHaveOutput(aDataStick)) {
+				ItemStack aStickOutput = getDataStickOutput(aDataStick);
+				GT_Recipe_AssemblyLine aIntendedRecipe = findAssemblyLineRecipeByOutput(aStickOutput);
+				if (aStickHash.equals(generateRecipeHash(aIntendedRecipe))) {
+					return false;
+				}				
+			}
+		}	
+		return true;
+	}
+
+	/**
+	 * @param aDataStick - The Data Stick to check.
+	 * @return Does this have a Recipe Hash String at all?
+	 */
+	public static boolean doesDataStickHaveRecipeHash(ItemStack aDataStick) {
+		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound()) {
+			NBTTagCompound aNBT = aDataStick.getTagCompound();
+			if (aNBT.hasKey("Data.Recipe.Hash")) {
+				return true;
+			}
+		}
+		return false;
+	}
+
+	/**
+	 * @param aDataStick - The Data Stick to procces.
+	 * @return The stored Recipe Hash String on the Data Stick, will return an invalid Hash if one is not found. <p>
+	 * Check with isValidHash(). <p> 
+	 * Will not return Null.
+	 */
+	public static String getHashFromDataStack(ItemStack aDataStick) {
+		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound()) {
+			NBTTagCompound aNBT = aDataStick.getTagCompound();
+			if (aNBT.hasKey("Data.Recipe.Hash")) {
+				return aNBT.getString("Data.Recipe.Hash");
+
+			}
+		}
+		return "Invalid.Recipe.Hash";
+	}
+
+	/**
+	 * 
+	 * @param aDataStick - The Data Stick to update.
+	 * @param aRecipeHash - The Recipe Hash String to update with.
+	 * @return Did we update the Recipe Hash String on the Data Stick?
+	 */
+	public static boolean writeRecipeHashToDataStick(ItemStack aDataStick, String aRecipeHash) {
+		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound()) {
+			NBTTagCompound aNBT = aDataStick.getTagCompound();
+			aNBT.setString("Data.Recipe.Hash", aRecipeHash);
+			aDataStick.setTagCompound(aNBT);
+			return true;
+		}
+		return false;
+	}
+
+	/**
+	 * @param aRecipe - The recipe to generate a Recipe Hash String from.
+	 * @return The Recipe Hash String.
+	 */
+	public static String generateRecipeHash(GT_Recipe_AssemblyLine aRecipe) {
+		String aHash = "Invalid.Recipe.Hash";		
+		if (aRecipe != null) {
+			aHash = "Hash."+aRecipe.hashCode();
+		}		
+		return aHash;
+	}
+
+	/**
+	 * @param aHash - Recipe hash String, may be null but will just be treated as invalid.
+	 * @return Is this Recipe Hash String valid?
+	 */
+	public static boolean isValidHash(String aHash) {
+		if (aHash != null && aHash.length() > 0) {
+			if (!aHash.equals("Invalid.Recipe.Hash") && aHash.contains("Hash.")) {
+				return true;
+			}
+		}
+		return false;
+	}
+
+	/**
+	 * @param aStack - The ItemStack to check.
+	 * @return Is this ItemStack a Data Stick?
+	 */
+	public static boolean isItemDataStick(ItemStack aStack) {
+		return GT_Utility.isStackValid(aStack) && ItemList.Tool_DataStick.isStackEqual(aStack, false, true);
+	}
+
+	/**
+	 * 
+	 * @param aDataStick - The Data Stick to update.
+	 * @param aNewRecipe - The New GT_Recipe_AssemblyLine recipe to update it with.
+	 * @return Did we set the new recipe data & Recipe Hash String on the Data Stick?
+	 */
+	public static boolean setAssemblyLineRecipeOnDataStick(ItemStack aDataStick, GT_Recipe_AssemblyLine aNewRecipe) {
+		if (isItemDataStick(aDataStick)) {
+			String s = aNewRecipe.mOutput.getDisplayName();
+			if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
+				s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mOutput.getDisplayName());
+				if (s == null)
+					s = aNewRecipe.mOutput.getDisplayName();
+			}
+
+			//remove possible old NBTTagCompound
+			aDataStick.setTagCompound(new NBTTagCompound());
+			GT_Utility.ItemNBT.setBookTitle(aDataStick, s + " Construction Data");
+
+			NBTTagCompound tNBT = aDataStick.getTagCompound();
+			if (tNBT == null) {
+				tNBT = new NBTTagCompound();
+			}
+
+			tNBT.setTag("output", aNewRecipe.mOutput.writeToNBT(new NBTTagCompound()));
+			tNBT.setInteger("time", aNewRecipe.mDuration);
+			tNBT.setInteger("eu", aNewRecipe.mEUt);
+			for (int i = 0; i < aNewRecipe.mInputs.length; i++) {
+				tNBT.setTag("" + i, aNewRecipe.mInputs[i].writeToNBT(new NBTTagCompound()));
+			}
+			for (int i = 0; i < aNewRecipe.mOreDictAlt.length; i++) {
+				if (aNewRecipe.mOreDictAlt[i] != null && aNewRecipe.mOreDictAlt[i].length > 0) {
+					tNBT.setInteger("a" + i, aNewRecipe.mOreDictAlt[i].length);
+					for (int j = 0; j < aNewRecipe.mOreDictAlt[i].length; j++) {
+						tNBT.setTag("a" + i + ":" + j, aNewRecipe.mOreDictAlt[i][j].writeToNBT(new NBTTagCompound()));
+					}
+				}
+			}
+			for (int i = 0; i < aNewRecipe.mFluidInputs.length; i++) {
+				tNBT.setTag("f" + i, aNewRecipe.mFluidInputs[i].writeToNBT(new NBTTagCompound()));
+			}
+			tNBT.setString("author", "Assembling Line Recipe Generator");
+			NBTTagList tNBTList = new NBTTagList();
+			s = aNewRecipe.mOutput.getDisplayName();
+			if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
+				s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mOutput.getDisplayName());
+				if (s == null)
+					s = aNewRecipe.mOutput.getDisplayName();
+			}
+			tNBTList.appendTag(new NBTTagString("Construction plan for " + aNewRecipe.mOutput.stackSize + " " + s + ". Needed EU/t: " + aNewRecipe.mEUt + " Production time: " + (aNewRecipe.mDuration / 20)));
+			for (int i = 0; i < aNewRecipe.mInputs.length; i++) {
+				if (aNewRecipe.mOreDictAlt[i] != null) {
+					int count = 0;
+					StringBuilder tBuilder = new StringBuilder("Input Bus " + (i + 1) + ": ");
+					for (ItemStack tStack : aNewRecipe.mOreDictAlt[i]) {
+						if (tStack != null) {
+							s = tStack.getDisplayName();
+							if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
+								s = GT_Assemblyline_Server.lServerNames.get(tStack.getDisplayName());
+								if (s == null)
+									s = tStack.getDisplayName();
+							}
+
+
+							tBuilder.append(count == 0 ? "" : "\nOr ").append(tStack.stackSize).append(" ").append(s);
+							count++;
+						}
+					}
+					if (count > 0) tNBTList.appendTag(new NBTTagString(tBuilder.toString()));
+				} else if (aNewRecipe.mInputs[i] != null) {
+					s = aNewRecipe.mInputs[i].getDisplayName();
+					if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
+						s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mInputs[i].getDisplayName());
+						if (s == null)
+							s = aNewRecipe.mInputs[i].getDisplayName();
+					}
+					tNBTList.appendTag(new NBTTagString("Input Bus " + (i + 1) + ": " + aNewRecipe.mInputs[i].stackSize + " " + s));
+				}
+			}
+			for (int i = 0; i < aNewRecipe.mFluidInputs.length; i++) {
+				if (aNewRecipe.mFluidInputs[i] != null) {
+					s = aNewRecipe.mFluidInputs[i].getLocalizedName();
+					if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
+						s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mFluidInputs[i].getLocalizedName());
+						if (s == null)
+							s = aNewRecipe.mFluidInputs[i].getLocalizedName();
+					}
+					tNBTList.appendTag(new NBTTagString("Input Hatch " + (i + 1) + ": " + aNewRecipe.mFluidInputs[i].amount + "L " + s));
+				}
+			}
+			tNBT.setTag("pages", tNBTList);		
+			aDataStick.setTagCompound(tNBT);	
+			// Set recipe hash
+			writeRecipeHashToDataStick(aDataStick, generateRecipeHash(aNewRecipe));
+			return true;
+		}
+		return false;
+	}
+
+}