1 files changed, 270 insertions, 0 deletions
diff --git a/src/main/kotlin/features/debug/itemeditor/LegacyItemExporter.kt b/src/main/kotlin/features/debug/itemeditor/LegacyItemExporter.kt
new file mode 100644
index 0000000..3cd1ce8
--- /dev/null
+++ b/src/main/kotlin/features/debug/itemeditor/LegacyItemExporter.kt
@@ -0,0 +1,270 @@
+package moe.nea.firmament.features.debug.itemeditor
+
+import kotlinx.serialization.json.JsonArray
+import kotlinx.serialization.json.JsonElement
+import kotlinx.serialization.json.buildJsonObject
+import kotlinx.serialization.json.put
+import kotlin.concurrent.thread
+import net.minecraft.component.DataComponentTypes
+import net.minecraft.item.ItemStack
+import net.minecraft.nbt.NbtCompound
+import net.minecraft.nbt.NbtElement
+import net.minecraft.nbt.NbtInt
+import net.minecraft.nbt.NbtOps
+import net.minecraft.nbt.NbtString
+import net.minecraft.text.Text
+import net.minecraft.util.Unit
+import moe.nea.firmament.Firmament
+import moe.nea.firmament.annotations.Subscribe
+import moe.nea.firmament.events.ClientStartedEvent
+import moe.nea.firmament.features.debug.ExportedTestConstantMeta
+import moe.nea.firmament.repo.SBItemStack
+import moe.nea.firmament.util.HypixelPetInfo
+import moe.nea.firmament.util.LegacyTagWriter.Companion.toLegacyString
+import moe.nea.firmament.util.StringUtil.words
+import moe.nea.firmament.util.directLiteralStringContent
+import moe.nea.firmament.util.extraAttributes
+import moe.nea.firmament.util.getLegacyFormatString
+import moe.nea.firmament.util.json.toJsonArray
+import moe.nea.firmament.util.mc.displayNameAccordingToNbt
+import moe.nea.firmament.util.mc.loreAccordingToNbt
+import moe.nea.firmament.util.mc.toNbtList
+import moe.nea.firmament.util.skyBlockId
+import moe.nea.firmament.util.skyblock.Rarity
+import moe.nea.firmament.util.transformEachRecursively
+import moe.nea.firmament.util.unformattedString
+
+class LegacyItemExporter private constructor(var itemStack: ItemStack) {
+	init {
+	    require(!itemStack.isEmpty)
+	}
+	var lore = itemStack.loreAccordingToNbt
+	var name = itemStack.displayNameAccordingToNbt
+	val extraAttribs = itemStack.extraAttributes.copy()
+	val legacyNbt = NbtCompound()
+	val warnings = mutableListOf<String>()
+
+	// TODO: check if lore contains non 1.8.9 able hex codes and emit lore in overlay files if so
+
+	fun preprocess() {
+		// TODO: split up preprocess steps into preprocess actions that can be toggled in a ui
+		extraAttribs.remove("timestamp")
+		extraAttribs.remove("uuid")
+		extraAttribs.remove("modifier")
+		extraAttribs.getString("petInfo").ifPresent { petInfoJson ->
+			var petInfo = Firmament.json.decodeFromString<HypixelPetInfo>(petInfoJson)
+			petInfo = petInfo.copy(candyUsed = 0, heldItem = null, exp = 0.0, active = null, uuid = null)
+			extraAttribs.putString("petInfo", Firmament.tightJson.encodeToString(petInfo))
+		}
+		itemStack.skyBlockId?.let {
+			extraAttribs.putString("id", it.neuItem)
+		}
+		trimLore()
+		itemStack.loreAccordingToNbt = itemStack.item.defaultStack.loreAccordingToNbt
+		itemStack.remove(DataComponentTypes.CUSTOM_NAME)
+	}
+
+	fun trimLore() {
+		val rarityIdx = lore.indexOfLast {
+			val firstWordInLine = it.unformattedString.words().filter { it.length > 2 }.firstOrNull()
+			firstWordInLine?.let(Rarity::fromString) != null
+		}
+		if (rarityIdx >= 0) {
+			lore = lore.subList(0, rarityIdx + 1)
+		}
+
+		trimStats()
+
+		deleteLineUntilNextSpace { it.startsWith("Held Item: ") }
+		deleteLineUntilNextSpace { it.startsWith("Progress to Level ") }
+		deleteLineUntilNextSpace { it.startsWith("MAX LEVEL") }
+		deleteLineUntilNextSpace { it.startsWith("Click to view recipe!") }
+		collapseWhitespaces()
+
+		name = name.transformEachRecursively {
+			var string = it.directLiteralStringContent ?: return@transformEachRecursively it
+			string = string.replace("Lvl \\d+".toRegex(), "Lvl {LVL}")
+			Text.literal(string).setStyle(it.style)
+		}
+
+		if (lore.isEmpty())
+			lore = listOf(Text.empty())
+	}
+
+	private fun trimStats() {
+		val lore = this.lore.toMutableList()
+		for (index in lore.indices) {
+			val value = lore[index]
+			val statLine = SBItemStack.parseStatLine(value)
+			if (statLine == null) break
+			val v = value.copy()
+			require(value.directLiteralStringContent == "")
+			v.siblings.removeIf { it.directLiteralStringContent!!.contains("(") }
+			val last = v.siblings.last()
+			v.siblings[v.siblings.lastIndex] =
+				Text.literal(last.directLiteralStringContent!!.trimEnd())
+					.setStyle(last.style)
+			lore[index] = v
+		}
+		this.lore = lore
+	}
+
+	fun collapseWhitespaces() {
+		lore = (listOf(null as Text?) + lore).zipWithNext()
+			.filter { !it.first?.unformattedString.isNullOrBlank() || !it.second?.unformattedString.isNullOrBlank() }
+			.map { it.second!! }
+	}
+
+	fun deleteLineUntilNextSpace(search: (String) -> Boolean) {
+		val idx = lore.indexOfFirst { search(it.unformattedString) }
+		if (idx < 0) return
+		val l = lore.toMutableList()
+		val p = l.subList(idx, l.size)
+		val nextBlank = p.indexOfFirst { it.unformattedString.isEmpty() }
+		if (nextBlank < 0)
+			p.clear()
+		else
+			p.subList(0, nextBlank).clear()
+		lore = l
+	}
+
+	fun processNbt() {
+		// TODO: calculate hideflags
+		legacyNbt.put("HideFlags", NbtInt.of(254))
+		copyUnbreakable()
+		copyItemModel()
+		copyExtraAttributes()
+		copyLegacySkullNbt()
+		copyDisplay()
+		copyEnchantments()
+		copyEnchantGlint()
+		// TODO: copyDisplay
+	}
+
+	private fun copyItemModel() {
+		val itemModel = itemStack.get(DataComponentTypes.ITEM_MODEL) ?: return
+		legacyNbt.put("ItemModel", NbtString.of(itemModel.toString()))
+	}
+
+	private fun copyDisplay() {
+		legacyNbt.put("display", NbtCompound().apply {
+			put("Lore", lore.map { NbtString.of(it.getLegacyFormatString(trimmed = true)) }.toNbtList())
+			putString("Name", name.getLegacyFormatString(trimmed = true))
+		})
+	}
+
+	fun exportModernSnbt(): NbtElement {
+		val overlay = ItemStack.CODEC.encodeStart(NbtOps.INSTANCE, itemStack)
+			.orThrow
+		val overlayWithVersion =
+			ExportedTestConstantMeta.SOURCE_CODEC.encode(ExportedTestConstantMeta.current, NbtOps.INSTANCE, overlay)
+				.orThrow
+		return overlayWithVersion
+	}
+
+	fun prepare() {
+		preprocess()
+		processNbt()
+		itemStack.extraAttributes = extraAttribs
+	}
+
+	fun exportJson(): JsonElement {
+		return buildJsonObject {
+			val (itemId, damage) = legacyifyItemStack()
+			put("itemid", itemId)
+			put("displayname", name.getLegacyFormatString(trimmed = true))
+			put("nbttag", legacyNbt.toLegacyString())
+			put("damage", damage)
+			put("lore", lore.map { it.getLegacyFormatString(trimmed = true) }.toJsonArray())
+			val sbId = itemStack.skyBlockId
+			if (sbId == null)
+				warnings.add("Could not find skyblock id")
+			put("internalname", sbId?.neuItem)
+			put("clickcommand", "")
+			put("crafttext", "")
+			put("modver", "Firmament ${Firmament.version.friendlyString}")
+			put("infoType", "")
+			put("info", JsonArray(listOf()))
+		}
+
+	}
+
+	companion object {
+		fun createExporter(itemStack: ItemStack): LegacyItemExporter {
+			return LegacyItemExporter(itemStack.copy()).also { it.prepare() }
+		}
+
+		@Subscribe
+		fun load(event: ClientStartedEvent) {
+			thread(start = true, name = "ItemExporter Meta Load Thread") {
+				LegacyItemData.itemLut
+			}
+		}
+	}
+
+	fun copyEnchantGlint() {
+		if (itemStack.get(DataComponentTypes.ENCHANTMENT_GLINT_OVERRIDE) == true) {
+			val ench = legacyNbt.getListOrEmpty("ench")
+			legacyNbt.put("ench", ench)
+		}
+	}
+
+	private fun copyUnbreakable() {
+		if (itemStack.get(DataComponentTypes.UNBREAKABLE) == Unit.INSTANCE) {
+			legacyNbt.putBoolean("Unbreakable", true)
+		}
+	}
+
+	fun copyEnchantments() {
+		val enchantments = itemStack.get(DataComponentTypes.ENCHANTMENTS)?.takeIf { !it.isEmpty } ?: return
+		val enchTag = legacyNbt.getListOrEmpty("ench")
+		legacyNbt.put("ench", enchTag)
+		enchantments.enchantmentEntries.forEach { entry ->
+			val id = entry.key.key.get().value
+			val legacyId = LegacyItemData.enchantmentLut[id]
+			if (legacyId == null) {
+				warnings.add("Could not find legacy enchantment id for ${id}")
+				return@forEach
+			}
+			enchTag.add(NbtCompound().apply {
+				putShort("lvl", entry.intValue.toShort())
+				putShort(
+					"id",
+					legacyId.id.toShort()
+				)
+			})
+		}
+	}
+
+	fun copyExtraAttributes() {
+		legacyNbt.put("ExtraAttributes", extraAttribs)
+	}
+
+	fun copyLegacySkullNbt() {
+		val profile = itemStack.get(DataComponentTypes.PROFILE) ?: return
+		legacyNbt.put("SkullOwner", NbtCompound().apply {
+			profile.id.ifPresent {
+				putString("Id", it.toString())
+			}
+			putBoolean("hypixelPopulated", true)
+			put("Properties", NbtCompound().apply {
+				profile.properties().forEach { prop, value ->
+					val list = getListOrEmpty(prop)
+					put(prop, list)
+					list.add(NbtCompound().apply {
+						value.signature?.let {
+							putString("Signature", it)
+						}
+						putString("Value", value.value)
+						putString("Name", value.name)
+					})
+				}
+			})
+		})
+	}
+
+	fun legacyifyItemStack(): LegacyItemData.LegacyItemType {
+		// TODO: add a default here
+		return LegacyItemData.itemLut[itemStack.item]!!
+	}
+}