1 files changed, 84 insertions, 0 deletions
diff --git a/src/main/java/at/hannibal2/skyhanni/utils/ItemNameResolver.kt b/src/main/java/at/hannibal2/skyhanni/utils/ItemNameResolver.kt
new file mode 100644
index 000000000..d17f579ed
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/utils/ItemNameResolver.kt
@@ -0,0 +1,84 @@
+package at.hannibal2.skyhanni.utils
+
+import at.hannibal2.skyhanni.utils.NEUInternalName.Companion.asInternalName
+import at.hannibal2.skyhanni.utils.NumberUtil.romanToDecimal
+import at.hannibal2.skyhanni.utils.StringUtils.matchMatcher
+import at.hannibal2.skyhanni.utils.StringUtils.removeColor
+import io.github.moulberry.notenoughupdates.util.ItemResolutionQuery
+
+object ItemNameResolver {
+    private val itemNameCache = mutableMapOf<String, NEUInternalName>() // item name -> internal name
+
+    internal fun getInternalNameOrNull(itemName: String): NEUInternalName? {
+        val lowercase = itemName.lowercase()
+        itemNameCache[lowercase]?.let {
+            return it
+        }
+
+        if (itemName == "§cmissing repo item") {
+                return itemNameCache.getOrPut(lowercase) { NEUInternalName.MISSING_ITEM }
+        }
+
+        resolveEnchantmentByName(itemName)?.let {
+            return itemNameCache.getOrPut(lowercase) { fixEnchantmentName(it) }
+        }
+
+        val internalName = ItemResolutionQuery.findInternalNameByDisplayName(itemName, true)?.let {
+
+            // This fixes a NEU bug with §9Hay Bale (cosmetic item)
+            // TODO remove workaround when this is fixed in neu
+            val rawInternalName = if (it == "HAY_BALE") "HAY_BLOCK" else it
+            rawInternalName.asInternalName()
+        } ?: run {
+            getInternalNameOrNullIgnoreCase(itemName)
+        } ?: return null
+
+        itemNameCache[lowercase] = internalName
+        return internalName
+    }
+
+    // Taken and edited from NEU
+    private fun resolveEnchantmentByName(enchantmentName: String) =
+        UtilsPatterns.enchantmentNamePattern.matchMatcher(enchantmentName) {
+            val name = group("name").trim { it <= ' ' }
+            val ultimate = group("format").lowercase().contains("§l")
+            ((if (ultimate && name != "Ultimate Wise") "ULTIMATE_" else "")
+                + turboCheck(name).replace(" ", "_").replace("-", "_").uppercase()
+                + ";" + group("level").romanToDecimal())
+        }
+
+    private fun turboCheck(text: String): String {
+        if (text == "Turbo-Cocoa") return "Turbo-Coco"
+        if (text == "Turbo-Cacti") return "Turbo-Cactus"
+        return text
+    }
+
+    // Workaround for duplex
+    private val duplexPattern = "ULTIMATE_DUPLEX;(?<tier>.*)".toPattern()
+
+    private fun fixEnchantmentName(originalName: String): NEUInternalName {
+        duplexPattern.matchMatcher(originalName) {
+            val tier = group("tier")
+            return "ULTIMATE_REITERATE;$tier".asInternalName()
+        }
+        // TODO USE SH-REPO
+        return originalName.asInternalName()
+    }
+
+    private fun getInternalNameOrNullIgnoreCase(itemName: String): NEUInternalName? {
+        val lowercase = itemName.removeColor().lowercase()
+        itemNameCache[lowercase]?.let {
+            return it
+        }
+
+        if (NEUItems.allItemsCache.isEmpty()) {
+            NEUItems.allItemsCache = NEUItems.readAllNeuItems()
+        }
+        NEUItems.allItemsCache[lowercase]?.let {
+            itemNameCache[lowercase] = it
+            return it
+        }
+
+        return null
+    }
+}