/*
 * 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)
{
}

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 = AsUnderlyingType(_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);
    }

    TraitMgr::LearnResult validationResult = TraitMgr::ValidateConfig(classTalentsRequestNewConfig.Config, _player);
    if (validationResult != TraitMgr::LearnResult::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);
}