From ab12e77cf72bcd361a380c8d1ed37d12fe0e1bbb Mon Sep 17 00:00:00 2001
From: Shauren <shauren.trinity@gmail.com>
Date: Tue, 19 Jan 2021 22:48:46 +0100
Subject: Core/Commands: Restore instance management commands

---
 src/server/game/Instances/InstanceLockMgr.cpp | 56 ++++++++++++++++++++++++++-
 src/server/game/Instances/InstanceLockMgr.h   | 27 +++++++++++++
 2 files changed, 82 insertions(+), 1 deletion(-)

(limited to 'src/server/game/Instances')

diff --git a/src/server/game/Instances/InstanceLockMgr.cpp b/src/server/game/Instances/InstanceLockMgr.cpp
index 7443a7f192d..97687ef8a8e 100644
--- a/src/server/game/Instances/InstanceLockMgr.cpp
+++ b/src/server/game/Instances/InstanceLockMgr.cpp
@@ -29,7 +29,7 @@ InstanceLockData::InstanceLockData() = default;
 InstanceLockData::~InstanceLockData() = default;
 
 InstanceLock::InstanceLock(uint32 mapId, Difficulty difficultyId, InstanceResetTimePoint expiryTime, uint32 instanceId)
-    : _mapId(mapId), _difficultyId(difficultyId), _instanceId(instanceId), _expiryTime(expiryTime), _extended(false)
+    : _mapId(mapId), _difficultyId(difficultyId), _instanceId(instanceId), _expiryTime(expiryTime), _extended(false), _isInUse(false)
 {
 }
 
@@ -460,6 +460,60 @@ std::pair<InstanceResetTimePoint, InstanceResetTimePoint> InstanceLockMgr::Updat
     return { InstanceResetTimePoint::min(), InstanceResetTimePoint::min() };
 }
 
+void InstanceLockMgr::ResetInstanceLocksForPlayer(ObjectGuid const& playerGuid, Optional<uint32> mapId, Optional<Difficulty> difficulty,
+    std::vector<InstanceLock const*>* locksReset, std::vector<InstanceLock const*>* locksFailedToReset)
+{
+    auto playerLocksItr = _instanceLocksByPlayer.find(playerGuid);
+    if (playerLocksItr == _instanceLocksByPlayer.end())
+        return;
+
+    for (PlayerLockMap::value_type const& playerLockPair : playerLocksItr->second)
+    {
+        if (playerLockPair.second->IsInUse())
+        {
+            locksFailedToReset->push_back(playerLockPair.second.get());
+            continue;
+        }
+
+        if (mapId && *mapId != playerLockPair.second->GetMapId())
+            continue;
+
+        if (difficulty && *difficulty != playerLockPair.second->GetDifficultyId())
+            continue;
+
+        locksReset->push_back(playerLockPair.second.get());
+    }
+
+    if (!locksReset->empty())
+    {
+        CharacterDatabaseTransaction trans = CharacterDatabase.BeginTransaction();
+        for (InstanceLock const* instanceLock : *locksReset)
+        {
+            MapDb2Entries entries(instanceLock->GetMapId(), instanceLock->GetDifficultyId());
+            InstanceResetTimePoint newExpiryTime = GetNextResetTime(entries) - Seconds(entries.MapDifficulty->GetRaidDuration());
+            // set reset time to last reset time
+            const_cast<InstanceLock*>(instanceLock)->SetExpiryTime(newExpiryTime);
+            const_cast<InstanceLock*>(instanceLock)->SetExtended(false);
+
+            CharacterDatabasePreparedStatement* stmt = CharacterDatabase.GetPreparedStatement(CHAR_UPD_CHARACTER_INSTANCE_LOCK_FORCE_EXPIRE);
+            stmt->setUInt64(0, uint64(std::chrono::system_clock::to_time_t(newExpiryTime)));
+            stmt->setUInt64(1, playerGuid.GetCounter());
+            stmt->setUInt32(2, entries.MapDifficulty->MapID);
+            stmt->setUInt32(3, entries.MapDifficulty->LockID);
+            trans->Append(stmt);
+        }
+        CharacterDatabase.CommitTransaction(trans);
+    }
+}
+
+InstanceLocksStatistics InstanceLockMgr::GetStatistics() const
+{
+    InstanceLocksStatistics statistics;
+    statistics.InstanceCount = _instanceLockDataById.size();
+    statistics.PlayerCount = _instanceLocksByPlayer.size();
+    return statistics;
+}
+
 InstanceResetTimePoint InstanceLockMgr::GetNextResetTime(MapDb2Entries const& entries)
 {
     tm dateTime = *GameTime::GetDateAndTime();
diff --git a/src/server/game/Instances/InstanceLockMgr.h b/src/server/game/Instances/InstanceLockMgr.h
index 3f8aebf6d25..b860335b48f 100644
--- a/src/server/game/Instances/InstanceLockMgr.h
+++ b/src/server/game/Instances/InstanceLockMgr.h
@@ -112,6 +112,9 @@ public:
 
     InstanceResetTimePoint GetEffectiveExpiryTime() const;
 
+    bool IsInUse() const { return _isInUse; }
+    void SetInUse(bool inUse) { _isInUse = inUse; }
+
 private:
     uint32 _mapId;
     Difficulty _difficultyId;
@@ -119,6 +122,7 @@ private:
     std::chrono::system_clock::time_point _expiryTime;
     bool _extended;
     InstanceLockData _data;
+    bool _isInUse;
 };
 
 struct SharedInstanceLockData : InstanceLockData
@@ -193,6 +197,12 @@ struct InstanceLockUpdateEvent
     Optional<uint32> EntranceWorldSafeLocId;
 };
 
+struct InstanceLocksStatistics
+{
+    uint32 InstanceCount = 0;   // Number of existing ID-based locks
+    uint32 PlayerCount = 0;     // Number of players that have any lock
+};
+
 class TC_GAME_API InstanceLockMgr
 {
 public:
@@ -264,6 +274,23 @@ public:
     */
     std::pair<InstanceResetTimePoint, InstanceResetTimePoint> UpdateInstanceLockExtensionForPlayer(ObjectGuid const& playerGuid, MapDb2Entries const& entries, bool extended);
 
+    /**
+       @brief Resets instances that match given filter - for use in GM commands
+       @param playerGuid Guid of player whose locks will be removed
+       @param mapId (Optional) Map id of instance locks to reset
+       @param difficulty (Optional) Difficulty of instance locks to reset
+       @param locksReset All locks that were reset
+       @param locksFailedToReset Locks that could not be reset because they are used by existing instance map
+    */
+    void ResetInstanceLocksForPlayer(ObjectGuid const& playerGuid, Optional<uint32> mapId, Optional<Difficulty> difficulty,
+        std::vector<InstanceLock const*>* locksReset, std::vector<InstanceLock const*>* locksFailedToReset);
+
+    /**
+       @brief Retrieves instance lock statistics - for use in GM commands
+       @return Statistics info
+    */
+    InstanceLocksStatistics GetStatistics() const;
+
     static InstanceLockMgr& Instance();
 
     static InstanceResetTimePoint GetNextResetTime(MapDb2Entries const& entries);
-- 
cgit v1.2.3