aboutsummaryrefslogtreecommitdiff
path: root/avatar.cpp
blob: 040bf9bbbc044429f23e9267f996f01bc9aa0877 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
/******************************************************************************
 * Copyright (C) 2017 Kitsune Ral <kitsune-ral@users.sf.net>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include "avatar.h"

#include "jobs/mediathumbnailjob.h"
#include "events/eventcontent.h"
#include "connection.h"

#include <QtGui/QPainter>
#include <QtCore/QPointer>

using namespace QMatrixClient;

class Avatar::Private
{
    public:
        Private(Connection* c, QIcon di) : _connection(c), _defaultIcon(di) { }
        QImage get(QSize size, Avatar::notifier_t notifier) const;

        Connection* _connection;
        const QIcon _defaultIcon;
        QUrl _url;

        // The below are related to image caching, hence mutable
        mutable QImage _originalImage;
        mutable std::vector<QPair<QSize, QImage>> _scaledImages;
        mutable QSize _requestedSize;
        mutable bool _valid = false;
        mutable QPointer<MediaThumbnailJob> _ongoingRequest = nullptr;
        mutable std::vector<notifier_t> notifiers;
};

Avatar::Avatar(Connection* connection, QIcon defaultIcon)
    : d(new Private { connection, std::move(defaultIcon) })
{ }

Avatar::~Avatar() = default;

QImage Avatar::get(int dimension, notifier_t notifier) const
{
    return d->get({dimension, dimension}, notifier);
}

QImage Avatar::get(int width, int height, notifier_t notifier) const
{
    return d->get({width, height}, notifier);
}

QString Avatar::mediaId() const
{
    return d->_url.authority() + d->_url.path();
}

QImage Avatar::Private::get(QSize size, Avatar::notifier_t notifier) const
{
    // FIXME: Alternating between longer-width and longer-height requests
    // is a sure way to trick the below code into constantly getting another
    // image from the server because the existing one is alleged unsatisfactory.
    // This is plain abuse by the client, though; so not critical for now.
    if( ( !(_valid || _ongoingRequest)
          || size.width() > _requestedSize.width()
          || size.height() > _requestedSize.height() ) && _url.isValid() )
    {
        qCDebug(MAIN) << "Getting avatar from" << _url.toString();
        _requestedSize = size;
        if (isJobRunning(_ongoingRequest))
            _ongoingRequest->abandon();
        notifiers.emplace_back(std::move(notifier));
        _ongoingRequest = _connection->getThumbnail(_url, size);
        QObject::connect( _ongoingRequest, &MediaThumbnailJob::success, [this]
        {
            _valid = true;
            _originalImage = _ongoingRequest->scaledThumbnail(_requestedSize);
            _scaledImages.clear();
            for (auto n: notifiers)
                n();
        });
    }

    if( _originalImage.isNull() )
    {
        if (_defaultIcon.isNull())
            return _originalImage;

        QPainter p { &_originalImage };
        _defaultIcon.paint(&p, { QPoint(), _defaultIcon.actualSize(size) });
    }

    for (auto p: _scaledImages)
        if (p.first == size)
            return p.second;
    auto result = _originalImage.scaled(size,
                    Qt::KeepAspectRatio, Qt::SmoothTransformation);
    _scaledImages.emplace_back(size, result);
    return result;
}

QUrl Avatar::url() const { return d->_url; }

bool Avatar::updateUrl(const QUrl& newUrl)
{
    if (newUrl == d->_url)
        return false;

    // FIXME: Make it a library-wide constant and maybe even make the URL checker
    // a Connection(?) method.
    if (newUrl.scheme() != "mxc"  || newUrl.path().count('/') != 1)
    {
        qCWarning(MAIN) << "Malformed avatar URL:" << newUrl.toDisplayString();
        return false;
    }
    d->_url = newUrl;
    d->_valid = false;
    return true;
}