1 files changed, 43 insertions, 71 deletions

diff --git a/src/server/scripts/Northrend/zone_storm_peaks.cpp b/src/server/scripts/Northrend/zone_storm_peaks.cpp
index dae2ffd36f6..d9528446bcf 100644
--- a/src/server/scripts/Northrend/zone_storm_peaks.cpp
+++ b/src/server/scripts/Northrend/zone_storm_peaks.cpp
@@ -356,34 +356,20 @@ public:
 
 enum FreedProtoDrake
 {
+    NPC_DRAKE                           = 29709,
+
     AREA_VALLEY_OF_ANCIENT_WINTERS      = 4437,
+
     TEXT_EMOTE                          = 0,
+
     SPELL_KILL_CREDIT_PRISONER          = 55144,
     SPELL_SUMMON_LIBERATED              = 55073,
-    SPELL_KILL_CREDIT_DRAKE             = 55143
-};
+    SPELL_KILL_CREDIT_DRAKE             = 55143,
 
-const Position FreedDrakeWaypoints[16] =
-{
-    {7294.96f, -2418.733f, 823.869f, 0.0f},
-    {7315.984f, -2331.46f, 826.3972f, 0.0f},
-    {7271.826f, -2271.479f, 833.5917f, 0.0f},
-    {7186.253f, -2218.475f, 847.5632f, 0.0f},
-    {7113.195f, -2164.288f, 850.2301f, 0.0f},
-    {7078.018f, -2063.106f, 854.7581f, 0.0f},
-    {7073.221f, -1983.382f, 861.9246f, 0.0f},
-    {7061.455f, -1885.899f, 865.119f, 0.0f},
-    {7033.32f, -1826.775f, 876.2578f, 0.0f},
-    {6999.902f, -1784.012f, 897.4521f, 0.0f},
-    {6954.913f, -1747.043f, 897.4521f, 0.0f},
-    {6933.856f, -1720.698f, 882.2022f, 0.0f},
-    {6932.729f, -1687.306f, 866.1189f, 0.0f},
-    {6952.458f, -1663.802f, 849.8133f, 0.0f},
-    {7002.819f, -1651.681f, 831.397f, 0.0f},
-    {7026.531f, -1649.239f, 828.8406f, 0.0f}
+    EVENT_CHECK_AREA                    = 1,
+    EVENT_REACHED_HOME                  = 2,
 };
 
-
 class npc_freed_protodrake : public CreatureScript
 {
 public:
@@ -393,74 +379,60 @@ public:
     {
         npc_freed_protodrakeAI(Creature* creature) : VehicleAI(creature) {}
 
-        bool autoMove;
-        bool wpReached;
-        uint16 CheckTimer;
-        uint16 countWP;
+        EventMap events;
 
         void Reset()
         {
-            autoMove = false;
-            wpReached = false;
-            CheckTimer = 5000;
-            countWP = 0;
+            events.ScheduleEvent(EVENT_CHECK_AREA, 5000);
         }
 
         void MovementInform(uint32 type, uint32 id)
         {
-            if (type != POINT_MOTION_TYPE)
+            if (type != WAYPOINT_MOTION_TYPE)
                 return;
 
-            if (id < 15)
-            {
-                ++countWP;
-                wpReached = true;
-            }
-            else
+            if (id == 15)
             // drake reached village
-            {
-                // get player that rides drake (from seat 0)
-                Unit* player = me->GetVehicleKit()->GetPassenger(0);
-                if (player && player->GetTypeId() == TYPEID_PLAYER)
-                {
-                    // for each prisoner on drake, give credit
-                    for (uint8 i = 1; i < 4; ++i)
-                        if (Unit* prisoner = me->GetVehicleKit()->GetPassenger(i))
-                        {
-                            if (prisoner->GetTypeId() != TYPEID_UNIT)
-                                return;
-                            prisoner->CastSpell(player, SPELL_KILL_CREDIT_PRISONER, true);
-                            prisoner->CastSpell(prisoner, SPELL_SUMMON_LIBERATED, true);
-                            prisoner->ExitVehicle();
-                        }
-                    me->CastSpell(me, SPELL_KILL_CREDIT_DRAKE, true);
-                    player->ExitVehicle();
-                }
-            }
+            events.ScheduleEvent(EVENT_REACHED_HOME, 2000);
         }
 
         void UpdateAI(const uint32 diff)
         {
-            if (!autoMove)
+            events.Update(diff);
+
+            switch (events.ExecuteEvent())
             {
-                if (CheckTimer < diff)
-                {
-                    CheckTimer = 5000;
+                case EVENT_CHECK_AREA:
                     if (me->GetAreaId() == AREA_VALLEY_OF_ANCIENT_WINTERS)
                     {
-                        Talk(TEXT_EMOTE, me->GetVehicleKit()->GetPassenger(0)->GetGUID());
-                        autoMove = true;
-                        wpReached = true;
+                        if (Vehicle* vehicle = me->GetVehicleKit())
+                            if (Unit* passenger = vehicle->GetPassenger(0))
+                            {
+                                Talk(TEXT_EMOTE, passenger->GetGUID());
+                                me->GetMotionMaster()->MovePath(NPC_DRAKE, false);
+                            }
                     }
-                }
-                else
-                    CheckTimer -= diff;
-            }
-
-            if (wpReached && autoMove)
-            {
-                wpReached = false;
-                me->GetMotionMaster()->MovePoint(countWP, FreedDrakeWaypoints[countWP]);
+                    else
+                        events.ScheduleEvent(EVENT_CHECK_AREA, 5000);
+                    break;
+                case EVENT_REACHED_HOME:
+                    Unit* player = me->GetVehicleKit()->GetPassenger(0);
+                    if (player && player->GetTypeId() == TYPEID_PLAYER)
+                    {
+                        // for each prisoner on drake, give credit
+                        for (uint8 i = 1; i < 4; ++i)
+                            if (Unit* prisoner = me->GetVehicleKit()->GetPassenger(i))
+                            {
+                                if (prisoner->GetTypeId() != TYPEID_UNIT)
+                                    return;
+                                prisoner->CastSpell(player, SPELL_KILL_CREDIT_PRISONER, true);
+                                prisoner->CastSpell(prisoner, SPELL_SUMMON_LIBERATED, true);
+                                prisoner->ExitVehicle();
+                            }
+                        me->CastSpell(me, SPELL_KILL_CREDIT_DRAKE, true);
+                        player->ExitVehicle();
+                    }
+                    break;
             }
         }
     };