Added support for respeccing in Retail's Specializations UI

Various Retail-only settings no longer show up in Classic

Author: Numynum

main.lua

@@ -44,8 +44,12 @@ end
 
 --- @type Button[]
 DialogKey.playerChoiceButtons = {}
+--- @type Button[]
+DialogKey.specButtons = {}
 DialogKey.activeOverrideBindings = {}
 
+DialogKey.dummyButton = CreateFrame("Button")
+
 function DialogKey:OnInitialize()
     DialogKeyNumyDB = DialogKeyNumyDB or {}
     --- @type DialogKeyDB
@@ -86,6 +90,9 @@ function DialogKey:ADDON_LOADED(_, addon)
     if addon == 'Blizzard_PlayerChoice' then
         self:SecureHook(PlayerChoiceFrame, "TryShow", "OnPlayerChoiceShow")
         self:SecureHookScript(PlayerChoiceFrame, "OnHide", "OnPlayerChoiceHide")
+    elseif addon == 'Blizzard_PlayerSpells' then
+        self:SecureHookScript(PlayerSpellsFrame.SpecFrame, "OnShow", "OnSpecFrameShow")
+        self:SecureHookScript(PlayerSpellsFrame.SpecFrame, "OnHide", "OnSpecFrameHide")
     end
 end
 
@@ -186,6 +193,24 @@ function DialogKey:OnPlayerChoiceHide()
     self.playerChoiceButtons = {}
 end
 
+function DialogKey:OnSpecFrameShow()
+    --- @type FramePool<Frame, ClassSpecContentFrameTemplate>
+    local framePool = PlayerSpellsFrame.SpecFrame.SpecContentFramePool
+
+    self.specButtons = {}
+    for specContentFrame in framePool:EnumerateActive() do
+        --- @type ClassSpecContentFrameTemplate
+        local specContentFrame = specContentFrame
+        self.specButtons[specContentFrame.specIndex] = specContentFrame.ActivateButton
+        local text = self.db.handleSpecFrame and (specContentFrame.specIndex .. ' ' .. TALENT_SPEC_ACTIVATE) or TALENT_SPEC_ACTIVATE
+        specContentFrame.ActivateButton:SetText(text)
+    end
+end
+
+function DialogKey:OnSpecFrameHide()
+    self.specButtons = {}
+end
+
 --- @param GossipFrame GossipFrame
 function DialogKey:OnGossipFrameUpdate(GossipFrame)
     local scrollbox = GossipFrame.GreetingPanel.ScrollBox
@@ -369,6 +394,8 @@ function DialogKey:ShouldIgnoreInput()
         and not self:GetFirstVisibleCustomFrame()
         -- Ignore input if no player choice buttons are visible
         and not next(self.playerChoiceButtons)
+        -- Ignore input if no spec buttons are visible
+        and not next (self.specButtons)
     then
         return true
     end
@@ -495,6 +522,17 @@ function DialogKey:HandleKey(key)
         end
     end
 
+    -- Spec Frame
+    if self.db.handleSpecFrame and next(self.specButtons) then
+        local button = self.specButtons[keynum]
+        if button then
+            -- blocks keybind for currently selected spec index
+            if not button:IsVisible() then button = self.dummyButton end
+            self:SetClickbuttonBinding(button, key)
+            return
+        end
+    end
+
     -- GossipFrame
     if (doAction or self.db.numKeysForGossip) and GossipFrame.GreetingPanel:IsVisible() then
         while keynum and keynum > 0 and keynum <= #self.frames do

options.lua

@@ -28,6 +28,7 @@ ns.defaultOptions = {
     handleCraftingOrders = true,
     handlePlayerChoice = true,
     numKeysForPlayerChoice = true,
+    handleSpecFrame = true,
     postAuctions = false,
     ignoreInProgressQuests = true,
 }
@@ -267,19 +268,29 @@ function ns:GetOptionsTable()
                         name = wrapName("Crafting Orders"),
                         desc = "Handle Crafting Orders: Start them, Craft them, Complete them",
                         descStyle = "inline", width = "full", type = "toggle",
+                        hidden = WOW_PROJECT_ID ~= WOW_PROJECT_MAINLINE,
                     },
                     handlePlayerChoice = {
                         order = increment(),
                         name = wrapName("Player Choice"),
                         desc = "Use keybinding to select the first Player Choice option",
                         descStyle = "inline", width = "full", type = "toggle",
+                        hidden = not C_AddOns.DoesAddOnExist("Blizzard_PlayerChoiceUI"),
                     },
                     numKeysForPlayerChoice = {
                         order = increment(),
                         name = wrapName("Number keys for Player Choice"),
                         desc = "Use the number keys (1 -> 0) to select Player Choices",
                         disabled = function() return not db.handlePlayerChoice end,
                         descStyle = "inline", width = "full", type = "toggle",
+                        hidden = not C_AddOns.DoesAddOnExist("Blizzard_PlayerChoiceUI"),
+                    },
+                    handleSpecFrame = {
+                        order = increment(),
+                        name = wrapName(SPECIALIZATION),
+                        desc = "Use the number keys (1 -> 4) to select a specialization while the specialization tab is open",
+                        descStyle = "inline", width = "full", type = "toggle",
+                        hidden = WOW_PROJECT_ID ~= WOW_PROJECT_MAINLINE,
                     },
                 },
             },