X-Git-Url: https://gitweb.stoutner.com/?a=blobdiff_plain;f=src%2Fviews%2FBrowserView.cpp;h=f17a237d6d33d962db9315eb3a9602504f79bddc;hb=e28b208d6f953d24bd05927a16775d103714fd36;hp=440c694ab02a7ca130bfc58dbeb85ba9fa2a1b8b;hpb=588db73b94af7b596b0e532f4557aa8b6c41f5c3;p=PrivacyBrowserPC.git diff --git a/src/views/BrowserView.cpp b/src/views/BrowserView.cpp index 440c694..f17a237 100644 --- a/src/views/BrowserView.cpp +++ b/src/views/BrowserView.cpp @@ -21,15 +21,25 @@ #include "BrowserView.h" #include "Settings.h" #include "ui_BrowserView.h" +#include "databases/CookiesDatabase.h" +#include "databases/DomainsDatabase.h" +#include "dialogs/SaveDialog.h" #include "filters/MouseEventFilter.h" -#include "helpers/DomainsDatabaseHelper.h" #include "helpers/SearchEngineHelper.h" #include "helpers/UserAgentHelper.h" #include "interceptors/UrlRequestInterceptor.h" #include "windows/BrowserWindow.h" -// Qt framework headers. +// KDE Framework headers. +#include +#include + +// Qt toolkit headers. #include +#include +#include +#include +#include // Initialize the public static variables. QString BrowserView::webEngineDefaultUserAgent = QStringLiteral(""); @@ -58,6 +68,9 @@ BrowserView::BrowserView(QWidget *parent) : QWidget(parent) // Set the WebEngine page. webEngineViewPointer->setPage(webEnginePagePointer); + // Handle full screen requests. + connect(webEnginePagePointer, SIGNAL(fullScreenRequested(QWebEngineFullScreenRequest)), this, SLOT(fullScreenRequested(QWebEngineFullScreenRequest))); + // Get handles for the aspects of the WebEngine. webEngineHistoryPointer = webEnginePagePointer->history(); webEngineSettingsPointer = webEngineViewPointer->settings(); @@ -72,20 +85,43 @@ BrowserView::BrowserView(QWidget *parent) : QWidget(parent) // Set the local storage filter. webEngineCookieStorePointer->setCookieFilter([this](const QWebEngineCookieStore::FilterRequest &filterRequest) { - // qDebug() << "Page URL: " << filterRequest.firstPartyUrl << ", Local storage URL: " << filterRequest.origin << ", Is third-party: " << filterRequest.thirdParty; + //qDebug().noquote().nospace() << "Page URL: " << filterRequest.firstPartyUrl << ", Local storage URL: " << filterRequest.origin << ", Is third-party: " << filterRequest.thirdParty; // Block all third party local storage requests, including the sneaky ones that don't register a first party URL. if (filterRequest.thirdParty || (filterRequest.firstPartyUrl == QStringLiteral(""))) + { + //qDebug() << "Request blocked."; + + // Return false. return false; + } + /* TODO. Waiting for a solution to . // Check each tab to see if this local storage request should be allowed. for (PrivacyWebEngine *privacyWebEnginePointer : *privacyWebEngineListPointer) { + //qDebug().noquote().nospace() << "Local storage: " << privacyWebEnginePointer->localStorageEnabled << ". WebEngine URL: " << webEngineViewPointer->url().host() << ". Request Host: " << filterRequest.firstPartyUrl.host(); + // Allow this local storage request if it comes from a tab with local storage enabled. if (privacyWebEnginePointer->localStorageEnabled && (webEngineViewPointer->url().host() == filterRequest.firstPartyUrl.host())) + { + //qDebug() << "Request allowed."; + + // Return true. return true; + } + } + */ + + // Allow the request if it is first party and local storage is enabled. + if (!filterRequest.thirdParty && currentPrivacyWebEnginePointer->localStorageEnabled) + { + // Return true. + return true; } + //qDebug() << "Request blocked."; + // Block any remaining local storage requests. return false; }); @@ -94,6 +130,13 @@ BrowserView::BrowserView(QWidget *parent) : QWidget(parent) connect(webEngineCookieStorePointer, SIGNAL(cookieAdded(QNetworkCookie)), this, SLOT(cookieAdded(QNetworkCookie))); connect(webEngineCookieStorePointer, SIGNAL(cookieRemoved(QNetworkCookie)), this, SLOT(cookieRemoved(QNetworkCookie))); + // Get a list of durable cookies. + QList *durableCookiesListPointer = CookiesDatabase::getCookies(); + + // Add the durable cookies to the store. + for (QNetworkCookie *cookiePointer : *durableCookiesListPointer) + addCookieToStore(*cookiePointer); + // Store a copy of the WebEngine default user agent. webEngineDefaultUserAgent = webEngineProfilePointer->httpUserAgent(); @@ -124,6 +167,9 @@ BrowserView::BrowserView(QWidget *parent) : QWidget(parent) // Set the URL request interceptor. webEngineProfilePointer->setUrlRequestInterceptor(urlRequestInterceptorPointer); + // Handle file downloads. + connect(webEngineProfilePointer, SIGNAL(downloadRequested(QWebEngineDownloadItem *)), this, SLOT(showSaveDialog(QWebEngineDownloadItem *))); + // Reapply the domain settings when the host changes. connect(urlRequestInterceptorPointer, SIGNAL(applyDomainSettings(QString)), this, SLOT(applyDomainSettingsWithoutReloading(QString))); @@ -152,7 +198,8 @@ BrowserView::~BrowserView() webEnginePagePointer->deleteLater(); } -void BrowserView::addCookieToStore(QNetworkCookie &cookie) const +// The cookie is copied instead of referenced so that changes made to the cookie do not create a race condition with the display of the cookie in the dialog. +void BrowserView::addCookieToStore(QNetworkCookie cookie) const { // Create a url. QUrl url; @@ -178,7 +225,7 @@ void BrowserView::applyApplicationSettings() searchEngineUrl = SearchEngineHelper::getSearchUrl(Settings::searchEngine()); // Emit the update search engine actions signal. - emit updateSearchEngineActions(Settings::searchEngine()); + emit updateSearchEngineActions(Settings::searchEngine(), true); } // This exists as a separate function from `applyDomainSettings()` so it can be listed as a slot and function without the need for a boolean argument. @@ -201,7 +248,7 @@ void BrowserView::applyDomainSettingsWithoutReloading(const QString &hostname) void BrowserView::applyDomainSettings(const QString &hostname, const bool reloadWebsite) { // Get the record for the hostname. - QSqlQuery domainQuery = DomainsDatabaseHelper::getDomainQuery(hostname); + QSqlQuery domainQuery = DomainsDatabase::getDomainQuery(hostname); // Check if the hostname has domain settings. if (domainQuery.isValid()) // The hostname has domain settings. @@ -210,10 +257,10 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload QSqlRecord domainRecord = domainQuery.record(); // Set the JavaScript status. - switch (domainRecord.field(DomainsDatabaseHelper::JAVASCRIPT).value().toInt()) + switch (domainRecord.field(DomainsDatabase::JAVASCRIPT).value().toInt()) { // Set the default JavaScript status. - case (DomainsDatabaseHelper::SYSTEM_DEFAULT): + case (DomainsDatabase::SYSTEM_DEFAULT): { webEngineSettingsPointer->setAttribute(QWebEngineSettings::JavascriptEnabled, Settings::javaScriptEnabled()); @@ -221,7 +268,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Disable JavaScript. - case (DomainsDatabaseHelper::DISABLED): + case (DomainsDatabase::DISABLED): { webEngineSettingsPointer->setAttribute(QWebEngineSettings::JavascriptEnabled, false); @@ -229,7 +276,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Enable JavaScript. - case (DomainsDatabaseHelper::ENABLED): + case (DomainsDatabase::ENABLED): { webEngineSettingsPointer->setAttribute(QWebEngineSettings::JavascriptEnabled, true); @@ -238,10 +285,10 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Set the local storage status. - switch (domainRecord.field(DomainsDatabaseHelper::LOCAL_STORAGE).value().toInt()) + switch (domainRecord.field(DomainsDatabase::LOCAL_STORAGE).value().toInt()) { // Set the default local storage status. - case (DomainsDatabaseHelper::SYSTEM_DEFAULT): + case (DomainsDatabase::SYSTEM_DEFAULT): { currentPrivacyWebEnginePointer->localStorageEnabled = Settings::localStorageEnabled(); @@ -249,7 +296,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Disable local storage. - case (DomainsDatabaseHelper::DISABLED): + case (DomainsDatabase::DISABLED): { currentPrivacyWebEnginePointer->localStorageEnabled = false; @@ -257,7 +304,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Enable local storage. - case (DomainsDatabaseHelper::ENABLED): + case (DomainsDatabase::ENABLED): { currentPrivacyWebEnginePointer->localStorageEnabled = true; @@ -266,10 +313,10 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Set the DOM storage status. - switch (domainRecord.field(DomainsDatabaseHelper::DOM_STORAGE).value().toInt()) + switch (domainRecord.field(DomainsDatabase::DOM_STORAGE).value().toInt()) { // Set the default DOM storage status. - case (DomainsDatabaseHelper::SYSTEM_DEFAULT): + case (DomainsDatabase::SYSTEM_DEFAULT): { webEngineSettingsPointer->setAttribute(QWebEngineSettings::LocalStorageEnabled, Settings::domStorageEnabled()); @@ -277,7 +324,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Disable DOM storage. - case (DomainsDatabaseHelper::DISABLED): + case (DomainsDatabase::DISABLED): { webEngineSettingsPointer->setAttribute(QWebEngineSettings::LocalStorageEnabled, false); @@ -285,7 +332,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Enable DOM storage. - case (DomainsDatabaseHelper::ENABLED): + case (DomainsDatabase::ENABLED): { webEngineSettingsPointer->setAttribute(QWebEngineSettings::LocalStorageEnabled, true); @@ -294,13 +341,13 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload } // Set the user agent. - webEngineProfilePointer->setHttpUserAgent(UserAgentHelper::getResultingDomainSettingsUserAgent(domainRecord.field(DomainsDatabaseHelper::USER_AGENT).value().toString())); + webEngineProfilePointer->setHttpUserAgent(UserAgentHelper::getResultingDomainSettingsUserAgent(domainRecord.field(DomainsDatabase::USER_AGENT).value().toString())); // Check if a custom zoom factor is set. - if (domainRecord.field(DomainsDatabaseHelper::ZOOM_FACTOR).value().toInt()) + if (domainRecord.field(DomainsDatabase::ZOOM_FACTOR).value().toInt()) { // Store the current zoom factor. - currentZoomFactor = domainRecord.field(DomainsDatabaseHelper::CUSTOM_ZOOM_FACTOR).value().toDouble(); + currentZoomFactor = domainRecord.field(DomainsDatabase::CUSTOM_ZOOM_FACTOR).value().toDouble(); } else { @@ -312,7 +359,7 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload webEngineViewPointer->setZoomFactor(currentZoomFactor); // Apply the domain settings palette to the URL line edit. - emit updateDomainSettingsIndicator(true, domainRecord.field(DomainsDatabaseHelper::DOMAIN_NAME).value().toString()); + emit updateDomainSettingsIndicator(true, domainRecord.field(DomainsDatabase::DOMAIN_NAME).value().toString()); } else // The hostname does not have domain settings. { @@ -342,14 +389,12 @@ void BrowserView::applyDomainSettings(const QString &hostname, const bool reload emit updateJavaScriptAction(webEngineSettingsPointer->testAttribute(QWebEngineSettings::JavascriptEnabled)); emit updateLocalStorageAction(currentPrivacyWebEnginePointer->localStorageEnabled); emit updateDomStorageAction(webEngineSettingsPointer->testAttribute(QWebEngineSettings::LocalStorageEnabled)); - emit updateUserAgentActions(webEngineProfilePointer->httpUserAgent()); + emit updateUserAgentActions(webEngineProfilePointer->httpUserAgent(), true); emit updateZoomFactorAction(webEngineViewPointer->zoomFactor()); // Reload the website if requested. if (reloadWebsite) - { webEngineViewPointer->reload(); - } } void BrowserView::applyOnTheFlySearchEngine(QAction *searchEngineActionPointer) @@ -362,6 +407,9 @@ void BrowserView::applyOnTheFlySearchEngine(QAction *searchEngineActionPointer) // Store the search engine string. searchEngineUrl = SearchEngineHelper::getSearchUrl(searchEngineName); + + // Update the search engine actionas. + emit updateSearchEngineActions(searchEngineName, false); } void BrowserView::applyOnTheFlyUserAgent(QAction *userAgentActionPointer) const @@ -375,6 +423,9 @@ void BrowserView::applyOnTheFlyUserAgent(QAction *userAgentActionPointer) const // Apply the user agent. webEngineProfilePointer->setHttpUserAgent(UserAgentHelper::getUserAgentFromTranslatedName(userAgentName)); + // Update the user agent actions. + emit updateUserAgentActions(webEngineProfilePointer->httpUserAgent(), false); + // Reload the website. webEngineViewPointer->reload(); } @@ -425,6 +476,15 @@ void BrowserView::forward() const webEngineViewPointer->forward(); } +void BrowserView::fullScreenRequested(QWebEngineFullScreenRequest fullScreenRequest) const +{ + // Make it so. + emit fullScreenRequested(fullScreenRequest.toggleOn()); + + // Accept the request. + fullScreenRequest.accept(); +} + void BrowserView::home() const { // Load the homepage. @@ -500,7 +560,7 @@ void BrowserView::loadUrlFromLineEdit(QString url) const void BrowserView::mouseBack() const { // Go back if possible. - if (webEngineHistoryPointer->canGoBack()) + if (webEngineViewPointer->isActiveWindow() && webEngineHistoryPointer->canGoBack()) { // Clear the URL line edit focus. emit clearUrlLineEditFocus(); @@ -513,7 +573,7 @@ void BrowserView::mouseBack() const void BrowserView::mouseForward() const { // Go forward if possible. - if (webEngineHistoryPointer->canGoForward()) + if (webEngineViewPointer->isActiveWindow() && webEngineHistoryPointer->canGoForward()) { // Clear the URL line edit focus. emit clearUrlLineEditFocus(); @@ -529,12 +589,121 @@ void BrowserView::pageLinkHovered(const QString &linkUrl) const emit linkHovered(linkUrl); } +void BrowserView::print() const +{ + // Create a printer. + QPrinter printer; + + // Set the resolution to be 300 dpi. + printer.setResolution(300); + + // Create a printer dialog. + QPrintDialog printDialog(&printer, webEngineViewPointer); + + // Display the dialog and print the page if instructed. + if (printDialog.exec() == QDialog::Accepted) + printWebpage(&printer); +} + +void BrowserView::printPreview() const +{ + // Create a printer. + QPrinter printer; + + // Set the resolution to be 300 dpi. + printer.setResolution(300); + + // Create a print preview dialog. + QPrintPreviewDialog printPreviewDialog(&printer, webEngineViewPointer); + + // Generate the print preview. + connect(&printPreviewDialog, SIGNAL(paintRequested(QPrinter *)), this, SLOT(printWebpage(QPrinter *))); + + // Display the dialog. + printPreviewDialog.exec(); +} + +void BrowserView::printWebpage(QPrinter *printerPointer) const +{ + // Create an event loop. For some reason, the print preview doesn't produce any output unless it is run inside an event loop. + QEventLoop eventLoop; + + // Print the webpage, converting the callback above into a `QWebEngineCallback`. + // Printing requires that the printer be a pointer, not a reference, or it will crash with much cursing. + webEnginePagePointer->print(printerPointer, [&eventLoop](bool printSuccess) + { + // Instruct the compiler to ignore the unused parameter. + (void) printSuccess; + + // Quit the loop. + eventLoop.quit(); + }); + + // Execute the loop. + eventLoop.exec(); +} + void BrowserView::refresh() const { // Reload the website. webEngineViewPointer->reload(); } +void BrowserView::showSaveDialog(QWebEngineDownloadItem *downloadItemPointer) const +{ + // Instantiate the save dialog. + SaveDialog *saveDialogPointer = new SaveDialog(downloadItemPointer); + + // Connect the save button. + connect(saveDialogPointer, SIGNAL(showSaveFilePickerDialog(QUrl &, QString &)), this, SLOT(showSaveFilePickerDialog(QUrl &, QString &))); + + // Show the dialog. + saveDialogPointer->show(); +} + +void BrowserView::showSaveFilePickerDialog(QUrl &downloadUrl, QString &suggestedFileName) +{ + // Get the download location. + QString downloadDirectory = Settings::downloadLocation(); + + // Resolve the system download directory if specified. + if (downloadDirectory == QStringLiteral("System Download Directory")) + downloadDirectory = QStandardPaths::writableLocation(QStandardPaths::DownloadLocation); + + // Create a save file dialog. + QFileDialog *saveFileDialogPointer = new QFileDialog(this, i18nc("Save file dialog caption", "Save File"), downloadDirectory); + + // Tell the dialog to use a save button. + saveFileDialogPointer->setAcceptMode(QFileDialog::AcceptSave); + + // Populate the file name from the download item pointer. + saveFileDialogPointer->selectFile(suggestedFileName); + + // Prevent interaction with the parent window while the dialog is open. + saveFileDialogPointer->setWindowModality(Qt::WindowModal); + + // Process the saving of the file. The save file dialog pointer must be captured directly instead of by reference or nasty crashes occur. + auto saveFile = [saveFileDialogPointer, &downloadUrl] () { + // Get the save location. The dialog box should only allow the selecting of one file location. + QUrl saveLocation = saveFileDialogPointer->selectedUrls().value(0); + + // Create a file copy job. `-1` creates the file with default permissions. + KIO::FileCopyJob *fileCopyJobPointer = KIO::file_copy(downloadUrl, saveLocation, -1, KIO::Overwrite); + + // Set the download job to display any error messages. + fileCopyJobPointer->uiDelegate()->setAutoErrorHandlingEnabled(true); + + // Start the download. + fileCopyJobPointer->start(); + }; + + // Handle clicks on the save button. + connect(saveFileDialogPointer, &QDialog::accepted, this, saveFile); + + // Show the dialog. + saveFileDialogPointer->show(); +} + void BrowserView::toggleDomStorage() const { // Toggle DOM storage.