From c9a2f082e281161c50d1aa797af26c94be3e0456 Mon Sep 17 00:00:00 2001
From: EpicKnarvik97 <kristian.knarvik@knett.no>
Date: Tue, 9 Jan 2024 20:24:22 +0100
Subject: [PATCH] Adds item clearing on teleportation

---
 .../clearonworldguard/WorldGuardListener.java | 46 ++++++++++++++++---
 1 file changed, 39 insertions(+), 7 deletions(-)

diff --git a/src/main/java/net/knarcraft/clearonworldguard/WorldGuardListener.java b/src/main/java/net/knarcraft/clearonworldguard/WorldGuardListener.java
index 44bb9bd..c860833 100644
--- a/src/main/java/net/knarcraft/clearonworldguard/WorldGuardListener.java
+++ b/src/main/java/net/knarcraft/clearonworldguard/WorldGuardListener.java
@@ -8,13 +8,16 @@ import com.sk89q.worldguard.protection.regions.ProtectedRegion;
 import com.sk89q.worldguard.protection.regions.RegionContainer;
 import com.sk89q.worldguard.protection.regions.RegionQuery;
 import org.bukkit.Bukkit;
+import org.bukkit.Location;
 import org.bukkit.World;
 import org.bukkit.configuration.ConfigurationSection;
 import org.bukkit.configuration.file.FileConfiguration;
+import org.bukkit.entity.Player;
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.EventPriority;
 import org.bukkit.event.Listener;
 import org.bukkit.event.player.PlayerMoveEvent;
+import org.bukkit.event.player.PlayerTeleportEvent;
 import org.jetbrains.annotations.NotNull;
 
 import java.util.HashMap;
@@ -66,10 +69,10 @@ public class WorldGuardListener implements Listener {
 
     /**
      * Loads clear regions for a world
-     * 
-     * @param world <p>The world to load regions for</p>
+     *
+     * @param world         <p>The world to load regions for</p>
      * @param worldsSection <p>The configuration section listing all worlds and regions</p>
-     * @param worldId <p>The user-specified identifier for the currently processed world</p>
+     * @param worldId       <p>The user-specified identifier for the currently processed world</p>
      */
     private void loadRegions(@NotNull World world, @NotNull ConfigurationSection worldsSection, @NotNull String worldId) {
         // Get a region manager for the world
@@ -99,17 +102,46 @@ public class WorldGuardListener implements Listener {
         if (event.getTo() == null || event.getFrom().getWorld() == null) {
             return;
         }
-        World playerWorld = event.getFrom().getWorld();
 
-        ApplicableRegionSet setFrom = query.getApplicableRegions(BukkitAdapter.adapt(event.getFrom()));
-        ApplicableRegionSet setTo = query.getApplicableRegions(BukkitAdapter.adapt(event.getTo()));
+        clearIfClearRegionChange(event.getFrom(), event.getTo(), event.getPlayer());
+    }
+
+    @EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
+    public void onTeleport(@NotNull PlayerTeleportEvent event) {
+        Location fromLocation = event.getFrom();
+        Location toLocation = event.getTo();
+
+        if (toLocation == null || fromLocation.equals(toLocation)) {
+            return;
+        }
+        clearIfClearRegionChange(fromLocation, toLocation, event.getPlayer());
+    }
+
+    /**
+     * Clears the given player's inventory if they are leaving or entering a clear region
+     *
+     * @param fromLocation <p>The location the player moved or teleported from</p>
+     * @param toLocation   <p>The location the player moved or teleported to</p>
+     * @param player       <p>The player that's moving</p>
+     */
+    private void clearIfClearRegionChange(@NotNull Location fromLocation, @NotNull Location toLocation,
+                                          @NotNull Player player) {
+        World playerWorld = fromLocation.getWorld();
+        if (playerWorld == null) {
+            ClearOnWorldGuard.logger().log(Level.WARNING, "Unable to check region change, as location " +
+                    fromLocation + " has no world.");
+            return;
+        }
+
+        ApplicableRegionSet setFrom = query.getApplicableRegions(BukkitAdapter.adapt(fromLocation));
+        ApplicableRegionSet setTo = query.getApplicableRegions(BukkitAdapter.adapt(toLocation));
 
         Set<ProtectedRegion> fromRegions = getOccupiedClearRegions(playerWorld, setFrom.getRegions());
         Set<ProtectedRegion> toRegions = getOccupiedClearRegions(playerWorld, setTo.getRegions());
 
         // If the player is in one or more clear regions, clear unless the clear regions are the same
         if ((!fromRegions.isEmpty() || !toRegions.isEmpty()) && !fromRegions.equals(toRegions)) {
-            event.getPlayer().getInventory().clear();
+            player.getInventory().clear();
         }
     }