Cockatrice/cockatrice/src/server/user/user_list_manager.cpp

#include "user_list_manager.h"

#include "../../client/game_logic/abstract_client.h"
#include "../../client/sound_engine.h"
#include "../pending_command.h"
#include "pb/event_add_to_list.pb.h"
#include "pb/event_remove_from_list.pb.h"
#include "pb/event_user_joined.pb.h"
#include "pb/event_user_left.pb.h"
#include "pb/response_list_users.pb.h"
#include "pb/session_commands.pb.h"
#include "user_info_box.h"

UserListManager::UserListManager(AbstractClient *_client, QWidget *parent)
    : QWidget(parent), client(_client), ownUserInfo(nullptr)
{
    connect(client, &AbstractClient::userJoinedEventReceived, this, &UserListManager::processUserJoinedEvent);
    connect(client, &AbstractClient::userLeftEventReceived, this, &UserListManager::processUserLeftEvent);
    connect(client, &AbstractClient::buddyListReceived, this, &UserListManager::buddyListReceived);
    connect(client, &AbstractClient::ignoreListReceived, this, &UserListManager::ignoreListReceived);
    connect(client, &AbstractClient::addToListEventReceived, this, &UserListManager::processAddToListEvent);
    connect(client, &AbstractClient::removeFromListEventReceived, this, &UserListManager::processRemoveFromListEvent);
    connect(client, &AbstractClient::userInfoChanged, this, &UserListManager::setOwnUserInfo);
}

UserListManager::~UserListManager()
{
    handleDisconnect();
}

void UserListManager::handleConnect()
{
    populateInitialOnlineUsers();
}

void UserListManager::handleDisconnect()
{
    onlineUsers.clear();
    buddyUsers.clear();
    ignoredUsers.clear();

    delete ownUserInfo;
    ownUserInfo = nullptr;
}

void UserListManager::setOwnUserInfo(const ServerInfo_User &userInfo)
{
    ownUserInfo = new ServerInfo_User(userInfo);
}

void UserListManager::populateInitialOnlineUsers()
{
    PendingCommand *pend = client->prepareSessionCommand(Command_ListUsers());
    connect(pend, qOverload<const Response &, const CommandContainer &, const QVariant &>(&PendingCommand::finished),
            this, &UserListManager::processListUsersResponse);
    client->sendCommand(pend);
}

void UserListManager::processListUsersResponse(const Response &response)
{
    const auto &resp = response.GetExtension(Response_ListUsers::ext);

    const int userListSize = resp.user_list_size();
    for (int i = 0; i < userListSize; ++i) {
        const ServerInfo_User &info = resp.user_list(i);
        const QString &userName = QString::fromStdString(info.name());
        onlineUsers.insert(userName, info);
    }
}

void UserListManager::processUserJoinedEvent(const Event_UserJoined &event)
{
    const auto &info = event.user_info();
    const QString &userName = QString::fromStdString(info.name());
    onlineUsers.insert(userName, info);
}

void UserListManager::processUserLeftEvent(const Event_UserLeft &event)
{
    const auto &userName = QString::fromStdString(event.name());
    onlineUsers.remove(userName);
}

void UserListManager::buddyListReceived(const QList<ServerInfo_User> &_buddyList)
{
    for (const auto &user : _buddyList) {
        const auto &userName = QString::fromStdString(user.name());
        buddyUsers.insert(userName, user);
    }
}

void UserListManager::ignoreListReceived(const QList<ServerInfo_User> &_ignoreList)
{
    for (const auto &user : _ignoreList) {
        const auto &userName = QString::fromStdString(user.name());
        ignoredUsers.insert(userName, user);
    }
}

void UserListManager::processAddToListEvent(const Event_AddToList &event)
{
    const auto &user = event.user_info();
    const auto &userName = QString::fromStdString(user.name());

    const auto &userListType = QString::fromStdString(event.list_name());

    QMap<QString, ServerInfo_User> userMap;
    if (userListType == "buddy") {
        userMap = buddyUsers;
    } else if (userListType == "ignore") {
        userMap = ignoredUsers;
    } else {
        return;
    }

    userMap.insert(userName, user);
}

void UserListManager::processRemoveFromListEvent(const Event_RemoveFromList &event)
{
    const auto &userListType = QString::fromStdString(event.list_name());
    const auto &userName = QString::fromStdString(event.user_name());

    QMap<QString, ServerInfo_User> userMap;
    if (userListType == "buddy") {
        userMap = buddyUsers;
    } else if (userListType == "ignore") {
        userMap = ignoredUsers;
    } else {
        return;
    }

    userMap.remove(userName);
}

bool UserListManager::isOwnUserRegistered() const
{
    return ownUserInfo != nullptr && (ownUserInfo->user_level() & ServerInfo_User::IsRegistered) != 0;
}

QString UserListManager::getOwnUsername() const
{
    return ownUserInfo != nullptr ? QString::fromStdString(ownUserInfo->name()) : QString();
}

bool UserListManager::isUserBuddy(const QString &userName) const
{
    return buddyUsers.contains(userName);
}

bool UserListManager::isUserIgnored(const QString &userName) const
{
    return ignoredUsers.contains(userName);
}

const ServerInfo_User *UserListManager::getOnlineUser(const QString &userName) const
{
    const QString &userNameToMatchLower = userName.toLower();

    const auto it =
        std::find_if(onlineUsers.begin(), onlineUsers.end(), [&userNameToMatchLower](const ServerInfo_User &user) {
            return userNameToMatchLower == QString::fromStdString(user.name()).toLower();
        });

    if (it != onlineUsers.end()) {
        return &*it;
    }

    return nullptr;
};