feat: Add LivingDeathEvent and EntityAddedEvent

fabric/src/main/java/net/blay09/mods/balm/fabric/event/FabricBalmCommonEvents.java

@@ -4,7 +4,9 @@
 import net.blay09.mods.balm.api.event.*;
 import net.blay09.mods.balm.api.event.server.ServerStartedEvent;
 import net.blay09.mods.balm.api.event.server.ServerStoppedEvent;
+import net.fabricmc.fabric.api.entity.event.v1.ServerLivingEntityEvents;
 import net.fabricmc.fabric.api.entity.event.v1.ServerPlayerEvents;
+import net.fabricmc.fabric.api.event.lifecycle.v1.ServerEntityEvents;
 import net.fabricmc.fabric.api.event.lifecycle.v1.ServerLifecycleEvents;
 import net.fabricmc.fabric.api.event.lifecycle.v1.ServerTickEvents;
 import net.fabricmc.fabric.api.event.player.PlayerBlockBreakEvents;
@@ -119,5 +121,17 @@ public static void registerEvents(FabricBalmEvents events) {
             final PlayerRespawnEvent event = new PlayerRespawnEvent(oldPlayer, newPlayer);
             events.fireEventHandlers(event);
         }));
+
+        events.registerEvent(EntityAddedEvent.class, () -> ServerLivingEntityEvents.ALLOW_DEATH.register((entity, damageSource, damageAmount) -> {
+            final var event = new LivingDeathEvent(entity, damageSource);
+            events.fireEventHandlers(event);
+            return !event.isCanceled();
+        }));
+
+        events.registerEvent(EntityAddedEvent.class, () -> ServerEntityEvents.ENTITY_LOAD.register((entity, level) -> {
+            final EntityAddedEvent event = new EntityAddedEvent(entity, level);
+            events.fireEventHandlers(event);
+            // TODO cannot cancel on fabric
+        }));
     }
 }

forge/src/main/java/net/blay09/mods/balm/forge/event/ForgeBalmCommonEvents.java

@@ -11,6 +11,7 @@
 import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraftforge.common.MinecraftForge;
 import net.minecraftforge.event.TickEvent;
+import net.minecraftforge.event.entity.EntityJoinLevelEvent;
 import net.minecraftforge.event.entity.item.ItemTossEvent;
 import net.minecraftforge.event.entity.player.AttackEntityEvent;
 import net.minecraftforge.event.entity.player.PlayerEvent;
@@ -255,6 +256,26 @@ public static void registerEvents(ForgeBalmEvents events) {
                 events.fireEventHandlers(priority, event);
             });
         });
+
+        events.registerEvent(LivingDeathEvent.class, priority -> {
+            MinecraftForge.EVENT_BUS.addListener(ForgeBalmEvents.toForge(priority), (net.minecraftforge.event.entity.living.LivingDeathEvent orig) -> {
+                final LivingDeathEvent event = new LivingDeathEvent(orig.getEntity(), orig.getSource());
+                events.fireEventHandlers(priority, event);
+                if (event.isCanceled()) {
+                    orig.setCanceled(true);
+                }
+            });
+        });
+
+        events.registerEvent(EntityAddedEvent.class, priority -> {
+            MinecraftForge.EVENT_BUS.addListener(ForgeBalmEvents.toForge(priority), (EntityJoinLevelEvent orig) -> {
+                final EntityAddedEvent event = new EntityAddedEvent(orig.getEntity(), orig.getLevel());
+                events.fireEventHandlers(priority, event);
+                if (event.isCanceled()) {
+                    orig.setCanceled(true);
+                }
+            });
+        });
     }
 
 }

shared/src/main/java/net/blay09/mods/balm/api/event/EntityAddedEvent.java

@@ -0,0 +1,22 @@
+package net.blay09.mods.balm.api.event;
+
+import net.minecraft.world.entity.Entity;
+import net.minecraft.world.level.Level;
+
+public class EntityAddedEvent extends BalmEvent {
+    private final Entity entity;
+    private final Level level;
+
+    public EntityAddedEvent(Entity entity, Level level) {
+        this.entity = entity;
+        this.level = level;
+    }
+
+    public Entity getEntity() {
+        return entity;
+    }
+
+    public Level getLevel() {
+        return level;
+    }
+}

shared/src/main/java/net/blay09/mods/balm/api/event/LivingDeathEvent.java

@@ -0,0 +1,22 @@
+package net.blay09.mods.balm.api.event;
+
+import net.minecraft.world.damagesource.DamageSource;
+import net.minecraft.world.entity.LivingEntity;
+
+public class LivingDeathEvent extends BalmEvent {
+    private final LivingEntity entity;
+    private final DamageSource damageSource;
+
+    public LivingDeathEvent(LivingEntity entity, DamageSource damageSource) {
+        this.entity = entity;
+        this.damageSource = damageSource;
+    }
+
+    public LivingEntity getEntity() {
+        return entity;
+    }
+
+    public DamageSource getDamageSource() {
+        return damageSource;
+    }
+}