From 9880d28011873b4b3624fa2a69bd0b4edb5ad3c1 Mon Sep 17 00:00:00 2001
From: insunaa <insuna@no.capfr.fr>
Date: Mon, 1 Apr 2024 10:17:54 +0200
Subject: [PATCH] Chat/GM: Add unkillable command Closes
 https://github.com/cmangos/mangos-wotlk/pull/510

---
 src/game/Chat/Chat.cpp       |  1 +
 src/game/Chat/Chat.h         |  1 +
 src/game/Chat/Level3.cpp     | 15 +++++++++++++++
 src/game/Entities/Player.cpp | 13 +++++++++++++
 src/game/Entities/Player.h   |  6 ++++++
 5 files changed, 36 insertions(+)

diff --git a/src/game/Chat/Chat.cpp b/src/game/Chat/Chat.cpp
index 83b099c65b..9b9f699138 100644
--- a/src/game/Chat/Chat.cpp
+++ b/src/game/Chat/Chat.cpp
@@ -293,6 +293,7 @@ ChatCommand* ChatHandler::getCommandTable()
     {
         { "chat",           SEC_ADMINISTRATOR,  false, &ChatHandler::HandleGMChatCommand,              "", nullptr },
         { "fly",            SEC_ADMINISTRATOR,  false, &ChatHandler::HandleGMFlyCommand,               "", nullptr },
+        { "unkillable",     SEC_ADMINISTRATOR,  false, &ChatHandler::HandleGMUnkillableCommand,        "", nullptr },
         { "ingame",         SEC_ADMINISTRATOR,  true,  &ChatHandler::HandleGMListIngameCommand,        "", nullptr },
         { "list",           SEC_ADMINISTRATOR,  true,  &ChatHandler::HandleGMListFullCommand,          "", nullptr },
         { "mountup",        SEC_ADMINISTRATOR,  false, &ChatHandler::HandleGMMountUpCommand,           "", nullptr },
diff --git a/src/game/Chat/Chat.h b/src/game/Chat/Chat.h
index 41b15de971..47d260033c 100644
--- a/src/game/Chat/Chat.h
+++ b/src/game/Chat/Chat.h
@@ -347,6 +347,7 @@ class ChatHandler
         bool HandleGMCommand(char* args);
         bool HandleGMChatCommand(char* args);
         bool HandleGMFlyCommand(char* args);
+        bool HandleGMUnkillableCommand(char* args);
         bool HandleGMListFullCommand(char* args);
         bool HandleGMListIngameCommand(char* args);
         bool HandleGMMountUpCommand(char* args);
diff --git a/src/game/Chat/Level3.cpp b/src/game/Chat/Level3.cpp
index 84678e58b9..45d5f3b2a5 100644
--- a/src/game/Chat/Level3.cpp
+++ b/src/game/Chat/Level3.cpp
@@ -5458,6 +5458,21 @@ bool ChatHandler::HandleGMFlyCommand(char* args)
     return true;
 }
 
+bool ChatHandler::HandleGMUnkillableCommand(char* args)
+{
+    bool value;
+    if (!ExtractOnOff(&args, value))
+    {
+        SendSysMessage(LANG_USE_BOL);
+        SetSentErrorMessage(true);
+        return false;
+    }
+    Player* target = m_session->GetPlayer();
+    target->SetDeathPrevention(value);
+    PSendSysMessage("GM Unkillability %s.", value ? "enabled" : "disabled");
+    return true;
+}
+
 bool ChatHandler::HandlePDumpLoadCommand(char* args)
 {
     char* file = ExtractQuotedOrLiteralArg(&args);
diff --git a/src/game/Entities/Player.cpp b/src/game/Entities/Player.cpp
index 2a81d3b351..de78618f68 100644
--- a/src/game/Entities/Player.cpp
+++ b/src/game/Entities/Player.cpp
@@ -20049,6 +20049,19 @@ void Player::SendLootError(ObjectGuid guid, LootError error) const
     SendDirectMessage(data);
 }
 
+void Player::SetDeathPrevention(bool enable)
+{
+    if (enable)
+        m_ExtraFlags |= PLAYER_EXTRA_GM_UNKILLABLE;
+    else
+        m_ExtraFlags &= ~PLAYER_EXTRA_GM_UNKILLABLE;
+}
+
+bool Player::IsPreventingDeath() const
+{
+    return m_ExtraFlags & PLAYER_EXTRA_GM_UNKILLABLE;
+}
+
 void Player::ForceHealAndPowerUpdateInZone()
 {
     for (auto guid : m_clientGUIDs)
diff --git a/src/game/Entities/Player.h b/src/game/Entities/Player.h
index 3f5033519d..94c9c35c9e 100644
--- a/src/game/Entities/Player.h
+++ b/src/game/Entities/Player.h
@@ -489,6 +489,9 @@ enum PlayerExtraFlags
     // other states
     PLAYER_EXTRA_PVP_DEATH          = 0x0100,                // store PvP death status until corpse creating.
     PLAYER_EXTRA_WHISP_RESTRICTION  = 0x0200,
+
+    // death prevention
+    PLAYER_EXTRA_GM_UNKILLABLE         = 0x0400,
 };
 
 // 2^n values
@@ -2206,6 +2209,9 @@ class Player : public Unit
 
         void SendLootError(ObjectGuid guid, LootError error) const;
 
+        void SetDeathPrevention(bool enable);
+        bool IsPreventingDeath() const override;
+
         // cooldown system
         virtual void AddGCD(SpellEntry const& spellEntry, uint32 forcedDuration = 0, bool updateClient = false) override;
         virtual void AddCooldown(SpellEntry const& spellEntry, ItemPrototype const* itemProto = nullptr, bool permanent = false, uint32 forcedDuration = 0, bool ignoreCat = false) override;