100 lines
3.0 KiB
C++
100 lines
3.0 KiB
C++
#include "roomlistmodel.h"
|
|
|
|
#include <QtCore/QDebug>
|
|
#include <QtGui/QBrush>
|
|
#include <QtGui/QColor>
|
|
|
|
RoomListModel::RoomListModel(QObject* parent) : QAbstractListModel(parent) {}
|
|
|
|
RoomListModel::~RoomListModel() {}
|
|
|
|
void RoomListModel::setConnection(QMatrixClient::Connection* connection) {
|
|
Q_ASSERT(connection);
|
|
|
|
using QMatrixClient::Connection;
|
|
using QMatrixClient::Room;
|
|
beginResetModel();
|
|
m_connection = connection;
|
|
connect(connection, &Connection::loggedOut, this,
|
|
[=] { setConnection(connection); });
|
|
// connect(connection, &Connection::invitedRoom, this,
|
|
// &RoomListModel::updateRoom);
|
|
// connect(connection, &Connection::joinedRoom, this,
|
|
// &RoomListModel::updateRoom);
|
|
// connect(connection, &Connection::leftRoom, this, &RoomListModel::updateRoom);
|
|
connect(connection, &Connection::aboutToDeleteRoom, this,
|
|
&RoomListModel::deleteRoom);
|
|
|
|
for (auto r : connection->roomMap()) addRoom(r);
|
|
endResetModel();
|
|
}
|
|
|
|
QMatrixClient::Room* RoomListModel::roomAt(int row) { return m_rooms.at(row); }
|
|
|
|
void RoomListModel::addRoom(QMatrixClient::Room* room) {
|
|
beginInsertRows(QModelIndex(), m_rooms.count(), m_rooms.count());
|
|
connect(room, &QMatrixClient::Room::namesChanged, this,
|
|
&RoomListModel::namesChanged);
|
|
m_rooms.append(room);
|
|
endInsertRows();
|
|
}
|
|
|
|
void RoomListModel::deleteRoom(QMatrixClient::Room* room) {
|
|
const auto it = std::find(m_rooms.begin(), m_rooms.end(), room);
|
|
if (it == m_rooms.end()) return; // Already deleted, nothing to do
|
|
const int row = it - m_rooms.begin();
|
|
beginRemoveRows(QModelIndex(), row, row);
|
|
m_rooms.erase(it);
|
|
endRemoveRows();
|
|
}
|
|
|
|
int RoomListModel::rowCount(const QModelIndex& parent) const {
|
|
if (parent.isValid()) return 0;
|
|
return m_rooms.count();
|
|
}
|
|
|
|
QVariant RoomListModel::data(const QModelIndex& index, int role) const {
|
|
if (!index.isValid()) return QVariant();
|
|
|
|
if (index.row() >= m_rooms.count()) {
|
|
qDebug() << "UserListModel: something wrong here...";
|
|
return QVariant();
|
|
}
|
|
QMatrixClient::Room* room = m_rooms.at(index.row());
|
|
if (role == Qt::DisplayRole) {
|
|
return room->displayName();
|
|
}
|
|
if (role == Qt::ForegroundRole) {
|
|
if (room->highlightCount() > 0) return QBrush(QColor("orange"));
|
|
return QVariant();
|
|
}
|
|
if (role == Qt::DecorationRole) {
|
|
if (room->avatarUrl().toString() != "") {
|
|
return room->avatarUrl();
|
|
}
|
|
return QVariant();
|
|
}
|
|
if (role == Qt::StatusTipRole) {
|
|
return room->topic();
|
|
}
|
|
return QVariant();
|
|
}
|
|
|
|
void RoomListModel::namesChanged(QMatrixClient::Room* room) {
|
|
int row = m_rooms.indexOf(room);
|
|
emit dataChanged(index(row), index(row));
|
|
}
|
|
|
|
void RoomListModel::unreadMessagesChanged(QMatrixClient::Room* room) {
|
|
int row = m_rooms.indexOf(room);
|
|
emit dataChanged(index(row), index(row));
|
|
}
|
|
|
|
QHash<int, QByteArray> RoomListModel::roleNames() const {
|
|
QHash<int, QByteArray> roles;
|
|
roles[Qt::DisplayRole] = "name";
|
|
roles[Qt::DecorationRole] = "avatar";
|
|
roles[Qt::StatusTipRole] = "topic";
|
|
return roles;
|
|
}
|