mirror of
https://github.com/parnic/ice-hud.git
synced 2025-06-16 06:40:13 -05:00
55028c88fb
- on first run, forced reset all custom DogTags to the module default since some of the old default tags no longer work
148 lines
3.2 KiB
Lua
148 lines
3.2 KiB
Lua
local AceOO = AceLibrary("AceOO-2.0")
|
|
|
|
local PetHealth = AceOO.Class(IceUnitBar)
|
|
|
|
PetHealth.prototype.happiness = nil
|
|
|
|
|
|
-- Constructor --
|
|
function PetHealth.prototype:init()
|
|
PetHealth.super.prototype.init(self, "PetHealth", "pet")
|
|
|
|
self:SetDefaultColor("PetHealthHappy", 37, 164, 30)
|
|
self:SetDefaultColor("PetHealthContent", 164, 164, 30)
|
|
self:SetDefaultColor("PetHealthUnhappy", 164, 30, 30)
|
|
|
|
self.scalingEnabled = true
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
function PetHealth.prototype:GetDefaultSettings()
|
|
local settings = PetHealth.super.prototype.GetDefaultSettings(self)
|
|
|
|
settings["side"] = IceCore.Side.Left
|
|
settings["offset"] = -1
|
|
settings.scale = 0.7
|
|
settings["textVerticalOffset"] = 4
|
|
settings["upperText"] = "[PercentHP:Round]"
|
|
settings["lowerText"] = ""
|
|
|
|
return settings
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
--[[
|
|
function PetHealth.prototype:CreateFrame()
|
|
PetHealth.super.prototype.CreateFrame(self)
|
|
|
|
local point, relativeTo, relativePoint, xoff, yoff = self.frame.bottomUpperText:GetPoint()
|
|
if (point == "TOPLEFT") then
|
|
point = "BOTTOMLEFT"
|
|
else
|
|
point = "BOTTOMRIGHT"
|
|
end
|
|
|
|
self.frame.bottomUpperText:ClearAllPoints()
|
|
self.frame.bottomUpperText:SetPoint(point, relativeTo, relativePoint, 0, 0)
|
|
end
|
|
]]
|
|
|
|
function PetHealth.prototype:Enable(core)
|
|
PetHealth.super.prototype.Enable(self, core)
|
|
|
|
self:RegisterEvent("PET_UI_UPDATE", "CheckPet");
|
|
self:RegisterEvent("PLAYER_PET_CHANGED", "CheckPet");
|
|
self:RegisterEvent("PET_BAR_CHANGED", "CheckPet");
|
|
self:RegisterEvent("UNIT_PET", "CheckPet");
|
|
|
|
self:RegisterEvent("UNIT_HEALTH", "Update")
|
|
self:RegisterEvent("UNIT_MAXHEALTH", "Update")
|
|
|
|
self:RegisterEvent("UNIT_HAPPINESS", "PetHappiness")
|
|
|
|
self:CheckPet()
|
|
end
|
|
|
|
function PetHealth.prototype:PetHappiness(unit)
|
|
if (unit and (unit ~= self.unit)) then
|
|
return
|
|
end
|
|
|
|
self.happiness = GetPetHappiness()
|
|
self.happiness = self.happiness or 3 -- '3' means happy
|
|
self:Update(unit)
|
|
end
|
|
|
|
|
|
function PetHealth.prototype:CheckPet()
|
|
if (UnitExists(self.unit)) then
|
|
self.frame:Show()
|
|
self:PetHappiness(self.unit)
|
|
self:Update(self.unit)
|
|
else
|
|
self.frame:Hide()
|
|
end
|
|
end
|
|
|
|
|
|
function PetHealth.prototype:Update(unit)
|
|
PetHealth.super.prototype.Update(self)
|
|
if (unit and (unit ~= self.unit)) then
|
|
return
|
|
end
|
|
|
|
local color = "PetHealthHappy"
|
|
if (self.happiness == 2) then
|
|
color = "PetHealthContent"
|
|
elseif(self.happiness == 1) then
|
|
color = "PetHealthUnhappy"
|
|
end
|
|
|
|
if (self.moduleSettings.scaleColor) then
|
|
color = "ScaledHealthColor"
|
|
end
|
|
|
|
if not (self.alive) then
|
|
color = "Dead"
|
|
end
|
|
|
|
|
|
self:UpdateBar(self.health/self.maxHealth, color)
|
|
|
|
if not AceLibrary:HasInstance("LibDogTag-3.0") then
|
|
self:SetBottomText1(math.floor(self.healthPercentage * 100))
|
|
end
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
function PetHealth.prototype:GetOptions()
|
|
local opts = PetHealth.super.prototype.GetOptions(self)
|
|
|
|
opts["scaleHealthColor"] = {
|
|
type = "toggle",
|
|
name = "Color bar by health %",
|
|
desc = "Colors the health bar from MaxHealthColor to MinHealthColor based on current health %",
|
|
get = function()
|
|
return self.moduleSettings.scaleHealthColor
|
|
end,
|
|
set = function(value)
|
|
self.moduleSettings.scaleHealthColor = value
|
|
self:Redraw()
|
|
end,
|
|
disabled = function()
|
|
return not self.moduleSettings.enabled
|
|
end,
|
|
order = 41
|
|
}
|
|
|
|
return opts
|
|
end
|
|
|
|
|
|
|
|
-- Load us up
|
|
IceHUD.PetHealth = PetHealth:new()
|