e8fd612 - Add more auras and dot tracking functions

[wowui.git] / kehys / incoming.lua

diff --git a/kehys/incoming.lua b/kehys/incoming.lua
index 0e3f150..2e880d9 100644 (file)
--- a/kehys/incoming.lua
+++ b/kehys/incoming.lua
@@ -7,6 +7,8 @@ local _, addon = ...;
 local guids = addon.FrameGuids;
 local abilities = {
     [260741] = true, -- Jagged Nettles (Waycrest Manor)
+    [342675] = true, -- Bone Spear (Theater of Pain)
+    [320788] = true, -- Frozen Binds (Necrotic Wake)
 };
 
 local frame = CreateFrame("Frame", "kehysIncoming");
@@ -27,12 +29,10 @@ local function event(_, event, target, cast, id)
                 else
                     f.incoming[id] = 1;
                 end
-            elseif event == "UNIT_SPELLCAST_STOP" or event == "UNIT_SPELLCAST_CHANNEL_START" then
-                if f.incoming[id] then
-                    f.incoming[id] = f.incoming[id] - 1;
-                    if f.incoming[id] == 0 then
-                        f.incoming[id] = nil;
-                    end
+            elseif f.incoming[id] then
+                f.incoming[id] = f.incoming[id] - 1;
+                if f.incoming[id] == 0 then
+                    f.incoming[id] = nil;
                 end
             end
         end
@@ -44,7 +44,11 @@ local function setup()
     frame:SetScript("OnEvent", event);
     frame:RegisterEvent("UNIT_SPELLCAST_START");
     frame:RegisterEvent("UNIT_SPELLCAST_CHANNEL_START");
+    frame:RegisterEvent("UNIT_SPELLCAST_CHANNEL_STOP");
     frame:RegisterEvent("UNIT_SPELLCAST_STOP");
+    frame:RegisterEvent("UNIT_SPELLCAST_INTERRUPTED");
+    frame:RegisterEvent("UNIT_SPELLCAST_FAILED");
+    frame:RegisterEvent("UNIT_SPELLCAST_FAILED_QUIET");
     frame:RegisterEvent("PLAYER_REGEN_ENABLED");
 end