mirror of
https://github.com/Nheko-Reborn/nheko.git
synced 2024-11-21 18:50:47 +03:00
115 lines
3.1 KiB
C++
115 lines
3.1 KiB
C++
// SPDX-FileCopyrightText: Nheko Contributors
|
|
//
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#include "InviteesModel.h"
|
|
|
|
#include "Logging.h"
|
|
#include "MatrixClient.h"
|
|
#include "mtx/responses/profile.hpp"
|
|
#include "timeline/TimelineModel.h"
|
|
|
|
InviteesModel::InviteesModel(TimelineModel *room, QObject *parent)
|
|
: QAbstractListModel{parent}
|
|
, room_{room}
|
|
{
|
|
}
|
|
|
|
void
|
|
InviteesModel::addUser(QString mxid, QString displayName, QString avatarUrl)
|
|
{
|
|
for (const auto &invitee : std::as_const(invitees_))
|
|
if (invitee->mxid_ == mxid)
|
|
return;
|
|
|
|
beginInsertRows(QModelIndex(), invitees_.count(), invitees_.count());
|
|
|
|
auto invitee = new Invitee{mxid, displayName, avatarUrl, this};
|
|
auto indexOfInvitee = invitees_.count();
|
|
connect(invitee, &Invitee::userInfoLoaded, this, [this, indexOfInvitee]() {
|
|
emit dataChanged(index(indexOfInvitee), index(indexOfInvitee));
|
|
});
|
|
|
|
invitees_.push_back(invitee);
|
|
|
|
endInsertRows();
|
|
emit countChanged();
|
|
}
|
|
|
|
void
|
|
InviteesModel::removeUser(QString mxid)
|
|
{
|
|
for (int i = 0; i < invitees_.length(); ++i) {
|
|
if (invitees_[i]->mxid_ == mxid) {
|
|
beginRemoveRows(QModelIndex(), i, i);
|
|
invitees_.removeAt(i);
|
|
endRemoveRows();
|
|
emit countChanged();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
QHash<int, QByteArray>
|
|
InviteesModel::roleNames() const
|
|
{
|
|
return {{Mxid, "mxid"}, {DisplayName, "displayName"}, {AvatarUrl, "avatarUrl"}};
|
|
}
|
|
|
|
QVariant
|
|
InviteesModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if (!index.isValid() || index.row() >= (int)invitees_.size() || index.row() < 0)
|
|
return {};
|
|
|
|
switch (role) {
|
|
case Mxid:
|
|
return invitees_[index.row()]->mxid_;
|
|
case DisplayName:
|
|
return invitees_[index.row()]->displayName_;
|
|
case AvatarUrl:
|
|
return invitees_[index.row()]->avatarUrl_;
|
|
default:
|
|
return {};
|
|
}
|
|
}
|
|
|
|
QStringList
|
|
InviteesModel::mxids()
|
|
{
|
|
QStringList mxidList;
|
|
mxidList.reserve(invitees_.size());
|
|
for (auto &invitee : std::as_const(invitees_))
|
|
mxidList.push_back(invitee->mxid_);
|
|
return mxidList;
|
|
}
|
|
|
|
Invitee::Invitee(QString mxid, QString displayName, QString avatarUrl, QObject *parent)
|
|
: QObject{parent}
|
|
, mxid_{std::move(mxid)}
|
|
{
|
|
// checking for empty avatarUrl will cause profiles that don't have an avatar
|
|
// to needlessly be loaded. Can we make sure we either provide both or none?
|
|
if (displayName == "" && avatarUrl == "") {
|
|
http::client()->get_profile(
|
|
mxid_.toStdString(),
|
|
[this](const mtx::responses::Profile &res, mtx::http::RequestErr err) {
|
|
if (err) {
|
|
nhlog::net()->warn("failed to retrieve profile info");
|
|
emit userInfoLoaded();
|
|
return;
|
|
}
|
|
|
|
displayName_ = QString::fromStdString(res.display_name);
|
|
avatarUrl_ = QString::fromStdString(res.avatar_url);
|
|
|
|
emit userInfoLoaded();
|
|
});
|
|
} else {
|
|
displayName_ = displayName;
|
|
avatarUrl_ = avatarUrl;
|
|
emit userInfoLoaded();
|
|
}
|
|
}
|
|
|
|
#include "moc_InviteesModel.cpp"
|