1 files changed, 41 insertions, 0 deletions
diff --git a/mod/src/main/java/moe/ymc/acron/mixin/LivingEntityMixin.java b/mod/src/main/java/moe/ymc/acron/mixin/LivingEntityMixin.java
new file mode 100644
index 0000000..9e16569
--- /dev/null
+++ b/mod/src/main/java/moe/ymc/acron/mixin/LivingEntityMixin.java
@@ -0,0 +1,41 @@
+package moe.ymc.acron.mixin;
+
+import moe.ymc.acron.s2c.Entity;
+import moe.ymc.acron.s2c.EventQueue;
+import moe.ymc.acron.s2c.event.EventEntityDeath;
+import net.minecraft.entity.EntityType;
+import net.minecraft.entity.LivingEntity;
+import net.minecraft.entity.damage.DamageSource;
+import net.minecraft.entity.damage.DamageTracker;
+import net.minecraft.world.World;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(LivingEntity.class)
+public abstract class LivingEntityMixin extends net.minecraft.entity.Entity {
+    private static final Logger AC_LOGGER = LogManager.getLogger();
+
+    @Shadow public abstract DamageTracker getDamageTracker();
+
+    public LivingEntityMixin(EntityType<?> type, World world) {
+        super(type, world);
+    }
+
+    // The original onDeath() will call getDamageTracker().update(),
+    // which clears all recent damages, making the getDeathMessage()
+    // output always generic.
+    // Thus, we need to use @At("HEAD") to get the injection called
+    // before it does anything else.
+    @Inject(at = @At("HEAD"), method = "onDeath")
+    public void onDeath(DamageSource source, CallbackInfo ci) {
+        AC_LOGGER.debug("onDeath[{}]",
+                getUuid());
+        EventQueue.enqueue(new EventEntityDeath(new Entity(this),
+                getDamageTracker().getDeathMessage().getString()));
+    }
+}
+\ No newline at end of file