#include "StatusQ/QClipboardProxy.h"

#include <QBuffer>
#include <QClipboard>
#include <QGuiApplication>
#include <QImage>
#include <QMimeData>
#include <QUrl>
#include <QFile>

#include <algorithm>

QClipboardProxy::QClipboardProxy()
    : m_clipboard(QGuiApplication::clipboard())
{
    connect(m_clipboard, &QClipboard::changed, this, [this](QClipboard::Mode mode) {
        if (mode == QClipboard::Clipboard)
            emit contentChanged();
    });
}

bool QClipboardProxy::hasText() const
{
    return m_clipboard->mimeData()->hasText();
}

QString QClipboardProxy::text() const
{
    return m_clipboard->text();
}

bool QClipboardProxy::hasHtml() const
{
    return m_clipboard->mimeData()->hasHtml();
}

QString QClipboardProxy::html() const
{
    auto mimeData = m_clipboard->mimeData();
    return mimeData ? mimeData->html() : QString{};
}

bool QClipboardProxy::hasImage() const
{
    return m_clipboard->mimeData()->hasImage();
}

QImage QClipboardProxy::image() const
{
    return m_clipboard->image();
}

QString QClipboardProxy::imageBase64() const
{
    if (!hasImage())
        return {};

    const auto img = image();
    QByteArray byteArray;
    QBuffer buffer(&byteArray);
    img.save(&buffer, "JPG");
    return QByteArrayLiteral("data:image/jpeg;base64,") + byteArray.toBase64();
}

bool QClipboardProxy::hasUrls() const
{
    return m_clipboard->mimeData()->hasUrls();
}

QList<QUrl> QClipboardProxy::urls() const
{
    return m_clipboard->mimeData()->urls();
}

bool QClipboardProxy::isValidImageUrl(const QUrl& url, const QStringList& acceptedExtensions) const
{
    const auto strippedUrl = url.url(QUrl::RemoveAuthority | QUrl::RemoveFragment | QUrl::RemoveQuery);
    return std::any_of(acceptedExtensions.constBegin(), acceptedExtensions.constEnd(), [strippedUrl](const auto & ext) {
        return strippedUrl.endsWith(ext, Qt::CaseInsensitive);
    });
}

qint64 QClipboardProxy::getFileSize(const QUrl& url) const
{
    if (url.isLocalFile()) {
        return QFile(url.toLocalFile()).size();
    }
    return 0;
}

void QClipboardProxy::copyTextToClipboard(const QString &text)
{
    m_clipboard->setText(text);
}

void QClipboardProxy::clear()
{
    m_clipboard->clear();
}