ice-hud/modules/ShadowOrbs.lua

local L = LibStub("AceLocale-3.0"):GetLocale("IceHUD", false)
local ShadowOrbs = IceCore_CreateClass(IceClassPowerCounter)

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

	self:SetDefaultColor("ShadowOrbsNumeric", 218, 231, 31)

	self.numericColor = "ShadowOrbsNumeric"
	self.unitPower = SPELL_POWER_SHADOW_ORBS
	self.minLevel = SHADOW_ORBS_SHOW_LEVEL
	self.bTreatEmptyAsFull = true
	self.unit = "player"
	if IceHUD.WowVer >= 60000 then
		self.numConsideredFull = PRIEST_BAR_NUM_LARGE_ORBS
		-- pulled from PriestBar.xml in Blizzard's UI source
		self.runeCoords = { }
		for i=1, PRIEST_BAR_NUM_SMALL_ORBS do
			self.runeCoords[i] = {0.45703125, 0.60546875, 0.44531250, 0.73437500}
		end
		if IsSpellKnown(SHADOW_ORB_MINOR_TALENT_ID) then
			self.numRunes = PRIEST_BAR_NUM_SMALL_ORBS
		else
			self.numRunes = PRIEST_BAR_NUM_LARGE_ORBS
		end
	else
		self.numConsideredFull = PRIEST_BAR_NUM_ORBS

		-- pulled from PriestBar.xml in Blizzard's UI source
		self.runeCoords = { }
		for i=1, PRIEST_BAR_NUM_ORBS do
			self.runeCoords[i] = {0.45703125, 0.60546875, 0.44531250, 0.73437500}
		end
	end
	self.runeHeight = 36
	self.runeWidth = 36
	self.requiredSpec = SPEC_PRIEST_SHADOW
end

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

	if IceHUD.WowVer >= 60000 then
		if not IsSpellKnown(SHADOW_ORB_MINOR_TALENT_ID) then
			self:RegisterEvent("SPELLS_CHANGED", "CheckHasMoreOrbs")
		end
	end
end

function ShadowOrbs.prototype:CheckHasMoreOrbs(event)
	if IsSpellKnown(SHADOW_ORB_MINOR_TALENT_ID) then
		self:UnregisterEvent("SPELLS_CHANGED")
		self:UpdateRunePower()
	end
end

function ShadowOrbs.prototype:GetOptions()
	local opts = ShadowOrbs.super.prototype.GetOptions(self)

	opts.hideBlizz.desc = L["Hides Blizzard Shadow Orb frame and disables all events related to it.\n\nNOTE: Blizzard attaches the shadow orb UI to the player's unitframe, so if you have that hidden in PlayerHealth, then this won't do anything."]

	return opts
end

function ShadowOrbs.prototype:GetRuneTexture(rune)
	if not rune or rune ~= tonumber(rune) then
		return
	end
	return "Interface\\PlayerFrame\\Priest-ShadowUI"
end

function ShadowOrbs.prototype:ShowBlizz()
	PriestBarFrame:Show()

	PriestBarFrame:GetScript("OnLoad")(PriestBarFrame)
end

function ShadowOrbs.prototype:HideBlizz()
	PriestBarFrame:Hide()

	PriestBarFrame:UnregisterAllEvents()
end

function ShadowOrbs.prototype:UpdateRunePower()
	local numRunes = UnitPowerMax(self.unit, self.unitPower)

	if self.fakeNumRunes ~= nil and self.fakeNumRunes > 0 then
		numRunes = self.fakeNumRunes
	end

	if numRunes ~= self.numRunes then
		if numRunes < self.numRunes and #self.frame.graphical >= numRunes then
			for i=numRunes + 1, #self.frame.graphical do
				self.frame.graphical[i]:Hide()
			end
		end
		self.numRunes = numRunes

		self:CreateRuneFrame()
	end
	ShadowOrbs.super.prototype.UpdateRunePower(self)
end

-- Load us up
local _, unitClass = UnitClass("player")
if (unitClass == "PRIEST" and PriestBarFrame) then
	IceHUD.ShadowOrbs = ShadowOrbs:new()
end