Minor improvements to leveling builds (closes #32)

Author: Numynum

core/BlizzardLoadoutChanger.lua

@@ -25,9 +25,9 @@ function Module:UpdateCurrentConfigID()
         or (C_ClassTalents.GetStarterBuildActive() and starterConfigID);
 end
 
---- @return FRAME?
+--- @return PlayerSpellsFrame_TalentsFrame|nil
 function Module:GetTalentFrame()
-    return (ClassTalentFrame and ClassTalentFrame.TalentsTab) or (PlayerSpellsFrame and PlayerSpellsFrame.TalentsFrame);
+    return PlayerSpellsFrame and PlayerSpellsFrame.TalentsFrame;
 end
 
 function Module:TryRefreshTalentUI()

core/TalentLoadoutManager.lua

@@ -21,7 +21,7 @@ local SERIALIZATION_VALUE_SEPARATOR = ns.SERIALIZATION_VALUE_SEPARATOR;
 if not _G.TLM then _G.TLM = TLM; end
 --@end-debug@
 
---- @type TLM_ImportExportV1|TLM_ImportExportV2
+--- @type TLM_ImportExportV2
 local ImportExport = ns.ImportExport;
 
 --- @type TLM_IcyVeinsImport
@@ -281,21 +281,11 @@ function TLM:GetActiveBlizzardLoadoutConfigID()
     if not self.playerSpecID then return nil; end
 
     local lastSelected = C_ClassTalents.GetLastSelectedSavedConfigID(self.playerSpecID);
-    local selectionID =
-        (
-            ClassTalentFrame
-            and ClassTalentFrame.TalentsTab
-            and ClassTalentFrame.TalentsTab.LoadoutDropDown
-            and ClassTalentFrame.TalentsTab.LoadoutDropDown.GetSelectionID
-            and ClassTalentFrame.TalentsTab.LoadoutDropDown:GetSelectionID()
-        )
-        or (
-            PlayerSpellsFrame
-            and PlayerSpellsFrame.TalentsFrame
-            and PlayerSpellsFrame.TalentsFrame.LoadSystem
-            and PlayerSpellsFrame.TalentsFrame.LoadSystem.GetSelectionID
-            and PlayerSpellsFrame.TalentsFrame.LoadSystem:GetSelectionID()
-        );
+    local selectionID = PlayerSpellsFrame
+        and PlayerSpellsFrame.TalentsFrame
+        and PlayerSpellsFrame.TalentsFrame.LoadSystem
+        and PlayerSpellsFrame.TalentsFrame.LoadSystem.GetSelectionID
+        and PlayerSpellsFrame.TalentsFrame.LoadSystem:GetSelectionID();
 
     return selectionID or lastSelected or C_ClassTalents.GetActiveConfigID() or nil;
 end
@@ -849,8 +839,7 @@ function TLM:ApplyCustomLoadout(loadoutInfo, autoApply)
         self:Print("Failed to commit loadout.");
         return false;
     end
-    --- @type Frame|nil
-    local talentsTab = ClassTalentFrame and ClassTalentFrame.TalentsTab; ---@diagnostic disable-line: undefined-global
+    local talentsTab = PlayerSpellsFrame and PlayerSpellsFrame.TalentsFrame
     local talentsTabIsVisible = talentsTab and talentsTab.IsVisible and talentsTab:IsVisible();
     if autoApply and talentsTab and talentsTabIsVisible then
         local stagedNodes = C_Traits.GetStagedPurchases(activeConfigID);

modules/Leveling.lua

@@ -46,7 +46,21 @@ function Module:ReapplyLoadout()
     if not loadoutInfo then return end
 
     self:Print("Automatically re-applying loadout", loadoutInfo.displayName, ", go to /TLM to disable this behavior.");
+    local mapBefore = self:GetSpellIDMap();
     CharacterAPI:LoadLoadout(loadoutID, true);
+    local mapAfter = self:GetSpellIDMap();
+
+    if mapBefore and mapAfter then
+        for spellID, rankAfter in pairs(mapAfter) do
+            if mapBefore[spellID] ~= rankAfter then
+                if (mapBefore[spellID] or 0) == 0 then
+                    self:Print("New Talent Learned:", C_Spell.GetSpellLink(spellID));
+                else
+                    self:Print("Talent Upgraded:", C_Spell.GetSpellLink(spellID), "to rank", rankAfter);
+                end
+            end
+        end
+    end
 end
 
 function Module:AddToCombatLockdownQueue(func, ...)
@@ -66,3 +80,31 @@ function Module:PLAYER_REGEN_ENABLED()
     end
     wipe(self.CombatLockdownQueue);
 end
+
+function Module:GetSpellIDMap()
+    local map = {}
+
+    local configID = C_ClassTalents.GetActiveConfigID()
+    if configID == nil then return end
+
+    local configInfo = C_Traits.GetConfigInfo(configID)
+    if configInfo == nil then return end
+
+    for _, treeID in ipairs(configInfo.treeIDs) do
+        local nodes = C_Traits.GetTreeNodes(treeID)
+        for _, nodeID in ipairs(nodes) do
+            local nodeInfo = C_Traits.GetNodeInfo(configID, nodeID)
+            for _, entryID in ipairs(nodeInfo.entryIDs) do
+                local entryInfo = C_Traits.GetEntryInfo(configID, entryID)
+                if entryInfo and entryInfo.definitionID then
+                    local definitionInfo = C_Traits.GetDefinitionInfo(entryInfo.definitionID)
+                    if definitionInfo.spellID then
+                        map[definitionInfo.spellID] = nodeInfo.activeRank
+                    end
+                end
+            end
+        end
+    end
+
+    return map
+end