#include "ReactionsModel.h"

#include <MatrixClient.h>

#include "Logging.h"

QHash<int, QByteArray>
ReactionsModel::roleNames() const
{
        return {
          {Key, "key"},
          {Count, "counter"},
          {Users, "users"},
          {SelfReacted, "selfReacted"},
        };
}

int
ReactionsModel::rowCount(const QModelIndex &) const
{
        return static_cast<int>(reactions.size());
}

QVariant
ReactionsModel::data(const QModelIndex &index, int role) const
{
        const int i = index.row();
        if (i < 0 || i >= static_cast<int>(reactions.size()))
                return {};

        switch (role) {
        case Key:
                return QString::fromStdString(reactions[i].key);
        case Count:
                return static_cast<int>(reactions[i].reactions.size());
        case Users: {
                QString users;
                for (size_t r = 0; r < reactions[i].reactions.size(); r++) {
                        if (r != 0)
                                users += ", ";
                        users += QString::fromStdString(reactions[i].reactions[r].sender);
                }
                return users;
        }
        case SelfReacted:
                for (const auto &reaction : reactions[i].reactions)
                        if (reaction.sender == http::client()->user_id().to_string())
                                return true;
                return false;
        default:
                return {};
        }
}

void
ReactionsModel::addReaction(const mtx::events::RoomEvent<mtx::events::msg::Reaction> &reaction)
{
        int idx = 0;
        for (auto &storedReactions : reactions) {
                if (storedReactions.key == reaction.content.relates_to.key) {
                        storedReactions.reactions.push_back(reaction);
                        emit dataChanged(index(idx, 0), index(idx, 0));
                        return;
                }
                idx++;
        }

        beginInsertRows(QModelIndex(), idx, idx);
        reactions.push_back(KeyReaction{reaction.content.relates_to.key, {reaction}});
        endInsertRows();
}

void
ReactionsModel::removeReaction(const mtx::events::RoomEvent<mtx::events::msg::Reaction> &reaction)
{
        int idx = 0;
        for (auto &storedReactions : reactions) {
                if (storedReactions.key == reaction.content.relates_to.key) {
                        for (auto it = begin(storedReactions.reactions);
                             it != end(storedReactions.reactions);
                             ++it) {
                                if (it->event_id == reaction.event_id) {
                                        storedReactions.reactions.erase(it);
                                        break;
                                }
                        }

                        if (storedReactions.reactions.size() == 0) {
                                beginRemoveRows(QModelIndex(), idx, idx);
                                reactions.erase(reactions.begin() + idx);
                                endRemoveRows();
                        } else
                                emit dataChanged(index(idx, 0), index(idx, 0));
                        return;
                }
                idx++;
        }
}