WoWInterface SVN TradeTabs

local TradeTabs = CreateFrame("Frame","TradeTabs")

local tradeSpells = { -- Spell order in this table determines the tab order

        28596, -- Alchemy

        29844, -- Blacksmithing

        28029, -- Enchanting

        30350, -- Engineering

        45357, -- Inscription

        28897, -- Jewel Crafting

        32549, -- Leatherworking

        53428, -- Runeforging

        2656,  -- Smelting

        26790, -- Tailoring

        33359, -- Cooking

        27028, -- First Aid

        13262, -- Disenchant

        51005, -- Milling

        31252, -- Prospecting

        818,   -- Basic Campfire

}

function TradeTabs:OnEvent(event,...)

        self:UnregisterEvent(event)

        if not IsLoggedIn() then

                self:RegisterEvent(event)

        elseif InCombatLockdown() then

                self:RegisterEvent("PLAYER_REGEN_ENABLED")

        else

                self:Initialize()

        end

end

function TradeTabs:Initialize()

        if self.initialized or not IsAddOnLoaded("Blizzard_TradeSkillUI") then return end -- Shouldn't need this, but I'm paranoid

        for i=1,#tradeSpells do

                local n = GetSpellInfo(tradeSpells[i])

                tradeSpells[n] = -1

                tradeSpells[i] = n

        end

        local parent = TradeSkillFrame

        if SkilletFrame then

                parent = SkilletFrame

                self:UnregisterAllEvents()

        end

        for i=1,MAX_SPELLS do

                local n = GetSpellName(i,"spell")

                if tradeSpells[n] then

                        tradeSpells[n] = i

                end

        end

        local prev

        for i,spell in ipairs(tradeSpells) do

                local spellid = tradeSpells[spell]

                if type(spellid) == "number" and spellid > 0 then

                        local tab = self:CreateTab(spell,spellid,parent)

                        local point,relPoint,x,y = "TOPLEFT","BOTTOMLEFT",0,-17

                        if not prev then

                                prev,relPoint,x,y = parent,"TOPRIGHT",-33,-44

                                if parent == SkilletFrame then x = 0 end -- Special case. ew

                        end

                        tab:SetPoint(point,prev,relPoint,x,y)

                        prev = tab

                end

        end

        self.initialized = true

end

local function onEnter(self)

    GameTooltip:SetOwner(self,"ANCHOR_RIGHT") GameTooltip:SetText(self.tooltip)

    self:GetParent():LockHighlight()

end

local function onLeave(self)

    GameTooltip:Hide()

    self:GetParent():UnlockHighlight()

end

local function updateSelection(self)

        if IsCurrentSpell(self.spellID,"spell") then

                self:SetChecked(true)

                self.clickStopper:Show()

        else

                self:SetChecked(false)

                self.clickStopper:Hide()

        end

end

local function createClickStopper(button)

    local f = CreateFrame("Frame",nil,button)

    f:SetAllPoints(button)

    f:EnableMouse(true)

    f:SetScript("OnEnter",onEnter)

    f:SetScript("OnLeave",onLeave)

    button.clickStopper = f

    f.tooltip = button.tooltip

    f:Hide()

end

function TradeTabs:CreateTab(spell,spellID,parent)

    local button = CreateFrame("CheckButton",nil,parent,"SpellBookSkillLineTabTemplate,SecureActionButtonTemplate")

    button.tooltip = spell

        button:Show()

    button:SetAttribute("type","spell")

    button:SetAttribute("spell",spell)

    button.spellID = spellID

    button:SetNormalTexture(GetSpellTexture(spellID, "spell"))

        button:SetScript("OnEvent",updateSelection)

        button:RegisterEvent("TRADE_SKILL_SHOW")

        button:RegisterEvent("TRADE_SKILL_CLOSE")

        button:RegisterEvent("CURRENT_SPELL_CAST_CHANGED")

    createClickStopper(button)

    updateSelection(button)

        return button

end

TradeTabs:RegisterEvent("TRADE_SKILL_SHOW")

TradeTabs:SetScript("OnEvent",TradeTabs.OnEvent)

TradeTabs:Initialize()