CandyDice 

assert(CDRogueTimers)
local BabbleSpell = AceLibrary("Babble-Spell-2.2")

KidneyShot = CDRogueTimers:NewModule("Kidney Shot")

local KSName = BabbleSpell["Kidney Shot"]

KidneyShot.category = "Stun"

function KidneyShot:OnEnable()
    --self:SetDebugging(true)
    self.rank = nil
    self:RegisterEvent("SPELLS_CHANGED", "ScanSkills")
    self:RegisterEvent("UNIT_SPELLCAST_SUCCEEDED")
    self.parser:RegisterEvent("CDRogueTimersKidneyShot", "CHAT_MSG_SPELL_PERIODIC_CREATURE_DAMAGE", function(event, info)
        return self:CHAT_MSG_SPELL_PERIODIC_CREATURE_DAMAGE(event, info)
    end)    
end

function KidneyShot:OnDisable()
    self.parser:UnRegisterAllEvents("CDRogueTimersKidneyShot")
end

function KidneyShot:UNIT_SPELLCAST_SUCCEEDED(unit, spell)
    if unit ~= "player" then return end
    if spell ~= KSName then return end
    self.combo = GetComboPoints()
end

function KidneyShot:CHAT_MSG_SPELL_PERIODIC_CREATURE_DAMAGE(event, info)
    if info.skill ~= KSName then return end
    if not self.rank then
        self:ScanSkills()
    end
    if not self.rank then return end
    if not self.combo then return end
    local duration = self.combo + (self.rank-1)
    self.combo = nil
    self:StartBar(duration)
end

function KidneyShot:ScanSkills()
    -- Scan spellbook to figure out what rank we have
    for ii =1,MAX_SPELLS do
        local name, rank = GetSpellName(ii, BOOKTYPE_SPELL)
        --self:Debug(name)
        if name == KSName then
            --- wrong wrong wrong... parse the rank out of the string
            local _,_,nrank = string.find(rank, 'Rank (%d)')
            self.rank = nrank
            return
        end
    end
end


CheapShot = CDRogueTimers:NewModule("Cheap Shot")

CheapShot.category = "Stun"

function CheapShot:OnEnable()
     self.parser:RegisterEvent("CDRogueTimersCheapShot", "CHAT_MSG_SPELL_PERIODIC_CREATURE_DAMAGE", function(event, info)
        return self:CHAT_MSG_SPELL_PERIODIC_CREATURE_DAMAGE(event, info)
    end)
end

function CheapShot:OnDisable()
    self.parser:UnRegisterAllEvents("CDRogueTimersCheapShot")
end

function CheapShot:CHAT_MSG_SPELL_PERIODIC_CREATURE_DAMAGE(event, info)
    if info.skill ~= BabbleSpell["Cheap Shot"] then return end
    if info.victim ~= UnitName("target") then return end
    self:StartBar(4)
end