[Feature] Add "classification:icon" tag (#191)

Krealle · Dec 17, 2024 · 0349dbb · 0349dbb
1 parent 3831d72
commit 0349dbb
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/Util/Utils.lua b/Util/Utils.lua
@@ -667,6 +667,9 @@ function Util:GetUnitSubgroup(unit)
     end
 end
 
+---@param unit UnitToken
+---@param localized boolean
+---@return string? classification
 function Util:GetUnitClassification(unit, localized)
     local classification = UnitClassification(unit)
     if localized then
@@ -675,6 +678,15 @@ function Util:GetUnitClassification(unit, localized)
     return classification
 end
 
+local gold, silver = "|A:nameplates-icon-elite-gold:16:16|a", "|A:nameplates-icon-elite-silver:16:16|a"
+local typeIcon = { elite = gold, worldboss = gold, rareelite = silver, rare = silver }
+
+---@param unit UnitToken
+---@return string? icon
+function Util:GetUnitClassificationIcon(unit)
+    return typeIcon[UnitClassification(unit)]
+end
+
 -------------------------------------------------
 -- MARK: Frames
 -------------------------------------------------

diff --git a/Widgets/Texts/CustomFormats.lua b/Widgets/Texts/CustomFormats.lua
@@ -716,6 +716,10 @@ end, "Group", "1-8")
 W:AddTag("classification", "UNIT_CLASSIFICATION_CHANGED", function(unit)
     return Util:GetUnitClassification(unit, true)
 end, "Classification", (L.rare .. ", " .. L.rareelite .. ", " .. L.elite .. ", " .. L.worldboss))
+W:AddTag("classification:icon", "UNIT_CLASSIFICATION_CHANGED", function(unit)
+    if UnitIsPlayer(unit) then return end
+    return Util:GetUnitClassificationIcon(unit)
+end, "Classification")
 
 -- MARK: Name
 for type, length in pairs(nameLengths) do