OmaRF/UpdateDispellableDebuffsHook.lua

   1 local normalBarColor = OmaRF.normalBarColor;
   2 local dispelBarColor = OmaRF.dispelBarColor;
   3 local normalBackColor = OmaRF.normalBackColor;
   4 local dispelBackColor = OmaRF.dispelBackColor;
   5 local UnitDebuff = UnitDebuff;
   6 local CompactUnitFrame_UpdateHealthColor = CompactUnitFrame_UpdateHealthColor;
   7 local unpack = unpack;
   8
   9 hooksecurefunc("CompactUnitFrame_UpdateDispellableDebuffs", function(frame)
  10     -- allowClassColorsForNPCs only in regular raid frames,
  11     -- match only to them
  12     if frame.optionTable.allowClassColorsForNPCs ~= nil then
  13         -- try to find dispellable debuff
  14         if UnitDebuff(frame.displayedUnit, 1, "RAID") ~= nil then
  15             frame.optionTable.healthBarColorOverride = dispelBarColor;
  16             frame.background:SetColorTexture(unpack(dispelBackColor));
  17         else
  18             frame.optionTable.healthBarColorOverride = normalBarColor;
  19             frame.background:SetColorTexture(unpack(normalBackColor));
  20         end
  21         -- update color
  22         CompactUnitFrame_UpdateHealthColor(frame);
  23     end
  24 end);