EnhanceCDs.lua

if select(2,UnitClass("player")) ~= "SHAMAN" then return end
if WOW_PROJECT_ID == WOW_PROJECT_CLASSIC and C_Seasons.GetActiveSeason() ~= 2 then return end

local isSOD = C_Seasons.GetActiveSeason() == 2

local _, TotemTimers = ...

local SpellIDs = TotemTimers.SpellIDs
local SpellTextures = TotemTimers.SpellTextures
local SpellNames = TotemTimers.SpellNames
local AvailableSpells = TotemTimers.AvailableSpells
local FlameShockID = SpellIDs.FlameShock
local FlameShockName = SpellNames[SpellIDs.FlameShock]

local cds = {}
TotemTimers.EnhanceCDs = cds

local role = 2

local function NoUpdate()
end

local ShieldName = SpellNames[SpellIDs.LightningShield]

local CDSpells = TotemTimers.CombatCooldownSpells

local FlameShockDuration = null
local Maelstrom, MaelstromButton, MaelstromIcon

local CenterProcAnchor = CreateFrame("Frame", "CenterProcAnchor", UIParent)
TotemTimers.EnhanceCDsCenterProcAnchor = CenterProcAnchor
CenterProcAnchor:Hide()
CenterProcAnchor:SetSize(130,65)
CenterProcAnchor:Show()

local LeftProcAnchor = CreateFrame("Frame", "LeftProcAnchor", UIParent)
TotemTimers.EnhanceCDsLeftProcAnchor = LeftProcAnchor
LeftProcAnchor:Hide()
LeftProcAnchor:SetSize(65, 130)



local RightProcAnchor = CreateFrame("Frame", nil, UIParent)
TotemTimers.EnhanceCDsRightProcAnchor = RightProcAnchor
RightProcAnchor:Hide()
RightProcAnchor:SetSize(65, 130)


local function ChangeCDOrder(self, _, _, spell)
    if InCombatLockdown() then return end
    if not spell then return end

    local spell1 = TotemTimers.GetBaseSpellID(spell)
    local spell2 = TotemTimers.GetBaseSpellID(self:GetAttribute("spell1"))

    local orderIndex1, orderIndex2

    for orderIndex, spellIndex in pairs(TotemTimers.ActiveProfile.EnhanceCDs_Order[TotemTimers.Specialization]) do
        if CDSpells[role][spellIndex] == spell1 then orderIndex1 = orderIndex end
        if CDSpells[role][spellIndex] == spell2 then orderIndex2 = orderIndex end
    end

    if orderIndex1 and orderIndex2 then
        TotemTimers.ActiveProfile.EnhanceCDs_Order[role][orderIndex1], TotemTimers.ActiveProfile.EnhanceCDs_Order[role][orderIndex2] =
            TotemTimers.ActiveProfile.EnhanceCDs_Order[role][orderIndex2], TotemTimers.ActiveProfile.EnhanceCDs_Order[role][orderIndex1]
    end

    TotemTimers.LayoutEnhanceCDs()
end
   
function TotemTimers.CreateEnhanceCDs()
    for i = 1,12 do
        cds[i] = XiTimers:new(1)
        cds[i].events = {"SPELL_UPDATE_COOLDOWN"}
        cds[i].dontFlash = true
        cds[i].timeStyle = "sec"
        cds[i].button.anchorframe = TotemTimers_EnhanceCDsFrame
        cds[i].button:SetAttribute("*type*", "spell")
        cds[i].reverseAlpha = true
        cds[i].alpha = 0.7
        cds[i].button.icons[1]:SetAlpha(1)
        cds[i].button:SetScript("OnEvent", TotemTimers.EnhanceCDEvents)
        cds[i].button:RegisterForClicks("LeftButtonUp", "RightButtonUp", "MiddleButtonUp")
    end

    FlameShockDuration = XiTimers:new(1)
    TotemTimers.FlameShockDuration = FlameShockDuration
    FlameShockDuration.button:Disable()
    FlameShockDuration.button.icons[1]:SetVertexColor(1,1,1)
    FlameShockDuration.button.bar:SetStatusBarColor(1,0.2,0.2,0.8)

    FlameShockDuration.button.icons[1]:SetTexture(SpellTextures[SpellIDs.FlameShock])
    FlameShockDuration.button.anchorframe = TotemTimers_EnhanceCDsFrame
    FlameShockDuration.dontAlpha = true
    FlameShockDuration.dontFlash = true
    FlameShockDuration.timeStyle = "sec"
    FlameShockDuration.button:SetAttribute("*type1", "spell")
    FlameShockDuration.button:SetAttribute("spell1", SpellNames[SpellIDs.FlameShock])
    FlameShockDuration.button.icons[1]:SetAlpha(1)
	FlameShockDuration.rangeCheck = SpellNames[SpellIDs.FlameShock]
	FlameShockDuration.manaCheck = SpellNames[SpellIDs.FlameShock]
    FlameShockDuration.button:SetScript("OnEvent", TotemTimers.FlameShockEvent)
    FlameShockDuration.events[1] = "COMBAT_LOG_EVENT_UNFILTERED"
    FlameShockDuration.events[2] = "UNIT_AURA"
    FlameShockDuration.events[3] = "PLAYER_REGEN_ENABLED"
    FlameShockDuration.events[4] = "PLAYER_REGEN_DISABLED"
    FlameShockDuration.events[5] = "PLAYER_TARGET_CHANGED"
	FlameShockDuration.forceBar = true
    FlameShockDuration.timerBarPos = nil
    FlameShockDuration.timerBars[1]:ClearAllPoints()
    FlameShockDuration.timerBars[1]:SetPoint("LEFT", FlameShockDuration.button, "LEFT")
    FlameShockDuration.timerBars[1]:SetFrameStrata("LOW")
    FlameShockDuration:SetBarColor(1,0.5,0)
	

    for i=1,#cds do
        cds[i].button:SetAttribute("_ondragstart",[[if IsShiftKeyDown() then
                                                    return "spell", self:GetAttribute("spell1")
                                              else control:CallMethod("StartMove") end]])
        cds[i].button:SetAttribute("_onreceivedrag",[[ if kind == "spell" then
                                                   control:CallMethod("ChangeCDOrder", value, ...)
                                                    return "clear"
                                              end]])
        cds[i].button.ChangeCDOrder = ChangeCDOrder
    end


    if WOW_PROJECT_ID > WOW_PROJECT_BURNING_CRUSADE_CLASSIC or C_Seasons.GetActiveSeason() == 2 then
        Maelstrom = XiTimers:new(1)
        TotemTimers.Maelstrom = Maelstrom

        --Maelstrom.button:Disable()
        Maelstrom.button.icons[1]:SetVertexColor(1,1,1)

        --Maelstrom.button.icons[1]:SetTexture(237584)
        Maelstrom.button.icons[1]:SetTexture(GetSpellTexture(SpellIDs.Maelstrom))
        Maelstrom.button.anchorframe = TotemTimers_EnhanceCDsFrame
        Maelstrom.dontAlpha = true
        Maelstrom.dontFlash = true
        Maelstrom.timeStyle = "sec"
        Maelstrom.button:SetAttribute("*type*", "spell")
        Maelstrom.button:RegisterForClicks("LeftButtonUp", "RightButtonUp")
        Maelstrom.button:SetAttribute("spell1", SpellIDs.LightningBolt)
        Maelstrom.button:SetAttribute("spell2", SpellIDs.ChainLightning)
        Maelstrom.button.icons[1]:SetAlpha(1)
        Maelstrom.rangeCheck = SpellNames[SpellIDs.LightningBolt]
        Maelstrom.manaCheck = SpellNames[SpellIDs.LightningBolt]
        Maelstrom.button:SetScript("OnEvent", TotemTimers.MaelstromEvent)
        Maelstrom.playerEvents[1] = "UNIT_AURA"
        Maelstrom.forceBar = true
        Maelstrom.timerBarPos = nil
        Maelstrom.timerBars[1]:ClearAllPoints()
        Maelstrom.timerBars[1]:SetPoint("LEFT", Maelstrom.button, "LEFT")
        Maelstrom.timerBars[1]:SetFrameStrata("LOW")
        Maelstrom:SetTimeWidth(100)
        --Maelstrom.timerBars[1]:SetStatusBarAtlas("_Shaman-MaelstromBar")
        --Maelstrom:SetBarColor(0.8,.8,1)
        Maelstrom:SetBarColor(RAID_CLASS_COLORS.SHAMAN.r, RAID_CLASS_COLORS.SHAMAN.g, RAID_CLASS_COLORS.SHAMAN.b, 1)
        --Maelstrom.timerBars[1].background:SetStatusBarAtlas("_Shaman-MaelstromBar")
        Maelstrom.timerBars[1].time:Hide()
        Maelstrom.button.icons[1]:SetAllPoints(Maelstrom.button)

        Maelstrom.animation.icon:SetTexture("Interface/AddOns/TotemTimers/textures/maelstrom_weapon")
        Maelstrom.animation.button:SetSize(72,36)


        Maelstrom.Update = function() end
        Maelstrom.Activate = function(self)
            XiTimers.Activate(self)
            TotemTimers.MaelstromEvent(self.button)
        end


        MaelstromButton = CreateFrame("Button", "TotemTimers_MaelstromBarButton", Maelstrom.button, "ActionButtonTemplate, SecureActionButtonTemplate")
        TotemTimers.MaelstromButton = MaelstromButton

        MaelstromButton:SetFrameLevel(Maelstrom.button:GetFrameLevel() + 10)
        MaelstromButton:SetPoint("LEFT", Maelstrom.button, "LEFT")
        MaelstromButton:SetSize(100, 17)
        --MaelstromButton:SetAllPoints(Maelstrom.timerBars[1])

        --MaelstromButton:SetNormalTexture(nil)
        _G[MaelstromButton:GetName().."NormalTexture"]:SetTexture(nil)
        MaelstromButton:SetHighlightTexture("Interface/AddOns/TotemTimers/textures/MaelstromHilight")
        MaelstromButton:SetPushedTexture("Interface/AddOns/TotemTimers/textures/MaelstromPushed")
        MaelstromButton:SetAttribute("*type*", "spell")
        MaelstromButton:RegisterForClicks("LeftButtonUp", "RightButtonUp")
        MaelstromButton:SetAttribute("spell1", SpellIDs.LightningBolt)
        MaelstromButton:SetAttribute("spell2", SpellIDs.ChainLightning)
        MaelstromButton.icon = TotemTimers_MaelstromBarButtonIcon

        MaelstromIcon = CreateFrame("Frame", "TotemTimers_MaelstromIcon", UIParent)
        TotemTimers.MaelstromIcon = MaelstromIcon
        Maelstrom.ChainOOCAlpha = TotemTimers.MaelstromIcon

        MaelstromIcon.icon = MaelstromIcon:CreateTexture(nil, "ARTWORK")
        MaelstromIcon.icon:SetAllPoints(MaelstromIcon)
        MaelstromIcon:SetFrameLevel(Maelstrom.button:GetFrameLevel() + 2)

        MaelstromIcon:ClearAllPoints()
        MaelstromIcon:SetWidth(100)
        MaelstromIcon:SetHeight(50)

        for _, icon in pairs({MaelstromIcon.icon, Maelstrom.timerBars[1].time}) do
            icon.AnimGroup = icon:CreateAnimationGroup()
            icon.AnimGroup:SetLooping("REPEAT")
            local scale = icon.AnimGroup:CreateAnimation("Scale")
            scale:SetDuration(0.4)
            scale:SetScale(1.05,1.05)
            scale:SetOrder(1)
            scale:SetSmoothing("IN_OUT")
            scale = icon.AnimGroup:CreateAnimation("Scale")
            scale:SetDuration(0.4)
            scale:SetScale(0.95,0.95)
            scale:SetOrder(2)
            scale:SetSmoothing("IN_OUT")
        end

        Maelstrom.animation.AnchoredButton = MaelstromIcon
    end
end

table.insert(TotemTimers.Modules, TotemTimers.CreateEnhanceCDs)

function TotemTimers.ConfigEnhanceCDs()
    role = TotemTimers.Specialization --GetSpecialization()
    if not role then role = 0 end
    
    for i=1,#cds do
        cds[i]:Deactivate()
    end
    FlameShockDuration:Deactivate()
    if Maelstrom then Maelstrom:Deactivate() end
    MaelstromIcon:Hide()

    if role == 0 or not TotemTimers.ActiveProfile.EnhanceCDs then return end

    if WOW_PROJECT_ID > WOW_PROJECT_WRATH_CLASSIC and role == 2 then
        for id, spell in ipairs(CDSpells[role]) do
            if spell == SpellIDs.StormStrike or spell == SpellIDs.PrimalStrike then
                if AvailableSpells[SpellIDs.StormStrike] then
                    CDSpells[role][id] = SpellIDs.StormStrike
                else
                    CDSpells[role][id] = SpellIDs.PrimalStrike
                end
                break
            end
        end
    end

    for i=1,#CDSpells[role] do
        local spell = CDSpells[role][i]
        local cd = cds[i]
        cd.button.cdspell = spell
        cd.button.icons[1]:SetTexture(SpellTextures[spell])
        cd.button:SetAttribute("spell1", spell)
        if spell == SpellIDs.FlameShock then
            cd.button:SetAttribute("spell2", SpellIDs.EarthShock)
            cd.button:SetAttribute("spell3", SpellIDs.FrostShock)
        elseif spell == SpellIDs.EarthShock then
            if SpellIDs.WayOfEarth and AvailableSpells[SpellIDs.WayOfEarth] then
                cd.button.cdspell = SpellIDs.EarthShockTank
            end
            cd.button:SetAttribute("spell2", SpellIDs.FlameShock)
            cd.button:SetAttribute("spell2", SpellIDs.FrostShock)
        else
		    cd.button:SetAttribute("spell2", nil)
		end
		cd.button:SetAttribute("spell3", nil)
        cd.rangeCheck = SpellNames[CDSpells[role][i]]
        cd.manaCheck = SpellNames[CDSpells[role][i]]
        cd.Update = nil
        cd.prohibitCooldown = false
        cd.button:SetAttribute("orderspell", CDSpells[role][i])

        cd.reverseAlpha = true
        cd.Stop = nil

        cd.events = {"SPELL_UPDATE_COOLDOWN"}
        cd.playerEvents = {}

        --cd.button.bar:SetStatusBarColor(0.6,0.6,1.0,0.5)

        --[[if spell == SpellIDs.FlameShock or spell == SpellIDs.EarthShock then
            cd.button:SetScript("OnEvent", TotemTimers.ShockEvent)
            cd.playerEvents[1] = "UNIT_AURA"
        else]]
        if spell == SpellIDs.StormStrike then
            cd.button:SetScript("OnEvent", TotemTimers.StormStrikeEvent)
            cd.events[2] = "UNIT_AURA"
            cd.events[3] = "PLAYER_TARGET_CHANGED"
        elseif spell == SpellIDs.Searing or spell == SpellIDs.Magma then
            cd.button:SetScript("OnEvent", TotemTimers.FireTotemEvent)
            cd.events[2] = "PLAYER_TOTEM_UPDATE"
        else
            cd.button:SetScript("OnEvent", TotemTimers.EnhanceCDEvents)
        end
    end

    if AvailableSpells[SpellIDs.FlameShock] and TotemTimers.ActiveProfile.EnhanceCDsFlameShockDuration_Specialization[role] then
        FlameShockDuration:Activate()
    end

    if (role == 2 or C_Seasons.GetActiveSeason() == 2) and Maelstrom
            and TotemTimers.AvailableTalents.Maelstrom and TotemTimers.ActiveProfile.EnhanceCDsMaelstrom
        then Maelstrom:Activate()
    end

    for i=1,#CDSpells[role] do
        if TotemTimers.ActiveProfile.EnhanceCDs_Spells[role][i] and AvailableSpells[cds[i].button.cdspell] then
            cds[i]:Activate()
        end
    end

end

local activeCDs = {}

local function ConvertCoords(frame1, frame2)
    return frame1:GetEffectiveScale()/frame2:GetEffectiveScale()
end


local function SizeToWidthHeight(size)
    size = min(3, size or 1)

    local spacing = TotemTimers.ActiveProfile.CooldownSpacing
    local height = 19 + (size-1) * 4

    --button + bar = 3 normal buttons + spacing, 4 btn for size 2, 5 btn for size 3
    local width = 108 + spacing * 2
    width = width + (size-1) * (36 + spacing)

    return width, height
end

local x = 0
function TotemTimers.LayoutEnhanceCDs()
    wipe(activeCDs)
    if role == 0 then return end

    for i=1,#CDSpells[role] do
        if cds[TotemTimers.ActiveProfile.EnhanceCDs_Order[role][i]].active then
            table.insert(activeCDs,cds[TotemTimers.ActiveProfile.EnhanceCDs_Order[role][i]])
        end
    end
    for i=1,12 do
        cds[i]:ClearAnchors()
        cds[i].button:ClearAllPoints()
    end
    FlameShockDuration.button:ClearAllPoints()
    if Maelstrom then Maelstrom.button:ClearAllPoints() end

    local numActiveCDs = #activeCDs

    if numActiveCDs == 0 then return end

    local topRow, bottomRow = numActiveCDs, 0

    if numActiveCDs > 5 and not TotemTimers.ActiveProfile.EnhanceCDsOneRow then
        topRow = floor(numActiveCDs / 2)
        bottomRow = ceil(numActiveCDs / 2)
    end

    local split = ceil(topRow / 2)
    if topRow % 2 == 0 then
        activeCDs[split]:SetPoint("RIGHT", TotemTimers_EnhanceCDsFrame, "CENTER", -3, 0)
    else
        activeCDs[split]:SetPoint("CENTER", TotemTimers_EnhanceCDsFrame)
    end

    for i=split-1, 1, -1 do
        activeCDs[i]:Anchor(activeCDs[i+1], "RIGHT")
    end
    for i = split + 1, topRow do
        activeCDs[i]:Anchor(activeCDs[i-1], "LEFT")
    end

    local bottomSplit
    if bottomRow > 0 then
        bottomSplit = topRow + ceil(bottomRow / 2)
        if bottomRow == topRow then
            activeCDs[bottomSplit]:Anchor(activeCDs[split], "TOP")
        else
            if topRow %2 == 0 then
                activeCDs[bottomSplit]:Anchor(activeCDs[split], "TOPLEFT", "BOTTOM", true)
            else
                activeCDs[bottomSplit]:Anchor(activeCDs[split], "TOPRIGHT", "BOTTOM", true)
            end
        end
        for i=bottomSplit-1, topRow+1, -1 do
            activeCDs[i]:Anchor(activeCDs[i+1], "RIGHT")
        end
        for i = bottomSplit + 1, numActiveCDs do
            activeCDs[i]:Anchor(activeCDs[i-1], "LEFT")
        end
    end

    local spacing = TotemTimers.ActiveProfile.CooldownSpacing

    local width, height = SizeToWidthHeight(FlameShockDuration.size)

    FlameShockDuration:SetTimeWidth(width)
    FlameShockDuration:SetTimeHeight(height)
    FlameShockDuration.button:SetSize(height, height)
    FlameShockDuration.button.icons[1]:SetAllPoints(FlameShockDuration.button)

    local fspoint, fsy = "BOTTOMLEFT", 0

    if TotemTimers.ActiveProfile.FlameShockDurationOnTop then
        fsy = 18 + spacing + activeCDs[1]:GetBorder("TOP")
    else
        fspoint = "TOPLEFT"
        local border = spacing + activeCDs[1]:GetBorder("BOTTOM")
        fsy = -18 - border
        if bottomRow > 0 then
            fsy = fsy - 36 - border
        end
    end

    FlameShockDuration.button:SetPoint(fspoint, TotemTimers_EnhanceCDsFrame, "CENTER", -width / 2, fsy)

    local cprocy = 18 + spacing + activeCDs[1]:GetBorder("TOP")

    if FlameShockDuration.active and TotemTimers.ActiveProfile.FlameShockDurationOnTop then
        cprocy = cprocy + spacing + height
    end

    CenterProcAnchor:ClearAllPoints()
    CenterProcAnchor:SetPoint("CENTER", TotemTimers_EnhanceCDsFrame, "CENTER", 0, cprocy + 12)

    LeftProcAnchor:ClearAllPoints()
    RightProcAnchor:ClearAllPoints()
    if bottomRow == 0 then
        LeftProcAnchor:SetPoint("RIGHT", activeCDs[1].button, "LEFT", LeftProcAnchor:GetWidth() / 3 * LeftProcAnchor:GetScale(), 0)
        RightProcAnchor:SetPoint("LEFT", activeCDs[numActiveCDs].button, "RIGHT", -RightProcAnchor:GetWidth() / 3 * RightProcAnchor:GetScale(), 0)
    end

    if Maelstrom then
        local msWidth, msHeight = SizeToWidthHeight(Maelstrom.size)
        Maelstrom:SetTimeWidth(msWidth)
        Maelstrom:SetTimeHeight(msHeight)
        Maelstrom.button:SetSize(msHeight, msHeight)
        Maelstrom.button.icons[1]:SetAllPoints(Maelstrom.button)
        MaelstromButton:SetSize(msWidth, msHeight)
        MaelstromIcon:SetSize(msWidth * 0.75, msWidth * 0.75 / 2)

        local msy = 18 + spacing + activeCDs[1]:GetBorder("TOP")

        if FlameShockDuration.active and TotemTimers.ActiveProfile.FlameShockDurationOnTop then
            msy = msy + spacing + height
        end

        Maelstrom.button:SetPoint("BOTTOMLEFT", TotemTimers_EnhanceCDsFrame, "CENTER", -msWidth / 2, msy)
        MaelstromIcon:SetPoint("CENTER", TotemTimers_EnhanceCDsFrame, "CENTER", 0, msy + 12)

    end
end

function TotemTimers.ActivateEnhanceCDs()
    TotemTimers.ConfigEnhanceCDs()
    TotemTimers.LayoutEnhanceCDs()
end

function TotemTimers.DeactivateEnhanceCDs()
    for k,v in pairs(cds) do
        v:Deactivate()
    end
    FlameShockDuration:Deactivate()
    if Maelstrom then Maelstrom:Deactivate() end
end

function TotemTimers.EnhanceCDEvents(self, event, spell)
    local settings = TotemTimers.ActiveProfile

    if event == "SPELL_UPDATE_COOLDOWN" then
        local cdspell = self.cdspell
        if isSOD then cdspell = SpellNames[cdspell] end
        local start, duration, enable = GetSpellCooldown(cdspell)
        if (not start and not duration) then --or (duration <= 1.5 and not InCombatLockdown()) then 			
            self.timer:Stop(1)
        else
            if duration <= 1.5 then
                self.timer:Stop(1)
            elseif duration > 1.5 then
                self.timer:Start(1,start+duration-GetTime(),duration)
            end
            CooldownFrame_Set(self.cooldown, start, duration, enable)
        end
    elseif event == "SPELL_ACTIVATION_OVERLAY_GLOW_SHOW" then
        if spell == self.glowSpell then
            ActionButton_ShowOverlayGlow(self)
        end
    elseif event == "SPELL_ACTIVATION_OVERLAY_GLOW_HIDE" then
        if spell == self.glowSpell then
            ActionButton_HideOverlayGlow(self)
        end
    end
end


function TotemTimers.FireTotemEvent(self,event,...)
    local element = ...
    if event == "PLAYER_TOTEM_UPDATE" then
        if element == 1 then
            local _, totem, startTime, duration, icon = GetTotemInfo(1)
            if icon == self.icon:GetTexture() and duration > 0 then
				self.icon:SetTexture(icon)
                self.timer:Start(1, duration)
            elseif self.timer.timers[1] > 0 then 
                self.timer:Stop(1)
            end
        end
    else
        -- TotemTimers.EnhanceCDEvents(self,event,...)
    end
end


function TotemTimers.ShieldChargeEvent(self, event, ...)
	if event == "UNIT_AURA" and select(1,...) =="player" then
		local name,_,texture,count,_,duration,endtime = UnitAura("player", ShieldName)
		if name then
			if role == 1 then
				self.timer:Start(1, count, 15)
			else
				self.timer:Start(1, endtime-GetTime(), duration)
			end
		elseif self.timer.timers[1]>0 then
			self.timer:Stop(1)
		end
	elseif event ~= "SPELL_UPDATE_COOLDOWN" then
        TotemTimers.EnhanceCDEvents(self,event,...)
    end
end


local FSAuraGUID = nil
local FSName, FSEv, FSSource, FSTarget, FSSpell, FSBuffType, FSDuration, FSExpires, FSID = nil

local function CheckFSBuff(self, unit)
	local _
	local fsFound = false	
	FSName = ""
	for i=1,40 do
        FSName,_,_,_,FSDuration,FSExpires,FSSource,_,_,FSID = UnitDebuff(unit, i)
        if FSName and FSName == FlameShockName and FSDuration and FSSource == "player" then
			self.timer:Start(1, -1 * GetTime() + FSExpires, FSDuration)
			fsFound = true
			break
        elseif not FSName then
            break
        end
	end
	if not fsFound then
		self.timer:Stop(1)
	end
end

function TotemTimers.FlameShockEvent(self,event,unit,...)
    if event == "COMBAT_LOG_EVENT_UNFILTERED" then
		local _
        FSEv, _, _, FSSource, _, _, FSTarget, _, _, _, FSSpell = ...
		if FSEv == "UNIT_DIED" then
            if FSAuraGUID and FSTarget == FSAuraGUID then
                self.timer:Stop(1)
                FSAuraGUID = nil
            end
        end
    elseif (event == "PLAYER_TARGET_CHANGED" and UnitExists("target"))
	  or (event == "UNIT_AURA" and unit == "target") then
        FSAuraGUID = UnitGUID("target")
        CheckFSBuff(self, "target")
	elseif event == "PET_BATTLE_OPENING_START" then
		if self:IsShown() then
			self.shownBeforePet = true
			self:Hide()
		end
	elseif event == "PET_BATTLE_CLOSE" then
		if self.shownBeforePet then
			self:Show()
			self.shownBeforePet = false
		end
    end
end


local StormStrike = SpellNames[SpellIDs.StormStrike]

function TotemTimers.StormStrikeEvent(self, event, unit, ...)
    if (event == "UNIT_AURA" and unit == "target") or event == "PLAYER_TARGET_CHANGED" then
        if UnitExists("target") then
            for i = 1,40 do
                local name,_,_,_,duration,endtime = UnitDebuff("target", i)
                if name then
                    if name == StormStrike then
                        local timeleft = endtime - GetTime()
                        self.timer:StartBarTimer(timeleft, duration)
                        return
                     end
                else
                    break
                end
            end
        end
        if self.timer.barTimer>0 then
            self.timer:StopBarTimer()
        end
    else
        TotemTimers.EnhanceCDEvents(self, event, unit, ...)
    end
end


local Focused = 43339
local ElementalFocus = 16164
local ClearCasting = 16246
local ShockBuffActive = {}
function TotemTimers.ShockEvent(self, event, unit, ...)
    if event == "UNIT_AURA" and unit == "player" then
        for i = 1,40 do
            local name,_,_,_,duration,endtime,_,_,_,spellID = UnitBuff("player", i)
            if spellID then
                if (spellID == Focused or spellID == ElementalFocus or spellID == ClearCasting) then
                    if not ShockBuffActive[self.timer.nr] then
                        ShockBuffActive[self.timer.nr] = true
                        ActionButton_ShowOverlayGlow(self)
                    end
                    return
                 end
            else
                break
            end
        end
        if ShockBuffActive[self.timer.nr] then
            ShockBuffActive[self.timer.nr] = false
            ActionButton_HideOverlayGlow(self)
        end
    else
        TotemTimers.EnhanceCDEvents(self, event, unit, ...)
    end
end


local actionToButton = {}
hooksecurefunc("ActionButton_Update", function(self)
    if self.action then actionToButton[self.action] = self end
end)

--[[local multiBarButtonNames = {
    [3] = "MultiBarRightButton",
    [4] = "MultiBarLeftButton",
    [5] = "MultiBarBottomRightButton",
    [6] = "MultiBarBottomLeftButton",
}]]

local function slotToButton(slot)
    if _G["BT4Button"..slot] then
        return _G["BT4Button"..slot]
    end
    if ElvUI then
        return _G["ElvUI_Bar"..ceil(slot/12).."Button"..((slot - 1) % 12 + 1)]
    end
    return actionToButton[slot]
    --[[if slot < 25 then
        if floor(slot/12) == GetActionBarPage() - 1 then
            return _G["ActionButton"..(slot % 12)]
        end
    elseif slot < 73 then
        return _G[multiBarButtonNames[ceil(slot/12)]..(slot % 12)]
    end]]
end


local function FindActionButtons(spellList)
    local buttons = {}
    for i = 1,#spellList do
        local spell = spellList[i]
        if AvailableSpells[spell] then
            local spellName = SpellNames[spell]
            local spellID = select(7, GetSpellInfo(spellName))

            local slots = C_ActionBar.FindSpellActionButtons(spellID);
            if slots then
                for slot = 1,#slots do
                    local button = slotToButton(slots[slot])
                    if button then
                        table.insert(buttons, button)
                    end
                end
            end
            for cd = 1,#cds do
                if cds[cd].button.cdspell == spell then
                    table.insert(buttons, cds[cd].button)
                end
            end
        end
    end
    return buttons
end
TotemTimers.FindActionButtons = FindActionButtons

local function ShowButtonsOverlayGlow(buttonList)
    for b = 1, #buttonList do
        ActionButton_ShowOverlayGlow(buttonList[b])
    end
end
TotemTimers.ShowButtonsOverlayGlow = ShowButtonsOverlayGlow

local function HideButtonsOverlayGlow(buttonList)
    for b = 1, #buttonList do
        ActionButton_HideOverlayGlow(buttonList[b])
    end
end
TotemTimers.HideButtonsOverlayGlow = HideButtonsOverlayGlow

local maelstromSpells = TotemTimers.MaelstromSpells
local maelstromSpellsButtons = nil

local MaelstromName = GetSpellInfo(SpellIDs.Maelstrom)
local lastMaelstromCount = 0

function TotemTimers.MaelstromEvent(self)
    local _,_,count = AuraUtil.FindAuraByName(MaelstromName, "player", "HELPFUL")
    local numberOnly = Maelstrom.NumberOnly

    if (not count or count < 5) and maelstromSpellsButtons and #maelstromSpellsButtons > 0 then
        HideButtonsOverlayGlow(maelstromSpellsButtons)
        maelstromSpellsButtons = nil
        ActionButton_HideOverlayGlow(TotemTimers.MaelstromButton)
    end

    if not count then
        Maelstrom:Stop(1)
        Maelstrom.timerBars[1].time:SetText("")
        Maelstrom.timerBars[1].time.AnimGroup:Stop()

        if not numberOnly then
            MaelstromIcon:Hide()
            MaelstromIcon.icon:SetTexture(nil)
            MaelstromIcon.icon.AnimGroup:Stop()
        end
    else
        Maelstrom.timerBars[1].time:SetText(count)
        Maelstrom:Start(1, count, 5)
        --Maelstrom:SetBarColor(.6 + count * .04, .6 + count * .04, .8 + count * .04)
        Maelstrom:SetBarColor(0, RAID_CLASS_COLORS.SHAMAN.g * 0.8 + count * .04, RAID_CLASS_COLORS.SHAMAN.b * 0.8 + count * .04)

        local animate
        if numberOnly then
            MaelstromIcon:Hide()
            MaelstromIcon.icon.AnimGroup:Stop()
            animate = Maelstrom.timerBars[1].time
        else
            MaelstromIcon:Show()
            MaelstromIcon.icon:SetTexture("Interface/AddOns/TotemTimers/textures/maelstrom_weapon"..(count < 5 and "_"..count or ""))
            animate = MaelstromIcon.icon
            ActionButton_HideOverlayGlow(TotemTimers.MaelstromButton)
        end


        if count < 5 then
            animate.AnimGroup:Stop()
        else
            animate.AnimGroup:Play()

            if lastMaelstromCount < count then
                XiTimers.PlayWarning(self, "Maelstrom")
                if Maelstrom.StopPulseOn5 then
                    Maelstrom.animation:Play()
                end
                if TotemTimers.ActiveProfile.OverlayGlow then
                    maelstromSpellsButtons = FindActionButtons(maelstromSpells)
                    ShowButtonsOverlayGlow(maelstromSpellsButtons)
                end
            end

            if numberOnly then
                ActionButton_ShowOverlayGlow(TotemTimers.MaelstromButton)
            end
        end

        lastMaelstromCount = count
    end
end



function gmb()
    maelstromSpellsButtons = FindActionButtons(TotemTimers.MaelstromSpells)
end

function smb()
    for b = 1, #maelstromSpellsButtons do
        ActionButton_ShowOverlayGlow(maelstromSpellsButtons[b])
    end
end

function hmb()
    for b = 1, #maelstromSpellsButtons do
        ActionButton_HideOverlayGlow(maelstromSpellsButtons[b])
    end
end