path: root/src/main/java/io/github/moulberry/notenoughupdates/util/ItemResolutionQuery.java

/*
 * 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.util;

import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;
import io.github.moulberry.notenoughupdates.NEUManager;
import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
import io.github.moulberry.notenoughupdates.core.util.StringUtils;
import lombok.var;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.Gui;
import net.minecraft.client.gui.inventory.GuiChest;
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.NBTTagCompound;

import javax.annotation.Nullable;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Locale;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class ItemResolutionQuery {

	private static final Pattern ENCHANTED_BOOK_NAME_PATTERN = Pattern.compile("^((?:§.)*)([^§]+) ([IVXL]+)$");
	private static final Pattern PET_PATTERN = Pattern.compile(".*(\\[Lvl .*\\] )§(.).*");
	private static final String EXTRA_ATTRIBUTES = "ExtraAttributes";
	private static final List<String> PET_RARITIES = Arrays.asList(
		"COMMON",
		"UNCOMMON",
		"RARE",
		"EPIC",
		"LEGENDARY",
		"MYTHIC"
	);
	private final NEUManager manager;
	private NBTTagCompound compound;
	private Item itemType;
	private int stackSize = -1;
	private Gui guiContext;
	private String knownInternalName;

	public ItemResolutionQuery(NEUManager manager) {
		this.manager = manager;
	}

	public ItemResolutionQuery withItemNBT(NBTTagCompound compound) {
		this.compound = compound;
		return this;
	}

	public ItemResolutionQuery withItemStack(ItemStack stack) {
		if (stack == null) return this;
		this.itemType = stack.getItem();
		this.compound = stack.getTagCompound();
		this.stackSize = stack.stackSize;
		return this;
	}

	public ItemResolutionQuery withGuiContext(Gui gui) {
		this.guiContext = gui;
		return this;
	}

	public ItemResolutionQuery withCurrentGuiContext() {
		this.guiContext = Minecraft.getMinecraft().currentScreen;
		return this;
	}

	public ItemResolutionQuery withKnownInternalName(String knownInternalName) {
		this.knownInternalName = knownInternalName;
		return this;
	}

	@Nullable
	public String resolveInternalName() {
		if (knownInternalName != null) {
			return knownInternalName;
		}
		String resolvedName = resolveFromSkyblock();
		if (resolvedName == null) {
			resolvedName = resolveContextualName();
		} else {
			switch (resolvedName.intern()) {
				case "PET":
					resolvedName = resolvePetName();
					break;
				case "RUNE":
				case "UNIQUE_RUNE":
					resolvedName = resolveRuneName();
					break;
				case "ENCHANTED_BOOK":
					resolvedName = resolveEnchantedBookNameFromNBT();
					break;
				case "PARTY_HAT_CRAB":
				case "PARTY_HAT_CRAB_ANIMATED":
					resolvedName = resolveCrabHatName();
					break;
				case "ABICASE":
					resolvedName = resolvePhoneCase();
					break;
				case "PARTY_HAT_SLOTH":
					resolvedName = resolveSlothHatName();
					break;
				case "POTION":
					resolvedName = resolvePotionName();
					break;
				case "BALLOON_HAT_2024":
					resolvedName = resolveBalloonHatName();
					break;
			}
		}

		return resolvedName;
	}

	@Nullable
	public JsonObject resolveToItemListJson() {
		String internalName = resolveInternalName();
		if (internalName == null) {
			return null;
		}
		return manager.getItemInformation().get(internalName);
	}

	@Nullable
	public ItemStack resolveToItemStack() {
		JsonObject jsonObject = resolveToItemListJson();
		if (jsonObject == null) return null;
		return manager.jsonToStack(jsonObject);
	}

	@Nullable
	public ItemStack resolveToItemStack(boolean useReplacements) {
		JsonObject jsonObject = resolveToItemListJson();
		if (jsonObject == null) return null;
		return manager.jsonToStack(jsonObject, false, useReplacements);
	}

	// <editor-fold desc="Resolution Helpers">
	private boolean isBazaar(IInventory chest) {
		if (chest.getDisplayName().getFormattedText().startsWith("Bazaar ➜ ")) {
			return true;
		}
		int bazaarSlot = chest.getSizeInventory() - 5;
		if (bazaarSlot < 0) return false;
		ItemStack stackInSlot = chest.getStackInSlot(bazaarSlot);
		if (stackInSlot == null || stackInSlot.stackSize == 0) return false;
		// NBT lore, we do not care about rendered lore
		List<String> lore = ItemUtils.getLore(stackInSlot);
		return lore.contains("§7To Bazaar");
	}

	private String resolveContextualName() {
		if (!(guiContext instanceof GuiChest)) {
			return null;
		}
		GuiChest chest = (GuiChest) guiContext;
		ContainerChest inventorySlots = (ContainerChest) chest.inventorySlots;
		String guiName = inventorySlots.getLowerChestInventory().getDisplayName().getUnformattedText();
		boolean isOnBazaar = isBazaar(inventorySlots.getLowerChestInventory());
		String displayName = ItemUtils.getDisplayName(compound);
		if (displayName == null) return null;
		if (itemType == Items.enchanted_book && isOnBazaar && compound != null) {
			return resolveEnchantmentByName(displayName);
		}
		if (displayName.endsWith("Enchanted Book") && guiName.startsWith("Superpairs")) {
			for (String loreLine : ItemUtils.getLore(compound)) {
				String enchantmentIdCandidate = resolveEnchantmentByName(loreLine);
				if (enchantmentIdCandidate != null) return enchantmentIdCandidate;
			}
			return null;
		}
		if (guiName.equals("Catacombs RNG Meter")) {
			return resolveItemInCatacombsRngMeter();
		}
		if (guiName.startsWith("Choose Pet")) {
			return findInternalNameByDisplayName(displayName, false);
		}
		return null;
	}

	/**
	 * Search for an item by the display name
	 *
	 * @param displayName  The display name of the item we are searching
	 * @param mayBeMangled Whether the item name may be mangled (for example: reforges, stars)
	 * @return the internal neu item id of that item, or null
	 */
	public static String findInternalNameByDisplayName(String displayName, boolean mayBeMangled) {
		return filterInternalNameCandidates(
			findInternalNameCandidatesForDisplayName(displayName),
			displayName,
			mayBeMangled
		);
	}

	public static String filterInternalNameCandidates(
		Collection<String> candidateInternalNames,
		String displayName,
		boolean mayBeMangled
	) {
		boolean isPet = displayName.contains("[Lvl ");
		String petRarity = null;
		if (isPet) {
			Matcher matcher = PET_PATTERN.matcher(displayName);
			if (matcher.matches()) {
				displayName = displayName.replace(matcher.group(1), "");
				petRarity = matcher.group(2);
			}
		}
		var cleanDisplayName = StringUtils.cleanColour(displayName);
		var manager = NotEnoughUpdates.INSTANCE.manager;
		String bestMatch = null;
		int bestMatchLength = -1;
		for (String internalName : candidateInternalNames) {

			String unCleanItemDisplayName = manager.getDisplayName(internalName);
			var cleanItemDisplayName = StringUtils.cleanColour(unCleanItemDisplayName);
			if (</