diff options
Diffstat (limited to 'src/server/game/Handlers')
| -rw-r--r-- | src/server/game/Handlers/CharacterHandler.cpp | 12 | ||||
| -rw-r--r-- | src/server/game/Handlers/TraitHandler.cpp | 253 |
2 files changed, 261 insertions, 4 deletions
diff --git a/src/server/game/Handlers/CharacterHandler.cpp b/src/server/game/Handlers/CharacterHandler.cpp index 98adc84d6fe..9a7d20cc1a6 100644 --- a/src/server/game/Handlers/CharacterHandler.cpp +++ b/src/server/game/Handlers/CharacterHandler.cpp @@ -176,10 +176,6 @@ bool LoginQueryHolder::Initialize() stmt->setUInt64(0, lowGuid); res &= SetPreparedQuery(PLAYER_LOGIN_QUERY_LOAD_VOID_STORAGE, stmt); - stmt = CharacterDatabase.GetPreparedStatement(CHAR_SEL_CHARACTER_ACTIONS); - stmt->setUInt64(0, lowGuid); - res &= SetPreparedQuery(PLAYER_LOGIN_QUERY_LOAD_ACTIONS, stmt); - stmt = CharacterDatabase.GetPreparedStatement(CHAR_SEL_MAIL); stmt->setUInt64(0, lowGuid); res &= SetPreparedQuery(PLAYER_LOGIN_QUERY_LOAD_MAILS, stmt); @@ -331,6 +327,14 @@ bool LoginQueryHolder::Initialize() stmt->setUInt64(0, lowGuid); res &= SetPreparedQuery(PLAYER_LOGIN_QUERY_LOAD_GARRISON_FOLLOWER_ABILITIES, stmt); + stmt = CharacterDatabase.GetPreparedStatement(CHAR_SEL_CHAR_TRAIT_ENTRIES); + stmt->setUInt64(0, lowGuid); + res &= SetPreparedQuery(PLAYER_LOGIN_QUERY_LOAD_TRAIT_ENTRIES, stmt); + + stmt = CharacterDatabase.GetPreparedStatement(CHAR_SEL_CHAR_TRAIT_CONFIGS); + stmt->setUInt64(0, lowGuid); + res &= SetPreparedQuery(PLAYER_LOGIN_QUERY_LOAD_TRAIT_CONFIGS, stmt); + return res; } diff --git a/src/server/game/Handlers/TraitHandler.cpp b/src/server/game/Handlers/TraitHandler.cpp new file mode 100644 index 00000000000..ae61b159d62 --- /dev/null +++ b/src/server/game/Handlers/TraitHandler.cpp @@ -0,0 +1,253 @@ +/* + * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information + * + * This program is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License as published by the + * Free Software Foundation; either version 2 of the License, or (at your + * option) any later version. + * + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for + * more details. + * + * You should have received a copy of the GNU General Public License along + * with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "WorldSession.h" +#include "Battleground.h" +#include "DB2Stores.h" +#include "Player.h" +#include "SpellHistory.h" +#include "TraitMgr.h" +#include "TraitPackets.h" + +void WorldSession::HandleTraitsCommitConfig(WorldPackets::Traits::TraitsCommitConfig const& traitsCommitConfig) +{ + int32 configId = traitsCommitConfig.Config.ID; + UF::TraitConfig const* existingConfig = _player->GetTraitConfig(configId); + if (!existingConfig) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_UNKNOWN).Write()); + return; + } + + if (_player->IsInCombat()) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_AFFECTING_COMBAT).Write()); + return; + } + + if (_player->GetBattleground() && _player->GetBattleground()->GetStatus() == STATUS_IN_PROGRESS) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_IN_PVP_MATCH).Write()); + return; + } + + auto findEntry = [](WorldPackets::Traits::TraitConfig& config, int32 traitNodeId, int32 traitNodeEntryId) -> WorldPackets::Traits::TraitEntry* + { + auto entryItr = std::find_if(config.Entries.begin(), config.Entries.end(), [=](WorldPackets::Traits::TraitEntry const& traitEntry) + { + return traitEntry.TraitNodeID == traitNodeId && traitEntry.TraitNodeEntryID == traitNodeEntryId; + }); + return entryItr != config.Entries.end() ? &*entryItr : nullptr; + }; + + bool hasRemovedEntries = false; + WorldPackets::Traits::TraitConfig newConfigState(*existingConfig); + for (WorldPackets::Traits::TraitEntry const& newEntry : traitsCommitConfig.Config.Entries) + { + WorldPackets::Traits::TraitEntry* traitEntry = findEntry(newConfigState, newEntry.TraitNodeID, newEntry.TraitNodeEntryID); + if (traitEntry && traitEntry->Rank > newEntry.Rank) + { + TraitNodeEntry const* traitNode = sTraitNodeStore.LookupEntry(newEntry.TraitNodeID); + if (!traitNode) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_UNKNOWN).Write()); + return; + } + + TraitTreeEntry const* traitTree = sTraitTreeStore.LookupEntry(traitNode->TraitTreeID); + if (!traitTree) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_UNKNOWN).Write()); + return; + } + + if (traitTree->GetFlags().HasFlag(TraitTreeFlag::CannotRefund)) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_CANT_REMOVE_TALENT).Write()); + return; + } + + TraitNodeEntryEntry const* traitNodeEntry = sTraitNodeEntryStore.LookupEntry(newEntry.TraitNodeEntryID); + if (!traitNodeEntry) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_UNKNOWN).Write()); + return; + } + + TraitDefinitionEntry const* traitDefinition = sTraitDefinitionStore.LookupEntry(traitNodeEntry->TraitDefinitionID); + if (!traitDefinition) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, TALENT_FAILED_UNKNOWN).Write()); + return; + } + + if (traitDefinition->SpellID && _player->GetSpellHistory()->HasCooldown(traitDefinition->SpellID)) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, traitDefinition->SpellID, TALENT_FAILED_CANT_REMOVE_TALENT).Write()); + return; + } + + if (traitDefinition->VisibleSpellID && _player->GetSpellHistory()->HasCooldown(traitDefinition->VisibleSpellID)) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, traitDefinition->VisibleSpellID, TALENT_FAILED_CANT_REMOVE_TALENT).Write()); + return; + } + + hasRemovedEntries = true; + } + + if (traitEntry) + { + if (newEntry.Rank) + traitEntry->Rank = newEntry.Rank; + else + newConfigState.Entries.erase(std::remove_if(newConfigState.Entries.begin(), newConfigState.Entries.end(), [&newEntry](WorldPackets::Traits::TraitEntry const& traitEntry) + { + return traitEntry.TraitNodeID == newEntry.TraitNodeID && traitEntry.TraitNodeEntryID == newEntry.TraitNodeEntryID; + }), newConfigState.Entries.end()); + } + else + newConfigState.Entries.emplace_back() = newEntry; + } + + TalentLearnResult validationResult = TraitMgr::ValidateConfig(newConfigState, _player, true); + if (validationResult != TALENT_LEARN_OK) + { + SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, validationResult).Write()); + return; + } + + bool needsCastTime = newConfigState.Type == TraitConfigType::Combat && hasRemovedEntries; + + if (traitsCommitConfig.SavedLocalIdentifier) + newConfigState.LocalIdentifier = traitsCommitConfig.SavedLocalIdentifier; + else if (UF::TraitConfig const* savedConfig = _player->GetTraitConfig(traitsCommitConfig.SavedLocalIdentifier)) + newConfigState.LocalIdentifier = savedConfig->LocalIdentifier; + + _player->UpdateTraitConfig(std::move(newConfigState), traitsCommitConfig.SavedConfigID, needsCastTime); +} + +void WorldSession::HandleClassTalentsRequestNewConfig(WorldPackets::Traits::ClassTalentsRequestNewConfig& classTalentsRequestNewConfig) +{ + if (classTalentsRequestNewConfig.Config.Type != TraitConfigType::Combat) + return; + + if ((classTalentsRequestNewConfig.Config.CombatConfigFlags & TraitCombatConfigFlags::ActiveForSpec) != TraitCombatConfigFlags::None) + return; + + int64 configCount = std::count_if(_player->m_activePlayerData->TraitConfigs.begin(), _player->m_activePlayerData->TraitConfigs.end(), [](UF::TraitConfig const& traitConfig) + { + return static_cast<TraitConfigType>(*traitConfig.Type) == TraitConfigType::Combat + && (static_cast<TraitCombatConfigFlags>(*traitConfig.CombatConfigFlags) & TraitCombatConfigFlags::ActiveForSpec) == TraitCombatConfigFlags::None; + }); + if (configCount >= TraitMgr::MAX_COMBAT_TRAIT_CONFIGS) + return; + + auto findFreeLocalIdentifier = [&]() + { + int32 index = 1; + while (_player->m_activePlayerData->TraitConfigs.FindIndexIf([&](UF::TraitConfig const& traitConfig) + { + return static_cast<TraitConfigType>(*traitConfig.Type) == TraitConfigType::Combat + && traitConfig.ChrSpecializationID == int32(_player->GetPrimarySpecialization()) + && traitConfig.LocalIdentifier == index; + }) >= 0) + ++index; + + return index; + }; + + classTalentsRequestNewConfig.Config.ChrSpecializationID = _player->GetPrimarySpecialization(); + classTalentsRequestNewConfig.Config.LocalIdentifier = findFreeLocalIdentifier(); + + for (UF::TraitEntry const& grantedEntry : TraitMgr::GetGrantedTraitEntriesForConfig(classTalentsRequestNewConfig.Config, _player)) + { + auto entryItr = std::find_if(classTalentsRequestNewConfig.Config.Entries.begin(), classTalentsRequestNewConfig.Config.Entries.end(), + [&](WorldPackets::Traits::TraitEntry const& entry) { return entry.TraitNodeID == grantedEntry.TraitNodeID && entry.TraitNodeEntryID == grantedEntry.TraitNodeEntryID; }); + + WorldPackets::Traits::TraitEntry& newEntry = entryItr != classTalentsRequestNewConfig.Config.Entries.end() ? *entryItr : classTalentsRequestNewConfig.Config.Entries.emplace_back(); + newEntry.TraitNodeID = grantedEntry.TraitNodeID; + newEntry.TraitNodeEntryID = grantedEntry.TraitNodeEntryID; + newEntry.Rank = grantedEntry.Rank; + newEntry.GrantedRanks = grantedEntry.GrantedRanks; + if (TraitNodeEntryEntry const* traitNodeEntry = sTraitNodeEntryStore.LookupEntry(grantedEntry.TraitNodeEntryID)) + if (newEntry.Rank + newEntry.GrantedRanks > traitNodeEntry->MaxRanks) + newEntry.Rank = std::max(0, traitNodeEntry->MaxRanks - newEntry.GrantedRanks); + } + + TalentLearnResult validationResult = TraitMgr::ValidateConfig(classTalentsRequestNewConfig.Config, _player); + if (validationResult != TALENT_LEARN_OK) + return; + + _player->CreateTraitConfig(classTalentsRequestNewConfig.Config); +} + +void WorldSession::HandleClassTalentsRenameConfig(WorldPackets::Traits::ClassTalentsRenameConfig& classTalentsRenameConfig) +{ + _player->RenameTraitConfig(classTalentsRenameConfig.ConfigID, classTalentsRenameConfig.Name.Move()); +} + +void WorldSession::HandleClassTalentsDeleteConfig(WorldPackets::Traits::ClassTalentsDeleteConfig const& classTalentsDeleteConfig) +{ + _player->DeleteTraitConfig(classTalentsDeleteConfig.ConfigID); +} + +void WorldSession::HandleClassTalentsSetStarterBuildActive(WorldPackets::Traits::ClassTalentsSetStarterBuildActive const& classTalentsSetStarterBuildActive) +{ + UF::TraitConfig const* traitConfig = _player->GetTraitConfig(classTalentsSetStarterBuildActive.ConfigID); + if (!traitConfig) + return; + + if (static_cast<TraitConfigType>(*traitConfig->Type) != TraitConfigType::Combat) + return; + + if (!EnumFlag(static_cast<TraitCombatConfigFlags>(*traitConfig->CombatConfigFlags)).HasFlag(TraitCombatConfigFlags::ActiveForSpec)) + return; + + if (classTalentsSetStarterBuildActive.Active) + { + WorldPackets::Traits::TraitConfig newConfigState(*traitConfig); + + auto findFreeLocalIdentifier = [&]() + { + int32 index = 1; + while (_player->m_activePlayerData->TraitConfigs.FindIndexIf([&](UF::TraitConfig const& traitConfig) + { + return static_cast<TraitConfigType>(*traitConfig.Type) == TraitConfigType::Combat + && traitConfig.ChrSpecializationID == int32(_player->GetPrimarySpecialization()) + && traitConfig.LocalIdentifier == index; + }) >= 0) + ++index; + + return index; + }; + + TraitMgr::InitializeStarterBuildTraitConfig(newConfigState, _player); + newConfigState.CombatConfigFlags |= TraitCombatConfigFlags::StarterBuild; + newConfigState.LocalIdentifier = findFreeLocalIdentifier(); + + _player->UpdateTraitConfig(std::move(newConfigState), 0, true); + } + else + _player->SetTraitConfigUseStarterBuild(classTalentsSetStarterBuildActive.ConfigID, false); +} + +void WorldSession::HandleClassTalentsSetUsesSharedActionBars(WorldPackets::Traits::ClassTalentsSetUsesSharedActionBars const& classTalentsSetUsesSharedActionBars) +{ + _player->SetTraitConfigUseSharedActionBars(classTalentsSetUsesSharedActionBars.ConfigID, classTalentsSetUsesSharedActionBars.UsesShared, + classTalentsSetUsesSharedActionBars.IsLastSelectedSavedConfig); +} |