-
Notifications
You must be signed in to change notification settings - Fork 0
/
mainwindow.cpp
90 lines (80 loc) · 3.07 KB
/
mainwindow.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "mainwindow.h"
#define db qDebug() << this <<
MainWindow::MainWindow(QWidget *parent)
: QMainWindow(parent)
, m_view(new QWebEngineView(this))
{
setCentralWidget(m_view);
Interceptor = new WebEngineURLInterceptor;
connect(m_view, &QWebEngineView::loadFinished, this, &MainWindow::loadFinished);
connect(m_view, &QWebEngineView::urlChanged, this, &MainWindow::urlChanged);
connect(QWebEngineProfile::defaultProfile(), SIGNAL(downloadRequested(QWebEngineDownloadItem*)),
this, SLOT(downloadRequested(QWebEngineDownloadItem*)));
m_view->page()->setUrlRequestInterceptor(Interceptor);
lParser.loadFiles();
cssParser.loadFiles();
sParser.loadFiles();
}
void MainWindow::loadURL(QUrl URL)
{
m_view->page()->load(URL);
}
void MainWindow::insertStyleSheet(const QString &name, const QString &source)
{
QWebEngineScript script;
QString s = QString::fromLatin1("(function() {"\
" css = document.createElement('style');"\
" css.type = 'text/css';"\
" css.id = '%1';"\
" document.head.appendChild(css);"\
" css.innerText = '%2';"\
"})()").arg(name).arg(source.simplified());
script.setName(name);
script.setSourceCode(s);
script.setInjectionPoint(QWebEngineScript::DocumentReady);
script.setRunsOnSubFrames(true);
script.setWorldId(QWebEngineScript::ApplicationWorld);
m_view->page()->scripts().insert(script);
m_view->page()->runJavaScript(s, QWebEngineScript::ApplicationWorld);
}
void MainWindow::loadFinished(bool b)
{
if(b == true)
{
QTimer::singleShot(1500, this, &MainWindow::loadScripts);
}
}
void MainWindow::loadScripts()
{
QString url = m_view->url().toString();
db url;
lParser.runLogins(url, m_view->page());
// QString js = lParser.getLoginJavascript(url);
// if(!js.isEmpty())
// m_view->page()->runJavaScript(js);
// QString scripts = sParser.getScriptJavascript(url);
// if(!scripts.isEmpty())
// m_view->page()->runJavaScript(scripts);
}
void MainWindow::urlChanged(QUrl url)
{
QString css = cssParser.getUserCss(url.toString());
if(!css.isEmpty())
insertStyleSheet(url.toString(), css);
QTimer::singleShot(5500, this, &MainWindow::loadScripts);
}
void MainWindow::downloadRequested(QWebEngineDownloadItem * download)
{
if (download->savePageFormat() != QWebEngineDownloadItem::UnknownSaveFormat)
{
qDebug() << "Format: " << download->savePageFormat();
qDebug() << "Path: " << download->downloadDirectory();
// If you want to modify something like the default path or the format
// download->setSavePageFormat(...);
QFileDialog fd;
auto path = fd.getSaveFileName(nullptr,"Save File",QApplication::applicationDirPath());
download->setDownloadDirectory(path);
// Check your url to accept/reject the download
download->accept();
}
}