4 files changed, 111 insertions, 4 deletions
diff --git a/src/main/kotlin/util/mc/ArmorUtil.kt b/src/main/kotlin/util/mc/ArmorUtil.kt
new file mode 100644
index 0000000..fd1867c
--- /dev/null
+++ b/src/main/kotlin/util/mc/ArmorUtil.kt
@@ -0,0 +1,8 @@
+package moe.nea.firmament.util.mc
+
+import net.minecraft.entity.EquipmentSlot
+import net.minecraft.entity.LivingEntity
+
+val LivingEntity.iterableArmorItems
+	get() = EquipmentSlot.entries.asSequence()
+		.map { it to getEquippedStack(it) }
diff --git a/src/main/kotlin/util/mc/NbtPrism.kt b/src/main/kotlin/util/mc/NbtPrism.kt
new file mode 100644
index 0000000..f034210
--- /dev/null
+++ b/src/main/kotlin/util/mc/NbtPrism.kt
@@ -0,0 +1,91 @@
+package moe.nea.firmament.util.mc
+
+import com.google.gson.Gson
+import com.google.gson.JsonArray
+import com.google.gson.JsonElement
+import com.google.gson.JsonPrimitive
+import com.mojang.brigadier.StringReader
+import com.mojang.brigadier.arguments.ArgumentType
+import com.mojang.brigadier.arguments.StringArgumentType
+import com.mojang.brigadier.context.CommandContext
+import com.mojang.brigadier.suggestion.Suggestions
+import com.mojang.brigadier.suggestion.SuggestionsBuilder
+import com.mojang.serialization.JsonOps
+import java.util.concurrent.CompletableFuture
+import kotlin.collections.indices
+import kotlin.collections.map
+import kotlin.jvm.optionals.getOrNull
+import net.minecraft.nbt.NbtCompound
+import net.minecraft.nbt.NbtElement
+import net.minecraft.nbt.NbtList
+import net.minecraft.nbt.NbtOps
+import net.minecraft.nbt.NbtString
+import moe.nea.firmament.util.Base64Util
+
+class NbtPrism(val path: List<String>) {
+	companion object {
+		fun fromElement(path: JsonElement): NbtPrism? {
+			if (path is JsonArray) {
+				return NbtPrism(path.map { (it as JsonPrimitive).asString })
+			} else if (path is JsonPrimitive && path.isString) {
+				return NbtPrism(path.asString.split("."))
+			}
+			return null
+		}
+	}
+
+	object Argument : ArgumentType<NbtPrism> {
+		override fun parse(reader: StringReader): NbtPrism? {
+			return fromElement(JsonPrimitive(StringArgumentType.string().parse(reader)))
+		}
+
+		override fun getExamples(): Collection<String?>? {
+			return listOf("some.nbt.path", "some.other.*", "some.path.*json.in.a.json.string")
+		}
+	}
+
+	override fun toString(): String {
+		return "Prism($path)"
+	}
+
+	fun access(root: NbtElement): Collection<NbtElement> {
+		var rootSet = mutableListOf(root)
+		var switch = mutableListOf<NbtElement>()
+		for (pathSegment in path) {
+			if (pathSegment == ".") continue
+			if (pathSegment != "*" && pathSegment.startsWith("*")) {
+				if (pathSegment == "*json") {
+					for (element in rootSet) {
+						val eString = element.asString().getOrNull() ?: continue
+						val element = Gson().fromJson(eString, JsonElement::class.java)
+						switch.add(JsonOps.INSTANCE.convertTo(NbtOps.INSTANCE, element))
+					}
+				} else if (pathSegment == "*base64") {
+					for (element in rootSet) {
+						val string = element.asString().getOrNull() ?: continue
+						switch.add(NbtString.of(Base64Util.decodeString(string)))
+					}
+				}
+			}
+			for (element in rootSet) {
+				if (element is NbtList) {
+					if (pathSegment == "*")
+						switch.addAll(element)
+					val index = pathSegment.toIntOrNull() ?: continue
+					if (index !in element.indices) continue
+					switch.add(element[index])
+				}
+				if (element is NbtCompound) {
+					if (pathSegment == "*")
+						element.keys.mapTo(switch) { element.get(it)!! }
+					switch.add(element.get(pathSegment) ?: continue)
+				}
+			}
+			val temp = switch
+			switch = rootSet
+			rootSet = temp
+			switch.clear()
+		}
+		return rootSet
+	}
+}
diff --git a/src/main/kotlin/util/mc/PlayerUtil.kt b/src/main/kotlin/util/mc/PlayerUtil.kt
new file mode 100644
index 0000000..53ef1f4
--- /dev/null
+++ b/src/main/kotlin/util/mc/PlayerUtil.kt
@@ -0,0 +1,7 @@
+package moe.nea.firmament.util.mc
+
+import net.minecraft.entity.EquipmentSlot
+import net.minecraft.entity.player.PlayerEntity
+
+
+val PlayerEntity.mainHandStack get() = this.getEquippedStack(EquipmentSlot.MAINHAND)
diff --git a/src/main/kotlin/util/mc/SNbtFormatter.kt b/src/main/kotlin/util/mc/SNbtFormatter.kt
index e773927..e2c24f6 100644
--- a/src/main/kotlin/util/mc/SNbtFormatter.kt
+++ b/src/main/kotlin/util/mc/SNbtFormatter.kt
@@ -1,5 +1,6 @@
 package moe.nea.firmament.util.mc
 
+import net.minecraft.nbt.AbstractNbtList
 import net.minecraft.nbt.NbtByte
 import net.minecraft.nbt.NbtByteArray
 import net.minecraft.nbt.NbtCompound
@@ -38,7 +39,7 @@ class SNbtFormatter private constructor() : NbtElementVisitor {
 
 
 	override fun visitString(element: NbtString) {
-		result.append(NbtString.escape(element.asString()))
+		result.append(NbtString.escape(element.value))
 	}
 
 	override fun visitByte(element: NbtByte) {
@@ -65,18 +66,18 @@ class SNbtFormatter private constructor() : NbtElementVisitor {
 		result.append(element.doubleValue()).append("d")
 	}
 
-	private fun visitArrayContents(array: List<NbtElement>) {
+	private fun visitArrayContents(array: AbstractNbtList) {
 		array.forEachIndexed { index, element ->
 			writeIndent()
 			element.accept(this)
-			if (array.size != index + 1) {
+			if (array.size() != index + 1) {
 				result.append(",")
 			}
 			result.append("\n")
 		}
 	}
 
-	private fun writeArray(arrayTypeTag: String, array: List<NbtElement>) {
+	private fun writeArray(arrayTypeTag: String, array: AbstractNbtList) {
 		result.append("[").append(arrayTypeTag).append("\n")
 		pushIndent()
 		visitArrayContents(array)