feat: Add dye logger for armor stands

1 files changed, 19 insertions, 12 deletions

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
 }