matrixion/src/AvatarProvider.cc

96 lines
3 KiB
C++
Raw Normal View History

/*
* nheko Copyright (C) 2017 Konstantinos Sideris <siderisk@auth.gr>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "AvatarProvider.h"
2017-10-28 15:46:39 +03:00
#include "MatrixClient.h"
2017-11-30 14:53:28 +03:00
QSharedPointer<MatrixClient> AvatarProvider::client_;
std::map<QString, AvatarData> AvatarProvider::avatars_;
std::map<QString, std::vector<std::function<void(QImage)>>> AvatarProvider::toBeResolved_;
2017-08-20 13:47:22 +03:00
void
AvatarProvider::init(QSharedPointer<MatrixClient> client)
{
2017-09-10 12:59:21 +03:00
client_ = client;
}
2017-08-20 13:47:22 +03:00
void
AvatarProvider::updateAvatar(const QString &uid, const QImage &img)
{
if (toBeResolved_.find(uid) != toBeResolved_.end()) {
2018-01-03 19:05:49 +03:00
auto callbacks = toBeResolved_[uid];
2017-09-10 12:59:21 +03:00
// Update all the timeline items with the resolved avatar.
2018-01-03 19:05:49 +03:00
for (const auto callback : callbacks)
callback(img);
2018-01-25 15:34:15 +03:00
toBeResolved_.erase(uid);
2017-09-10 12:59:21 +03:00
}
2018-01-30 21:39:06 +03:00
auto avatarData = &avatars_[uid];
avatarData->img = img;
}
2017-08-20 13:47:22 +03:00
void
2018-01-03 19:05:49 +03:00
AvatarProvider::resolve(const QString &userId, std::function<void(QImage)> callback)
{
if (avatars_.find(userId) == avatars_.end())
2017-10-28 21:11:40 +03:00
return;
2017-10-28 21:11:40 +03:00
auto img = avatars_[userId].img;
2017-10-28 21:11:40 +03:00
if (!img.isNull()) {
2018-01-03 19:05:49 +03:00
callback(img);
2017-09-10 12:59:21 +03:00
return;
}
2017-10-28 21:11:40 +03:00
// Add the current timeline item to the waiting list for this avatar.
if (toBeResolved_.find(userId) == toBeResolved_.end()) {
client_->fetchUserAvatar(avatars_[userId].url,
[userId](QImage image) { updateAvatar(userId, image); },
[userId](QString error) {
qWarning()
<< error << ": failed to retrieve user avatar"
<< userId;
});
std::vector<std::function<void(QImage)>> items;
items.emplace_back(callback);
toBeResolved_.emplace(userId, items);
2017-10-28 21:11:40 +03:00
} else {
toBeResolved_[userId].emplace_back(callback);
2017-09-10 12:59:21 +03:00
}
}
2017-08-20 13:47:22 +03:00
void
AvatarProvider::setAvatarUrl(const QString &userId, const QUrl &url)
{
2017-10-28 21:11:40 +03:00
AvatarData data;
data.url = url;
avatars_.emplace(userId, data);
}
2017-08-20 13:47:22 +03:00
void
AvatarProvider::clear()
{
2017-10-28 21:11:40 +03:00
avatars_.clear();
2017-09-10 12:59:21 +03:00
toBeResolved_.clear();
}