/*
 * 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 "profileview.h"
#include "ui_profileview.h"
#include <webprofile.h>
#include <QWebEngineSettings>
#include <QWebEngineCookieStore>
#include <QDateTime>

void ProfileView::connectSetting(QCheckBox *checkBox, QWebEngineSettings *settings, QWebEngineSettings::WebAttribute attr)
{
    checkBox->setChecked(settings->testAttribute(attr));
    connect(checkBox, &QCheckBox::clicked, [this, settings, attr](bool checked) {
        this->m_isChanged = true;
        settings->setAttribute(attr, checked);
    });
}

ProfileView::ProfileView(WebProfile *profile, QWidget *parent)
    : QWidget(parent)
    , ui(new Ui::ProfileView)
{
    Q_CHECK_PTR(profile);
    m_profile = profile;
    ui->setupUi(this);

    // general tab
    ui->name->setText(profile->name());
    ui->offTheRecord->setChecked(profile->isOffTheRecord());
    ui->configurationPath->setText(profile->configurationPath());

    // http tab
    ui->userAgent->setPlainText(m_profile->httpUserAgent());
    connect(ui->userAgent, &QPlainTextEdit::textChanged, profile, [=]() {
        this->m_isChanged = true;
        profile->setHttpUserAgent(ui->userAgent->toPlainText());
    });

    ui->acceptLanguage->setPlainText(m_profile->httpAcceptLanguage());
    connect(ui->acceptLanguage, &QPlainTextEdit::textChanged, profile, [=]() {
        this->m_isChanged = true;
        profile->setHttpAcceptLanguage(ui->acceptLanguage->toPlainText());
    });

    ui->cacheType->setCurrentIndex(m_profile->httpCacheType());
    connect(ui->cacheType, QOverload<int>::of(&QComboBox::currentIndexChanged), profile, [=](int index) {
        this->m_isChanged = true;
        profile->setHttpCacheType(static_cast<QWebEngineProfile::HttpCacheType>(index));
    });

    ui->cacheSize->setText(QString::number(m_profile->httpCacheMaximumSize()));
    connect(ui->cacheSize, &QLineEdit::textChanged, profile, [=](const QString &text) {
        this->m_isChanged = true;
        profile->setHttpCacheMaximumSize(text.toInt());
    });

    ui->cookiePolicy->setCurrentIndex(m_profile->persistentCookiesPolicy());
    connect(ui->cookiePolicy, QOverload<int>::of(&QComboBox::currentIndexChanged), profile, [=](int index) {
        this->m_isChanged = true;
        profile->setPersistentCookiesPolicy(static_cast<QWebEngineProfile::PersistentCookiesPolicy>(index));
    });

    connect(ui->clearCache_pushButton, &QPushButton::clicked, profile, &QWebEngineProfile::clearHttpCache);
    connect(ui->clearHistory_pushButton, &QPushButton::clicked, profile, &QWebEngineProfile::clearAllVisitedLinks);

    ui->storagePath_lineEdit->setText(m_profile->persistentStoragePath());
    ui->cachePath_lineEdit->setText(m_profile->cachePath());

    // settings tab
    loadSettings(profile->settings());

    // cookies tab
    loadCookies(profile->cookieStore());
    for(const auto &c : profile->cookies()) {
        cookieAdded(c);
    }
}

ProfileView::~ProfileView()
{
    delete ui;
}

bool ProfileView::isChanged() const
{
    return m_isChanged;
}

void ProfileView::loadSettings(QWebEngineSettings *settings)
{
    // settings
    connectSetting(ui->autoloadImages, settings, QWebEngineSettings::AutoLoadImages);
    connectSetting(ui->autoloadIcons, settings, QWebEngineSettings::AutoLoadIconsForPage);

    connectSetting(ui->javascriptEnabled, settings, QWebEngineSettings::JavascriptEnabled);
    connectSetting(ui->javascriptCanAccessClipboard, settings, QWebEngineSettings::JavascriptCanAccessClipboard);
    connectSetting(ui->javascriptCanOpenWindows, settings, QWebEngineSettings::JavascriptCanOpenWindows);
    connectSetting(ui->javascriptCanActivateWindows, settings, QWebEngineSettings::AllowWindowActivationFromJavaScript);

    connectSetting(ui->linksIncludedInFocusChain, settings, QWebEngineSettings::LinksIncludedInFocusChain);
    connectSetting(ui->spatialNavigationEnabled, settings, QWebEngineSettings::SpatialNavigationEnabled);
    connectSetting(ui->focusOnNavigationEnabled, settings, QWebEngineSettings::FocusOnNavigationEnabled);
    connectSetting(ui->touchIconsEnabled, settings, QWebEngineSettings::TouchIconsEnabled);

    connectSetting(ui->showScrollBars, settings, QWebEngineSettings::ShowScrollBars);
    connectSetting(ui->scrollAnimatorEnabled, settings, QWebEngineSettings::ScrollAnimatorEnabled);
    connectSetting(ui->errorPagesEnabled, settings, QWebEngineSettings::ErrorPageEnabled);

    connectSetting(ui->localContentCanAccessRemoteUrls, settings, QWebEngineSettings::LocalContentCanAccessRemoteUrls);
    connectSetting(ui->localContentCanAccessFileUrls, settings, QWebEngineSettings::LocalContentCanAccessFileUrls);

    connectSetting(ui->localStorageEnabled, settings, QWebEngineSettings::LocalStorageEnabled);
    connectSetting(ui->webglEnabled, settings, QWebEngineSettings::WebGLEnabled);
    connectSetting(ui->canvasEnabled, settings, QWebEngineSettings::Accelerated2dCanvasEnabled);

    connectSetting(ui->xssAuditingEnabled, settings, QWebEngineSettings::XSSAuditingEnabled);
    connectSetting(ui->hyperlinkAuditingEnabled, settings, QWebEngineSettings::HyperlinkAuditingEnabled);

    connectSetting(ui->allowRunningInsecureContent, settings, QWebEngineSettings::AllowRunningInsecureContent);
    connectSetting(ui->allowGeolocationOnInsecureContent, settings, QWebEngineSettings::AllowGeolocationOnInsecureOrigins);
    connectSetting(ui->ppapiPluginsEnabled, settings, QWebEngineSettings::PluginsEnabled);
    connectSetting(ui->fullscreenSupportEnabled, settings, QWebEngineSettings::FullScreenSupportEnabled);
    connectSetting(ui->screenCaptureEnabled, settings, QWebEngineSettings::ScreenCaptureEnabled);
    connectSetting(ui->printElementBackgrounds, settings, QWebEngineSettings::PrintElementBackgrounds);
}

void ProfileView::loadCookies(QWebEngineCookieStore *store)
{
    //
    connect(store, &QWebEngineCookieStore::cookieAdded, this, &ProfileView::cookieAdded);
    connect(store, &QWebEngineCookieStore::cookieRemoved, this, &ProfileView::cookieRemoved);

    connect(ui->cookies_reload, &QPushButton::clicked, store, [=]() {
        ui->cookies->clearContents();
        ui->cookies->setRowCount(0);
        store->loadAllCookies();
    });

    connect(ui->cookies_delete, &QPushButton::clicked, store, [=]() {
        for(auto &list : ui->cookies->selectedRanges()) {
            for(int i = list.bottomRow(); i >= list.topRow(); --i) {
                auto cookie = ui->cookies->item(i, 0)->data(Qt::UserRole).value<QNetworkCookie>();
                store->deleteCookie(cookie);
            }
        }
    });
    connect(ui->cookies_deleteSession, &QPushButton::clicked, store, &QWebEngineCookieStore::deleteSessionCookies);
    connect(ui->cookies_deleteAll, &QPushButton::clicked, store, &QWebEngineCookieStore::deleteAllCookies);
}

void ProfileView::cookieAdded(const QNetworkCookie &cookie)
{
    auto index = ui->cookies->rowCount();
    ui->cookies->setRowCount(index + 1);

    auto *item = new QTableWidgetItem(QString(cookie.name()));
    item->setData(Qt::UserRole, QVariant::fromValue(cookie));
    ui->cookies->setItem(index, 0, item);
    ui->cookies->setItem(index, 1, new QTableWidgetItem(cookie.domain()));
    ui->cookies->setItem(index, 2, new QTableWidgetItem(cookie.path()));
    if(cookie.isSessionCookie())
        ui->cookies->setItem(index, 3, new QTableWidgetItem(tr("session")));
    else
        ui->cookies->setItem(index, 3, new QTableWidgetItem(cookie.expirationDate().toString(Qt::RFC2822Date)));
}

void ProfileView::cookieRemoved(const QNetworkCookie &cookie)
{
    for(int i = 0; i < ui->cookies->rowCount(); ++i) {
        auto *item = ui->cookies->item(i, 0);
        if(item->data(Qt::UserRole).value<QNetworkCookie>() == cookie) {
            //qDebug("removing cookie on row %i", i);
            ui->cookies->removeRow(i);
            break;
        }
    }
}