Core/BossPrototype: Allow registering SPELL_DISPEL and `SPELL_INTER…

…RUPT` with a spell ID
BigWigsMods · Aug 7, 2024 · caa2ffe · caa2ffe
1 parent 49ba81a
commit caa2ffe
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/Core/BossPrototype.lua b/Core/BossPrototype.lua
@@ -711,6 +711,24 @@ do
 						end
 					end
 				end
+			elseif event == "SPELL_DISPEL" or event == "SPELL_INTERRUPT" then
+				for i = #enabledModules, 1, -1 do
+					local self = enabledModules[i]
+					local m = eventMap[self][event]
+					if m and (m[extraSpellId] or m["*"]) then
+						local func = m[extraSpellId] or m["*"]
+						-- DEVS! Please ask if you need args attached to the table that we've missed out!
+						args.sourceGUID, args.sourceName, args.sourceFlags, args.sourceRaidFlags = sourceGUID, sourceName, sourceFlags, sourceRaidFlags
+						args.destGUID, args.destName, args.destFlags, args.destRaidFlags = destGUID, destName, destFlags, destRaidFlags
+						args.spellId, args.spellName, args.spellSchool = spellId, spellName, spellSchool
+						args.time, args.extraSpellId, args.extraSpellName, args.amount = time, extraSpellId, amount, amount
+						if type(func) == "function" then
+							func(args)
+						else
+							self[func](self, args)
+						end
+					end
+				end
 			else
 				for i = #enabledModules, 1, -1 do
 					local self = enabledModules[i]

diff --git a/Core/BossPrototype_Classic.lua b/Core/BossPrototype_Classic.lua
@@ -784,6 +784,24 @@ do
 						end
 					end
 				end
+			elseif event == "SPELL_DISPEL" or event == "SPELL_INTERRUPT" then
+				for i = #enabledModules, 1, -1 do
+					local self = enabledModules[i]
+					local m = eventMap[self][event]
+					if m and (m[extraSpellId] or m["*"]) then
+						local func = m[extraSpellId] or m["*"]
+						-- DEVS! Please ask if you need args attached to the table that we've missed out!
+						args.sourceGUID, args.sourceName, args.sourceFlags, args.sourceRaidFlags = sourceGUID, sourceName, sourceFlags, sourceRaidFlags
+						args.destGUID, args.destName, args.destFlags, args.destRaidFlags = destGUID, destName, destFlags, destRaidFlags
+						args.spellId, args.spellName, args.spellSchool = spellId, spellName, spellSchool
+						args.time, args.extraSpellId, args.extraSpellName, args.amount = time, extraSpellId, amount, amount
+						if type(func) == "function" then
+							func(args)
+						else
+							self[func](self, args)
+						end
+					end
+				end
 			else
 				for i = #enabledModules, 1, -1 do
 					local self = enabledModules[i]