/*
 * 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;
    }

    TraitMgr::LearnResult validationResult = TraitMgr::ValidateConfig(newConfigState, _player, true);
    if (validationResult != TraitMgr::LearnResult::Ok)
    {
        SendPacket(WorldPackets::Traits::TraitConfigCommitFailed(configId, 0, AsUnderlyingType(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 = 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);
}