combatindicator: Support units that affect player combat

elements/combatindicator.lua

@@ -25,7 +25,8 @@ A default texture will be applied if the widget is a Texture and doesn't have a
 local _, ns = ...
 local oUF = ns.oUF
 
-local function Update(self, event)
+local function Update(self, event, unit)
+	if(not unit or self.unit ~= unit) then return end
 	local element = self.CombatIndicator
 
 	--[[ Callback: CombatIndicator:PreUpdate()
@@ -37,7 +38,7 @@ local function Update(self, event)
 		element:PreUpdate()
 	end
 
-	local inCombat = UnitAffectingCombat('player')
+	local inCombat = UnitAffectingCombat(unit)
 	if(inCombat) then
 		element:Show()
 	else
@@ -71,12 +72,17 @@ end
 
 local function Enable(self, unit)
 	local element = self.CombatIndicator
-	if(element and UnitIsUnit(unit, 'player')) then
+	if(element) then
 		element.__owner = self
 		element.ForceUpdate = ForceUpdate
 
-		self:RegisterEvent('PLAYER_REGEN_DISABLED', Path, true)
-		self:RegisterEvent('PLAYER_REGEN_ENABLED', Path, true)
+		if(unit == 'player') then
+			self:RegisterEvent('PLAYER_REGEN_DISABLED', Path, true)
+			self:RegisterEvent('PLAYER_REGEN_ENABLED', Path, true)
+		else
+			self:RegisterEvent('UNIT_COMBAT', Path)
+			self:RegisterEvent('UNIT_FLAGS', Path)
+		end
 
 		if(element:IsObjectType('Texture') and not element:GetTexture()) then
 			element:SetTexture([[Interface\CharacterFrame\UI-StateIcon]])
@@ -92,8 +98,13 @@ local function Disable(self)
 	if(element) then
 		element:Hide()
 
-		self:UnregisterEvent('PLAYER_REGEN_DISABLED', Path)
-		self:UnregisterEvent('PLAYER_REGEN_ENABLED', Path)
+		if(unit == 'player') then
+			self:UnregisterEvent('PLAYER_REGEN_DISABLED', Path, true)
+			self:UnregisterEvent('PLAYER_REGEN_ENABLED', Path, true)
+		else
+			self:UnregisterEvent('UNIT_COMBAT', Path)
+			self:UnregisterEvent('UNIT_FLAGS', Path)
+		end
 	end
 end