WoWInterface SVN bdGrid

--[[ Element: Eclipse Bar

 Handle updating and visibility of the Druid eclipse state status bars.

 Widget

 EclipseBar - A table to hold the sub-widgets.

 Sub-Widgets

 LunarBar - A StatusBar used to represent the lunar power state.

 SolarBar - A StatusBar used to represent the solar power state.

 Notes

 The default StatusBar texture will be applied if the UI widget doesn't have a

 status bar texture or color defined.

 Examples

   local EclipseBar = CreateFrame('Frame', nil, self)

   EclipseBar:SetPoint('BOTTOM', self, 'TOP')

   EclipseBar:SetSize(160, 20)

   -- Position and size

   local LunarBar = CreateFrame('StatusBar', nil, EclipseBar)

   LunarBar:SetPoint('LEFT')

   LunarBar:SetSize(160, 20)

   local SolarBar = CreateFrame('StatusBar', nil, EclipseBar)

   SolarBar:SetPoint('LEFT', LunarBar:GetStatusBarTexture(), 'RIGHT')

   SolarBar:SetSize(160, 20)

   -- Register with oUF

   EclipseBar.LunarBar = LunarBar

   EclipseBar.SolarBar = SolarBar

   self.EclipseBar = EclipseBar

 Hooks and Callbacks

 Override(self) - Used to completely override the internal update function.

                  Removing the table key entry will make the element fall-back

                  to its internal function again.

]]

if(select(2, UnitClass('player')) ~= 'DRUID') then return end

local parent, ns = ...

local oUF = ns.oUF

local ECLIPSE_BAR_SOLAR_BUFF = GetSpellInfo(171744)

local ECLIPSE_BAR_LUNAR_BUFF = GetSpellInfo(171743)

local SPELL_POWER_ECLIPSE = SPELL_POWER_ECLIPSE

local MOONKIN_FORM = MOONKIN_FORM

local UNIT_POWER = function(self, event, unit, powerType)

        if(self.unit ~= unit or (event == 'UNIT_POWER_FREQUENT' and powerType ~= 'ECLIPSE')) then return end

        local eb = self.EclipseBar

        local power = UnitPower('player', SPELL_POWER_ECLIPSE)

        local maxPower = UnitPowerMax('player', SPELL_POWER_ECLIPSE)

        if(eb.LunarBar) then

                eb.LunarBar:SetMinMaxValues(-maxPower, maxPower)

                eb.LunarBar:SetValue(power)

        end

        if(eb.SolarBar) then

                eb.SolarBar:SetMinMaxValues(-maxPower, maxPower)

                eb.SolarBar:SetValue(power * -1)

        end

        if(eb.PostUpdatePower) then

                --[[ :PostUpdatePower(unit)

                 Callback which is called after lunar and solar bar was updated.

                 Arguments

                 self - The widget that holds the eclipse frame.

                 unit - The unit that has the widget.

                 power - The unit's current power.

                 maxPower - The unit's maximum power.

                ]]

                return eb:PostUpdatePower(unit, power, maxPower)

        end

end

local UPDATE_VISIBILITY = function(self, event)

        local eb = self.EclipseBar

        -- check form/mastery

        local showBar

        local form = GetShapeshiftFormID()

        if(not form) then

                local ptt = GetSpecialization()

                if(ptt and ptt == 1) then -- player has balance spec

                        showBar = true

                end

        elseif(form == MOONKIN_FORM) then

                showBar = true

        end

        if(UnitHasVehicleUI'player') then

                showBar = false

        end

        if(showBar) then

                eb:Show()

        else

                eb:Hide()

        end

        if(eb.PostUpdateVisibility) then

                --[[ :PostUpdateVisibility(unit)

                 Callback which is called after the eclipse frame was shown or hidden.

                 Arguments

                 self - The widget that holds the eclipse frame.

                 unit - The unit that has the widget.

                ]]

                return eb:PostUpdateVisibility(self.unit)

        end

end

local UNIT_AURA = function(self, event, unit)

        if(self.unit ~= unit) then return end

        local eb = self.EclipseBar

        local hasSolarEclipse = not not UnitBuff(unit, ECLIPSE_BAR_SOLAR_BUFF)

        local hasLunarEclipse = not not UnitBuff(unit, ECLIPSE_BAR_LUNAR_BUFF)

        if(eb.hasSolarEclipse == hasSolarEclipse and eb.hasLunarEclipse == hasLunarEclipse) then return end

        eb.hasSolarEclipse = hasSolarEclipse

        eb.hasLunarEclipse = hasLunarEclipse

        if(eb.PostUnitAura) then

                --[[ :PostUnitAura(unit)

                 Callback which is called after the eclipse state was checked.

                 Arguments

                 self - The widget that holds the eclipse frame.

                 unit - The unit that has the widget.

                ]]

                return eb:PostUnitAura(unit)

        end

end

local ECLIPSE_DIRECTION_CHANGE = function(self, event, direction)

        local eb = self.EclipseBar

        eb.directionIsLunar = direction == "moon"

        eb.direction = direction

        if(eb.PostDirectionChange) then

                --[[ :PostDirectionChange(unit)

                 Callback which is called after eclipse direction was changed.

                 Arguments

                 self - The widget that holds the eclipse frame.

                 unit - The unit that has the widget.

                ]]

                return eb:PostDirectionChange(self.unit)

        end

end

local Update = function(self, event, ...)

        UNIT_POWER(self, event, ...)

        UNIT_AURA(self, event, ...)

        ECLIPSE_DIRECTION_CHANGE(self, event)

        return UPDATE_VISIBILITY(self, event)

end

local ForceUpdate = function(element)

        return Update(element.__owner, 'ForceUpdate', element.__owner.unit, 'ECLIPSE')

end

local function Enable(self)

        local eb = self.EclipseBar

        if(eb) then

                eb.__owner = self

                eb.ForceUpdate = ForceUpdate

                if(eb.LunarBar and eb.LunarBar:IsObjectType'StatusBar' and not eb.LunarBar:GetStatusBarTexture()) then

                        eb.LunarBar:SetStatusBarTexture([[Interface\TargetingFrame\UI-StatusBar]])

                end

                if(eb.SolarBar and eb.SolarBar:IsObjectType'StatusBar' and not eb.SolarBar:GetStatusBarTexture()) then

                        eb.SolarBar:SetStatusBarTexture([[Interface\TargetingFrame\UI-StatusBar]])

                end

                self:RegisterEvent('ECLIPSE_DIRECTION_CHANGE', ECLIPSE_DIRECTION_CHANGE, true)

                self:RegisterEvent('PLAYER_TALENT_UPDATE', UPDATE_VISIBILITY, true)

                self:RegisterEvent('UNIT_AURA', UNIT_AURA)

                self:RegisterEvent('UNIT_POWER_FREQUENT', UNIT_POWER)

                self:RegisterEvent('UPDATE_SHAPESHIFT_FORM', UPDATE_VISIBILITY, true)

                return true

        end

end

local function Disable(self)

        local eb = self.EclipseBar

        if(eb) then

                eb:Hide()

                self:UnregisterEvent('ECLIPSE_DIRECTION_CHANGE', ECLIPSE_DIRECTION_CHANGE)

                self:UnregisterEvent('PLAYER_TALENT_UPDATE', UPDATE_VISIBILITY)

                self:UnregisterEvent('UNIT_AURA', UNIT_AURA)

                self:UnregisterEvent('UNIT_POWER_FREQUENT', UNIT_POWER)

                self:UnregisterEvent('UPDATE_SHAPESHIFT_FORM', UPDATE_VISIBILITY)

        end

end

oUF:AddElement('EclipseBar', Update, Enable, Disable)