1 files changed, 90 insertions, 0 deletions
diff --git a/mod/src/main/kotlin/moe/nea/ledger/ItemUtil.kt b/mod/src/main/kotlin/moe/nea/ledger/ItemUtil.kt
new file mode 100644
index 0000000..a3d8162
--- /dev/null
+++ b/mod/src/main/kotlin/moe/nea/ledger/ItemUtil.kt
@@ -0,0 +1,90 @@
+package moe.nea.ledger
+
+import net.minecraft.inventory.IInventory
+import net.minecraft.item.ItemStack
+import net.minecraft.nbt.NBTTagCompound
+
+
+fun ItemStack.getExtraAttributes(): NBTTagCompound {
+	val nbt = this.tagCompound ?: return NBTTagCompound()
+	return nbt.getCompoundTag("ExtraAttributes")
+}
+
+fun ItemStack?.getInternalId(): ItemId? {
+	if (this == null) return null
+	val extraAttributes = getExtraAttributes()
+	var id = extraAttributes.getString("id")
+	id = id.takeIf { it.isNotBlank() }
+	if (id == "PET") {
+		id = getPetId() ?: id
+	}
+	if (id == "ENCHANTED_BOOK") {
+		id = getEnchanments().entries.singleOrNull()?.let {
+			"${it.key};${it.value}".uppercase()
+		}
+	}
+	return id?.let(::ItemId)
+}
+
+fun ItemStack.getEnchanments(): Map<String, Int> {
+	val enchantments = getExtraAttributes().getCompoundTag("enchantments")
+	return enchantments.keySet.associateWith { enchantments.getInteger(it) }
+}
+
+class PetInfo {
+	var type: String? = null
+	var tier: String? = null
+}
+
+fun ItemStack.getPetId(): String? {
+	val petInfoStr = getExtraAttributes().getString("petInfo")
+	val petInfo = runCatching {
+		Ledger.gson.fromJson(petInfoStr,
+		                     PetInfo::class.java)
+	}.getOrNull() // TODO: error reporting to sentry
+	if (petInfo?.type == null || petInfo.tier == null) return null
+	return petInfo.type + ";" + rarityToIndex(petInfo.tier ?: "")
+}
+
+fun rarityToIndex(rarity: String): Int {
+	return when (rarity) {
+		"COMMON" -> 0
+		"UNCOMMON" -> 1
+		"RARE" -> 2
+		"EPIC" -> 3
+		"LEGENDARY" -> 4
+		"MYTHIC" -> 5
+		else -> -1
+	}
+}
+
+fun ItemStack.getLore(): List<String> {
+	val nbt = this.tagCompound ?: NBTTagCompound()
+	val extraAttributes = nbt.getCompoundTag("display")
+	val lore = extraAttributes.getTagList("Lore", 8)
+	return (0 until lore.tagCount()).map { lore.getStringTagAt(it) }
+}
+
+
+fun IInventory.asIterable(): Iterable<ItemStack?> = object : Iterable<ItemStack?> {
+	override fun iterator(): Iterator<ItemStack?> {
+		return object : Iterator<ItemStack?> {
+			var i = 0
+			override fun hasNext(): Boolean {
+				return i < this@asIterable.sizeInventory
+			}
+
+			override fun next(): ItemStack? {
+				if (!hasNext()) throw NoSuchElementException("$i is out of range for inventory ${this@asIterable}")
+				return this@asIterable.getStackInSlot(i++)
+			}
+		}
+	}
+}
+
+fun ItemStack.getDisplayNameU(): String {
+	val nbt = this.tagCompound ?: NBTTagCompound()
+	val extraAttributes = nbt.getCompoundTag("display")
+	return extraAttributes.getString("Name")
+}
+