function to check talent

[Dajova]

ok so im trying to fix a function for my mod that can determine what spec i am based on the spellid it checks for (eg. spec specific spells, like Lava Lash for Shaman)

However, i've come to a brick wall here :/ This is the current function i am using atm.

local function CheckForKnownTalent(spellid)
    local wanted_name = GetSpellInfo(spellid)
    if not wanted_name then return nil end
    local num_tabs = GetNumTalentTabs()
    for t=1, num_tabs do
        local num_talents = GetNumTalents(t)
        for i=1, num_talents do
            local name_talent, _, _, _, current_rank = GetTalentInfo(t,i)
            if name_talent and (name_talent == wanted_name) then
                if current_rank and (current_rank > 0) then
                    return true
                else
                    return false
                end
            end
        end
    end
    return false
end

EDIT: Oh and this is the call for the function

local spell1
local spell2
if CheckForKnownTalent(78674) then
    spell1 = GetSpellInfo(81192) -- Lunar Shower
elseif CheckForKnownTalent(33878) then
  spell2 = GetSpellInfo(33745) -- Lacerate  
end

I don't know what i am doing wrong, cause it returns... well, i dont know, cause i dont even get a error

[brotherhobbes]

If you just want the general spec, you can use something like

if GetPrimaryTalentTree() == 2 then
-- enhancement
end

Example: for shaman, 1 = ele, 2 = enh, 3 = resto, 0 = you haven't selected yet

The function name is misleading, it returns what your current active spec is, not what you have assigned to your Primary/Secondary spec.

Anyway, this should be much easier for you to use.

[Dajova]

Ah ok. Looks fair enough.
This is what i have atm, should work.

   if unit == 'player' and GetPrimaryTalentTree() == 1 then
      local _,_,_,count = UnitBuff('player', GetSpellInfo(81192))