From 94f96bbd40b5d1acea6535b974fb1cc2afb37935 Mon Sep 17 00:00:00 2001
From: WildInterloper <156627888+WildInterloper@users.noreply.github.com>
Date: Tue, 5 Mar 2024 00:31:18 -0500
Subject: [PATCH] Added permission check to PlayerArmor listener

---
 pom.xml                                       |  2 +-
 .../Listeners/PlayerArmor.java                | 20 +++++++++----------
 2 files changed, 10 insertions(+), 12 deletions(-)

diff --git a/pom.xml b/pom.xml
index d0b2b1a..d47a073 100644
--- a/pom.xml
+++ b/pom.xml
@@ -6,7 +6,7 @@
 
   <groupId>me.NVus</groupId>
   <artifactId>NVus_Prison_Setup</artifactId>
-  <version>0.1-SNAPSHOT</version>
+  <version>0.1.1-SNAPSHOT</version>
   <packaging>jar</packaging>
 
   <name>PrisonSetup</name>
diff --git a/src/main/java/me/nvus/nvus_prison_setup/Listeners/PlayerArmor.java b/src/main/java/me/nvus/nvus_prison_setup/Listeners/PlayerArmor.java
index 8c9f2a5..21a6e7c 100644
--- a/src/main/java/me/nvus/nvus_prison_setup/Listeners/PlayerArmor.java
+++ b/src/main/java/me/nvus/nvus_prison_setup/Listeners/PlayerArmor.java
@@ -1,5 +1,6 @@
 package me.nvus.nvus_prison_setup.Listeners;
 
+import org.bukkit.entity.Player;
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.EventPriority;
 import org.bukkit.event.Listener;
@@ -10,18 +11,15 @@ public class PlayerArmor implements Listener {
 
     // Prevent Armor Unequip??
     @EventHandler(priority = EventPriority.LOWEST)
-    public void onInventoryClick(InventoryClickEvent event)
-    {
-        if (event.getClickedInventory().getType() == InventoryType.PLAYER)
-        {
-            if (event.getSlotType() == InventoryType.SlotType.ARMOR)
-            {
-                event.setCancelled(true);
+    public void onInventoryClick(InventoryClickEvent event) {
+        if (event.getClickedInventory() != null && event.getClickedInventory().getType() == InventoryType.PLAYER) {
+            if (event.getSlotType() == InventoryType.SlotType.ARMOR) {
+                Player player = (Player) event.getWhoClicked();
+                // Check if the player is an actual prisoner before cancelling ;)
+                if (player.hasPermission("nvus.prisoner")) {
+                    event.setCancelled(true);
+                }
             }
         }
     }
-
-
-
-
 }