BlizzNameplatesPlus/modules/classcolors.lua

BNP = BNP or {}

local fallbackColors = {
  WARRIOR = { 0.78, 0.61, 0.43 },
  MAGE = { 0.41, 0.80, 0.94 },
  ROGUE = { 1.00, 0.96, 0.41 },
  DRUID = { 1.00, 0.49, 0.04 },
  HUNTER = { 0.67, 0.83, 0.45 },
  SHAMAN = { 0.00, 0.44, 0.87 },
  PRIEST = { 1.00, 1.00, 1.00 },
  WARLOCK = { 0.58, 0.51, 0.79 },
  PALADIN = { 0.96, 0.55, 0.73 },
}

local function GetColor(class)
  if RAID_CLASS_COLORS and RAID_CLASS_COLORS[class] then
    local c = RAID_CLASS_COLORS[class]
    return c.r, c.g, c.b
  end

  local c = fallbackColors[class]
  if c then return c[1], c[2], c[3] end
end

local function Enabled()
  if BNP.AreClassColorsEnabled then return BNP:AreClassColorsEnabled() end
  return not BNP_DB or BNP_DB.classColors ~= false
end

local function ClearClassCache(plate)
  if not plate then return end
  plate.BNPClassColorApplied = nil
  plate.BNPClassColorToken = nil
  plate.BNPClassGUID = nil
  plate.BNPClassR = nil
  plate.BNPClassG = nil
  plate.BNPClassB = nil
end

local function ResolvePlateUnit(plate)
  if not plate or not plate.GetName then return nil, nil end

  local token = plate:GetName(1)
  if not token or not UnitExists(token) then
    return token, nil
  end

  local exists, guid = UnitExists(token)
  if exists then
    return token, guid
  end

  return token, nil
end

local function ApplyClassColor(plate)
  if not Enabled() then
    ClearClassCache(plate)
    return false
  end

  if not plate or not plate.healthbar then return false end

  local token, guid = ResolvePlateUnit(plate)

  -- BNP only owns PLAYER class colors.
  -- NPC/neutral/friendly-NPC colors stay completely under Blizzard control.
  if not token or not guid or not UnitIsPlayer(token) then
    ClearClassCache(plate)
    return false
  end

  local _, class = UnitClass(token)
  local r, g, b = GetColor(class)
  if not r then
    ClearClassCache(plate)
    return false
  end

  plate.BNPClassColorApplied = true
  plate.BNPClassColorToken = token
  plate.BNPClassGUID = guid
  plate.BNPClassR = r
  plate.BNPClassG = g
  plate.BNPClassB = b

  plate.healthbar:SetStatusBarColor(r, g, b)
  return true
end

BNP.ApplyClassColor = ApplyClassColor

if BNP.libnameplate then
  table.insert(BNP.libnameplate.OnInit, function(plate)
    local current = plate or this
    if not current then return end
    ClearClassCache(current)
    ApplyClassColor(current)
  end)

  table.insert(BNP.libnameplate.OnShow, function(plate)
    local current = plate or this
    if not current then return end
    ClearClassCache(current)
    ApplyClassColor(current)
  end)
end

-- Blizzard may rewrite player colors during combat/threat updates.
-- Re-apply only while the plate still resolves to the SAME player GUID.
-- If a recycled frame now represents an NPC/neutral unit, drop the cache
-- immediately and never touch its Blizzard color.
local elapsed = 0
local updater = CreateFrame("Frame")
updater:SetScript("OnUpdate", function()
  elapsed = elapsed + arg1
  if elapsed < 0.05 then return end
  elapsed = 0

  if not Enabled() then return end

  local plate
  for plate in pairs(BNP.plates) do
    if plate:IsShown() and plate.healthbar then
      local token, guid = ResolvePlateUnit(plate)

      if plate.BNPClassGUID then
        if not token or not guid or guid ~= plate.BNPClassGUID or not UnitIsPlayer(token) then
          ClearClassCache(plate)
        else
          local r, g, b = plate.healthbar:GetStatusBarColor()
          if math.abs(r - plate.BNPClassR) > 0.01
            or math.abs(g - plate.BNPClassG) > 0.01
            or math.abs(b - plate.BNPClassB) > 0.01 then
            plate.healthbar:SetStatusBarColor(
              plate.BNPClassR,
              plate.BNPClassG,
              plate.BNPClassB
            )
          end
        end
      elseif token and guid and UnitIsPlayer(token) then
        ApplyClassColor(plate)
      end
    end
  end
end)