72 lines
2.2 KiB
C++
72 lines
2.2 KiB
C++
|
#include "accountlistmodel.h"
|
||
|
|
|
||
|
|
AccountListModel::AccountListModel(QObject* parent)
|
||
|
|
: QAbstractListModel(parent) {}
|
||
|
|
|
||
|
|
void AccountListModel::setController(Controller* value) {
|
||
|
|
if (m_controller != value) {
|
||
|
|
beginResetModel();
|
||
|
|
m_connections.clear();
|
||
|
|
|
||
|
|
m_controller = value;
|
||
|
|
|
||
|
|
for (auto c : m_controller->connections()) m_connections.append(c);
|
||
|
|
|
||
|
|
connect(m_controller, &Controller::connectionAdded, this,
|
||
|
|
[=](Connection* conn) {
|
||
|
|
beginInsertRows(QModelIndex(), m_connections.count(),
|
||
|
|
m_connections.count());
|
||
|
|
m_connections.append(conn);
|
||
|
|
endInsertRows();
|
||
|
|
});
|
||
|
|
connect(m_controller, &Controller::connectionDropped, this,
|
||
|
|
[=](Connection* conn) {
|
||
|
|
const auto it =
|
||
|
|
std::find(m_connections.begin(), m_connections.end(), conn);
|
||
|
|
if (it == m_connections.end())
|
||
|
|
return; // Already deleted, nothing to do
|
||
|
|
const int row = it - m_connections.begin();
|
||
|
|
beginRemoveRows(QModelIndex(), row, row);
|
||
|
|
m_connections.erase(it);
|
||
|
|
endRemoveRows();
|
||
|
|
});
|
||
|
|
emit controllerChanged();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
QVariant AccountListModel::data(const QModelIndex& index, int role) const {
|
||
|
|
if (!index.isValid()) return QVariant();
|
||
|
|
|
||
|
|
if (index.row() >= m_controller->connections().count()) {
|
||
|
|
qDebug()
|
||
|
|
<< "UserListModel, something's wrong: index.row() >= m_users.count()";
|
||
|
|
return QVariant();
|
||
|
|
}
|
||
|
|
auto m_connection = m_controller->connections().at(index.row());
|
||
|
|
if (role == NameRole) {
|
||
|
|
return m_connection->user()->displayname();
|
||
|
|
}
|
||
|
|
if (role == AvatarRole) {
|
||
|
|
return m_connection->user()->avatar(64);
|
||
|
|
}
|
||
|
|
if (role == ConnectionRole) {
|
||
|
|
return QVariant::fromValue(m_connection);
|
||
|
|
}
|
||
|
|
|
||
|
|
return QVariant();
|
||
|
|
}
|
||
|
|
|
||
|
|
int AccountListModel::rowCount(const QModelIndex& parent) const {
|
||
|
|
if (parent.isValid()) return 0;
|
||
|
|
|
||
|
|
return m_controller->connections().count();
|
||
|
|
}
|
||
|
|
|
||
|
|
QHash<int, QByteArray> AccountListModel::roleNames() const {
|
||
|
|
QHash<int, QByteArray> roles;
|
||
|
|
roles[NameRole] = "name";
|
||
|
|
roles[AvatarRole] = "avatar";
|
||
|
|
roles[ConnectionRole] = "connection";
|
||
|
|
return roles;
|
||
|
|
}
|