1 files changed, 143 insertions, 0 deletions
diff --git a/TF2 Crates/src/main/java/tf2crates/crate/RandomTinkersTool.java b/TF2 Crates/src/main/java/tf2crates/crate/RandomTinkersTool.java
new file mode 100755
index 0000000..c79ea86
--- /dev/null
+++ b/TF2 Crates/src/main/java/tf2crates/crate/RandomTinkersTool.java
@@ -0,0 +1,143 @@
+package tf2crates.crate;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+import net.minecraft.enchantment.EnchantmentHelper;
+import net.minecraft.init.Items;
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+import tconstruct.library.TConstructRegistry;
+import tconstruct.library.crafting.ModifyBuilder;
+import tconstruct.library.crafting.ToolBuilder;
+import tconstruct.library.crafting.ToolRecipe;
+import tconstruct.library.modifier.ItemModifier;
+import tconstruct.library.tools.ToolCore;
+import tconstruct.library.tools.ToolMaterial;
+import tconstruct.library.util.IToolPart;
+import tlhpoeCore.util.MathUtil;
+
+public class RandomTinkersTool extends RandomLoot {
+	private boolean				isLegendary;
+
+	private ToolRecipe			toolType;
+
+	private ArrayList<Integer>	materialIDs;
+
+	public RandomTinkersTool(boolean isLegendary, ToolRecipe toolType) {
+		super(null, isLegendary);
+
+		this.isLegendary = isLegendary;
+		this.toolType = toolType;
+	}
+
+	@Override
+	public String getDisplayName() {
+		return "A Random " + (isLegendary ? " Legendary" : "")
+				+ " Tinker's " + toolType.getType().getLocalizedToolName();
+	}
+
+	@Override
+	public ItemStack getLoot() {
+		if (materialIDs == null) {
+			prepareMaterials();
+		}
+
+		// determine type
+		ToolRecipe recipe = toolType;
+		ToolCore type = recipe.getType();
+
+		ItemStack tool = null;
+		int tries = 0;
+		// try to build the tool
+		do {
+			tries++;
+			// get components
+			ItemStack[] parts = new ItemStack[] { null, null, null, null };
+
+			Item[] items = new Item[4];
+			items[0] = type.getHeadItem();
+			items[1] = type.getHandleItem();
+			items[2] = type.getAccessoryItem();
+			items[3] = type.getExtraItem();
+
+			for (int i = 0; i < 4; i++) {
+				if (items[i] == null) {
+					continue;
+				}
+
+				do {
+					// get a material
+					Integer matId = materialIDs
+							.get(MathUtil.nextInt(materialIDs.size()));
+
+					parts[i] = new ItemStack(items[i], 1, matId);
+				} while (((IToolPart) items[i])
+						.getMaterialID(parts[i]) == -1);
+			}
+			// build the tool
+			tool = ToolBuilder.instance.buildTool(parts[0], parts[1],
+					parts[2], parts[3], "");
+		} while (tool == null && tries < 200);
+
+		if (tool == null) {
+			ItemStack oops = new ItemStack(Items.stick);
+
+			oops.setStackDisplayName("Crate Malfunction");
+
+			return oops;
+		}
+
+		int modCount = MathUtil.nextInt(3);
+
+		for (int i = modCount; i > 0; i--) {
+			doModifyWeapon(tool);
+		}
+
+		if (isLegendary) {
+			EnchantmentHelper.addRandomEnchantment(MathUtil.getRandom(),
+					tool, MathUtil.getRandomIntegerBetween(25, 30));
+		}
+
+		return tool;
+	}
+
+	private void doModifyWeapon(ItemStack tool) {
+		List<ItemModifier> modifiers = ModifyBuilder.instance.itemModifiers;
+
+		ItemModifier mod = modifiers
+				.get(MathUtil.nextInt(modifiers.size()));
+
+		ItemStack[] stacks = stackFromModifier(mod);
+
+		if (!mod.matches(stacks, tool)) {
+			return;
+		}
+
+		mod.addMatchingEffect(tool);
+
+		mod.modify(stacks, tool);
+	}
+
+	private static ItemStack[] stackFromModifier(ItemModifier modifier) {
+		ItemStack[] stack = new ItemStack[modifier.stacks.size()];
+		int i = 0;
+		for (Object s : modifier.stacks)
+			stack[i++] = (ItemStack) s;
+
+		return stack;
+	}
+
+	private void prepareMaterials() {
+		materialIDs = new ArrayList<Integer>(); // converted to list for
+												// indexed access for
+												// random
+
+		for (Map.Entry<Integer,
+				ToolMaterial> entry : TConstructRegistry.toolMaterials
+						.entrySet()) {
+			materialIDs.add(entry.getKey());
+		}
+	}
+}