feat: Add dye logger for armor standsHEADmaster

4 files changed, 72 insertions, 14 deletions

diff --git a/src/main/kotlin/Firmament.kt b/src/main/kotlin/Firmament.kt
index f43018b..0191036 100644
--- a/src/main/kotlin/Firmament.kt
+++ b/src/main/kotlin/Firmament.kt
@@ -120,9 +120,8 @@ object Firmament {
 	@JvmStatic
 	fun onClientInitialize() {
 		FeatureManager.subscribeEvents()
-		var tick = 0
 		ClientTickEvents.END_CLIENT_TICK.register(ClientTickEvents.EndTick { instance ->
-			TickEvent.publish(TickEvent(tick++))
+			TickEvent.publish(TickEvent(MC.currentTick++))
 		})
 		IDataHolder.registerEvents()
 		RepoManager.initialize()
diff --git a/src/main/kotlin/events/EntityUpdateEvent.kt b/src/main/kotlin/events/EntityUpdateEvent.kt
index d091984..27a90f9 100644
--- a/src/main/kotlin/events/EntityUpdateEvent.kt
+++ b/src/main/kotlin/events/EntityUpdateEvent.kt
@@ -1,9 +1,11 @@
-
 package moe.nea.firmament.events
 
+import com.mojang.datafixers.util.Pair
 import net.minecraft.entity.Entity
+import net.minecraft.entity.EquipmentSlot
 import net.minecraft.entity.LivingEntity
 import net.minecraft.entity.data.DataTracker
+import net.minecraft.item.ItemStack
 import net.minecraft.network.packet.s2c.play.EntityAttributesS2CPacket
 
 /**
@@ -13,19 +15,24 @@ import net.minecraft.network.packet.s2c.play.EntityAttributesS2CPacket
  * *after* the values have been applied to the entity.
  */
 sealed class EntityUpdateEvent : FirmamentEvent() {
-    companion object : FirmamentEventBus<EntityUpdateEvent>()
+	companion object : FirmamentEventBus<EntityUpdateEvent>()
+
+	abstract val entity: Entity
 
-    abstract val entity: Entity
+	data class AttributeUpdate(
+		override val entity: LivingEntity,
+		val attributes: List<EntityAttributesS2CPacket.Entry>,
+	) : EntityUpdateEvent()
 
-    data class AttributeUpdate(
-        override val entity: LivingEntity,
-        val attributes: List<EntityAttributesS2CPacket.Entry>,
-    ) : EntityUpdateEvent()
+	data class TrackedDataUpdate(
+		override val entity: Entity,
+		val trackedValues: List<DataTracker.SerializedEntry<*>>,
+	) : EntityUpdateEvent()
 
-    data class TrackedDataUpdate(
-        override val entity: Entity,
-        val trackedValues: List<DataTracker.SerializedEntry<*>>,
-    ) : EntityUpdateEvent()
+	data class EquipmentUpdate(
+		override val entity: Entity,
+		val newEquipment: List<Pair<EquipmentSlot, ItemStack>>,
+	) : EntityUpdateEvent()
 
-// TODO: onEntityPassengersSet, onEntityAttach?, onEntityEquipmentUpdate, onEntityStatusEffect
+// TODO: onEntityPassengersSet, onEntityAttach?, onEntityStatusEffect
 }
diff --git a/src/main/kotlin/features/debug/AnimatedClothingScanner.kt b/src/main/kotlin/features/debug/AnimatedClothingScanner.kt
new file mode 100644
index 0000000..11b47a9
--- /dev/null
+++ b/src/main/kotlin/features/debug/AnimatedClothingScanner.kt
@@ -0,0 +1,51 @@
+package moe.nea.firmament.features.debug
+
+import net.minecraft.component.DataComponentTypes
+import net.minecraft.entity.Entity
+import moe.nea.firmament.annotations.Subscribe
+import moe.nea.firmament.commands.thenExecute
+import moe.nea.firmament.commands.thenLiteral
+import moe.nea.firmament.events.CommandEvent
+import moe.nea.firmament.events.EntityUpdateEvent
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.skyBlockId
+import moe.nea.firmament.util.tr
+
+object AnimatedClothingScanner {
+
+	var observedEntity: Entity? = null
+
+	@OptIn(ExperimentalStdlibApi::class)
+	@Subscribe
+	fun onUpdate(event: EntityUpdateEvent) {
+		if (event.entity != observedEntity) return
+		if (event is EntityUpdateEvent.EquipmentUpdate) {
+			event.newEquipment.forEach {
+				val id = it.second.skyBlockId?.neuItem
+				val colour = it.second.get(DataComponentTypes.DYED_COLOR)
+					?.rgb?.toHexString(HexFormat.UpperCase)
+					?.let { " #$it" } ?: ""
+				MC.sendChat(tr("firmament.fitstealer.update",
+				               "[FIT CHECK][${MC.currentTick}] ${it.first.asString()} => ${id}${colour}"))
+			}
+		}
+	}
+
+	@Subscribe
+	fun onSubCommand(event: CommandEvent.SubCommand) {
+		event.subcommand("dev") {
+			thenLiteral("stealthisfit") {
+				thenExecute {
+					observedEntity =
+						if (observedEntity == null) MC.instance.targetedEntity else null
+
+					MC.sendChat(
+						observedEntity?.let {
+							tr("firmament.fitstealer.targeted", "Observing the equipment of ${it.name}.")
+						} ?: tr("firmament.fitstealer.targetlost", "No longer logging equipment."),
+					)
+				}
+			}
+		}
+	}
+}
diff --git a/src/main/kotlin/util/MC.kt b/src/main/kotlin/util/MC.kt
index a0d2fc0..c1a5e65 100644
--- a/src/main/kotlin/util/MC.kt
+++ b/src/main/kotlin/util/MC.kt
@@ -113,6 +113,7 @@ object MC {
 	val defaultRegistries: RegistryWrapper.WrapperLookup by lazy { BuiltinRegistries.createWrapperLookup() }
 	inline val currentOrDefaultRegistries get() = currentRegistries ?: defaultRegistries
 	val defaultItems: RegistryWrapper.Impl<Item> by lazy { defaultRegistries.getOrThrow(RegistryKeys.ITEM) }
+	var currentTick = 0
 	var lastWorld: World? = null
 		get() {
 			field = world ?: field