Scripts/Gundrak: Drakkari Colossus cleanup (#31275)

Drakkari Colossus implementation cleanup

2 files changed, 45 insertions, 0 deletions

diff --git a/src/server/game/Miscellaneous/CommonHelpers.cpp b/src/server/game/Miscellaneous/CommonHelpers.cpp
index e1a30cb6869..77b2f32ef41 100644
--- a/src/server/game/Miscellaneous/CommonHelpers.cpp
+++ b/src/server/game/Miscellaneous/CommonHelpers.cpp
@@ -17,10 +17,14 @@
 
 #include "CommonHelpers.h"
 #include "Common.h"
+#include "Creature.h"
+#include "CreatureAI.h"
 #include "Item.h"
 #include "ItemTemplate.h"
+#include "ObjectAccessor.h"
 #include "Player.h"
 #include "SharedDefines.h"
+#include "UnitAI.h"
 
 enum PlayerSpecializationIconicSpellIds
 {
@@ -322,3 +326,26 @@ bool Trinity::Helpers::Entity::IsPlayerRangedAttacker(Player const* who)
             return (Trinity::Helpers::Entity::GetPlayerSpecialization(who) == SPEC_DRUID_BALANCE);
     }
 }
+
+Trinity::Helpers::Events::SetAggresiveStateEvent::SetAggresiveStateEvent(Creature* owner, bool startCombat/* = true*/, ObjectGuid summonerGUID/* = ObjectGuid::Empty*/) : _owner(owner), _startCombat(startCombat), _summonerGUID(summonerGUID)
+{
+}
+
+bool Trinity::Helpers::Events::SetAggresiveStateEvent::Execute(uint64 /*time*/, uint32 /*diff*/)
+{
+    _owner->SetReactState(REACT_AGGRESSIVE);
+    if (_startCombat)
+    {
+        if (Unit* currentVictim = _owner->SelectVictim())
+        {
+            if (_owner->IsAIEnabled())
+                _owner->AI()->AttackStart(currentVictim);
+        }
+        else if (!_summonerGUID.IsEmpty())
+            if (Creature* summoner = ObjectAccessor::GetCreature(*_owner, _summonerGUID))
+                if (summoner->IsEngaged() && summoner->IsAIEnabled() && _owner->IsAIEnabled())
+                    if (Unit* target = summoner->AI()->SelectTarget(SelectTargetMethod::Random, 0, 0.0f, true))
+                        _owner->AI()->AttackStart(target);
+    }
+    return true;
+}
diff --git a/src/server/game/Miscellaneous/CommonHelpers.h b/src/server/game/Miscellaneous/CommonHelpers.h
index a38acb8b968..228484fbc3d 100644
--- a/src/server/game/Miscellaneous/CommonHelpers.h
+++ b/src/server/game/Miscellaneous/CommonHelpers.h
@@ -18,8 +18,11 @@
 #ifndef TRINITY_COMMONHELPERS_H
 #define TRINITY_COMMONHELPERS_H
 
+#include "EventProcessor.h"
 #include "Define.h"
+#include "ObjectGuid.h"
 
+class Creature;
 class Player;
 
 namespace Trinity
@@ -33,6 +36,21 @@ namespace Trinity
             TC_GAME_API bool IsPlayerHealer(Player const* who);
             bool IsPlayerRangedAttacker(Player const* who);
         }
+        namespace Events
+        {
+            class SetAggresiveStateEvent : public BasicEvent
+            {
+                public:
+                    SetAggresiveStateEvent(Creature* owner, bool startCombat = true, ObjectGuid summonerGUID = ObjectGuid::Empty);
+
+                    bool Execute(uint64 /*time*/, uint32 /*diff*/) override;
+
+                private:
+                    Creature* _owner;
+                    bool const _startCombat;
+                    ObjectGuid const _summonerGUID;
+            };
+        }
     }
 }