aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/events/EntityUpdateEvent.kt
blob: fec2fa5128632cef141468fd07c0fa462a1179c8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
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
import moe.nea.firmament.annotations.Subscribe
import moe.nea.firmament.util.MC

/**
 * This event is fired when some entity properties are updated.
 * It is not fired for common changes like position, but is for less common ones,
 * like health, tracked data, names, equipment. It is always fired
 * *after* the values have been applied to the entity.
 */
sealed class EntityUpdateEvent : FirmamentEvent() {
	companion object : FirmamentEventBus<EntityUpdateEvent>() {
		@Subscribe
		fun onPlayerInventoryUpdate(event: PlayerInventoryUpdate) {
			val p = MC.player ?: return
			val updatedSlots = listOf(
				EquipmentSlot.HEAD to 39,
				EquipmentSlot.CHEST to 38,
				EquipmentSlot.LEGS to 37,
				EquipmentSlot.FEET to 36,
				EquipmentSlot.OFFHAND to 40,
				EquipmentSlot.MAINHAND to p.inventory.selectedSlot, // TODO: also equipment update when you swap your selected slot perhaps
			).mapNotNull { (slot, stackIndex) ->
				val slotIndex = p.playerScreenHandler.getSlotIndex(p.inventory, stackIndex).asInt
				event.getOrNull(slotIndex)?.let {
					Pair.of(slot, it)
				}
			}
			if (updatedSlots.isNotEmpty())
				publish(EquipmentUpdate(p, updatedSlots))
		}
	}

	abstract val entity: Entity

	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 EquipmentUpdate(
		override val entity: Entity,
		val newEquipment: List<Pair<EquipmentSlot, ItemStack>>,
	) : EntityUpdateEvent()

// TODO: onEntityPassengersSet, onEntityAttach?, onEntityStatusEffect
}