From 343fa0055e7d9c5ab14d1f553c08229ed234c8b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=83=91=E4=BD=A9=E8=8C=B9?= <zhengpeiru21@protonmail.com>
Date: Fri, 24 Jun 2022 18:45:04 -0600
Subject: [PATCH] Implement retroactive quest appearance collection

---
 conf/transmog.conf.dist    | 12 ++++++++++++
 src/Transmogrification.cpp | 12 ++++++++++++
 src/Transmogrification.h   |  5 +++++
 src/transmog_scripts.cpp   | 23 +++++++++++++++++++++++
 4 files changed, 52 insertions(+)

diff --git a/conf/transmog.conf.dist b/conf/transmog.conf.dist
index dda1312..e15558e 100644
--- a/conf/transmog.conf.dist
+++ b/conf/transmog.conf.dist
@@ -28,6 +28,16 @@
 #                     This allows these appearances to be used later if the configuration is changed.
 #        Default:     1
 #
+#    Transmogrification.RetroActiveAppearances
+#        Description: Enables/Disables checking all completed quests for uncollected appearances.
+#                     Occurs only once per player.
+#        Default:    1
+#
+#    Transmogrification.ResetRetroActiveAppearancesFlag
+#        Description: Resets the flag indicating whether the retroactive appearance check has been run.
+#                     Occurs for each character on log in.
+#        Default:    0
+#
 #    Transmogrification.EnableTransmogInfo
 #        Description: Enables / Disables the info button for transmogrification
 #        Default:    1
@@ -50,6 +60,8 @@ Transmogrification.Enable = 1
 Transmogrification.UseCollectionSystem = 1
 Transmogrification.AllowHiddenTransmog = 1
 Transmogrification.TrackUnusableItems = 1
+Transmogrification.RetroActiveAppearances = 1
+Transmogrification.ResetRetroActiveAppearancesFlag = 0
 
 Transmogrification.EnableTransmogInfo = 1
 Transmogrification.TransmogNpcText = 601083
diff --git a/src/Transmogrification.cpp b/src/Transmogrification.cpp
index 8cefe9f..20756c1 100644
--- a/src/Transmogrification.cpp
+++ b/src/Transmogrification.cpp
@@ -797,6 +797,8 @@ void Transmogrification::LoadConfig(bool reload)
     UseCollectionSystem = sConfigMgr->GetOption<bool>("Transmogrification.UseCollectionSystem", true);
     AllowHiddenTransmog = sConfigMgr->GetOption<bool>("Transmogrification.AllowHiddenTransmog", true);
     TrackUnusableItems = sConfigMgr->GetOption<bool>("Transmogrification.TrackUnusableItems", true);
+    RetroActiveAppearances = sConfigMgr->GetOption<bool>("Transmogrification.RetroActiveAppearances", true);
+    ResetRetroActiveAppearances = sConfigMgr->GetOption<bool>("Transmogrification.ResetRetroActiveAppearancesFlag", false);
 
     IsTransmogEnabled = sConfigMgr->GetOption<bool>("Transmogrification.Enable", true);
 
@@ -886,6 +888,16 @@ bool Transmogrification::GetTrackUnusableItems() const
     return TrackUnusableItems;
 }
 
+bool Transmogrification::EnableRetroActiveAppearances() const
+{
+    return RetroActiveAppearances;
+}
+
+bool Transmogrification::EnableResetRetroActiveAppearances() const
+{
+    return ResetRetroActiveAppearances;
+}
+
 bool Transmogrification::IsEnabled() const
 {
     return IsTransmogEnabled;
diff --git a/src/Transmogrification.h b/src/Transmogrification.h
index 43b0ea0..06aa1d6 100644
--- a/src/Transmogrification.h
+++ b/src/Transmogrification.h
@@ -27,6 +27,7 @@ struct ItemTemplate;
 enum TransmogSettings
 {
     SETTING_HIDE_TRANSMOG = 0,
+    SETTING_RETROACTIVE_CHECK = 1
 };
 
 enum TransmogAcoreStrings // Language.h might have same entries, appears when executing SQL, change if needed
@@ -133,6 +134,8 @@ public:
     bool UseCollectionSystem;
     bool AllowHiddenTransmog;
     bool TrackUnusableItems;
+    bool RetroActiveAppearances;
+    bool ResetRetroActiveAppearances;
 
     bool IsTransmogEnabled;
 
@@ -184,6 +187,8 @@ public:
     bool GetUseCollectionSystem() const;
     bool GetAllowHiddenTransmog() const;
     bool GetTrackUnusableItems() const;
+    bool EnableRetroActiveAppearances() const;
+    bool EnableResetRetroActiveAppearances() const;
     [[nodiscard]] bool IsEnabled() const;
 };
 #define sTransmogrification Transmogrification::instance()
diff --git a/src/transmog_scripts.cpp b/src/transmog_scripts.cpp
index 33a3af3..2f7565a 100644
--- a/src/transmog_scripts.cpp
+++ b/src/transmog_scripts.cpp
@@ -517,6 +517,21 @@ private:
             CharacterDatabase.Execute( "INSERT INTO custom_unlocked_appearances (account_id, item_template_id) VALUES ({}, {})", accountId, itemId);
         }
     }
+
+    void CheckRetroActiveQuestAppearances(Player* player)
+    {
+        QueryResult result = CharacterDatabase.Query("SELECT `quest` FROM `character_queststatus` WHERE `status` = 3 AND `guid` = {}", player->GetGUID().GetCounter());
+        if (result)
+        {
+            do
+            {
+                uint32 questId = (*result)[0].Get<uint32>();
+                Quest* questTemplate = const_cast<Quest*>(sObjectMgr->GetQuestTemplate(questId));
+                OnPlayerCompleteQuest(player, questTemplate);
+            } while (result->NextRow());
+        }
+        player->UpdatePlayerSetting("mod-transmog", SETTING_RETROACTIVE_CHECK, 1);
+    }
 public:
     PS_Transmogrification() : PlayerScript("Player_Transmogrify") { }
 
@@ -597,6 +612,14 @@ public:
 
     void OnLogin(Player* player) override
     {
+        if (sT->EnableResetRetroActiveAppearances())
+        {
+            player->UpdatePlayerSetting("mod-transmog", SETTING_RETROACTIVE_CHECK, 0);
+        }
+        if (sT->EnableRetroActiveAppearances() && !(player->GetPlayerSetting("mod-transmog", SETTING_RETROACTIVE_CHECK).value))
+        {
+            CheckRetroActiveQuestAppearances(player);
+        }
         ObjectGuid playerGUID = player->GetGUID();
         sT->entryMap.erase(playerGUID);
         QueryResult result = CharacterDatabase.Query("SELECT GUID, FakeEntry FROM custom_transmogrification WHERE Owner = {}", player->GetGUID().GetCounter());