Using TC structure allowing easier patches importing

2026-01-16 10:30:27 +00:00 · 2017-12-20 20:48:35 +01:00
parent 0fc4a6a153
commit 17332304fd
1197 changed files with 0 additions and 69 deletions
--- a/src/server/game/Tools/CharacterDatabaseCleaner.cpp
+++ b/src/server/game/Tools/CharacterDatabaseCleaner.cpp
@@ -0,0 +1,146 @@
+/*
+ * Copyright (C) 2016+     AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license: http://github.com/azerothcore/azerothcore-wotlk/LICENSE-GPL2
+ * Copyright (C) 2008-2016 TrinityCore <http://www.trinitycore.org/>
+ * Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
+ */
+
+#include "Common.h"
+#include "CharacterDatabaseCleaner.h"
+#include "World.h"
+#include "Database/DatabaseEnv.h"
+#include "SpellMgr.h"
+#include "DBCStores.h"
+
+void CharacterDatabaseCleaner::CleanDatabase()
+{
+    // config to disable
+    if (!sWorld->getBoolConfig(CONFIG_CLEAN_CHARACTER_DB))
+        return;
+
+    sLog->outString("Cleaning character database...");
+
+    uint32 oldMSTime = getMSTime();
+
+    // check flags which clean ups are necessary
+    QueryResult result = CharacterDatabase.PQuery("SELECT value FROM worldstates WHERE entry = %d", WS_CLEANING_FLAGS);
+    if (!result)
+        return;
+
+    uint32 flags = (*result)[0].GetUInt32();
+
+    // clean up
+    if (flags & CLEANING_FLAG_ACHIEVEMENT_PROGRESS)
+        CleanCharacterAchievementProgress();
+
+    if (flags & CLEANING_FLAG_SKILLS)
+        CleanCharacterSkills();
+
+    if (flags & CLEANING_FLAG_SPELLS)
+        CleanCharacterSpell();
+
+    if (flags & CLEANING_FLAG_TALENTS)
+        CleanCharacterTalent();
+
+    if (flags & CLEANING_FLAG_QUESTSTATUS)
+        CleanCharacterQuestStatus();
+
+    // NOTE: In order to have persistentFlags be set in worldstates for the next cleanup,
+    // you need to define them at least once in worldstates.
+    flags &= sWorld->getIntConfig(CONFIG_PERSISTENT_CHARACTER_CLEAN_FLAGS);
+    CharacterDatabase.DirectPExecute("UPDATE worldstates SET value = %u WHERE entry = %d", flags, WS_CLEANING_FLAGS);
+
+    sWorld->SetCleaningFlags(flags);
+
+    sLog->outString(">> Cleaned character database in %u ms", GetMSTimeDiffToNow(oldMSTime));
+    sLog->outString();
+}
+
+void CharacterDatabaseCleaner::CheckUnique(const char* column, const char* table, bool (*check)(uint32))
+{
+    QueryResult result = CharacterDatabase.PQuery("SELECT DISTINCT %s FROM %s", column, table);
+    if (!result)
+    {
+        sLog->outString("Table %s is empty.", table);
+        return;
+    }
+
+    bool found = false;
+    std::ostringstream ss;
+    do
+    {
+        Field* fields = result->Fetch();
+
+        uint32 id = fields[0].GetUInt32();
+
+        if (!check(id))
+        {
+            if (!found)
+            {
+                ss << "DELETE FROM " << table << " WHERE " << column << " IN (";
+                found = true;
+            }
+            else
+                ss << ',';
+
+            ss << id;
+        }
+    }
+    while (result->NextRow());
+
+    if (found)
+    {
+        ss << ')';
+        CharacterDatabase.Execute(ss.str().c_str());
+    }
+}
+
+bool CharacterDatabaseCleaner::AchievementProgressCheck(uint32 criteria)
+{
+    return sAchievementCriteriaStore.LookupEntry(criteria);
+}
+
+void CharacterDatabaseCleaner::CleanCharacterAchievementProgress()
+{
+    CheckUnique("criteria", "character_achievement_progress", &AchievementProgressCheck);
+}
+
+bool CharacterDatabaseCleaner::SkillCheck(uint32 skill)
+{
+    return sSkillLineStore.LookupEntry(skill);
+}
+
+void CharacterDatabaseCleaner::CleanCharacterSkills()
+{
+    CheckUnique("skill", "character_skills", &SkillCheck);
+}
+
+bool CharacterDatabaseCleaner::SpellCheck(uint32 spell_id)
+{
+    return sSpellMgr->GetSpellInfo(spell_id) && !GetTalentSpellPos(spell_id);
+}
+
+void CharacterDatabaseCleaner::CleanCharacterSpell()
+{
+    CheckUnique("spell", "character_spell", &SpellCheck);
+}
+
+bool CharacterDatabaseCleaner::TalentCheck(uint32 talent_id)
+{
+    TalentEntry const* talentInfo = sTalentStore.LookupEntry(talent_id);
+    if (!talentInfo)
+        return false;
+
+    return sTalentTabStore.LookupEntry(talentInfo->TalentTab);
+}
+
+void CharacterDatabaseCleaner::CleanCharacterTalent()
+{
+    CharacterDatabase.DirectPExecute("DELETE FROM character_talent WHERE specMask >= %u", 1 << MAX_TALENT_SPECS);
+    CheckUnique("spell", "character_talent", &TalentCheck);
+}
+
+void CharacterDatabaseCleaner::CleanCharacterQuestStatus()
+{
+    CharacterDatabase.DirectExecute("DELETE FROM character_queststatus WHERE status = 0");
+}
+