From a691235f47c2a5208c97a80c3bc86dbb44d7ff35 Mon Sep 17 00:00:00 2001
From: NichtStudioCode <51272202+NichtStudioCode@users.noreply.github.com>
Date: Sat, 30 Sep 2023 18:08:26 +0200
Subject: [PATCH] Fix #32

---
 .../xyz/xenondevs/invui/window/WindowManager.java    | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/invui-core/src/main/java/xyz/xenondevs/invui/window/WindowManager.java b/invui-core/src/main/java/xyz/xenondevs/invui/window/WindowManager.java
index 06de271..bd8b224 100644
--- a/invui-core/src/main/java/xyz/xenondevs/invui/window/WindowManager.java
+++ b/invui-core/src/main/java/xyz/xenondevs/invui/window/WindowManager.java
@@ -8,11 +8,9 @@ import org.bukkit.event.EventPriority;
 import org.bukkit.event.Listener;
 import org.bukkit.event.entity.EntityPickupItemEvent;
 import org.bukkit.event.entity.PlayerDeathEvent;
-import org.bukkit.event.inventory.InventoryClickEvent;
-import org.bukkit.event.inventory.InventoryCloseEvent;
-import org.bukkit.event.inventory.InventoryDragEvent;
-import org.bukkit.event.inventory.InventoryOpenEvent;
+import org.bukkit.event.inventory.*;
 import org.bukkit.event.player.PlayerQuitEvent;
+import org.bukkit.inventory.EntityEquipment;
 import org.bukkit.inventory.Inventory;
 import org.jetbrains.annotations.Nullable;
 import xyz.xenondevs.invui.InvUI;
@@ -114,6 +112,12 @@ public class WindowManager implements Listener {
         AbstractWindow window = (AbstractWindow) getOpenWindow((Player) event.getWhoClicked());
         if (window != null) {
             window.handleClickEvent(event);
+            
+            if (event.getClick() == ClickType.SWAP_OFFHAND && event.isCancelled()) {
+                EntityEquipment equipment = event.getWhoClicked().getEquipment();
+                System.out.println(equipment.getItemInOffHand());
+                equipment.setItemInOffHand(equipment.getItemInOffHand());
+            }
         }
     }