Add shiny pig tracker

2 files changed, 104 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/firmament/repo/ItemNameLookup.kt b/src/main/kotlin/moe/nea/firmament/repo/ItemNameLookup.kt
new file mode 100644
index 0000000..4bdf616
--- /dev/null
+++ b/src/main/kotlin/moe/nea/firmament/repo/ItemNameLookup.kt
@@ -0,0 +1,103 @@
+/*
+ * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+package moe.nea.firmament.repo
+
+import io.github.moulberry.repo.IReloadable
+import io.github.moulberry.repo.NEURepository
+import io.github.moulberry.repo.data.NEUItem
+import java.util.NavigableMap
+import java.util.TreeMap
+import moe.nea.firmament.util.SkyblockId
+import moe.nea.firmament.util.removeColorCodes
+import moe.nea.firmament.util.skyblockId
+
+object ItemNameLookup : IReloadable {
+
+    fun getItemNameChunks(name: String): Set<String> {
+        return name.removeColorCodes().split(" ").filterTo(mutableSetOf()) { it.isNotBlank() }
+    }
+
+    var nameMap: NavigableMap<String, out Set<SkyblockId>> = TreeMap()
+
+    override fun reload(repository: NEURepository) {
+        val nameMap = TreeMap<String, MutableSet<SkyblockId>>()
+        repository.items.items.values.forEach { item ->
+            getAllNamesForItem(item).forEach { name ->
+                val chunks = getItemNameChunks(name)
+                chunks.forEach { chunk ->
+                    val set = nameMap.getOrPut(chunk, ::mutableSetOf)
+                    set.add(item.skyblockId)
+                }
+            }
+        }
+        this.nameMap = nameMap
+    }
+
+    fun getAllNamesForItem(item: NEUItem): Set<String> {
+        val names = mutableSetOf<String>()
+        names.add(item.displayName)
+        if (item.displayName.contains("Enchanted Book")) {
+            val enchantName = item.lore.firstOrNull()
+            if (enchantName != null) {
+                names.add(enchantName)
+            }
+        }
+        return names
+    }
+
+    fun findItemCandidatesByName(name: String): MutableSet<SkyblockId> {
+        val candidates = mutableSetOf<SkyblockId>()
+        for (chunk in getItemNameChunks(name)) {
+            val set = nameMap[chunk] ?: emptySet()
+            candidates.addAll(set)
+        }
+        return candidates
+    }
+
+
+    fun guessItemByName(
+        /**
+         * The display name of the item. Color codes will be ignored.
+         */
+        name: String,
+        /**
+         * Whether the [name] may contain other text, such as reforges, master stars and such.
+         */
+        mayBeMangled: Boolean
+    ): SkyblockId? {
+        val cleanName = name.removeColorCodes()
+        return findBestItemFromCandidates(
+            findItemCandidatesByName(cleanName),
+            cleanName,
+            true
+        )
+    }
+
+    fun findBestItemFromCandidates(
+        candidates: Iterable<SkyblockId>,
+        name: String, mayBeMangled: Boolean
+    ): SkyblockId? {
+        val expectedClean = name.removeColorCodes()
+        var bestMatch: SkyblockId? = null
+        var bestMatchLength = -1
+        for (candidate in candidates) {
+            val item = RepoManager.getNEUItem(candidate) ?: continue
+            for (name in getAllNamesForItem(item)) {
+                val actualClean = name.removeColorCodes()
+                val matches = if (mayBeMangled) expectedClean == actualClean
+                else expectedClean.contains(actualClean)
+                if (!matches) continue
+                if (actualClean.length > bestMatchLength) {
+                    bestMatch = candidate
+                    bestMatchLength = actualClean.length
+                }
+            }
+        }
+        return bestMatch
+    }
+
+}
diff --git a/src/main/kotlin/moe/nea/firmament/repo/RepoManager.kt b/src/main/kotlin/moe/nea/firmament/repo/RepoManager.kt
index a1e73b1..db17e6a 100644
--- a/src/main/kotlin/moe/nea/firmament/repo/RepoManager.kt
+++ b/src/main/kotlin/moe/nea/firmament/repo/RepoManager.kt
@@ -63,6 +63,7 @@ object RepoManager {
     val neuRepo: NEURepository = NEURepository.of(RepoDownloadManager.repoSavedLocation).apply {
         registerReloadListener(ItemCache)
         registerReloadListener(ExpLadders)
+        registerReloadListener(ItemNameLookup)
         registerReloadListener {
             Firmament.coroutineScope.launch(MinecraftDispatcher) {
                 if (!trySendClientboundUpdateRecipesPacket()) {