Added closing check for inventories and menu sessions

JVerbruggen · Jun 23, 2024 · ef8d072 · ef8d072
1 parent 9dfc826
commit ef8d072
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/...om/jverbruggen/jrides/control/uiinterface/menu/button/event/ButtonClickEventListener.java b/...om/jverbruggen/jrides/control/uiinterface/menu/button/event/ButtonClickEventListener.java
@@ -107,6 +107,10 @@ private void onItemTowardsMenu(InventoryInteractEvent event, Inventory destinati
     @EventHandler
     public void onClose(InventoryCloseEvent event){
         Player player = playerManager.getPlayer((org.bukkit.entity.Player) event.getPlayer());
-        menuSessionManager.removeOpenMenu(player);
+        Menu menu = menuSessionManager.getOpenMenu(player);
+
+        if(menu.matchesInventory(event.getView())){
+            menuSessionManager.removeOpenMenu(player);
+        }
     }
 }
diff --git a/src/main/java/com/jverbruggen/jrides/models/menu/BaseMenu.java b/src/main/java/com/jverbruggen/jrides/models/menu/BaseMenu.java
@@ -20,6 +20,7 @@
 import com.jverbruggen.jrides.api.JRidesPlayer;
 import org.bukkit.Bukkit;
 import org.bukkit.inventory.Inventory;
+import org.bukkit.inventory.InventoryView;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -60,6 +61,11 @@ public Inventory getInventoryFor(JRidesPlayer player){
         return inventory;
     }
 
+    @Override
+    public boolean matchesInventory(InventoryView inventoryView) {
+        return inventoryView.getTitle().equalsIgnoreCase(inventoryTitle);
+    }
+
     @Override
     public Menu addButton(MenuButton button) {
         buttons.put(button.getUuid(), button);

diff --git a/src/main/java/com/jverbruggen/jrides/models/menu/Menu.java b/src/main/java/com/jverbruggen/jrides/models/menu/Menu.java
@@ -20,6 +20,7 @@
 import com.jverbruggen.jrides.api.JRidesPlayer;
 import com.jverbruggen.jrides.models.entity.Player;
 import org.bukkit.inventory.Inventory;
+import org.bukkit.inventory.InventoryView;
 
 import java.util.Map;
 import java.util.UUID;
@@ -30,6 +31,7 @@ public interface Menu {
     void removeSession(JRidesPlayer player);
 
     Inventory getInventoryFor(JRidesPlayer player);
+    boolean matchesInventory(InventoryView inventoryView);
     Menu addButton(MenuButton button);
     MenuButton getButton(UUID buttonUUID);
     void sendUpdate();