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

local SPELL_POWER_RAGE = SPELL_POWER_RAGE
local SPELL_POWER_FOCUS = SPELL_POWER_FOCUS
local SPELL_POWER_ENERGY = SPELL_POWER_ENERGY
if Enum and Enum.PowerType then
	SPELL_POWER_RAGE = Enum.PowerType.Rage
	SPELL_POWER_FOCUS = Enum.PowerType.Focus
	SPELL_POWER_ENERGY = Enum.PowerType.Energy
end

-- Constructor --
function FocusMana.prototype:init()
	FocusMana.super.prototype.init(self, "FocusMana", "focus")

	self:SetDefaultColor("FocusMana", 52, 64, 221)
	self:SetDefaultColor("FocusRage", 235, 44, 26)
	self:SetDefaultColor("FocusEnergy", 228, 242, 31)
	self:SetDefaultColor("FocusFocus", 242, 149, 98)
end


function FocusMana.prototype:GetDefaultSettings()
	local settings = FocusMana.super.prototype.GetDefaultSettings(self)

	settings["enabled"] = false
	settings["side"] = IceCore.Side.Right
	settings["offset"] = -2
	settings["scale"] = 0.7
	settings["upperText"] = "[PercentMP:Round]"
	settings["lowerText"] = ""
	settings["barVerticalOffset"] = 35

	return settings
end


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

	if IceHUD.WowVer >= 40000 then
		self:RegisterEvent(IceHUD.UnitPowerEvent, "UpdateEvent")
		if IceHUD.WowVer < 80000 then
			self:RegisterEvent("UNIT_MAXPOWER", "UpdateEvent")
		end
	else
		self:RegisterEvent("UNIT_MANA", "UpdateEvent")
		self:RegisterEvent("UNIT_MAXMANA", "UpdateEvent")
		self:RegisterEvent("UNIT_RAGE", "UpdateEvent")
		self:RegisterEvent("UNIT_MAXRAGE", "UpdateEvent")
		self:RegisterEvent("UNIT_ENERGY", "UpdateEvent")
		self:RegisterEvent("UNIT_MAXENERGY", "UpdateEvent")
		self:RegisterEvent("UNIT_FOCUS", "UpdateEvent")
		self:RegisterEvent("UNIT_MAXFOCUS", "UpdateEvent")
	end
	self:RegisterEvent("UNIT_AURA", "UpdateEvent")
	self:RegisterEvent("UNIT_FLAGS", "UpdateEvent")
	self:RegisterEvent("PLAYER_FOCUS_CHANGED", "UpdateFocus")

	self:Update(self.unit)
end

function FocusMana.prototype:UpdateFocus()
	self:Update(self.unit)
end

function FocusMana.prototype:UpdateEvent(event, unit)
	self:Update(unit)
end

function FocusMana.prototype:Update(unit)
	FocusMana.super.prototype.Update(self)
	if (unit and (unit ~= self.unit)) then
		return
	end

	if ((not UnitExists(unit)) or (self.maxMana == 0)) then
		self:Show(false)
		return
	else
		self:Show(true)
	end


	local manaType = UnitPowerType(self.unit)

	local color = "FocusMana"
	if (self.moduleSettings.scaleManaColor) then
		color = "ScaledManaColor"
	end
	if (manaType == SPELL_POWER_RAGE) then
		color = "FocusRage"
	elseif (manaType == SPELL_POWER_FOCUS) then
		color = "FocusFocus"
	elseif (manaType == SPELL_POWER_ENERGY) then
		color = "FocusEnergy"
	end

	if (self.tapped) then
		color = "Tapped"
	end

	self:UpdateBar(self.manaPercentage, color)

	if not IceHUD.IceCore:ShouldUseDogTags() then
		self:SetBottomText1(math.floor(self.manaPercentage * 100))
		self:SetBottomText2(self:GetFormattedText(self.mana, self.maxMana), color)
	end
end


-- OVERRIDE
function FocusMana.prototype:GetOptions()
	local opts = FocusMana.super.prototype.GetOptions(self)

	opts["scaleManaColor"] = {
		type = "toggle",
		name = L["Color bar by mana %"],
		desc = L["Colors the mana bar from MaxManaColor to MinManaColor based on current mana %"],
		get = function()
			return self.moduleSettings.scaleManaColor
		end,
		set = function(info, value)
			self.moduleSettings.scaleManaColor = value
			self:Redraw()
		end,
		disabled = function()
			return not self.moduleSettings.enabled
		end,
		order = 51
	}

	return opts
end


-- Load us up
if FocusUnit then
	IceHUD.FocusMana = FocusMana:new()
end
