/*
 * This file is part of smolbote. It's copyrighted by the contributors recorded
 * in the version control history of the file, available from its original
 * location: https://neueland.iserlohn-fortress.net/smolbote.hg
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#include "webpage.h"
#include <QMessageBox>
#include <QWebEngineFullScreenRequest>

#include <QLayout>

QString feature_toString(QWebEnginePage::Feature feature)
{
    switch(feature) {
    case QWebEnginePage::Notifications:
        return QObject::tr("Notifications");
    case QWebEnginePage::Geolocation:
        return QObject::tr("Geolocation");
    case QWebEnginePage::MediaAudioCapture:
        return QObject::tr("Audio Capture");
    case QWebEnginePage::MediaVideoCapture:
        return QObject::tr("Video Capture");
    case QWebEnginePage::MediaAudioVideoCapture:
        return QObject::tr("Audio and Video Capture");
    case QWebEnginePage::MouseLock:
        return QObject::tr("Mouse Lock");
    case QWebEnginePage::DesktopVideoCapture:
        return QObject::tr("Desktop Video Capture");
    case QWebEnginePage::DesktopAudioVideoCapture:
        return QObject::tr("Desktop Audio and Video Capture");
    }
}

WebPage::WebPage(QWebEngineProfile *profile, QObject *parent)
    : QWebEnginePage(profile, parent)
{
    connect(this, &WebPage::fullScreenRequested, this, [](QWebEngineFullScreenRequest request) {
        request.accept();
    });

    connect(this, &QWebEnginePage::featurePermissionRequested, this, &WebPage::featurePermissionDialog);
}

bool WebPage::certificateError(const QWebEngineCertificateError &certificateError)
{
    QMessageBox messageBox;

    messageBox.setWindowTitle(tr("SSL Error"));
    if(certificateError.isOverridable())
        messageBox.setIcon(QMessageBox::Warning);
    else
        messageBox.setIcon(QMessageBox::Critical);

    messageBox.setText(tr("An SSL error has occurred on <strong>%1</strong>").arg(certificateError.url().toString()));
    messageBox.setInformativeText(tr("<p>%1</p>"
                                     "<p>This error %2 be overridden.</p>")
                                      .arg(certificateError.errorDescription(),
                                          certificateError.isOverridable() ? tr("can") : tr("cannot")));
    messageBox.setDetailedText(tr("Error code: %1").arg(certificateError.error()));

    if(certificateError.isOverridable()) {
        messageBox.setStandardButtons(QMessageBox::Ignore | QMessageBox::Abort);
        messageBox.setDefaultButton(QMessageBox::Ignore);
    } else
        messageBox.setStandardButtons(QMessageBox::Abort);

    auto resp = messageBox.exec();

    return resp == QMessageBox::Ignore;
}

bool WebPage::acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame)
{

    /* Workaround for https://bugreports.qt.io/browse/QTBUG-68224
     * Only affects 5.11.0; should be fixed in 5.11.1
     */
#ifdef QTBUG_68224_WORKAROUND
    auto *layout = this->view()->layout();
    auto count = layout->count();

    if(count > 1) {
        for (int i = 0; i < count; ++i) {
            auto *item = layout->itemAt(i);
            if(item == nullptr)
                continue;
            auto *widget = item->widget();
            if(widget != this->view()->focusProxy()) {
                //qDebug("Removing widget");
                layout->removeWidget(widget);
            }
        }
    }
#endif

    return true;
}

void WebPage::featurePermissionDialog(const QUrl &securityOrigin, QWebEnginePage::Feature feature)
{
    QMessageBox messageBox;

    messageBox.setWindowTitle(tr("Feature permission request"));
    messageBox.setIcon(QMessageBox::Question);
    messageBox.setText(tr("<p>The webpage <strong>%1</strong> has requested permission to access: %2</p>"
                          "<p>Allow this feature?</p>")
                           .arg(securityOrigin.toString(), feature_toString(feature)));

    messageBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
    messageBox.setDefaultButton(QMessageBox::No);

    if(messageBox.exec() == QMessageBox::Yes)
        setFeaturePermission(securityOrigin, feature, QWebEnginePage::PermissionGrantedByUser);
    else
        setFeaturePermission(securityOrigin, feature, QWebEnginePage::PermissionDeniedByUser);
}