/*
 * Copyright (C) 2008-2014 TrinityCore <http://www.trinitycore.org/>
 *
 * 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 "SessionManager.h"

bool Battlenet::SessionManager::StartNetwork(boost::asio::io_service& service, std::string const& bindIp, uint16 port)
{
    if (!BaseSocketMgr::StartNetwork(service, bindIp, port))
        return false;

    _acceptor->AsyncAcceptManaged(&OnSocketAccept);
    return true;
}

NetworkThread<Battlenet::Session>* Battlenet::SessionManager::CreateThreads() const
{
    return new NetworkThread<Session>[GetNetworkThreadCount()];
}

void Battlenet::SessionManager::OnSocketAccept(tcp::socket&& sock)
{
    sSessionMgr.OnSocketOpen(std::forward<tcp::socket>(sock));
}

void Battlenet::SessionManager::AddSession(Session* session)
{
    std::unique_lock<boost::shared_mutex> lock(_sessionMutex);
    _sessions[{ session->GetAccountId(), session->GetGameAccountId() }] = session;
    _sessionsByAccountId[session->GetAccountId()].push_back(session);
}

void Battlenet::SessionManager::RemoveSession(Session* session)
{
    std::unique_lock<boost::shared_mutex> lock(_sessionMutex);
    auto itr = _sessions.find({ session->GetAccountId(), session->GetGameAccountId() });
    // Remove old session only if it was not overwritten by reconnecting
    if (itr != _sessions.end() && itr->second == session)
        _sessions.erase(itr);

    _sessionsByAccountId[session->GetAccountId()].remove(session);
}

Battlenet::Session* Battlenet::SessionManager::GetSession(uint32 accountId, uint32 gameAccountId) const
{
    auto itr = _sessions.find({ accountId, gameAccountId });
    if (itr != _sessions.end())
        return itr->second;

    return nullptr;
}

std::list<Battlenet::Session*> Battlenet::SessionManager::GetSessions(uint32 accountId) const
{
    std::list<Session*> sessions;
    auto itr = _sessionsByAccountId.find(accountId);
    if (itr != _sessionsByAccountId.end())
        sessions = itr->second;

    return sessions;
}