1 files changed, 149 insertions, 0 deletions
diff --git a/src/main/kotlin/util/SkyblockId.kt b/src/main/kotlin/util/SkyblockId.kt
new file mode 100644
index 0000000..59b1d2c
--- /dev/null
+++ b/src/main/kotlin/util/SkyblockId.kt
@@ -0,0 +1,149 @@
+
+
+@file:UseSerializers(DashlessUUIDSerializer::class)
+
+package moe.nea.firmament.util
+
+import io.github.moulberry.repo.data.NEUItem
+import io.github.moulberry.repo.data.Rarity
+import java.util.UUID
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.UseSerializers
+import kotlinx.serialization.json.Json
+import net.minecraft.component.DataComponentTypes
+import net.minecraft.component.type.NbtComponent
+import net.minecraft.item.ItemStack
+import net.minecraft.nbt.NbtCompound
+import net.minecraft.util.Identifier
+import moe.nea.firmament.repo.set
+import moe.nea.firmament.util.json.DashlessUUIDSerializer
+
+/**
+ * A skyblock item id, as used by the NEU repo.
+ * This is not exactly the format used by HyPixel, but is mostly the same.
+ * Usually this id splits an id used by HyPixel into more sub items. For example `PET` becomes `$PET_ID;$PET_RARITY`,
+ * with those values extracted from other metadata.
+ */
+@JvmInline
+@Serializable
+value class SkyblockId(val neuItem: String) {
+    val identifier
+        get() = Identifier.of("skyblockitem",
+                              neuItem.lowercase().replace(";", "__")
+                                  .replace(":", "___")
+                                  .replace(illlegalPathRegex) {
+                                      it.value.toCharArray()
+                                          .joinToString("") { "__" + it.code.toString(16).padStart(4, '0') }
+                                  })
+
+    override fun toString(): String {
+        return neuItem
+    }
+
+    /**
+     * A bazaar stock item id, as returned by the HyPixel bazaar api endpoint.
+     * These are not equivalent to the in-game ids, or the NEU repo ids, and in fact, do not refer to items, but instead
+     * to bazaar stocks. The main difference from [SkyblockId]s is concerning enchanted books. There are probably more,
+     * but for now this holds.
+     */
+    @JvmInline
+    @Serializable
+    value class BazaarStock(val bazaarId: String) {
+        fun toRepoId(): SkyblockId {
+            bazaarEnchantmentRegex.matchEntire(bazaarId)?.let {
+                return SkyblockId("${it.groupValues[1]};${it.groupValues[2]}")
+            }
+            return SkyblockId(bazaarId.replace(":", "-"))
+        }
+    }
+
+    companion object {
+        val COINS: SkyblockId = SkyblockId("SKYBLOCK_COIN")
+        private val bazaarEnchantmentRegex = "ENCHANTMENT_(\\D*)_(\\d+)".toRegex()
+        val NULL: SkyblockId = SkyblockId("null")
+        val PET_NULL: SkyblockId = SkyblockId("null_pet")
+        private val illlegalPathRegex = "[^a-z0-9_.-/]".toRegex()
+    }
+}
+
+val NEUItem.skyblockId get() = SkyblockId(skyblockItemId)
+
+@Serializable
+data class HypixelPetInfo(
+    val type: String,
+    val tier: Rarity,
+    val exp: Double = 0.0,
+    val candyUsed: Int = 0,
+    val uuid: UUID? = null,
+) {
+    val skyblockId get() = SkyblockId("${type.uppercase()};${tier.ordinal}")
+}
+
+private val jsonparser = Json { ignoreUnknownKeys = true }
+
+val ItemStack.extraAttributes: NbtCompound
+    get() {
+        val customData = get(DataComponentTypes.CUSTOM_DATA) ?: run {
+            val component = NbtComponent.of(NbtCompound())
+            set(DataComponentTypes.CUSTOM_DATA, component)
+            component
+        }
+        return customData.nbt
+    }
+
+val ItemStack.skyblockUUIDString: String?
+    get() = extraAttributes.getString("uuid")?.takeIf { it.isNotBlank() }
+
+val ItemStack.skyblockUUID: UUID?
+    get() = skyblockUUIDString?.let { UUID.fromString(it) }
+
+val ItemStack.petData: HypixelPetInfo?
+    get() {
+        val jsonString = extraAttributes.getString("petInfo")
+        if (jsonString.isNullOrBlank()) return null
+        return runCatching { jsonparser.decodeFromString<HypixelPetInfo>(jsonString) }
+            .getOrElse { return null }
+    }
+
+fun ItemStack.setSkyBlockFirmamentUiId(uiId: String) = setSkyBlockId(SkyblockId("FIRMAMENT_UI_$uiId"))
+fun ItemStack.setSkyBlockId(skyblockId: SkyblockId): ItemStack {
+    this.extraAttributes["id"] = skyblockId.neuItem
+    return this
+}
+
+val ItemStack.skyBlockId: SkyblockId?
+    get() {
+        return when (val id = extraAttributes.getString("id")) {
+            "" -> {
+                null
+            }
+
+            "PET" -> {
+                petData?.skyblockId ?: SkyblockId.PET_NULL
+            }
+
+            "RUNE", "UNIQUE_RUNE" -> {
+                val runeData = extraAttributes.getCompound("runes")
+                val runeKind = runeData.keys.singleOrNull()
+                if (runeKind == null) SkyblockId("RUNE")
+                else SkyblockId("${runeKind.uppercase()}_RUNE;${runeData.getInt(runeKind)}")
+            }
+
+            "ABICASE" -> {
+                SkyblockId("ABICASE_${extraAttributes.getString("model").uppercase()}")
+            }
+
+            "ENCHANTED_BOOK" -> {
+                val enchantmentData = extraAttributes.getCompound("enchantments")
+                val enchantName = enchantmentData.keys.singleOrNull()
+                if (enchantName == null) SkyblockId("ENCHANTED_BOOK")
+                else SkyblockId("${enchantName.uppercase()};${enchantmentData.getInt(enchantName)}")
+            }
+
+            // TODO: PARTY_HAT_CRAB{,_ANIMATED,_SLOTH},POTION
+            else -> {
+                SkyblockId(id)
+            }
+        }
+    }
+