Only teleport when damage cause is void

GeyserMC · Jan 19, 2025 · acf0853 · acf0853
1 parent d4415e0
commit acf0853
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/main/java/org/geysermc/globallinkserver/handler/TeleportToSpawnHandler.java b/src/main/java/org/geysermc/globallinkserver/handler/TeleportToSpawnHandler.java
@@ -11,6 +11,7 @@
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.Listener;
 import org.bukkit.event.entity.EntityDamageEvent;
+import org.bukkit.event.entity.PlayerDeathEvent;
 import org.bukkit.event.player.PlayerJoinEvent;
 import org.jspecify.annotations.NullMarked;
 
@@ -33,10 +34,19 @@ public void onPlayerJoin(PlayerJoinEvent event) {
     public void onEntityDamage(EntityDamageEvent event) {
         if (event.getEntity() instanceof Player player) {
             event.setCancelled(true);
-            teleport(player);
+
+            if (event.getCause() == EntityDamageEvent.DamageCause.VOID) {
+                teleport(player);
+            }
         }
     }
 
+    @EventHandler
+    public void onPlayerDeath(PlayerDeathEvent event) {
+        event.setCancelled(true);
+        teleport(event.getEntity());
+    }
+
     private void teleport(Player player) {
         player.teleport(spawn);
         player.setFallDistance(0);