6 files changed, 172 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/FirmamentDataComponentTypes.kt b/src/main/kotlin/util/mc/FirmamentDataComponentTypes.kt
index 012f52e..0866665 100644
--- a/src/main/kotlin/util/mc/FirmamentDataComponentTypes.kt
+++ b/src/main/kotlin/util/mc/FirmamentDataComponentTypes.kt
@@ -1,12 +1,15 @@
 package moe.nea.firmament.util.mc
 
 import com.mojang.serialization.Codec
+import io.netty.buffer.ByteBuf
 import net.minecraft.component.ComponentType
+import net.minecraft.network.codec.PacketCodec
 import net.minecraft.registry.Registries
 import net.minecraft.registry.Registry
 import moe.nea.firmament.Firmament
 import moe.nea.firmament.annotations.Subscribe
 import moe.nea.firmament.events.ClientInitEvent
+import moe.nea.firmament.repo.MiningRepoData
 
 object FirmamentDataComponentTypes {
 
@@ -26,11 +29,32 @@ object FirmamentDataComponentTypes {
 		)
 	}
 
+	fun <T> errorCodec(message: String): PacketCodec<in ByteBuf, T> =
+		object : PacketCodec<ByteBuf, T> {
+			override fun decode(buf: ByteBuf?): T? {
+				error(message)
+			}
+
+			override fun encode(buf: ByteBuf?, value: T?) {
+				error(message)
+			}
+		}
+
+	fun <T, B : ComponentType.Builder<T>> B.neverEncode(message: String = "This element should never be encoded or decoded"): B {
+		packetCodec(errorCodec(message))
+		codec(null)
+		return this
+	}
+
 	val IS_BROKEN = register<Boolean>(
 		"is_broken"
 	) {
 		it.codec(Codec.BOOL.fieldOf("is_broken").codec())
 	}
 
+	val CUSTOM_MINING_BLOCK_DATA = register<MiningRepoData.CustomMiningBlock>("custom_mining_block") {
+		it.neverEncode()
+	}
+
 
 }
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)
diff --git a/src/main/kotlin/util/mc/asFakeServer.kt b/src/main/kotlin/util/mc/asFakeServer.kt
new file mode 100644
index 0000000..d3811bd
--- /dev/null
+++ b/src/main/kotlin/util/mc/asFakeServer.kt
@@ -0,0 +1,37 @@
+package moe.nea.firmament.util.mc
+
+import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource
+import net.minecraft.server.command.CommandOutput
+import net.minecraft.server.command.ServerCommandSource
+import net.minecraft.text.Text
+
+fun FabricClientCommandSource.asFakeServer(): ServerCommandSource {
+	val source = this
+	return ServerCommandSource(
+		object : CommandOutput {
+			override fun sendMessage(message: Text?) {
+				source.player.sendMessage(message, false)
+			}
+
+			override fun shouldReceiveFeedback(): Boolean {
+				return true
+			}
+
+			override fun shouldTrackOutput(): Boolean {
+				return true
+			}
+
+			override fun shouldBroadcastConsoleToOps(): Boolean {
+				return true
+			}
+		},
+		source.position,
+		source.rotation,
+		null,
+		0,
+		"FakeServerCommandSource",
+		Text.literal("FakeServerCommandSource"),
+		null,
+		source.player
+	)
+}