fix most issues with switching recorded leveling builds

Author: Numynum

TalentTreeViewer/TalentViewer.lua

@@ -572,8 +572,7 @@ function TalentViewer:ClearLevelingBuild()
     table.insert(self.levelingBuilds[self.selectedSpecId], info);
     self.recordingInfo.buildID = #self.levelingBuilds[self.selectedSpecId];
 
-    self:RecalculateCurrentStartingOffsets();
-    self:GetTalentFrame():SetLevelingBuildID(self.recordingInfo.buildID);
+    self:StartRecordingLevelingBuild();
 end
 
 function TalentViewer:IsRecordingLevelingBuild()
@@ -700,12 +699,12 @@ function TalentViewer:InitLevelingBuildUIs()
                 end,
             });
         end
-        for buildID, buildEntries in ipairs(self.levelingBuilds[self.selectedSpecId] or {}) do
+        for buildID, buildInfo in ipairs(self.levelingBuilds[self.selectedSpecId] or {}) do
             table.insert(menu, {
                 text = string.format(
                     'Leveling build %d (%d points spent)',
                     buildID,
-                    #buildEntries
+                    #buildInfo.entries[1] + #buildInfo.entries[2]
                 ),
                 func = function(_, buildID)
                     self:ApplyLevelingBuild(buildID, currentValue, true);

TalentTreeViewer/TalentViewerUIMixin.lua

@@ -745,7 +745,7 @@ function TalentViewerUIMixin:UpdateLevelingBuildHighlights()
     end
 end
 
---- @return nil|TalentViewer_LevelingBuildEntry[]
+--- @return nil|table<number, TalentViewer_LevelingBuildEntry> # [level] = entry
 function TalentViewerUIMixin:GetLevelingBuildInfo(buildID)
     return TalentViewer:GetLevelingBuild(buildID);
 end
@@ -754,9 +754,10 @@ function TalentViewerUIMixin:GetNextLevelingBuildPurchase(buildID)
     local info = self:GetLevelingBuildInfo(buildID);
     if not info then return; end
 
-    for _, entryInfo in ipairs(info) do
-        local nodeInfo = self:GetAndCacheNodeInfo(entryInfo.nodeID);
-        if nodeInfo.ranksPurchased < entryInfo.targetRank then
+    for i = 10, ns.MAX_LEVEL do
+        local entryInfo = info[i];
+        local nodeInfo = entryInfo and self:GetAndCacheNodeInfo(entryInfo.nodeID);
+        if nodeInfo and nodeInfo.ranksPurchased < entryInfo.targetRank then
             return entryInfo.nodeID, entryInfo.entryID;
         end
     end
@@ -802,14 +803,16 @@ function TalentViewerUIMixin:ApplyLevelingBuild(level, lockLevelingBuild)
         end
     end
 
-    --- @type table<number, number[]>
-    local byNodes = {};
-    for index, entry in ipairs(info) do
-        byNodes[entry.nodeID] = byNodes[entry.nodeID] or {};
-        table.insert(byNodes[entry.nodeID], index);
-    end
     for _, button in ipairs(self.levelingOrderButtons) do
-        button:SetOrder(byNodes[button:GetParent():GetNodeID()] or {});
+        button:SetOrder({});
+    end
+    for entryLevel = 10, ns.MAX_LEVEL do
+        local entryInfo = info[entryLevel];
+        if entryInfo then
+            local nodeID = entryInfo.nodeID;
+            local button = self:GetTalentButtonByNodeID(nodeID);
+            button.LevelingOrder:AppendToOrder(entryLevel);
+        end
     end
 
     self:UpdateTreeCurrencyInfo();