From f86aeefcc819d71d54782688322a85b3c0daaed4 Mon Sep 17 00:00:00 2001
From: Adrian L Lange <git@p3lim.net>
Date: Tue, 6 Sep 2022 21:27:32 +0200
Subject: [PATCH] Gather skill levels on both classic and retail

Fixes #14
---
 LibProcessable.lua | 44 +++++++++++---------------------------------
 1 file changed, 11 insertions(+), 33 deletions(-)

diff --git a/LibProcessable.lua b/LibProcessable.lua
index 8291855..29e3db0 100644
--- a/LibProcessable.lua
+++ b/LibProcessable.lua
@@ -330,46 +330,24 @@ function lib:GetProfessionSkillLines(professionID)
 	return professionSkillLines and CopyTable(professionSkillLines)
 end
 
-local classicIDs = {
-	[(GetSpellInfo(2259))]  = 171, -- Alchemy
-	[(GetSpellInfo(2018))]  = 164, -- Blacksmithing
-	[(GetSpellInfo(7411))]  = 333, -- Enchanting
-	[(GetSpellInfo(4036))]  = 202, -- Engineering
-	[(GetSpellInfo(9134))]  = 182, -- Herbalism (spell from gloves with +5 herbalism)
-	[(GetSpellInfo(2108))]  = 165, -- Leatherworking
-	[(GetSpellInfo(2575))]  = 186, -- Mining
-	[(GetSpellInfo(8613))]  = 393, -- Skinning
-	[(GetSpellInfo(3908))]  = 197, -- Tailoring
-	[(GetSpellInfo(25229)) or 0] = 755, -- Jewelcrafting
-	[(GetSpellInfo(45357)) or 0] = 773, -- Inscription
-}
-
 local Handler = CreateFrame('Frame')
 Handler:RegisterEvent('SKILL_LINES_CHANGED')
 Handler:SetScript('OnEvent', function(self, event, ...)
 	table.wipe(professions)
 
-	if(CLASSIC) then
-		-- all professions are spells in the first spellbook tab
-		local _, _, offset, numSpells = GetSpellTabInfo(1)
-		for index = offset + 1, offset + numSpells do
-			-- iterate through all the spells to find the professions
-			local professionID = classicIDs[(GetSpellBookItemName(index, BOOKTYPE_SPELL))]
-			if(professionID) then
-				professions[professionID] = true
+	for _, professionIndex in next, {GetProfessions()} do
+		local _, _, skillLevel, _, _, _, professionID = GetProfessionInfo(professionIndex)
+		if data.professionSkillLines[professionID] then
+			if CLASSIC then
+				professions[professionID] = skillLevel
+			else
+				professions[professionID] = {}
+				for expansion, skillLine in next, data.professionSkillLines[professionID] do
+					local _, currentRank = C_TradeSkillUI.GetTradeSkillLineInfoByID(skillLine)
+					professions[professionID][expansion] = currentRank
+				end
 			end
 		end
-	else
-		local first, second = GetProfessions()
-		if(first) then
-			local _, _, _, _, _, _, professionID = GetProfessionInfo(first)
-			professions[professionID] = true
-		end
-
-		if(second) then
-			local _, _, _, _, _, _, professionID = GetProfessionInfo(second)
-			professions[professionID] = true
-		end
 	end
 end)