diff options
| -rw-r--r-- | src/server/game/Entities/Unit/Unit.cpp | 5 | ||||
| -rw-r--r-- | src/server/game/Entities/Unit/Unit.h | 1 |
2 files changed, 6 insertions, 0 deletions
diff --git a/src/server/game/Entities/Unit/Unit.cpp b/src/server/game/Entities/Unit/Unit.cpp index e4e24b34ce4..32e3362285f 100644 --- a/src/server/game/Entities/Unit/Unit.cpp +++ b/src/server/game/Entities/Unit/Unit.cpp @@ -544,6 +544,11 @@ void Unit::AtStartOfEncounter() Unit::ProcSkillsAndAuras(this, nullptr, PROC_FLAG_ENCOUNTER_START, PROC_FLAG_NONE, PROC_SPELL_TYPE_MASK_ALL, PROC_SPELL_PHASE_NONE, PROC_HIT_NONE, nullptr, nullptr, nullptr); } +void Unit::AtEndOfEncounter() +{ + RemoveAurasWithInterruptFlags(SpellAuraInterruptFlags2::EndOfEncounter); +} + void Unit::UpdateSplineMovement(uint32 t_diff) { if (movespline->Finalized()) diff --git a/src/server/game/Entities/Unit/Unit.h b/src/server/game/Entities/Unit/Unit.h index e48e920f056..9875c5eab70 100644 --- a/src/server/game/Entities/Unit/Unit.h +++ b/src/server/game/Entities/Unit/Unit.h @@ -2019,6 +2019,7 @@ class TC_GAME_API Unit : public WorldObject public: void AtStartOfEncounter(); + void AtEndOfEncounter(); private: |