Attachment #578010 for bug #686842

View | Details | Raw Unified | Return to bug 686842
Collapse All | Expand All




/*
   Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public License
   along with this library; see the file COPYING.LIB.  If not, write to
   the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/
#include "mailwebenginepage.h"
#include <QFontDatabase>
#include <QWebEngineSettings>
#include <QWebEngineProfile>

using namespace MessageViewer;

MailWebEnginePage::MailWebEnginePage(QObject *parent)
    : WebEngineViewer::WebEnginePage(parent)
{
    initialize();
}

MailWebEnginePage::MailWebEnginePage(QWebEngineProfile *profile, QObject *parent)
    : WebEngineViewer::WebEnginePage(profile, parent)
{
    initialize();
}

MailWebEnginePage::~MailWebEnginePage()
{
}

void MailWebEnginePage::initialize()
{
    settings()->setAttribute(QWebEngineSettings::JavascriptEnabled, false);
    settings()->setAttribute(QWebEngineSettings::PluginsEnabled, false);
    settings()->setAttribute(QWebEngineSettings::JavascriptCanOpenWindows, false);
    settings()->setAttribute(QWebEngineSettings::JavascriptCanAccessClipboard, false);
    settings()->setAttribute(QWebEngineSettings::LocalStorageEnabled, false);
    settings()->setAttribute(QWebEngineSettings::XSSAuditingEnabled, false);
    settings()->setAttribute(QWebEngineSettings::LocalContentCanAccessRemoteUrls, false);
    settings()->setAttribute(QWebEngineSettings::LocalContentCanAccessFileUrls, false);
    settings()->setAttribute(QWebEngineSettings::ErrorPageEnabled, false);
    settings()->setAttribute(QWebEngineSettings::HyperlinkAuditingEnabled, false);
    settings()->setAttribute(QWebEngineSettings::FullScreenSupportEnabled, false);
    settings()->setAttribute(QWebEngineSettings::WebGLEnabled, false);
    settings()->setAttribute(QWebEngineSettings::AutoLoadIconsForPage, false);
    settings()->setAttribute(QWebEngineSettings::Accelerated2dCanvasEnabled, false);
    settings()->setAttribute(QWebEngineSettings::WebGLEnabled, false);

    settings()->setAttribute(QWebEngineSettings::FocusOnNavigationEnabled, false);
    settings()->setAttribute(QWebEngineSettings::AllowRunningInsecureContent, false);
    profile()->setPersistentCookiesPolicy(QWebEngineProfile::NoPersistentCookies);
    profile()->setHttpCacheType(QWebEngineProfile::MemoryHttpCache);

    const QFontInfo font(QFontDatabase().systemFont(QFontDatabase::GeneralFont));
    settings()->setFontFamily(QWebEngineSettings::StandardFont, font.family());
    settings()->setFontSize(QWebEngineSettings::DefaultFontSize, font.pixelSize());
    connect(this, &QWebEnginePage::featurePermissionRequested,
            this, &MailWebEnginePage::slotFeaturePermissionRequested);
}

void MailWebEnginePage::setPrintElementBackground(bool printElementBackground)
{
    settings()->setAttribute(QWebEngineSettings::PrintElementBackgrounds, printElementBackground);
}

void MailWebEnginePage::slotFeaturePermissionRequested(const QUrl &url, QWebEnginePage::Feature feature)
{
    //Denied all permissions.
    setFeaturePermission(url, feature, QWebEnginePage::PermissionDeniedByUser);
}




/*
   Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public License
   along with this library; see the file COPYING.LIB.  If not, write to
   the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/
#ifndef MAILWEBENGINEPAGE_H
#define MAILWEBENGINEPAGE_H

#include "messageviewer_export.h"
#include <WebEngineViewer/WebEnginePage>

namespace MessageViewer {
class MESSAGEVIEWER_EXPORT MailWebEnginePage : public WebEngineViewer::WebEnginePage
{
    Q_OBJECT
public:
    explicit MailWebEnginePage(QObject *parent = nullptr);
    virtual ~MailWebEnginePage() = default;
    ~MailWebEnginePage();

    void setPrintElementBackground(bool printElementBackground);

private:
    void slotFeaturePermissionRequested(const QUrl &url, QWebEnginePage::Feature feature);
    void initialize();
};
}
#endif // MAILWEBENGINEPAGE_H




/*
   Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public License
   along with this library; see the file COPYING.LIB.  If not, write to
   the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/
#include "mailwebengineview.h"
#include "mailwebenginepage.h"
#include "webengineviewer/webengineaccesskey.h"
#include "webengineviewer/webenginescript.h"
#include "messageviewer/messageviewersettings.h"
#include "../urlhandlermanager.h"
#include "loadexternalreferencesurlinterceptor/loadexternalreferencesurlinterceptor.h"
#include "blockexternalresourcesurlinterceptor/blockexternalresourcesurlinterceptor.h"
#include "blockmailtrackingurlinterceptor/blockmailtrackingurlinterceptor.h"
#include "cidreferencesurlinterceptor/cidreferencesurlinterceptor.h"
#include <WebEngineViewer/InterceptorManager>
#include <WebEngineViewer/WebEngineManageScript>

#include "scamdetection/scamdetectionwebengine.h"
#include "scamdetection/scamcheckshorturl.h"
#include <QContextMenuEvent>
#include <WebEngineViewer/WebHitTest>

#include <QWebEngineProfile>
#include <QPrinter>

#include <WebEngineViewer/WebHitTestResult>

using namespace MessageViewer;
template<typename Arg, typename R, typename C>
struct InvokeWrapper {
    R *receiver;
    void (C::*memberFunction)(Arg);
    void operator()(Arg result)
    {
        (receiver->*memberFunction)(result);
    }
};

template<typename Arg, typename R, typename C>

InvokeWrapper<Arg, R, C> invoke(R *receiver, void (C::*memberFunction)(Arg))
{
    InvokeWrapper<Arg, R, C> wrapper = {receiver, memberFunction};
    return wrapper;
}

class MessageViewer::MailWebEngineViewPrivate
{
public:
    MailWebEngineViewPrivate()
    {
    }

    QUrl mHoveredUrl;
    QPoint mLastClickPosition;
    ScamDetectionWebEngine *mScamDetection = nullptr;
    WebEngineViewer::WebEngineAccessKey *mWebViewAccessKey = nullptr;
    MessageViewer::LoadExternalReferencesUrlInterceptor *mExternalReference = nullptr;
    MailWebEnginePage *mPageEngine = nullptr;
    WebEngineViewer::InterceptorManager *mNetworkAccessManager = nullptr;
    MessageViewer::ViewerPrivate *mViewer = nullptr;
    MessageViewer::BlockMailTrackingUrlInterceptor *mBlockMailTrackingUrl = nullptr;
    bool mCanStartDrag = false;
};

MailWebEngineView::MailWebEngineView(KActionCollection *ac, QWidget *parent)
    : WebEngineViewer::WebEngineView(parent)
    , d(new MessageViewer::MailWebEngineViewPrivate)
{
    d->mPageEngine = new MailWebEnginePage(this);
    setPage(d->mPageEngine);
    d->mWebViewAccessKey = new WebEngineViewer::WebEngineAccessKey(this, this);
    d->mWebViewAccessKey->setActionCollection(ac);
    d->mScamDetection = new ScamDetectionWebEngine(this);
    connect(d->mScamDetection, &ScamDetectionWebEngine::messageMayBeAScam, this,
            &MailWebEngineView::messageMayBeAScam);
    connect(d->mWebViewAccessKey, &WebEngineViewer::WebEngineAccessKey::openUrl, this,
            &MailWebEngineView::openUrl);
    connect(this, &MailWebEngineView::loadFinished, this, &MailWebEngineView::slotLoadFinished);

    d->mNetworkAccessManager = new WebEngineViewer::InterceptorManager(this, ac, this);
    d->mExternalReference = new MessageViewer::LoadExternalReferencesUrlInterceptor(this);
    d->mNetworkAccessManager->addInterceptor(d->mExternalReference);
    MessageViewer::CidReferencesUrlInterceptor *cidReference
        = new MessageViewer::CidReferencesUrlInterceptor(this);
    d->mNetworkAccessManager->addInterceptor(cidReference);
    MessageViewer::BlockExternalResourcesUrlInterceptor *blockExternalUrl
        = new MessageViewer::BlockExternalResourcesUrlInterceptor(this);
    connect(blockExternalUrl, &BlockExternalResourcesUrlInterceptor::formSubmittedForbidden, this,
            &MailWebEngineView::formSubmittedForbidden);
    d->mNetworkAccessManager->addInterceptor(blockExternalUrl);

    d->mBlockMailTrackingUrl
        = new MessageViewer::BlockMailTrackingUrlInterceptor(this);
    connect(d->mBlockMailTrackingUrl, &BlockMailTrackingUrlInterceptor::mailTrackingFound, this,
            &MailWebEngineView::mailTrackingFound);

    setFocusPolicy(Qt::WheelFocus);
    connect(d->mPageEngine, &MailWebEnginePage::urlClicked, this, &MailWebEngineView::openUrl);
    connect(
        page(), &QWebEnginePage::scrollPositionChanged, d->mWebViewAccessKey,
        &WebEngineViewer::WebEngineAccessKey::hideAccessKeys);
    initializeScripts();
}

MailWebEngineView::~MailWebEngineView()
{
    delete d;
}

void MailWebEngineView::readConfig()
{
    if (MessageViewer::MessageViewerSettings::self()->mailTrackingUrlEnabled()) {
        d->mNetworkAccessManager->addInterceptor(d->mBlockMailTrackingUrl);
    } else {
        d->mNetworkAccessManager->removeInterceptor(d->mBlockMailTrackingUrl);
    }
}

void MailWebEngineView::setLinkHovered(const QUrl &url)
{
    //TODO we need to detect image url too.
    d->mHoveredUrl = url;
}

void MailWebEngineView::runJavaScriptInWordId(const QString &script)
{
    page()->runJavaScript(script, WebEngineViewer::WebEngineManageScript::scriptWordId());
}

void MailWebEngineView::setViewer(MessageViewer::ViewerPrivate *viewer)
{
    d->mViewer = viewer;
}

void MailWebEngineView::initializeScripts()
{
    initializeJQueryScript();
}

void MailWebEngineView::contextMenuEvent(QContextMenuEvent *e)
{
    WebEngineViewer::WebHitTest *webHit = d->mPageEngine->hitTestContent(e->pos());
    connect(webHit, &WebEngineViewer::WebHitTest::finished, this,
            &MailWebEngineView::slotWebHitFinished);
}

void MailWebEngineView::slotWebHitFinished(const WebEngineViewer::WebHitTestResult &result)
{
    Q_EMIT popupMenu(result);
}

void MailWebEngineView::scrollUp(int pixels)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::scrollUp(pixels));
}

void MailWebEngineView::scrollDown(int pixels)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::scrollDown(pixels));
}

void MailWebEngineView::selectAll()
{
    page()->triggerAction(QWebEnginePage::SelectAll);
}

void MailWebEngineView::slotZoomChanged(qreal zoom)
{
    setZoomFactor(zoom);
}

void MailWebEngineView::scamCheck()
{
    d->mScamDetection->scanPage(page());
}

void MailWebEngineView::slotShowDetails()
{
    d->mScamDetection->showDetails();
}

void MailWebEngineView::forwardKeyReleaseEvent(QKeyEvent *e)
{
    if (MessageViewer::MessageViewerSettings::self()->accessKeyEnabled()) {
        d->mWebViewAccessKey->keyReleaseEvent(e);
    }
}

void MailWebEngineView::forwardMousePressEvent(QMouseEvent *event)
{
    if (d->mViewer && !d->mHoveredUrl.isEmpty()) {
        if (event->button() == Qt::LeftButton && (event->modifiers() & Qt::ShiftModifier)) {
            // special processing for shift+click
            if (URLHandlerManager::instance()->handleShiftClick(d->mHoveredUrl, d->mViewer)) {
                event->accept();
                return;
            }
        }
        if (event->button() == Qt::LeftButton) {
            d->mCanStartDrag = URLHandlerManager::instance()->willHandleDrag(d->mHoveredUrl,
                                                                             d->mViewer);
            d->mLastClickPosition = event->pos();
        }
    }
}

void MailWebEngineView::forwardMouseMoveEvent(QMouseEvent *event)
{
    if (d->mViewer && !d->mHoveredUrl.isEmpty()) {
        // If we are potentially handling a drag, deal with that.
        if (d->mCanStartDrag && (event->buttons() & Qt::LeftButton)) {
            if ((d->mLastClickPosition - event->pos()).manhattanLength()
                > QApplication::startDragDistance()) {
                if (URLHandlerManager::instance()->handleDrag(d->mHoveredUrl, d->mViewer)) {
                    // If the URL handler manager started a drag, don't handle this in the future
                    d->mCanStartDrag = false;
                }
            }
            event->accept();
        }
    }
}

void MailWebEngineView::forwardMouseReleaseEvent(QMouseEvent *event)
{
    Q_UNUSED(event);
    d->mCanStartDrag = false;
}

void MailWebEngineView::forwardKeyPressEvent(QKeyEvent *e)
{
    if (e && hasFocus()) {
        if (MessageViewer::MessageViewerSettings::self()->accessKeyEnabled()) {
            d->mWebViewAccessKey->keyPressEvent(e);
        }
    }
}

void MailWebEngineView::forwardWheelEvent(QWheelEvent *e)
{
    if (MessageViewer::MessageViewerSettings::self()->accessKeyEnabled()) {
        d->mWebViewAccessKey->wheelEvent(e);
    }
    if (QApplication::keyboardModifiers() & Qt::ControlModifier) {
        const int numDegrees = e->delta() / 8;
        const int numSteps = numDegrees / 15;
        Q_EMIT wheelZoomChanged(numSteps);
        e->accept();
    }
}

void MailWebEngineView::resizeEvent(QResizeEvent *e)
{
    if (MessageViewer::MessageViewerSettings::self()->accessKeyEnabled()) {
        d->mWebViewAccessKey->resizeEvent(e);
    }
    QWebEngineView::resizeEvent(e);
}

void MailWebEngineView::saveMainFrameScreenshotInFile(const QString &filename)
{
    //TODO need to verify it
    QImage image(size(), QImage::Format_ARGB32_Premultiplied);
    image.fill(Qt::transparent);

    QPainter painter(&image);
    painter.setRenderHint(QPainter::Antialiasing, true);
    painter.setRenderHint(QPainter::TextAntialiasing, true);
    painter.setRenderHint(QPainter::SmoothPixmapTransform, true);
    render(&painter);
    painter.end();
    image.save(filename);
}

void MailWebEngineView::showAccessKeys()
{
    d->mWebViewAccessKey->showAccessKeys();
}

void MailWebEngineView::hideAccessKeys()
{
    d->mWebViewAccessKey->hideAccessKeys();
}

void MailWebEngineView::isScrolledToBottom()
{
    page()->runJavaScript(WebEngineViewer::WebEngineScript::isScrolledToBottom(),
                          WebEngineViewer::WebEngineManageScript::scriptWordId(),
                          invoke(this, &MailWebEngineView::handleIsScrolledToBottom));
}

void MailWebEngineView::setElementByIdVisible(const QString &id, bool visible)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::setElementByIdVisible(id, visible));
}

void MailWebEngineView::removeAttachmentMarking(const QString &id)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::removeStyleToElement(id));
}

void MailWebEngineView::markAttachment(const QString &id, const QString &style)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::setStyleToElement(id, style));
}

void MailWebEngineView::scrollToAnchor(const QString &anchor)
{
    page()->runJavaScript(WebEngineViewer::WebEngineScript::searchElementPosition(anchor),
                          WebEngineViewer::WebEngineManageScript::scriptWordId(),
                          invoke(this, &MailWebEngineView::handleScrollToAnchor));
}

void MailWebEngineView::handleIsScrolledToBottom(const QVariant &result)
{
    bool scrolledToBottomResult = false;
    if (result.isValid()) {
        scrolledToBottomResult = result.toBool();
    }
    Q_EMIT pageIsScrolledToBottom(scrolledToBottomResult);
}

void MailWebEngineView::handleScrollToAnchor(const QVariant &result)
{
    if (result.isValid()) {
        const QList<QVariant> lst = result.toList();
        if (lst.count() == 2) {
            const QPoint pos(lst.at(0).toInt(), lst.at(1).toInt());
            runJavaScriptInWordId(WebEngineViewer::WebEngineScript::scrollToPosition(pos));
        }
    }
}

void MailWebEngineView::scrollPageDown(int percent)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::scrollPercentage(percent));
}

void MailWebEngineView::scrollPageUp(int percent)
{
    scrollPageDown(-percent);
}

void MailWebEngineView::scrollToRelativePosition(qreal pos)
{
    runJavaScriptInWordId(WebEngineViewer::WebEngineScript::scrollToRelativePosition(pos));
}

void MailWebEngineView::setAllowExternalContent(bool b)
{
    if (d->mExternalReference->allowExternalContent() != b) {
        d->mExternalReference->setAllowExternalContent(b);
        reload();
    }
}

QList<QAction *> MailWebEngineView::interceptorUrlActions(
    const WebEngineViewer::WebHitTestResult &result) const
{
    return d->mNetworkAccessManager->interceptorUrlActions(result);
}

void MailWebEngineView::slotLoadFinished()
{
    scamCheck();
}

void MailWebEngineView::setPrintElementBackground(bool printElementBackground)
{
    d->mPageEngine->setPrintElementBackground(printElementBackground);
}

bool MailWebEngineView::execPrintPreviewPage(QPrinter *printer, int timeout)
{
    return d->mPageEngine->execPrintPreviewPage(printer, timeout);
}




/*
   Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public License
   along with this library; see the file COPYING.LIB.  If not, write to
   the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/

#include "webenginepage.h"
#include "webhittest.h"
#include "webhittestresult.h"

#include <KLocalizedString>

#include <QEventLoop>
#include <QWebEngineDownloadItem>
#include <QPointer>
#include <QTimer>
#include <QFileDialog>
#include <QWebEngineProfile>
#include <QPrinter>
using namespace WebEngineViewer;

WebEnginePage::WebEnginePage(QObject *parent)
    : QWebEnginePage(new QWebEngineProfile, parent)
{
    // Create a private (off the record) QWebEngineProfile here to isolate the
    // browsing settings, and adopt it as a child so that it will be deleted
    // when we are destroyed.  The profile must remain active for as long as
    // any QWebEnginePage's belonging to it exist, see the API documentation
    // of QWebEnginePage::QWebEnginePage(QWebEngineProfile *, QObject *).
    // Deleting it as our child on destruction is safe.
    //
    // Do not try to save a line of code by setting the parent on construction:
    //
    //    WebEnginePage::WebEnginePage(QObject *parent)
    //      : QWebEnginePage(new QWebEngineProfile(this), parent)
    //
    // because the QWebEngineProfile constructor will call out to the QWebEnginePage
    // and crash because the QWebEnginePage is not fully constructed yet.
    profile()->setParent(this);

    init();
}

WebEnginePage::WebEnginePage(QWebEngineProfile *profile, QObject *parent)
    : QWebEnginePage(profile, parent)
{
    init();
}

WebEnginePage::~WebEnginePage()
{
}

void WebEnginePage::init()
{
    connect(profile(), &QWebEngineProfile::downloadRequested, this, &WebEnginePage::saveHtml);
}

WebEngineViewer::WebHitTest *WebEnginePage::hitTestContent(const QPoint &pos)
{
    return new WebHitTest(this, pos);
}

void WebEnginePage::saveHtml(QWebEngineDownloadItem *download)
{
    const QString fileName = QFileDialog::getSaveFileName(view(), i18n("Save HTML Page"));
    if (!fileName.isEmpty()) {
        download->setSavePageFormat(QWebEngineDownloadItem::SingleHtmlSaveFormat);
        download->setPath(fileName);
        download->accept();
    }
}

bool WebEnginePage::acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame)
{
    if (isMainFrame && type == NavigationTypeLinkClicked) {
        Q_EMIT urlClicked(url);
        return false;
    }
    return true;
}

void WebEnginePage::javaScriptConsoleMessage(QWebEnginePage::JavaScriptConsoleMessageLevel level, const QString &message, int lineNumber, const QString &sourceID)
{
    Q_UNUSED(level);
    Q_UNUSED(sourceID);
    //Don't convert to debug categories
    qDebug() << "WebEnginePage::javaScriptConsoleMessage lineNumber: " << lineNumber << " message: " << message;
    Q_EMIT showConsoleMessage(message);
}

bool WebEnginePage::execPrintPreviewPage(QPrinter *printer, int timeout)
{
    QPointer<QEventLoop> loop = new QEventLoop;
    bool result = false;
    QTimer::singleShot(timeout, loop.data(), &QEventLoop::quit);

    print(printer, [loop, &result](bool res) {
        if (loop && loop->isRunning()) {
            result = res;
            loop->quit();
        }
    });

    loop->exec();
    delete loop;

    return result;
}




/*
   Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public License
   along with this library; see the file COPYING.LIB.  If not, write to
   the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/

#ifndef WEBENGINEPAGE_H
#define WEBENGINEPAGE_H

#include <QWebEnginePage>
#include "webengineviewer_export.h"
class QWebEngineProfile;
class QWebEngineDownloadItem;
class QPrinter;
namespace WebEngineViewer {
class WebHitTest;
class WEBENGINEVIEWER_EXPORT WebEnginePage : public QWebEnginePage
{
    Q_OBJECT
public:
    /**
     * Constructor.
     *
     * A private QWebEngineProfile, only applying to this QWebEnginePage,
     * will be created to implement browser settings.  It can be accessed via
     * @c profile(), but it should not be shared or reused unless care is
     * taken that the profile is not deleted until all of the QWebEnginePage's
     * belonging to it are deleted first.
     *
     * @param parent The parent object
     **/
    explicit WebEnginePage(QObject *parent = nullptr);
    explicit WebEnginePage(QWebEngineProfile *profile, QObject *parent = nullptr);

    /**
     * Constructor.
     *
     * The specified QWebEngineProfile will be used.  See the description of
     * @c WebEnginePage(QObject *) and the API documentation of QWebEnginePage
     * for caution regarding the lifetime of the profile.
     *
     * @param profile The profile to be used
     * @param parent The parent object
     * @deprecated Use the single argument constructor, which creates and uses
     * a private profile.
     **/
#ifndef WEBENGINEVIEWER_NO_DEPRECATED
    explicit WEBENGINEVIEWER_DEPRECATED WebEnginePage(QWebEngineProfile *profile, QObject *parent = nullptr);
#endif

    /**
     * Destructor.  If there is a private QWebEngineProfile then it will also
     * be destroyed.
     **/
    virtual ~WebEnginePage() override = default;

    WebEngineViewer::WebHitTest *hitTestContent(const QPoint &pos);

    void saveHtml(QWebEngineDownloadItem *download);
    Q_REQUIRED_RESULT bool execPrintPreviewPage(QPrinter *printer, int timeout);

Q_SIGNALS:
    void urlClicked(const QUrl &url);
    void showConsoleMessage(const QString &message);

protected:
    bool acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame) override;
    void javaScriptConsoleMessage(JavaScriptConsoleMessageLevel level, const QString &message, int lineNumber, const QString &sourceID) override;

private:
    void init();
};
}

#endif // WEBENGINEPAGE_H

Return to bug 686842

Lines 1-81 Link Here

(-)a/messageviewer/src/viewer/webengine/mailwebenginepage.cpp (-10 lines)
1	/*	1	/*
2	Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>	2	Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>
3		3
4	This library is free software; you can redistribute it and/or	4	This library is free software; you can redistribute it and/or
5	modify it under the terms of the GNU Library General Public	5	modify it under the terms of the GNU Library General Public
6	License as published by the Free Software Foundation; either	6	License as published by the Free Software Foundation; either
7	version 2 of the License, or (at your option) any later version.	7	version 2 of the License, or (at your option) any later version.
8		8
9	This library is distributed in the hope that it will be useful,	9	This library is distributed in the hope that it will be useful,
10	but WITHOUT ANY WARRANTY; without even the implied warranty of	10	but WITHOUT ANY WARRANTY; without even the implied warranty of
11	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU	11	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12	Library General Public License for more details.	12	Library General Public License for more details.
13		13
14	You should have received a copy of the GNU Library General Public License	14	You should have received a copy of the GNU Library General Public License
15	along with this library; see the file COPYING.LIB. If not, write to	15	along with this library; see the file COPYING.LIB. If not, write to
16	the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,	16	the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
17	Boston, MA 02110-1301, USA.	17	Boston, MA 02110-1301, USA.
18	*/	18	*/
19	#include "mailwebenginepage.h"	19	#include "mailwebenginepage.h"
20	#include <QFontDatabase>	20	#include <QFontDatabase>
21	#include <QWebEngineSettings>	21	#include <QWebEngineSettings>
22	#include <QWebEngineProfile>	22	#include <QWebEngineProfile>
23		23
24	using namespace MessageViewer;	24	using namespace MessageViewer;
25		25
26	MailWebEnginePage::MailWebEnginePage(QObject *parent)	26	MailWebEnginePage::MailWebEnginePage(QObject *parent)
27	: WebEngineViewer::WebEnginePage(parent)	27	: WebEngineViewer::WebEnginePage(parent)
28	{	28	{
29	initialize();	29	initialize();
30	}	30	}
31		31
32	MailWebEnginePage::MailWebEnginePage(QWebEngineProfile profile, QObject parent)
33	: WebEngineViewer::WebEnginePage(profile, parent)
34	{
35	initialize();
36	}
37
38	MailWebEnginePage::~MailWebEnginePage()
39	{
40	}
41
42	void MailWebEnginePage::initialize()	32	void MailWebEnginePage::initialize()
43	{	33	{
44	settings()->setAttribute(QWebEngineSettings::JavascriptEnabled, false);	34	settings()->setAttribute(QWebEngineSettings::JavascriptEnabled, false);
45	settings()->setAttribute(QWebEngineSettings::PluginsEnabled, false);	35	settings()->setAttribute(QWebEngineSettings::PluginsEnabled, false);
46	settings()->setAttribute(QWebEngineSettings::JavascriptCanOpenWindows, false);	36	settings()->setAttribute(QWebEngineSettings::JavascriptCanOpenWindows, false);
47	settings()->setAttribute(QWebEngineSettings::JavascriptCanAccessClipboard, false);	37	settings()->setAttribute(QWebEngineSettings::JavascriptCanAccessClipboard, false);
48	settings()->setAttribute(QWebEngineSettings::LocalStorageEnabled, false);	38	settings()->setAttribute(QWebEngineSettings::LocalStorageEnabled, false);
49	settings()->setAttribute(QWebEngineSettings::XSSAuditingEnabled, false);	39	settings()->setAttribute(QWebEngineSettings::XSSAuditingEnabled, false);
50	settings()->setAttribute(QWebEngineSettings::LocalContentCanAccessRemoteUrls, false);	40	settings()->setAttribute(QWebEngineSettings::LocalContentCanAccessRemoteUrls, false);
51	settings()->setAttribute(QWebEngineSettings::LocalContentCanAccessFileUrls, false);	41	settings()->setAttribute(QWebEngineSettings::LocalContentCanAccessFileUrls, false);
52	settings()->setAttribute(QWebEngineSettings::ErrorPageEnabled, false);	42	settings()->setAttribute(QWebEngineSettings::ErrorPageEnabled, false);
53	settings()->setAttribute(QWebEngineSettings::HyperlinkAuditingEnabled, false);	43	settings()->setAttribute(QWebEngineSettings::HyperlinkAuditingEnabled, false);
54	settings()->setAttribute(QWebEngineSettings::FullScreenSupportEnabled, false);	44	settings()->setAttribute(QWebEngineSettings::FullScreenSupportEnabled, false);
55	settings()->setAttribute(QWebEngineSettings::WebGLEnabled, false);	45	settings()->setAttribute(QWebEngineSettings::WebGLEnabled, false);
56	settings()->setAttribute(QWebEngineSettings::AutoLoadIconsForPage, false);	46	settings()->setAttribute(QWebEngineSettings::AutoLoadIconsForPage, false);
57	settings()->setAttribute(QWebEngineSettings::Accelerated2dCanvasEnabled, false);	47	settings()->setAttribute(QWebEngineSettings::Accelerated2dCanvasEnabled, false);
58	settings()->setAttribute(QWebEngineSettings::WebGLEnabled, false);	48	settings()->setAttribute(QWebEngineSettings::WebGLEnabled, false);
59		49
60	settings()->setAttribute(QWebEngineSettings::FocusOnNavigationEnabled, false);	50	settings()->setAttribute(QWebEngineSettings::FocusOnNavigationEnabled, false);
61	settings()->setAttribute(QWebEngineSettings::AllowRunningInsecureContent, false);	51	settings()->setAttribute(QWebEngineSettings::AllowRunningInsecureContent, false);
62	profile()->setPersistentCookiesPolicy(QWebEngineProfile::NoPersistentCookies);	52	profile()->setPersistentCookiesPolicy(QWebEngineProfile::NoPersistentCookies);
63	profile()->setHttpCacheType(QWebEngineProfile::MemoryHttpCache);	53	profile()->setHttpCacheType(QWebEngineProfile::MemoryHttpCache);
64		54
65	const QFontInfo font(QFontDatabase().systemFont(QFontDatabase::GeneralFont));	55	const QFontInfo font(QFontDatabase().systemFont(QFontDatabase::GeneralFont));
66	settings()->setFontFamily(QWebEngineSettings::StandardFont, font.family());	56	settings()->setFontFamily(QWebEngineSettings::StandardFont, font.family());
67	settings()->setFontSize(QWebEngineSettings::DefaultFontSize, font.pixelSize());	57	settings()->setFontSize(QWebEngineSettings::DefaultFontSize, font.pixelSize());
68	connect(this, &QWebEnginePage::featurePermissionRequested,	58	connect(this, &QWebEnginePage::featurePermissionRequested,
69	this, &MailWebEnginePage::slotFeaturePermissionRequested);	59	this, &MailWebEnginePage::slotFeaturePermissionRequested);
70	}	60	}
71		61
72	void MailWebEnginePage::setPrintElementBackground(bool printElementBackground)	62	void MailWebEnginePage::setPrintElementBackground(bool printElementBackground)
73	{	63	{
74	settings()->setAttribute(QWebEngineSettings::PrintElementBackgrounds, printElementBackground);	64	settings()->setAttribute(QWebEngineSettings::PrintElementBackgrounds, printElementBackground);
75	}	65	}
76		66
77	void MailWebEnginePage::slotFeaturePermissionRequested(const QUrl &url, QWebEnginePage::Feature feature)	67	void MailWebEnginePage::slotFeaturePermissionRequested(const QUrl &url, QWebEnginePage::Feature feature)
78	{	68	{
79	//Denied all permissions.	69	//Denied all permissions.
80	setFeaturePermission(url, feature, QWebEnginePage::PermissionDeniedByUser);	70	setFeaturePermission(url, feature, QWebEnginePage::PermissionDeniedByUser);
81	}	71	}

Lines 1-106 Link Here

(-)a/webengineviewer/src/webenginepage.cpp (-5 / +17 lines)
1	/*	1	/*
2	Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>	2	Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>
3		3
4	This library is free software; you can redistribute it and/or	4	This library is free software; you can redistribute it and/or
5	modify it under the terms of the GNU Library General Public	5	modify it under the terms of the GNU Library General Public
6	License as published by the Free Software Foundation; either	6	License as published by the Free Software Foundation; either
7	version 2 of the License, or (at your option) any later version.	7	version 2 of the License, or (at your option) any later version.
8		8
9	This library is distributed in the hope that it will be useful,	9	This library is distributed in the hope that it will be useful,
10	but WITHOUT ANY WARRANTY; without even the implied warranty of	10	but WITHOUT ANY WARRANTY; without even the implied warranty of
11	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU	11	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12	Library General Public License for more details.	12	Library General Public License for more details.
13		13
14	You should have received a copy of the GNU Library General Public License	14	You should have received a copy of the GNU Library General Public License
15	along with this library; see the file COPYING.LIB. If not, write to	15	along with this library; see the file COPYING.LIB. If not, write to
16	the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,	16	the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
17	Boston, MA 02110-1301, USA.	17	Boston, MA 02110-1301, USA.
18	*/	18	*/
19		19
20	#include "webenginepage.h"	20	#include "webenginepage.h"
21	#include "webhittest.h"	21	#include "webhittest.h"
22	#include "webhittestresult.h"	22	#include "webhittestresult.h"
23		23
24	#include <KLocalizedString>	24	#include <KLocalizedString>
25		25
26	#include <QEventLoop>	26	#include <QEventLoop>
27	#include <QWebEngineDownloadItem>	27	#include <QWebEngineDownloadItem>
28	#include <QPointer>	28	#include <QPointer>
29	#include <QTimer>	29	#include <QTimer>
30	#include <QFileDialog>	30	#include <QFileDialog>
31	#include <QWebEngineProfile>	31	#include <QWebEngineProfile>
32	#include <QPrinter>	32	#include <QPrinter>
33	using namespace WebEngineViewer;	33	using namespace WebEngineViewer;
34		34
35	WebEnginePage::WebEnginePage(QObject *parent)	35	WebEnginePage::WebEnginePage(QObject *parent)
36	: QWebEnginePage(parent)	36	: QWebEnginePage(new QWebEngineProfile, parent)
37	{	37	{
		38	// Create a private (off the record) QWebEngineProfile here to isolate the
		39	// browsing settings, and adopt it as a child so that it will be deleted
		40	// when we are destroyed. The profile must remain active for as long as
		41	// any QWebEnginePage's belonging to it exist, see the API documentation
		42	// of QWebEnginePage::QWebEnginePage(QWebEngineProfile , QObject ).
		43	// Deleting it as our child on destruction is safe.
		44	//
		45	// Do not try to save a line of code by setting the parent on construction:
		46	//
		47	// WebEnginePage::WebEnginePage(QObject *parent)
		48	// : QWebEnginePage(new QWebEngineProfile(this), parent)
		49	//
		50	// because the QWebEngineProfile constructor will call out to the QWebEnginePage
		51	// and crash because the QWebEnginePage is not fully constructed yet.
		52	profile()->setParent(this);
		53
38	init();	54	init();
39	}	55	}
40		56
41	WebEnginePage::WebEnginePage(QWebEngineProfile profile, QObject parent)	57	WebEnginePage::WebEnginePage(QWebEngineProfile profile, QObject parent)
42	: QWebEnginePage(profile, parent)	58	: QWebEnginePage(profile, parent)
43	{	59	{
44	init();	60	init();
45	}	61	}
46		62
47	WebEnginePage::~WebEnginePage()
48	{
49	}
50
51	void WebEnginePage::init()	63	void WebEnginePage::init()
52	{	64	{
53	connect(profile(), &QWebEngineProfile::downloadRequested, this, &WebEnginePage::saveHtml);	65	connect(profile(), &QWebEngineProfile::downloadRequested, this, &WebEnginePage::saveHtml);
54	}	66	}
55		67
56	WebEngineViewer::WebHitTest *WebEnginePage::hitTestContent(const QPoint &pos)	68	WebEngineViewer::WebHitTest *WebEnginePage::hitTestContent(const QPoint &pos)
57	{	69	{
58	return new WebHitTest(this, pos);	70	return new WebHitTest(this, pos);
59	}	71	}
60		72
61	void WebEnginePage::saveHtml(QWebEngineDownloadItem *download)	73	void WebEnginePage::saveHtml(QWebEngineDownloadItem *download)
62	{	74	{
63	const QString fileName = QFileDialog::getSaveFileName(view(), i18n("Save HTML Page"));	75	const QString fileName = QFileDialog::getSaveFileName(view(), i18n("Save HTML Page"));
64	if (!fileName.isEmpty()) {	76	if (!fileName.isEmpty()) {
65	download->setSavePageFormat(QWebEngineDownloadItem::SingleHtmlSaveFormat);	77	download->setSavePageFormat(QWebEngineDownloadItem::SingleHtmlSaveFormat);
66	download->setPath(fileName);	78	download->setPath(fileName);
67	download->accept();	79	download->accept();
68	}	80	}
69	}	81	}
70		82
71	bool WebEnginePage::acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame)	83	bool WebEnginePage::acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame)
72	{	84	{
73	if (isMainFrame && type == NavigationTypeLinkClicked) {	85	if (isMainFrame && type == NavigationTypeLinkClicked) {
74	Q_EMIT urlClicked(url);	86	Q_EMIT urlClicked(url);
75	return false;	87	return false;
76	}	88	}
77	return true;	89	return true;
78	}	90	}
79		91
80	void WebEnginePage::javaScriptConsoleMessage(QWebEnginePage::JavaScriptConsoleMessageLevel level, const QString &message, int lineNumber, const QString &sourceID)	92	void WebEnginePage::javaScriptConsoleMessage(QWebEnginePage::JavaScriptConsoleMessageLevel level, const QString &message, int lineNumber, const QString &sourceID)
81	{	93	{
82	Q_UNUSED(level);	94	Q_UNUSED(level);
83	Q_UNUSED(sourceID);	95	Q_UNUSED(sourceID);
84	//Don't convert to debug categories	96	//Don't convert to debug categories
85	qDebug() << "WebEnginePage::javaScriptConsoleMessage lineNumber: " << lineNumber << " message: " << message;	97	qDebug() << "WebEnginePage::javaScriptConsoleMessage lineNumber: " << lineNumber << " message: " << message;
86	Q_EMIT showConsoleMessage(message);	98	Q_EMIT showConsoleMessage(message);
87	}	99	}
88		100
89	bool WebEnginePage::execPrintPreviewPage(QPrinter *printer, int timeout)	101	bool WebEnginePage::execPrintPreviewPage(QPrinter *printer, int timeout)
90	{	102	{
91	QPointer<QEventLoop> loop = new QEventLoop;	103	QPointer<QEventLoop> loop = new QEventLoop;
92	bool result = false;	104	bool result = false;
93	QTimer::singleShot(timeout, loop.data(), &QEventLoop::quit);	105	QTimer::singleShot(timeout, loop.data(), &QEventLoop::quit);
94		106
95	print(printer, [loop, &result](bool res) {	107	print(printer, [loop, &result](bool res) {
96	if (loop && loop->isRunning()) {	108	if (loop && loop->isRunning()) {
97	result = res;	109	result = res;
98	loop->quit();	110	loop->quit();
99	}	111	}
100	});	112	});
101		113
102	loop->exec();	114	loop->exec();
103	delete loop;	115	delete loop;
104		116
105	return result;	117	return result;
106	}	118	}

Lines 1-56 Link Here

(-)a/webengineviewer/src/webenginepage.h (-2 / +33 lines)
1	/*	1	/*
2	Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>	2	Copyright (C) 2016-2018 Laurent Montel <montel@kde.org>
3		3
4	This library is free software; you can redistribute it and/or	4	This library is free software; you can redistribute it and/or
5	modify it under the terms of the GNU Library General Public	5	modify it under the terms of the GNU Library General Public
6	License as published by the Free Software Foundation; either	6	License as published by the Free Software Foundation; either
7	version 2 of the License, or (at your option) any later version.	7	version 2 of the License, or (at your option) any later version.
8		8
9	This library is distributed in the hope that it will be useful,	9	This library is distributed in the hope that it will be useful,
10	but WITHOUT ANY WARRANTY; without even the implied warranty of	10	but WITHOUT ANY WARRANTY; without even the implied warranty of
11	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU	11	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12	Library General Public License for more details.	12	Library General Public License for more details.
13		13
14	You should have received a copy of the GNU Library General Public License	14	You should have received a copy of the GNU Library General Public License
15	along with this library; see the file COPYING.LIB. If not, write to	15	along with this library; see the file COPYING.LIB. If not, write to
16	the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,	16	the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
17	Boston, MA 02110-1301, USA.	17	Boston, MA 02110-1301, USA.
18	*/	18	*/
19		19
20	#ifndef WEBENGINEPAGE_H	20	#ifndef WEBENGINEPAGE_H
21	#define WEBENGINEPAGE_H	21	#define WEBENGINEPAGE_H
22		22
23	#include <QWebEnginePage>	23	#include <QWebEnginePage>
24	#include "webengineviewer_export.h"	24	#include "webengineviewer_export.h"
25	class QWebEngineProfile;	25	class QWebEngineProfile;
26	class QWebEngineDownloadItem;	26	class QWebEngineDownloadItem;
27	class QPrinter;	27	class QPrinter;
28	namespace WebEngineViewer {	28	namespace WebEngineViewer {
29	class WebHitTest;	29	class WebHitTest;
30	class WEBENGINEVIEWER_EXPORT WebEnginePage : public QWebEnginePage	30	class WEBENGINEVIEWER_EXPORT WebEnginePage : public QWebEnginePage
31	{	31	{
32	Q_OBJECT	32	Q_OBJECT
33	public:	33	public:
		34	/**
		35	* Constructor.
		36	*
		37	* A private QWebEngineProfile, only applying to this QWebEnginePage,
		38	* will be created to implement browser settings. It can be accessed via
		39	* @c profile(), but it should not be shared or reused unless care is
		40	* taken that the profile is not deleted until all of the QWebEnginePage's
		41	* belonging to it are deleted first.
		42	*
		43	* @param parent The parent object
		44	**/
34	explicit WebEnginePage(QObject *parent = nullptr);	45	explicit WebEnginePage(QObject *parent = nullptr);
35	explicit WebEnginePage(QWebEngineProfile profile, QObject parent = nullptr);
36		46
37	~WebEnginePage() override;	47	/**
		48	* Constructor.
		49	*
		50	* The specified QWebEngineProfile will be used. See the description of
		51	* @c WebEnginePage(QObject *) and the API documentation of QWebEnginePage
		52	* for caution regarding the lifetime of the profile.
		53	*
		54	* @param profile The profile to be used
		55	* @param parent The parent object
		56	* @deprecated Use the single argument constructor, which creates and uses
		57	* a private profile.
		58	**/
		59	#ifndef WEBENGINEVIEWER_NO_DEPRECATED
		60	explicit WEBENGINEVIEWER_DEPRECATED WebEnginePage(QWebEngineProfile profile, QObject parent = nullptr);
		61	#endif
		62
		63	/**
		64	* Destructor. If there is a private QWebEngineProfile then it will also
		65	* be destroyed.
		66	**/
		67	virtual ~WebEnginePage() override = default;
		68
38	WebEngineViewer::WebHitTest *hitTestContent(const QPoint &pos);	69	WebEngineViewer::WebHitTest *hitTestContent(const QPoint &pos);
39		70
40	void saveHtml(QWebEngineDownloadItem *download);	71	void saveHtml(QWebEngineDownloadItem *download);
41	Q_REQUIRED_RESULT bool execPrintPreviewPage(QPrinter *printer, int timeout);	72	Q_REQUIRED_RESULT bool execPrintPreviewPage(QPrinter *printer, int timeout);
42		73
43	Q_SIGNALS:	74	Q_SIGNALS:
44	void urlClicked(const QUrl &url);	75	void urlClicked(const QUrl &url);
45	void showConsoleMessage(const QString &message);	76	void showConsoleMessage(const QString &message);
46		77
47	protected:	78	protected:
48	bool acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame) override;	79	bool acceptNavigationRequest(const QUrl &url, NavigationType type, bool isMainFrame) override;
49	void javaScriptConsoleMessage(JavaScriptConsoleMessageLevel level, const QString &message, int lineNumber, const QString &sourceID) override;	80	void javaScriptConsoleMessage(JavaScriptConsoleMessageLevel level, const QString &message, int lineNumber, const QString &sourceID) override;
50		81
51	private:	82	private:
52	void init();	83	void init();
53	};	84	};
54	}	85	}
55		86
56	#endif // WEBENGINEPAGE_H	87	#endif // WEBENGINEPAGE_H