83 lines
1.9 KiB
C++
83 lines
1.9 KiB
C++
// SPDX-FileCopyrightText: 2022 Tobias Fella <tobias.fella@kde.org>
|
|
// SPDX-License-Identifier: LGPL-2.0-or-later
|
|
|
|
#include "statemodel.h"
|
|
|
|
StateModel::StateModel(QObject *parent)
|
|
: QAbstractListModel(parent)
|
|
{
|
|
}
|
|
|
|
QHash<int, QByteArray> StateModel::roleNames() const
|
|
{
|
|
return {
|
|
{TypeRole, "type"},
|
|
{EventCountRole, "eventCount"},
|
|
{StateKeyRole, "stateKey"},
|
|
};
|
|
}
|
|
QVariant StateModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
auto row = index.row();
|
|
switch (role) {
|
|
case TypeRole:
|
|
return m_stateEvents.keys()[row];
|
|
case EventCountRole:
|
|
return m_stateEvents.values()[row].count();
|
|
case StateKeyRole:
|
|
return m_stateEvents.values()[row][0];
|
|
}
|
|
return {};
|
|
}
|
|
|
|
int StateModel::rowCount(const QModelIndex &parent) const
|
|
{
|
|
Q_UNUSED(parent);
|
|
return m_stateEvents.count();
|
|
}
|
|
|
|
NeoChatRoom *StateModel::room() const
|
|
{
|
|
return m_room;
|
|
}
|
|
|
|
void StateModel::loadState()
|
|
{
|
|
beginResetModel();
|
|
m_stateEvents.clear();
|
|
if (!m_room) {
|
|
return;
|
|
}
|
|
const auto keys = m_room->currentState().events().keys();
|
|
for (const auto &[type, stateKey] : keys) {
|
|
if (!m_stateEvents.contains(type)) {
|
|
m_stateEvents[type] = {};
|
|
}
|
|
m_stateEvents[type] += stateKey;
|
|
}
|
|
endResetModel();
|
|
}
|
|
|
|
void StateModel::setRoom(NeoChatRoom *room)
|
|
{
|
|
m_room = room;
|
|
Q_EMIT roomChanged();
|
|
loadState();
|
|
|
|
connect(room, &NeoChatRoom::changed, this, [this] {
|
|
loadState();
|
|
});
|
|
}
|
|
|
|
QByteArray StateModel::stateEventJson(const QString &type, const QString &stateKey)
|
|
{
|
|
return QJsonDocument(m_room->currentState().get(type, stateKey)->fullJson()).toJson();
|
|
}
|
|
|
|
QByteArray StateModel::stateEventContentJson(const QString &type, const QString &stateKey)
|
|
{
|
|
return QJsonDocument(m_room->currentState().get(type, stateKey)->contentJson()).toJson();
|
|
}
|
|
|
|
#include "moc_statemodel.cpp"
|