Make status colors configurable

The aura based status colors can now be configured in frames.lua. This
is important in TBC where some healers don't care about e.g. magic
debuffs.

src/frames.lua

@@ -26,9 +26,12 @@ local colors = {
     disease = { 0.4, 0.2, 0.0 },
     poison = { 0.0, 0.7, 0.7 },
     curse = { 0.7, 0.0, 0.7 },
+    immune = { 0.0, 0.2, 0.4 },
+    bomb = { 1.0, 0.7, 0.7 },
     red = { 1, 0, 0 },
     blue = { 0.4, 0.4, 1 },
     light_blue = { 0.7, 0.7, 1 },
+    green = { 0.2, 0.9, 0.2 },
 }
 
 ---@param list number[]
@@ -43,6 +46,9 @@ end
 
 local AURAS_PRIEST_RENEW = set({ 139, 6074, 6075, 6076, 6077, 6078, 10927, 10928, 10929, 25315, 25221, 25222 })
 local AURAS_PRIEST_PWS = set({ 17, 592, 600, 3747, 6065, 6066, 10898, 10899, 10900, 10901, 25217, 25218 })
+local AURAS_DRUID_LIFEBLOOM = set({ 33763 })
+local AURAS_DRUID_REJUV = set({ 774, 1058, 1430, 2090, 2091, 3627, 8910, 9839, 9840, 9841, 25299, 26981, 26982 })
+local AURAS_DRUID_REGROWTH = set({ 8936, 8938, 8939, 8940, 8941, 9750, 9856, 9857, 9858, 26980 })
 
 local function RangeConfig()
     local _, class = UnitClass("player")
@@ -58,6 +64,12 @@ local function RangeConfig()
             enemy = "Smite",
             fade = 0.2,
         }
+    elseif class == "DRUID" then
+        return {
+            friendly = "Rejuvenation",
+            enemy = "Wrath",
+            fade = 0.2,
+        }
     else
         return {
             fade = 0.2,
@@ -173,6 +185,66 @@ local function MouseConfig()
                 data = "/cast [@UNIT,help]Leap of Faith;\n/stopspelltarget",
             },
         }
+    elseif class == "DRUID" then
+        return {
+            -- No modifier
+            { button = "mouse1", mods = {}, kind = "target" },
+            {
+                button = "mouse2",
+                mods = {},
+                kind = "macro",
+                data = "/use [@UNIT,help]Rejuvenation",
+            },
+            {
+                button = "wheel-up",
+                mods = {},
+                kind = "macro",
+                data = "/use [@UNIT,help]Lifebloom",
+            },
+            {
+                button = "wheel-down",
+                mods = {},
+                kind = "macro",
+                data = "/use [@UNIT,help]Regrowth",
+            },
+
+			-- ctrl
+			{
+				button = "wheel-up",
+				mods = { ctrl = true },
+				kind = "macro",
+				data = "/use [@UNIT,help]Abolish Poison"
+			},
+			{
+				button = "wheel-down",
+				mods = { ctrl = true },
+				kind = "macro",
+				data = "/use [@UNIT,help]Remove Curse"
+			},
+
+            -- alt
+            {
+                button = "mouse1",
+                mods = { alt = true },
+                kind = "macro",
+                data = "/use [@UNIT,help]Innervate",
+            },
+            {
+                button = "mouse2",
+                mods = { alt = true },
+                kind = "macro",
+                data = "/use [@UNIT,dead,help]Rebirth",
+            },
+
+            -- Shift
+			{
+                button = "wheel-down",
+                mods = { shift = true },
+                kind = "macro",
+                data = "/use [@UNIT,help]Swiftmend",
+            },
+            { button = "mouse2", mods = { shift = true }, kind = "togglemenu" },
+        }
     elseif class == "MONK" then
         return {
             -- No modifier
@@ -447,6 +519,64 @@ local function TriggerClassConfig()
                 },
             },
         }
+    elseif class == "DRUID" then
+        return {
+            {
+                kind = "AuraTrigger",
+                spellIds = AURAS_DRUID_LIFEBLOOM,
+                own = true, -- Lifebloom
+                indicator = {
+                    kind = "SquareIndicator",
+                    size = 17,
+                    point = "BOTTOMLEFT",
+                    x = 3,
+                    y = 3,
+                    color = colors.green,
+                    showStacks = true,
+                    fadeTime = 7.0,
+                    flashTime = 2,
+                },
+            },
+			{
+                kind = "AuraTrigger",
+                spellIds = AURAS_DRUID_REGROWTH,
+                own = true, -- Regrowth
+                indicator = {
+                    kind = "SquareIndicator",
+                    size = 17,
+                    point = "BOTTOMLEFT",
+                    x = 22,
+                    y = 3,
+                    color = colors.yellow,
+                    fadeTime = 21.0,
+                },
+            },
+            {
+                kind = "AuraTrigger",
+                spellIds = AURAS_DRUID_REJUV,
+                own = true, -- Rejuvenation
+                indicator = {
+                    kind = "SquareIndicator",
+                    size = 17,
+                    point = "BOTTOMLEFT",
+                    x = 41,
+                    y = 3,
+                    color = colors.violet,
+                    fadeTime = 12.0,
+                },
+            },
+            {
+                kind = "StatusTrigger",
+                status = "Burn",
+                invert = false,
+                indicator = {
+                    kind = "BorderIndicator",
+                    thickness = 3.0,
+                    color = colors.red,
+                    level = 0,
+                },
+            },
+        }
     elseif class == "PALADIN" then
         return {
             {
@@ -705,6 +835,51 @@ local function TriggerConfig()
     return triggers
 end
 
+local function StatusColorClassConfig()
+    local _, class = UnitClass("player")
+    if class == "DRUID" then
+        return {
+            { status = "Curse", color = colors.curse },
+            { status = "Poison", color = colors.poison },
+        }
+    elseif class == "MAGE" then
+        return {
+            { status = "Curse", color = colors.curse },
+        }
+    elseif class == "PALADIN" then
+        return {
+            { status = "Magic", color = colors.magic },
+            { status = "Poison", color = colors.poison },
+            { status = "Disease", color = colors.disease },
+        }
+    elseif class == "PRIEST" then
+        return {
+            { status = "Magic", color = colors.magic },
+            { status = "Disease", color = colors.disease },
+        }
+    elseif class == "SHAMAN" then
+        return {
+            { status = "Poison", color = colors.poison },
+            { status = "Disease", color = colors.disease },
+        }
+    end
+
+    return {}
+end
+
+local function StatusColorConfig()
+    local statusColors = {
+        { status = "Immune", color = colors.immune },
+        { status = "Bomb", color = colors.bomb },
+    }
+
+    for _, entry in ipairs(StatusColorClassConfig()) do
+        table.insert(statusColors, entry)
+    end
+
+    return statusColors
+end
+
 local types = omif.GetModule("types")
 local UnitFrame = types.UnitFrame
 local UnitGroup = types.UnitGroup

src/types/unitframe.lua

@@ -50,12 +50,6 @@ local colors = {
     low = { 1.0, 0.0, 0 },
     offline = { 0.7, 0.7, 0.7 },
     dead = { 0.7, 0.7, 0.7 },
-    magic = { 0.4, 0.4, 1.0 },
-    disease = { 0.4, 0.2, 0.0 },
-    poison = { 0.0, 0.7, 0.7 },
-    curse = { 0.7, 0.0, 0.7 },
-    immune = { 0.0, 0.2, 0.4 },
-    bomb = { 1.0, 0.7, 0.7 },
     cyan = { 0.0, 0.8, 0.8 },
     white = { 1.0, 1.0, 1.0 },
     shield = { 0.85, 0.95, 1.0 },
@@ -90,6 +84,8 @@ function UnitFrame:Init(unit, config)
     self.rangeEnemy = config.range.enemy
     self.rangeFade = config.range.fade
 
+    self.statusColors = config.statusColors
+
     local secure = self:CreateSecureFrame(width, height)
     secure:Hide()
 
@@ -126,8 +122,6 @@ function UnitFrame:Init(unit, config)
 
     self:CreateTriggers(config.triggers)
 
-    self:InitDebuffs()
-
     self:RegisterEvents()
     self:Enable()
 end
@@ -493,74 +487,6 @@ function UnitFrame:GROUP_ROSTER_UPDATE()
     end
 end
 
--- FIXME: This is a total hack job, just quickly want to get debuffs rendering on my unit frames to learn new dungeons
-function UnitFrame:InitDebuffs()
-    local icons = {}
-    self.debuffIndicators = icons
-    for i = 1, 5 do
-        local icon = IconIndicator:new(self, 17, "BOTTOMRIGHT", -2 - (i - 1) * 19, 2, true, 10.0, nil, true)
-        icons[i] = icon
-    end
-end
-
-local ignoredDebuffs = {
-    [57723] = true, -- Sated, shaman lust debuff (alliance)
-    [57724] = true, -- Sated, shaman lust debuff (horde)
-    [80354] = true, -- Temporal displacement
-    [390435] = true, -- Exhausstion
-    [264689] = true, -- Fatigued
-    [87024] = true, -- Cauterized
-    [45181] = true, -- Cheated Death
-    [41425] = true, -- hypothermia
-    [124275] = true, -- Light stagger
-    [124274] = true, -- moderate stagger
-    [124273] = true, -- heavy stagger
-    [25771] = true, -- Forbearance
-    [456447] = true, -- Battle-Scarred Veteran
-    [382912] = true, -- Well-Honed Instincts
-    [1223611] = true, -- Ethereal Exhaustion
-    [456779] = true, -- Will of the Dawn
-    [448005] = true, -- Light of the Martyr
-    [157131] = true, -- Recently Saved by the Light
-    [387847] = true,
-    [206151] = true, -- Challenger's Burden
-    [113942] = true, -- Demonic Gateway
-    [1216943] = true, -- void pulsar
-    [462661] = true, -- Voidbinding
-    [1239229] = true, -- k'areshi surge
-    [449042] = true, -- Radiant Light
-    -- blood draw
-}
-
-function UnitFrame:UpdateDebuffs()
-    local auras = {}
-    local i = 1
-    while #auras < 5 do
-        local aura = C_UnitAuras.GetAuraDataByIndex(self.unit, i, "HARMFUL")
-        if aura == nil then
-            break
-        end
-        if not ignoredDebuffs[aura.spellId] then
-            table.insert(auras, aura)
-        end
-        i = i + 1
-    end
-    for i = 1, 5 do
-        local aura = auras[i]
-        local icon = self.debuffIndicators[i]
-
-        if aura then
-            if icon.frame then
-                icon:Update(aura)
-            else
-                icon:Show(aura)
-            end
-        elseif icon.frame then
-            icon:Hide()
-        end
-    end
-end
-
 function UnitFrame:UNIT_AURA(unit, info)
     if self:HasUnitChanged() then
         self:UpdateAll(true)
@@ -569,8 +495,6 @@ function UnitFrame:UNIT_AURA(unit, info)
     if self.auras:Update(info) then
         self:UpdateHealthColor()
     end
-
-    self:UpdateDebuffs()
 end
 
 function UnitFrame:UNIT_HEALTH()
@@ -613,10 +537,20 @@ function UnitFrame:UNIT_POWER_UPDATE()
     self:UpdatePower()
 end
 
-function UnitFrame:UpdateRole(role)
+function UnitFrame:GetRole(role)
     if role == nil then
         role = UnitGroupRolesAssigned(self.unit)
     end
+
+    if GetPartyAssignment("MAINTANK", self.unit) then
+        role = "TANK"
+    end
+
+    return role
+end
+
+function UnitFrame:UpdateRole(role)
+    role = self:GetRole(role)
     if role == "HEALER" then
         self:EnablePower(Enum.PowerType.Mana)
     elseif role == "TANK" and select(3, UnitClass(self.unit)) == 6 then
@@ -674,7 +608,6 @@ function UnitFrame:UpdateAll(unitChanged)
     self:UpdateRange()
     self:UpdateName()
     self:UpdateRole() -- Also calls UpdatePower if power is visible
-    self:UpdateDebuffs()
 end
 
 function UnitFrame:UpdateHealth()
@@ -725,23 +658,24 @@ function UnitFrame:UpdateHealthColor()
 
     if not isFriend then
         self.hp:SetColor(unpack(colors.hostile))
-    elseif self.auras.statusCount["Immune"] then
-        self.hp:SetColor(unpack(colors.immune))
-    elseif self.auras.statusCount["Bomb"] then
-        self.hp:SetColor(unpack(colors.bomb))
-    elseif self.auras.statusCount["Magic"] then
-        self.hp:SetColor(unpack(colors.magic))
-    elseif self.auras.statusCount["Curse"] then
-        self.hp:SetColor(unpack(colors.curse))
-    elseif self.auras.statusCount["Poison"] then
-        self.hp:SetColor(unpack(colors.poison))
-    elseif self.auras.statusCount["Disease"] then
-        self.hp:SetColor(unpack(colors.disease))
-    elseif isFriend and UnitIsDead(unit) then
+        return
+    end
+
+    local statusColor = nil
+    for _, entry in ipairs(self.statusColors) do
+        if self.auras.statusCount[entry.status] then
+            statusColor = entry.color
+            break
+        end
+    end
+
+    if statusColor then
+        self.hp:SetColor(unpack(statusColor))
+    elseif UnitIsDead(unit) then
         self.hp:SetColor(unpack(colors.dead))
-    elseif isFriend and pct >= 0.90 then
+    elseif pct >= 0.90 then
         self.hp:SetColor(unpack(colors.healthy))
-    elseif isFriend and pct >= 0.75 then
+    elseif pct >= 0.75 then
         local progress = (pct - 0.75) / (0.90 - 0.75)
         self.hp:SetInterpolatedColor(colors.mid, colors.high, progress)
     else

src/types/unitgroup.lua

@@ -53,8 +53,8 @@ local function UnitFrameDefaultCompare(lhs, rhs)
         end
 
         -- tank < healer < damage < ?
-        local lrole = RoleValue(UnitGroupRolesAssigned(lunit))
-        local rrole = RoleValue(UnitGroupRolesAssigned(runit))
+        local lrole = RoleValue(lhs:GetRole())
+        local rrole = RoleValue(rhs:GetRole())
 
         if lrole ~= rrole then
             return lrole < rrole