matrixion/src/AvatarProvider.cc

67 lines
2.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/>.
*/
2018-04-21 21:18:57 +03:00
#include <QBuffer>
#include <QtConcurrent>
#include "AvatarProvider.h"
2018-04-21 16:34:50 +03:00
#include "Cache.h"
2017-10-28 15:46:39 +03:00
#include "MatrixClient.h"
2017-11-30 14:53:28 +03:00
2017-08-20 13:47:22 +03:00
void
2018-04-21 16:34:50 +03:00
AvatarProvider::resolve(const QString &room_id,
const QString &user_id,
2018-03-25 15:59:47 +03:00
QObject *receiver,
std::function<void(QImage)> callback)
{
2018-04-21 21:18:57 +03:00
const auto key = QString("%1 %2").arg(room_id).arg(user_id);
const auto avatarUrl = Cache::avatarUrl(room_id, user_id);
2018-04-21 16:34:50 +03:00
2018-05-08 20:30:09 +03:00
if (!Cache::AvatarUrls.contains(key) || !cache::client())
2017-10-28 21:11:40 +03:00
return;
2018-04-21 21:18:57 +03:00
if (avatarUrl.isEmpty())
return;
2018-05-08 20:30:09 +03:00
auto data = cache::client()->image(avatarUrl);
2018-04-21 21:18:57 +03:00
if (!data.isNull()) {
callback(QImage::fromData(data));
return;
2017-09-10 12:59:21 +03:00
}
auto proxy = http::client()->fetchUserAvatar(avatarUrl);
2018-03-25 23:05:44 +03:00
if (proxy.isNull())
2018-03-25 15:59:47 +03:00
return;
2018-03-25 15:59:47 +03:00
connect(proxy.data(),
&DownloadMediaProxy::avatarDownloaded,
receiver,
2018-04-21 21:18:57 +03:00
[user_id, proxy, callback, avatarUrl](const QImage &img) {
2018-03-25 15:59:47 +03:00
proxy->deleteLater();
2018-04-21 21:18:57 +03:00
QtConcurrent::run([img, avatarUrl]() {
QByteArray data;
QBuffer buffer(&data);
buffer.open(QIODevice::WriteOnly);
img.save(&buffer, "PNG");
2018-05-08 20:30:09 +03:00
cache::client()->saveImage(avatarUrl, data);
2018-04-21 21:18:57 +03:00
});
2018-03-25 15:59:47 +03:00
callback(img);
});
}