99 lines
3.2 KiB
C++
99 lines
3.2 KiB
C++
|
#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++;
|
||
|
|
}
|
||
|
|
}
|