nheko/src/JdenticonProvider.h

// SPDX-FileCopyrightText: 2021 Nheko Contributors
//
// SPDX-License-Identifier: GPL-3.0-or-later

#pragma once

#include <QImage>
#include <QQuickAsyncImageProvider>
#include <QQuickImageResponse>
#include <QThreadPool>

#include <mtx/common.hpp>

#include "jdenticoninterface.h"

namespace Jdenticon {
JdenticonInterface *
getJdenticonInterface();
}

class JdenticonResponse
  : public QQuickImageResponse
  , public QRunnable
{
public:
        JdenticonResponse(const QString &key, bool crop, double radius, const QSize &requestedSize);

        QQuickTextureFactory *textureFactory() const override
        {
                return QQuickTextureFactory::textureFactoryForImage(m_pixmap.toImage());
        }

        void run() override;

        QString m_key;
        bool m_crop;
        double m_radius;
        QSize m_requestedSize;
        QPixmap m_pixmap;
        JdenticonInterface *jdenticonInterface_ = nullptr;
};

class JdenticonProvider
  : public QObject
  , public QQuickAsyncImageProvider
{
        Q_OBJECT

public:
        static bool isAvailable() { return Jdenticon::getJdenticonInterface() != nullptr; }

public slots:
        QQuickImageResponse *requestImageResponse(const QString &id,
                                                  const QSize &requestedSize) override
        {
                auto id_      = id;
                bool crop     = true;
                double radius = 0;

                auto queryStart = id.lastIndexOf('?');
                if (queryStart != -1) {
                        id_            = id.left(queryStart);
                        auto query     = id.midRef(queryStart + 1);
                        auto queryBits = query.split('&');

                        for (auto b : queryBits) {
                                if (b.startsWith("radius=")) {
                                        radius = b.mid(7).toDouble();
                                }
                        }
                }

                JdenticonResponse *response =
                  new JdenticonResponse(id_, crop, radius, requestedSize);
                pool.start(response);
                return response;
        }

private:
        QThreadPool pool;
};