Core/Spells: Delay combat flagging for spell targets until spell missile lands

1 files changed, 7 insertions, 2 deletions

diff --git a/src/server/game/Combat/CombatManager.cpp b/src/server/game/Combat/CombatManager.cpp
index b5b8b24e068..2dabbc0ccb5 100644
--- a/src/server/game/Combat/CombatManager.cpp
+++ b/src/server/game/Combat/CombatManager.cpp
@@ -172,7 +172,7 @@ Unit* CombatManager::GetAnyTarget() const
     return nullptr;
 }
 
-bool CombatManager::SetInCombatWith(Unit* who)
+bool CombatManager::SetInCombatWith(Unit* who, bool suppressPvpSecond)
 {
     // Are we already in combat? If yes, refresh pvp combat
     auto it = _pvpRefs.find(who->GetGUID());
@@ -191,7 +191,12 @@ bool CombatManager::SetInCombatWith(Unit* who)
     // ...then create new reference
     CombatReference* ref;
     if (_owner->IsControlledByPlayer() && who->IsControlledByPlayer())
-        ref = new PvPCombatReference(_owner, who);
+    {
+        PvPCombatReference* refPvp = new PvPCombatReference(_owner, who);
+        if (suppressPvpSecond)
+            refPvp->SuppressFor(who);
+        ref = refPvp;
+    }
     else
         ref = new CombatReference(_owner, who);