aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/firmament/util/SkyblockId.kt
blob: 3ac1463edecf0e92dfb5d0df204e07ad855aee0d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
/*
 * SPDX-FileCopyrightText: 2023 Linnea Gräf <nea@nea.moe>
 * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

@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)
            }
        }
    }