Scripts/Stratholme: Portcullis Trap (#29596)

Co-authored-by: Shauren <shauren.trinity@gmail.com>

2 files changed, 119 insertions, 8 deletions

diff --git a/src/server/scripts/EasternKingdoms/Stratholme/instance_stratholme.cpp b/src/server/scripts/EasternKingdoms/Stratholme/instance_stratholme.cpp
index c1d21da6571..54cc338f123 100644
--- a/src/server/scripts/EasternKingdoms/Stratholme/instance_stratholme.cpp
+++ b/src/server/scripts/EasternKingdoms/Stratholme/instance_stratholme.cpp
@@ -34,11 +34,13 @@ EndScriptData */
 #include "MotionMaster.h"
 #include "Player.h"
 #include "stratholme.h"
+#include "Pet.h"
 
 enum InstanceEvents
 {
     EVENT_BARON_RUN         = 1,
-    EVENT_SLAUGHTER_SQUARE  = 2
+    EVENT_SLAUGHTER_SQUARE  = 2,
+    EVENT_RAT_TRAP_CLOSE    = 3,
 };
 
 enum StratholmeMisc
@@ -49,6 +51,25 @@ enum StratholmeMisc
 Position const timmyTheCruelSpawnPosition = { 3625.358f, -3188.108f, 130.3985f, 4.834562f };
 EllipseBoundary const beforeScarletGate(Position(3671.158f, -3181.79f), 60.0f, 40.0f);
 
+enum class StratholmeGateTrapType : uint8
+{
+    ScaletSide = 0,
+    UndeadSide = 1
+};
+
+Position const GateTrapPos[] =              // Positions of the two Gate Traps 3919.88 -3547.34 134.269
+{
+    { 3612.29f, -3335.39f, 124.077f },      // Scarlet side
+    { 3919.88f, -3545.34f, 134.269f }       // Undead side
+};
+
+struct GateTrapData
+{
+    std::array<ObjectGuid, 2> Gates;
+    GuidUnorderedSet Rats;
+    bool Triggered = false;
+};
+
 class instance_stratholme : public InstanceMapScript
 {
     public:
@@ -68,6 +89,8 @@ class instance_stratholme : public InstanceMapScript
 
                 timmySpawned = false;
                 scarletsKilled = 0;
+
+                events.ScheduleEvent(EVENT_RAT_TRAP_CLOSE, 15s);
             }
 
             uint32 EncounterState[MAX_ENCOUNTER];
@@ -95,6 +118,8 @@ class instance_stratholme : public InstanceMapScript
             GuidSet abomnationGUID;
             EventMap events;
 
+            std::array<GateTrapData, 2> TrapGates;
+
             void OnUnitDeath(Unit* who) override
             {
                 switch (who->GetEntry())
@@ -119,6 +144,20 @@ class instance_stratholme : public InstanceMapScript
                         }
                         break;
                     }
+                    case NPC_PLAGUED_RAT:
+                    {
+                        for (GateTrapData& trapGate : TrapGates)
+                        {
+                            auto el = trapGate.Rats.find(who->GetGUID());
+                            if (el != trapGate.Rats.end())
+                            {
+                                trapGate.Rats.erase(el);
+                                for (ObjectGuid gate : trapGate.Gates)
+                                    UpdateGoState(gate, GO_STATE_ACTIVE);
+                            }
+                        }
+                        break;
+                    }
                 }
             }
 
@@ -136,21 +175,39 @@ class instance_stratholme : public InstanceMapScript
                 return false;
             }
 
-            //if withRestoreTime true, then newState will be ignored and GO should be restored to original state after 10 seconds
-            void UpdateGoState(ObjectGuid goGuid, uint32 newState, bool withRestoreTime)
+            //if restoreTime is not 0, then newState will be ignored and GO should be restored to original state after "restoreTime" millisecond
+            void UpdateGoState(ObjectGuid goGuid, uint32 newState, uint32 restoreTime = 0u)
             {
                 if (!goGuid)
                     return;
-
                 if (GameObject* go = instance->GetGameObject(goGuid))
                 {
-                    if (withRestoreTime)
-                        go->UseDoorOrButton(10);
+                    if (restoreTime)
+                        go->UseDoorOrButton(restoreTime);
                     else
                         go->SetGoState((GOState)newState);
                 }
             }
 
+            void DoGateTrap(StratholmeGateTrapType type, Unit* where)
+            {
+                // close the gate, but in two minutes it will open on its own
+                for (ObjectGuid trapGateGuid : TrapGates[AsUnderlyingType(type)].Gates)
+                    UpdateGoState(trapGateGuid, GO_STATE_READY, 20 * IN_MILLISECONDS);
+
+                for (uint8 i = 0; i < 30; ++i)
+                {
+                    Position summonPos = where->GetRandomPoint(GateTrapPos[AsUnderlyingType(type)], 5.0f);
+                    if (Creature* creature = where->SummonCreature(NPC_PLAGUED_RAT, summonPos, TEMPSUMMON_DEAD_DESPAWN, 0s))
+                    {
+                        TrapGates[AsUnderlyingType(type)].Rats.insert(creature->GetGUID());
+                        creature->EngageWithTarget(where);
+                    }
+                }
+
+                TrapGates[AsUnderlyingType(type)].Triggered = true;
+            }
+
             void OnCreatureCreate(Creature* creature) override
             {
                 switch (creature->GetEntry())
@@ -242,6 +299,18 @@ class instance_stratholme : public InstanceMapScript
                     case GO_YSIDA_CAGE:
                         ysidaCageGUID = go->GetGUID();
                         break;
+                    case GO_PORT_TRAP_GATE_1:
+                        TrapGates[AsUnderlyingType(StratholmeGateTrapType::ScaletSide)].Gates[0] = go->GetGUID();
+                        break;
+                    case GO_PORT_TRAP_GATE_2:
+                        TrapGates[AsUnderlyingType(StratholmeGateTrapType::ScaletSide)].Gates[1] = go->GetGUID();
+                        break;
+                    case GO_PORT_TRAP_GATE_3:
+                        TrapGates[AsUnderlyingType(StratholmeGateTrapType::UndeadSide)].Gates[0] = go->GetGUID();
+                        break;
+                    case GO_PORT_TRAP_GATE_4:
+                        TrapGates[AsUnderlyingType(StratholmeGateTrapType::UndeadSide)].Gates[1] = go->GetGUID();
+                        break;
                 }
             }
 
@@ -504,6 +573,41 @@ class instance_stratholme : public InstanceMapScript
                                 TC_LOG_DEBUG("scripts", "Instance Stratholme: Black guard sentries spawned. Opening gates to baron.");
                             }
                             break;
+                        case EVENT_RAT_TRAP_CLOSE:
+                        {
+                            for (uint8 i = 0; i < std::size(GateTrapPos); ++i)
+                            {
+                                if (TrapGates[i].Triggered)
+                                    continue;
+
+                                Position const* gateTrapPos = &GateTrapPos[i];
+                                // Check that the trap is not on cooldown, if so check if player/pet is in range
+                                for (MapReference const& itr : instance->GetPlayers())
+                                {
+                                    Player* player = itr.GetSource();
+                                    if (player->IsGameMaster())
+                                        continue;
+
+                                    if (player->IsWithinDist2d(gateTrapPos, 5.5f))
+                                    {
+                                        DoGateTrap(StratholmeGateTrapType(i), player);
+                                        break;
+                                    }
+
+                                    Pet* pet = player->GetPet();
+                                    if (pet && pet->IsWithinDist2d(gateTrapPos, 5.5f))
+                                    {
+                                        DoGateTrap(StratholmeGateTrapType(i), pet);
+                                        break;
+                                    }
+                                }
+
+                            }
+                            //if you haven't already fallen into the trap, update it
+                            if (std::any_of(TrapGates.begin(), TrapGates.end(), [](GateTrapData const& trap) { return !trap.Triggered; }))
+                                events.ScheduleEvent(EVENT_RAT_TRAP_CLOSE, 1s);
+                            break;
+                        }
                         default:
                             break;
                     }
diff --git a/src/server/scripts/EasternKingdoms/Stratholme/stratholme.h b/src/server/scripts/EasternKingdoms/Stratholme/stratholme.h
index 93a07f13830..fa0554c5c67 100644
--- a/src/server/scripts/EasternKingdoms/Stratholme/stratholme.h
+++ b/src/server/scripts/EasternKingdoms/Stratholme/stratholme.h
@@ -63,7 +63,10 @@ enum STRCreatureIds
     NPC_CRIMSON_INITATE                 = 10420,
     NPC_CRIMSON_GALLANT                 = 10424,
 
-    NPC_TIMMY_THE_CRUEL                 = 10808
+    NPC_TIMMY_THE_CRUEL                 = 10808,
+
+    // Rat trap
+    NPC_PLAGUED_RAT                     = 10441,
 };
 
 enum STRGameobjectIds
@@ -79,7 +82,11 @@ enum STRGameobjectIds
     GO_PORT_GAUNTLET                    = 175374,  // port from gauntlet to slaugther
     GO_PORT_SLAUGTHER                   = 175373,  // port at slaugther
     GO_PORT_ELDERS                      = 175377,  // port at elders square
-    GO_YSIDA_CAGE                       = 181071
+    GO_YSIDA_CAGE                       = 181071,
+    GO_PORT_TRAP_GATE_1                 = 175351,  // Portcullis used in the gate traps (rats trap)
+    GO_PORT_TRAP_GATE_2                 = 175350,  // Scarlet side
+    GO_PORT_TRAP_GATE_3                 = 175355,  // Undead side
+    GO_PORT_TRAP_GATE_4                 = 175354,
 };
 
 enum STRQuestIds