ice-hud/modules/EclipseBar.lua

local AceOO = AceLibrary("AceOO-2.0")

local EclipseBar = AceOO.Class(IceBarElement)
EclipseBar.prototype.markerHeight = 6
EclipseBar.prototype.barUpdateColor = "EclipseLunar"

function EclipseBar.prototype:init()
	EclipseBar.super.prototype.init(self, "EclipseBar")

	self:SetDefaultColor("EclipseLunar", 31, 31, 231)
	self:SetDefaultColor("EclipseLunarActive", 0, 0, 255)
	self:SetDefaultColor("EclipseSolar", 190, 210, 31)
	self:SetDefaultColor("EclipseSolarActive", 238, 251, 31)
	self:SetDefaultColor("EclipseMarker", 255, 0, 0)
end

function EclipseBar.prototype:GetOptions()
	local opts = EclipseBar.super.prototype.GetOptions(self)
	opts.reverse.hidden = true
	return opts
end

function EclipseBar.prototype:GetDefaultSettings()
	local defaults =  EclipseBar.super.prototype.GetDefaultSettings(self)

	defaults.textVisible.lower = false
	defaults.offset = -1
	defaults.enabled = true
	defaults.usesDogTagStrings = false
	defaults.textVerticalOffset = 13
	defaults.textHorizontalOffset = 12
	defaults.shouldAnimate = false
	defaults.hideAnimationSettings = true
	defaults.lockUpperTextAlpha = false

	return defaults
end

function EclipseBar.prototype:Enable(core)
	EclipseBar.super.prototype.Enable(self, core)

	self:RegisterEvent("UPDATE_SHAPESHIFT_FORM", "UpdateShown")
	self:RegisterEvent("PLAYER_TALENT_UPDATE", "UpdateShown")
	self:RegisterEvent("MASTERY_UPDATE", "UpdateShown")
	self:RegisterEvent("UNIT_AURA", "UpdateEclipseBuffs")

	self.frame:SetScript("OnUpdate", function() self:Update() end)

	self:UpdateShown()
end

function EclipseBar.prototype:Disable(core)
	EclipseBar.super.prototype.Disable(self, core)
end

function EclipseBar.prototype:CreateFrame()
	EclipseBar.super.prototype.CreateFrame(self)

	self:CreateSolarBar()
	self:CreateMarker()
	self:UpdateShown()
	self:UpdateAlpha()
end

function EclipseBar.prototype:RegisterOnUpdate()
	return false
end

function EclipseBar.prototype:CreateSolarBar()
	if not (self.solarBar) then
		self.solarBar = CreateFrame("Frame", nil, self.frame)
	end

	local solarTop = not IceHUD:xor(self.moduleSettings.reverse, self.moduleSettings.inverse)

	self.solarBar:SetFrameStrata("BACKGROUND")
	self.solarBar:SetWidth(self.settings.barWidth + (self.moduleSettings.widthModifier or 0))
	self.solarBar:SetHeight(self.settings.barHeight)
	self.solarBar:ClearAllPoints()
	if solarTop then
		self.solarBar:SetPoint("TOPLEFT", self.frame, "TOPLEFT")
	else
		self.solarBar:SetPoint("BOTTOMLEFT", self.frame, "BOTTOMLEFT")
	end

	if not (self.solarBar.bar) then
		self.solarBar.bar = self.solarBar:CreateTexture(nil, "LOW")
	end

	self.solarBar.bar:SetTexture(IceElement.TexturePath .. self:GetMyBarTexture())
	self.solarBar.bar:SetAllPoints(self.solarBar)

	self.solarBar.bar:SetVertexColor(self:GetColor("EclipseSolar", 1))

	local pos = 0.5
	local min_y = 0
	local max_y = pos
	if not solarTop then
		min_y = 1-pos
		max_y = 1
	end

	if self.moduleSettings.side == IceCore.Side.Left then
		self.solarBar.bar:SetTexCoord(1, 0, min_y, max_y)
	else
		self.solarBar.bar:SetTexCoord(0, 1, min_y, max_y)
	end

	self.solarBar.bar:Show()
	self.solarBar:SetHeight(self.settings.barHeight * pos)
end

function EclipseBar.prototype:CreateMarker()
	local bIsNewBar = false
	if not (self.markerBar) then
		self.markerBar = CreateFrame("Frame", nil, self.frame)
		bIsNewBar = true
	end

	self.markerBar:SetFrameStrata("LOW")
	self.markerBar:SetWidth(self.settings.barWidth + (self.moduleSettings.widthModifier or 0))
	self.markerBar:SetHeight(self.markerHeight)
	self.markerBar:ClearAllPoints()

	if not (self.markerBar.bar) then
		self.markerBar.bar = self.markerBar:CreateTexture(nil, "LOW")
	end

	self.markerBar.bar:SetTexture(IceElement.TexturePath .. self:GetMyBarTexture())
	self.markerBar.bar:SetAllPoints(self.markerBar)

	self.markerBar.bar:SetVertexColor(self:GetColor("EclipseMarker", 1))

	if bIsNewBar then
		self:PositionMarker(0)
	end
end

function EclipseBar.prototype:PositionMarker(pos)
	if self.moduleSettings.inverse then
		pos = pos * -1
	end
	local coordPos = 0.5 + pos
	local adjustedBarHeight = self.settings.barHeight - (self.markerHeight)
	local heightScale = (self.markerHeight / self.settings.barHeight) / 2

	local min_y = 1-coordPos-heightScale
	local max_y = 1-coordPos+heightScale

	if self.moduleSettings.side == IceCore.Side.Left then
		self.markerBar.bar:SetTexCoord(1, 0, min_y, max_y)
	else
		self.markerBar.bar:SetTexCoord(0, 1, min_y, max_y)
	end

	self.markerBar.bar:Show()
	self.markerBar:SetPoint("CENTER", self.frame, "CENTER", 0, (self.settings.barHeight * pos))
end

function EclipseBar.prototype:UpdateShown()
	local form  = GetShapeshiftFormID();

	if form == MOONKIN_FORM or not form then
		if GetMasteryIndex(GetActiveTalentGroup(false, false)) == 1 then
			self:Show(true)
		else
			self:Show(false)
		end
	else
		self:Show(false)
	end
end

function EclipseBar.prototype:UseTargetAlpha(scale)
	return UnitPower("player", SPELL_POWER_ECLIPSE) ~= 0
end

function EclipseBar.prototype:UpdateEclipseBuffs()
	local buffStatus = IceHUD:HasBuffs("player", {ECLIPSE_BAR_SOLAR_BUFF_ID, ECLIPSE_BAR_LUNAR_BUFF_ID})
	local hasSolar = buffStatus[1]
	local hasLunar = buffStatus[2]

	if hasSolar then
		self.barUpdateColor = "EclipseSolarActive"
		self.solarBar.bar:SetVertexColor(self:GetColor("EclipseSolarActive", 1))
	elseif hasLunar then
		self.barUpdateColor = "EclipseLunarActive"
		self.solarBar.bar:SetVertexColor(self:GetColor("EclipseLunarActive", 1))
	else
		self.barUpdateColor = "EclipseLunar"
		self.solarBar.bar:SetVertexColor(self:GetColor("EclipseSolar", 1))
	end
end

function EclipseBar.prototype:UpdateEclipsePower()
	local power = UnitPower("player", SPELL_POWER_ECLIPSE)
	local maxPower = UnitPowerMax("player", SPELL_POWER_ECLIPSE)

	self:SetBottomText1(abs((power/maxPower) * 100))

	local pos = (power/maxPower) / 2
	self:PositionMarker(pos)
end

function EclipseBar.prototype:Update()
	EclipseBar.super.prototype.Update(self)

	self:UpdateEclipsePower()
	self:UpdateBar(0.5, self.barUpdateColor)
	self:UpdateAlpha()
end

local _, unitClass = UnitClass("player")
if (unitClass == "DRUID" and IceHUD.WowVer >= 40000) then
	IceHUD.EclipseBar = EclipseBar:new()
end