From a4d5fcfce0ec2c8982286a0c8d2606a177c33331 Mon Sep 17 00:00:00 2001
From: Permok <martinw1394@gmail.com>
Date: Fri, 17 Mar 2023 18:53:29 +0100
Subject: [PATCH] Remove current category from the "Add to Category" dropdown

---
 ImprovedTalentLoadouts.lua | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/ImprovedTalentLoadouts.lua b/ImprovedTalentLoadouts.lua
index 6b44b13..0a78379 100644
--- a/ImprovedTalentLoadouts.lua
+++ b/ImprovedTalentLoadouts.lua
@@ -1696,20 +1696,24 @@ local function LoadoutDropdownInitialize(_, level, menu, ...)
             end
         end
     elseif menu == "addToCategory" then
+        local isCategory = type(L_UIDROPDOWNMENU_MENU_VALUE) == "table"
+
         for _, categoryInfo in spairs(TalentLoadouts.globalDB.categories[currentSpecID], 
         function(t, a, b) if t[a] and t[b] and t[a].name and t[b].name then return t[a].name < t[b].name end end) do
-            LibDD:UIDropDownMenu_AddButton(
-                    {
-                        arg1 = categoryInfo,
-                        arg2 = L_UIDROPDOWNMENU_MENU_VALUE,
-                        colorCode = "|cFFab96b3",
-                        text = categoryInfo.name,
-                        minWidth = 170,
-                        fontObject = dropdownFont,
-                        notCheckable = 1,
-                        func = AddToCategory,
-                    },
-            level)
+            if not isCategory or categoryInfo.key ~= L_UIDROPDOWNMENU_MENU_VALUE.key then
+                LibDD:UIDropDownMenu_AddButton(
+                        {
+                            arg1 = categoryInfo,
+                            arg2 = L_UIDROPDOWNMENU_MENU_VALUE,
+                            colorCode = "|cFFab96b3",
+                            text = categoryInfo.name,
+                            minWidth = 170,
+                            fontObject = dropdownFont,
+                            notCheckable = 1,
+                            func = AddToCategory,
+                        },
+                level)
+            end
         end
     elseif menu == "removeFromCategory" then
         for _, categoryKey in ipairs(L_UIDROPDOWNMENU_MENU_VALUE.parents) do