Added repo parserer for the draconic table thingy (#365)

1 files changed, 482 insertions, 0 deletions

diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/dev/RepoExporters.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/dev/RepoExporters.java
new file mode 100644
index 00000000..ecd7a203
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/dev/RepoExporters.java
@@ -0,0 +1,482 @@
+/*
+ * Copyright (C) 2022 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscfeatures.dev;
+
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import com.google.gson.JsonArray;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+import com.google.gson.JsonPrimitive;
+import io.github.moulberry.notenoughupdates.NEUOverlay;
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
+import io.github.moulberry.notenoughupdates.itemeditor.NEUItemEditor;
+import io.github.moulberry.notenoughupdates.listener.RenderListener;
+import io.github.moulberry.notenoughupdates.util.ItemUtils;
+import io.github.moulberry.notenoughupdates.util.SBInfo;
+import io.github.moulberry.notenoughupdates.util.Utils;
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.gui.inventory.GuiChest;
+import net.minecraft.init.Blocks;
+import net.minecraft.init.Items;
+import net.minecraft.inventory.ContainerChest;
+import net.minecraft.inventory.IInventory;
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTTagList;
+import net.minecraft.util.ChatComponentText;
+import net.minecraft.util.EnumChatFormatting;
+import net.minecraft.util.StringUtils;
+
+import java.io.BufferedReader;
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.OutputStreamWriter;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Locale;
+import java.util.Map;
+import java.util.TreeMap;
+import java.util.stream.Collectors;
+
+public class RepoExporters {
+
+	private static final RepoExporters INSTANCE = new RepoExporters();
+
+	public static RepoExporters getInstance() {
+		return INSTANCE;
+	}
+
+	public void essenceExporter() {
+		try {
+			GuiChest eventGui = (GuiChest) Minecraft.getMinecraft().currentScreen;
+			ContainerChest cc = (ContainerChest) eventGui.inventorySlots;
+			IInventory lower = cc.getLowerChestInventory();
+			File file = new File(
+				Minecraft.getMinecraft().mcDataDir.getAbsolutePath(),
+				"config/notenoughupdates/repo/constants/essencecosts.json"
+			);
+			String fileContent;
+			fileContent = new BufferedReader(new InputStreamReader(
+				Files.newInputStream(file.toPath()),
+				StandardCharsets.UTF_8
+			))
+				.lines()
+				.collect(Collectors.joining(System.lineSeparator()));
+			String id = null;
+			JsonObject jsonObject = new JsonParser().parse(fileContent).getAsJsonObject();
+			JsonObject newEntry = new JsonObject();
+			for (int i = 0; i < 54; i++) {
+				ItemStack stack = lower.getStackInSlot(i);
+				if (!stack.getDisplayName().isEmpty() && stack.getItem() != Item.getItemFromBlock(Blocks.barrier) &&
+					stack.getItem() != Items.arrow) {
+					if (stack.getTagCompound().getCompoundTag("display").hasKey("Lore", 9)) {
+						int stars = Utils.getNumberOfStars(stack);
+						if (stars == 0) continue;
+
+						NBTTagList lore = stack.getTagCompound().getCompoundTag("display").getTagList("Lore", 8);
+						int costIndex = 10000;
+						id = NotEnoughUpdates.INSTANCE.manager
+							.createItemResolutionQuery()
+							.withItemStack(stack)
+							.resolveInternalName();
+						if (jsonObject.has(id)) {
+							jsonObject.remove(id);
+						}
+						for (int j = 0; j < lore.tagCount(); j++) {
+							String entry = lore.getStringTagAt(j);
+							if (entry.equals("§7Cost")) {
+								costIndex = j;
+							}
+							if (j > costIndex) {
+								entry = entry.trim();
+								int index = entry.lastIndexOf('x');
+								String item, amountString;
+								if (index < 0) {
+									item = entry.trim() + " x1";
+									amountString = "x1";
+								} else {
+									amountString = entry.substring(index);
+									item = entry.substring(0, index).trim();
+								}
+								item = item.substring(0, item.length() - 3);
+								int amount = Integer.parseInt(amountString.trim().replace("x", "").replace(",", ""));
+								if (item.endsWith("Essence")) {
+									int index2 = entry.indexOf("Essence");
+									String typeAndAmount = item.substring(0, index2).trim().substring(2);
+									int whitespaceIndex = typeAndAmount.indexOf(' ');
+									int essenceAmount = Integer.parseInt(typeAndAmount
+										.substring(0, whitespaceIndex)
+										.replace(",", ""));
+									newEntry.add("type", new JsonPrimitive(typeAndAmount.substring(whitespaceIndex + 1)));
+									if (stars == -1) {
+										newEntry.add("dungeonize", new JsonPrimitive(essenceAmount));
+									} else {
+										newEntry.add(String.valueOf(stars), new JsonPrimitive(essenceAmount));
+									}
+								} else if (item.endsWith("Coins")) {
+									int index2 = entry.indexOf("Coins");
+									String coinsAmount = item.substring(0, index2).trim().substring(2);
+									if (!newEntry.has("items")) {
+										newEntry.add("items", new JsonObject());
+									}
+									if (!newEntry.get("items").getAsJsonObject().has(String.valueOf(stars))) {
+										newEntry.get("items").getAsJsonObject().add(String.valueOf(stars), new JsonArray());
+									}
+									newEntry
+										.get("items")
+										.getAsJsonObject()
+										.get(String.valueOf(stars))
+										.getAsJsonArray()
+										.add(new JsonPrimitive("SKYBLOCK_COIN:" + coinsAmount.replace(",", "")));
+								} else {
+									String itemString = "_";
+									for (Map.Entry<String, JsonObject> itemEntry : NotEnoughUpdates.INSTANCE.manager
+										.getItemInformation()
+										.entrySet()) {
+
+										if (itemEntry.getValue().has("displayname")) {
+											String name = itemEntry.getValue().get("displayname").getAsString();
+											if (name.equals(item)) {
+												itemString = itemEntry.getKey() + ":" + amount;
+											}
+										}
+									}
+									if (!newEntry.has("items")) {
+										newEntry.add("items", new JsonObject());
+									}
+									if (!newEntry.get("items").getAsJsonObject().has(String.valueOf(stars))) {
+										newEntry.get("items").getAsJsonObject().add(String.valueOf(stars), new JsonArray());
+									}
+									newEntry
+										.get("items")
+										.getAsJsonObject()
+										.get(String.valueOf(stars))
+										.getAsJsonArray()
+										.add(new JsonPrimitive(itemString));
+								}
+							}
+						}
+						jsonObject.add(id, newEntry);
+					}
+				}
+			}
+			if (jsonObject.get(id).getAsJsonObject().has("items")) {
+				JsonObject itemsObj = jsonObject.get(id).getAsJsonObject().get("items").getAsJsonObject();
+				jsonObject.get(id).getAsJsonObject().remove("items");
+				jsonObject.get(id).getAsJsonObject().add("items", itemsObj);
+			}
+			Gson gson = new GsonBuilder().setPrettyPrinting().create();
+			try {
+				try (
+					BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
+						Files.newOutputStream(file.toPath()),
+						StandardCharsets.UTF_8
+					))
+				) {
+					writer.write(gson.toJson(jsonObject));
+					Utils.addChatMessage(EnumChatFormatting.AQUA + "Parsed and saved: " + EnumChatFormatting.WHITE + id);
+				}
+			} catch (IOException ignored) {
+				Utils.addChatMessage(EnumChatFormatting.RED + "Error while writing file.");
+			}
+		} catch (Exception e) {
+			e.printStackTrace();
+			Utils.addChatMessage(
+				EnumChatFormatting.RED + "Error while parsing inventory. Try again or check logs for details.");
+		}
+	}
+
+	public void draconicAlterExporter() {
+		try {
+			for (int i = 0; i < 54; i++) {
+				File file = null;
+				String fileContent = null;
+				JsonObject newEntry = new JsonObject();
+				JsonObject jsonObject = null;
+				String id = null;
+				GuiChest eventGui = (GuiChest) Minecraft.getMinecraft().currentScreen;
+				ContainerChest cc = (ContainerChest) eventGui.inventorySlots;
+				IInventory lower = cc.getLowerChestInventory();
+				ItemStack stack = lower.getStackInSlot(i);
+				if (stack == null) continue;
+				if (!stack.getDisplayName().isEmpty() && stack.getItem() != Item.getItemFromBlock(Blocks.barrier) &&
+					stack.getItem() != Items.arrow) {
+					if (stack.getTagCompound().getCompoundTag("display").hasKey("Lore", 9)) {
+
+						NBTTagList lore = stack.getTagCompound().getCompoundTag("display").getTagList("Lore", 8);
+						int costIndex = 10000;
+						id = StringUtils.stripControlCodes(stack.getDisplayName().replace(" ", "_").toUpperCase(Locale.US));
+						id = ItemUtils.fixDraconicId(id);
+						if (!NotEnoughUpdates.INSTANCE.manager.isValidInternalName(id)) continue;
+
+						file = new File(
+							Minecraft.getMinecraft().mcDataDir.getAbsolutePath(),
+							"config/notenoughupdates/repo/items/" + id + ".json"
+						);
+						fileContent = new BufferedReader(new InputStreamReader(
+							Files.newInputStream(file.toPath()),
+							StandardCharsets.UTF_8
+						))
+							.lines()
+							.collect(Collectors.joining(System.lineSeparator()));
+						jsonObject = new JsonParser().parse(fileContent).getAsJsonObject();
+
+						int essence = -1;
+						boolean funny = true;
+						for (int j = 0; j < lore.tagCount(); j++) {
+							String entry = lore.getStringTagAt(j);
+							if (entry.equals("§8§m-----------------")) {
+								costIndex = j;
+							}
+							if (j > costIndex) {
+								if (j == costIndex + 1) {
+									if (entry.startsWith("§7Dragon Essence: §d")) {
+										essence = Integer.parseInt(entry.substring("§7Dragon Essence: §d".length()));
+									} else {
+										funny = false;
+									}
+									continue;
+								} else if (j == costIndex + 2 && funny) continue;
+								entry = entry.trim();
+								if (!newEntry.has("dragon_items")) {
+									newEntry.add("dragon_items", new JsonArray());
+								}
+								newEntry
+									.get("dragon_items")
+									.getAsJsonArray()
+									.add(new JsonPrimitive(entry.trim()));
+							}
+						}
+						if (essence != -1) jsonObject.add("dragon_essence", new JsonPrimitive(essence));
+						jsonObject.add("dragon_items", newEntry.get("dragon_items"));
+					}
+				}
+				if (jsonObject == null) continue;
+				if (jsonObject.has("dragon_items")) {
+					JsonArray itemsObj = jsonObject.get("dragon_items").getAsJsonArray();
+					jsonObject.remove("dragon_items");
+					jsonObject.add("dragon_items", itemsObj);
+				}
+				Gson gson = new GsonBuilder().setPrettyPrinting().create();
+				try {
+					try (
+						BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
+							Files.newOutputStream(file.toPath()),
+							StandardCharsets.UTF_8
+						))
+					) {
+						writer.write(gson.toJson(jsonObject));
+						Utils.addChatMessage(
+							EnumChatFormatting.AQUA + "Parsed and saved: " + EnumChatFormatting.WHITE + id);
+					}
+				} catch (IOException ignored) {
+					Utils.addChatMessage(
+						EnumChatFormatting.RED + "Error while writing file.");
+				}
+			}
+		} catch (Exception e) {
+			e.printStackTrace();
+			Minecraft.getMinecraft().thePlayer.addChatMessage(new ChatComponentText(
+				EnumChatFormatting.RED + "Error while parsing inventory. Try again or check logs for details."));
+		}
+	}
+
+	private HashMap<String, String> cachedDefinitions;
+	private String correctingItem;
+
+	public boolean npcExporter() {
+		if (RenderListener.typing) {
+			RenderListener.typing = false;
+			cachedDefinitions.put(correctingItem, NEUOverlay.getTextField().getText());
+			NEUOverlay.getTextField().setText("");
+		}
+		if (Minecraft.getMinecraft().currentScreen instanceof GuiChest) {
+			Minecraft mc = Minecraft.getMinecraft();
+			GuiChest eventGui = (GuiChest) Minecraft.getMinecraft().currentScreen;
+			ContainerChest cc = (ContainerChest) eventGui.inventorySlots;
+			IInventory lower = cc.getLowerChestInventory();
+
+			try {
+				JsonObject newNPC = new JsonObject();
+				String displayName = lower.getDisplayName().getUnformattedText();
+				File file = new File(
+					Minecraft.getMinecraft().mcDataDir.getAbsolutePath(),
+					"config" + File.separator + "notenoughupdates" +
+						File.separator + "repo" + File.separator + "npc" + File.separator +
+						displayName.toUpperCase().replace(" ", "_") + ".json"
+				);
+				newNPC.add("itemid", new JsonPrimitive("minecraft:skull"));
+				newNPC.add("displayname", new JsonPrimitive("§9" + displayName + " (NPC)"));
+				newNPC.add("nbttag", new JsonPrimitive("TODO"));
+				newNPC.add("damage", new JsonPrimitive(3));
+
+				JsonArray newArray = new JsonArray();
+				newArray.add(new JsonPrimitive(""));
+				newNPC.add("lore", newArray);
+				newNPC.add("internalname", new JsonPrimitive(displayName.toUpperCase().replace(" ", "_") + "_NPC"));
+				newNPC.add("clickcommand", new JsonPrimitive("viewrecipe"));
+				newNPC.add("modver", new JsonPrimitive(NotEnoughUpdates.VERSION));
+				newNPC.add("infoType", new JsonPrimitive("WIKI_URL"));
+				JsonArray emptyInfoArray = new JsonArray();
+				emptyInfoArray.add(new JsonPrimitive("TODO"));
+				newNPC.add("info", emptyInfoArray);
+				newNPC.add("x", new JsonPrimitive((int) mc.thePlayer.posX));
+				newNPC.add("y", new JsonPrimitive((int) mc.thePlayer.posY + 2));
+				newNPC.add("z", new JsonPrimitive((int) mc.thePlayer.posZ));
+				newNPC.add("island", new JsonPrimitive(SBInfo.getInstance().getLocation()));
+
+				JsonArray recipesArray = new JsonArray();
+
+				TreeMap<String, JsonObject> itemInformation = NotEnoughUpdates.INSTANCE.manager.getItemInformation();
+				for (int i = 0; i < 45; i++) {
+					ItemStack stack = lower.getStackInSlot(i);
+					if (stack == null) continue;
+					if (stack.getDisplayName().isEmpty() || stack.getDisplayName().equals(" ")) continue;
+
+					String internalname = NotEnoughUpdates.INSTANCE.manager.getInternalnameFromNBT(stack.getTagCompound());
+					if (internalname == null) continue;
+					JsonObject currentRecipe = new JsonObject();
+					currentRecipe.add("type", new JsonPrimitive("npc_shop"));
+					JsonArray costArray = new JsonArray();
+					boolean inCost = false;
+					for (String s : NotEnoughUpdates.INSTANCE.manager.getLoreFromNBT(stack.getTagCompound())) {
+						if (s.equals("§7Cost")) {
+							inCost = true;
+							continue;
+						} else if (s.equals("§eClick to trade!")) {
+							inCost = false;
+						}
+						if (!inCost) continue;
+						String entry = StringUtils.stripControlCodes(s);
+						if (entry.isEmpty()) continue;
+						int coinIndex = entry.indexOf(" Coins");
+						if (coinIndex != -1) {
+							String amountString = entry.substring(0, coinIndex).replace(",", "");
+							costArray.add(new JsonPrimitive("SKYBLOCK_COIN:" + amountString));
+						} else {
+							if (cachedDefinitions == null) {
+								cachedDefinitions = new HashMap<>();
+							}
+
+							String item;
+							int amountIndex = entry.lastIndexOf(" x");
+							String amountString;
+							if (amountIndex == -1) {
+								amountString = "1";
+								item = entry.replace(" ", "_").toUpperCase();
+							} else {
+								amountString = entry.substring(amountIndex);
+								item = entry.substring(0, amountIndex).replace(" ", "_").toUpperCase();
+							}
+							amountString = amountString.replace(",", "").replace("x", "").trim();
+							if (itemInformation.containsKey(item)) {
+								costArray.add(new JsonPrimitive(item + ":" + amountString));
+							} else if (cachedDefinitions.containsKey(item)) {
+								costArray.add(new JsonPrimitive(cachedDefinitions.get(item) + ":" + amountString));
+							} else {
+								Utils.addChatMessage("Change the item ID of " + item + " to the correct one and press Enter.");
+								NEUOverlay.getTextField().setText(item);
+								RenderListener.typing = true;
+								correctingItem = item;
+								if (cachedDefinitions == null) {
+									cachedDefinitions = new HashMap<>();
+								}
+								return true;
+							}
+						}
+					}
+					currentRecipe.add("cost", costArray);
+					currentRecipe.add("result", new JsonPrimitive(internalname));
+					recipesArray.add(currentRecipe);
+					newNPC.add("recipes", recipesArray);
+				}
+				Gson gson = new GsonBuilder().setPrettyPrinting().create();
+				System.out.println(gson.toJson(newNPC));
+				try {
+					//noinspection ResultOfMethodCallIgnored
+					file.createNewFile();
+					try (
+						BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
+							Files.newOutputStream(file.toPath()),
+							StandardCharsets.UTF_8
+						))
+					) {
+						writer.write(gson.toJson(newNPC));
+						Utils.addChatMessage(
+							EnumChatFormatting.AQUA + "Parsed and saved: " + EnumChatFormatting.WHITE + displayName);
+					}
+				} catch (IOException ignored) {
+					Utils.addChatMessage(EnumChatFormatting.RED + "Error while writing file.");
+				}
+			} catch (Exception e) {
+				e.printStackTrace();
+				Utils.addChatMessage(
+					EnumChatFormatting.RED + "Error while parsing inventory. Try again or check logs for details");
+			}
+		}
+		return false;
+	}
+
+	public void essenceExporter2() {
+		GuiChest eventGui = (GuiChest) Minecraft.getMinecraft().currentScreen;
+		ContainerChest cc = (ContainerChest) eventGui.inventorySlots;
+		IInventory lower = cc.getLowerChestInventory();
+
+		for (int i = 9; i < 45; i++) {
+			ItemStack stack = lower.getStackInSlot(i);
+			if (stack == null) continue;
+			if (stack.getDisplayName().isEmpty() || stack.getDisplayName().equals(" ")) continue;
+			String internalName = NotEnoughUpdates.INSTANCE.manager.getInternalNameForItem(stack);
+			if (internalName == null) {
+				Utils.addChatMessage(
+					EnumChatFormatting.RED + "ERROR: Could not get internal name for: " + EnumChatFormatting.AQUA +
+						stack.getDisplayName());
+				continue;
+			}
+			JsonObject itemObject = NotEnoughUpdates.INSTANCE.manager.getJsonForItem(stack);
+			JsonArray lore = itemObject.get("lore").getAsJsonArray();
+			List<String> loreList = new ArrayList<>();
+			for (int j = 0; j < lore.size(); j++) loreList.add(lore.get(j).getAsString());
+			if (loreList.get(loreList.size() - 1).equals("§7§eClick to view upgrades!")) {
+				loreList.remove(loreList.size() - 1);
+				loreList.remove(loreList.size() - 1);
+			}
+
+			JsonArray newLore = new JsonArray();
+			for (String s : loreList) {
+				newLore.add(new JsonPrimitive(s));
+			}
+			itemObject.remove("lore");
+			itemObject.add("lore", newLore);
+
+			if (!NEUItemEditor.saveOnly(internalName, itemObject)) {
+				Utils.addChatMessage(
+					EnumChatFormatting.RED + "ERROR: Failed to save item: " + EnumChatFormatting.AQUA + stack.getDisplayName());
+			}
+		}
+		Utils.addChatMessage(EnumChatFormatting.AQUA + "Parsed page: " + lower.getDisplayName().getUnformattedText());
+	}
+}