From cff164b75eb9975c482f9087191a57c27d7d719d Mon Sep 17 00:00:00 2001
From: shagu <ericmauser@googlemail.com>
Date: Sat, 20 Jan 2018 14:42:27 +0100
Subject: [PATCH] unitframes: add simple aggro detection api

---
 api/unitframes.lua | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/api/unitframes.lua b/api/unitframes.lua
index 3e615a96..cbf37ba0 100644
--- a/api/unitframes.lua
+++ b/api/unitframes.lua
@@ -24,6 +24,21 @@ for i=1,4 do pfValidUnits["partypet" .. i .. "target"] = true end
 for i=1,40 do pfValidUnits["raid" .. i .. "target"] = true end
 for i=1,40 do pfValidUnits["raidpet" .. i .. "target"] = true end
 
+local aggrodata = { }
+function pfUI.api.UnitHasAggro(unit)
+  if UnitExists(unit) and UnitIsFriend(unit, "player") and ( not aggrodata[unit] or GetTime() < aggrodata[unit].check + 1 ) then
+    aggrodata[unit] = { check = GetTime(), state = 0}
+    for u in pairs(pfValidUnits) do
+      local t = u .. "target"
+      local tt = t .. "target"
+      if UnitExists(tt) and UnitIsUnit(tt, unit) and UnitCanAttack(t, tt) then
+        aggrodata[unit].state = aggrodata[unit].state + 1
+      end
+    end
+  end
+  return aggrodata[unit] and aggrodata[unit].state or 0
+end
+
 function pfUI.uf:UpdateFrameSize()
   local default_border = pfUI_config.appearance.border.default
   if pfUI_config.appearance.border.unitframes ~= "-1" then