110 lines
3.3 KiB
C++
110 lines
3.3 KiB
C++
// For license of this file, see <project-root-folder>/LICENSE.md.
|
|
|
|
#include "network-web/webengine/webenginepage.h"
|
|
|
|
#include "definitions/definitions.h"
|
|
#include "gui/webviewers/webengine/webengineviewer.h"
|
|
#include "miscellaneous/application.h"
|
|
#include "miscellaneous/settings.h"
|
|
#include "network-web/adblock/adblockmanager.h"
|
|
#include "network-web/adblock/adblockrequestinfo.h"
|
|
#include "network-web/webfactory.h"
|
|
|
|
#include <QString>
|
|
#include <QStringList>
|
|
#include <QTimer>
|
|
#include <QUrl>
|
|
#include <QUrlQuery>
|
|
#include <QWebEngineScript>
|
|
|
|
WebEnginePage::WebEnginePage(QObject* parent) : QWebEnginePage(qApp->web()->engineProfile(), parent) {
|
|
setBackgroundColor(Qt::GlobalColor::transparent);
|
|
|
|
connect(this, &QWebEnginePage::loadFinished, this, &WebEnginePage::hideUnwantedElements);
|
|
}
|
|
|
|
WebEngineViewer* WebEnginePage::view() const {
|
|
#if QT_VERSION_MAJOR == 6
|
|
return qobject_cast<WebEngineViewer*>(QWebEngineView::forPage(this));
|
|
#else
|
|
return qobject_cast<WebEngineViewer*>(QWebEnginePage::view());
|
|
#endif
|
|
}
|
|
|
|
QString WebEnginePage::pageHtml(const QString& url) {
|
|
QEventLoop loop;
|
|
QString html;
|
|
|
|
// Load initial DOM/page.
|
|
connect(this, &WebEnginePage::loadFinished, &loop, &QEventLoop::quit);
|
|
connect(this, &WebEnginePage::domIsIdle, &loop, &QEventLoop::quit);
|
|
|
|
load(url);
|
|
loop.exec();
|
|
|
|
// Page is loaded. Send artificial scroll-to-bottom and wait for changes to end.
|
|
// runJavaScript("window.resizeTo(3800, 2100);");
|
|
runJavaScript(IOFactory::readFile(BUILTIN_JS_FOLDER + QL1C('/') + OBSERVER_JS_FILE));
|
|
loop.exec();
|
|
|
|
toHtml([&](const QString& htm) {
|
|
html = htm;
|
|
loop.exit();
|
|
});
|
|
loop.exec();
|
|
|
|
return html;
|
|
}
|
|
|
|
void WebEnginePage::hideUnwantedElements() {
|
|
if (!qApp->web()->adBlock()->isEnabled()) {
|
|
return;
|
|
}
|
|
|
|
auto css = qApp->web()->adBlock()->elementHidingRulesForDomain(url());
|
|
|
|
if (!css.isEmpty()) {
|
|
auto js = qApp->web()->adBlock()->generateJsForElementHiding(css);
|
|
|
|
runJavaScript(js);
|
|
qDebugNN << LOGSEC_ADBLOCK << "Running domain-specific JS for element hiding rules.";
|
|
}
|
|
}
|
|
|
|
bool WebEnginePage::acceptNavigationRequest(const QUrl& url, NavigationType type, bool is_main_frame) {
|
|
if (type == NavigationType::NavigationTypeLinkClicked) {
|
|
bool open_externally_now =
|
|
qApp->settings()->value(GROUP(Browser), SETTING(Browser::OpenLinksInExternalBrowserRightAway)).toBool();
|
|
|
|
if (open_externally_now) {
|
|
qApp->web()->openUrlInExternalBrowser(url.toString());
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (is_main_frame) {
|
|
auto blocked = qApp->web()->adBlock()->block(AdblockRequestInfo(url));
|
|
|
|
if (blocked.m_blocked) {
|
|
// This website is entirely blocked.
|
|
setHtml(qApp->skins()->adBlockedPage(url.toString(), blocked.m_blockedByFilter),
|
|
QUrl::fromUserInput(QSL(INTERNAL_URL_ADBLOCKED)));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return QWebEnginePage::acceptNavigationRequest(url, type, is_main_frame);
|
|
}
|
|
|
|
void WebEnginePage::javaScriptConsoleMessage(JavaScriptConsoleMessageLevel level,
|
|
const QString& message,
|
|
int line_number,
|
|
const QString& source_id) {
|
|
Q_UNUSED(level)
|
|
|
|
qWarningNN << LOGSEC_JS << message << QSL(" (source: %1:%2)").arg(source_id, QString::number(line_number));
|
|
|
|
if (message.contains(QSL("iiddllee"))) {
|
|
emit domIsIdle();
|
|
}
|
|
}
|