Enable downloading of files that require login cookies. https://redmine.stoutner...

[PrivacyBrowserPC.git] / src / dialogs / SaveDialog.cpp

diff --git a/src/dialogs/SaveDialog.cpp b/src/dialogs/SaveDialog.cpp
index fff857d41e18679f2b61cf374152f8316ced57a8..39d2b1855bbb9d83a3dcd25909c7fa9717d9149d 100644 (file)
--- a/src/dialogs/SaveDialog.cpp
+++ b/src/dialogs/SaveDialog.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright © 2022 Soren Stoutner <soren@stoutner.com>.
+ * Copyright 2022 Soren Stoutner <soren@stoutner.com>.
  *
  * This file is part of Privacy Browser PC <https://www.stoutner.com/privacy-browser-pc>.
  *
@@ -30,7 +30,7 @@
 #include <QShortcut>
 #include <QStandardPaths>
 
-SaveDialog::SaveDialog(QWebEngineDownloadItem *downloadItemPointer)
+SaveDialog::SaveDialog(QUrl &url, QString &mimeTypeString, int totalBytes, QString fileName, bool nativeDownloader): downloadUrl(url), suggestedFileName(fileName)
 {
     // Set the dialog window title.
     setWindowTitle(i18nc("The save dialog window title", "Save"));
@@ -45,6 +45,7 @@ SaveDialog::SaveDialog(QWebEngineDownloadItem *downloadItemPointer)
     saveDialogUi.setupUi(this);
 
     // Get handles for the widgets.
+    QGraphicsView *mimeGraphicsViewPointer = saveDialogUi.mimeGraphicsView;
     QLabel *urlLabelPointer = saveDialogUi.urlLabel;
     QLabel *filetypeLabelPointer = saveDialogUi.fileTypeLabel;
     QLabel *mimeTypeLabelPointer = saveDialogUi.mimeTypeLabel;
@@ -52,43 +53,67 @@ SaveDialog::SaveDialog(QWebEngineDownloadItem *downloadItemPointer)
     QDialogButtonBox *dialogButtonBoxPointer = saveDialogUi.dialogButtonBox;
     QPushButton *saveButtonPointer = dialogButtonBoxPointer->button(QDialogButtonBox::Save);
 
-    // Get the URL and the suggested file name.
-    downloadUrl = downloadItemPointer->url();
-    suggestedFileName = downloadItemPointer->suggestedFileName();
-    QString mimeType = downloadItemPointer->mimeType();
-
     // Get a MIME type database.
     QMimeDatabase mimeDatabase;
 
+    // Get the MIME type.
+    QMimeType mimeType = mimeDatabase.mimeTypeForName(mimeTypeString);
+
+    // Get the MIME type icon.
+    QIcon mimeTypeIcon = QIcon::fromTheme(mimeType.iconName());
+
+    // Create a graphics scene.
+    QGraphicsScene *mimeGraphicsScenePointer = new QGraphicsScene(this);
+
+    // Set the graphics scene.
+    mimeGraphicsViewPointer->setScene(mimeGraphicsScenePointer);
+
+    // Set the background of the graphics view to be the same as the window
+    mimeGraphicsViewPointer->setBackgroundRole(QPalette::Window);
+
+    // Add the MIME type icon to the scene.
+    mimeGraphicsScenePointer->addPixmap(mimeTypeIcon.pixmap(64, 64));
+
     // Populate the labels.
     urlLabelPointer->setText("<b>" + downloadUrl.toString() + "</b>");
-    filetypeLabelPointer->setText("<b>" + mimeDatabase.mimeTypeForName(mimeType).comment() + "</b>");
-    mimeTypeLabelPointer->setText("<b>" + mimeType + "</b>");
+    filetypeLabelPointer->setText("<b>" + mimeType.comment() + "</b>");
+    mimeTypeLabelPointer->setText("<b>" + mimeTypeString + "</b>");
 
     // Populate the download size label.
-    if (downloadItemPointer->totalBytes() == -1)  // The file size is unknown.
+    if (totalBytes == -1)  // The file size is unknown.
         sizeLabelPointer->setText(i18nc("Unknown download file size.  The bold style should be preserved.", "<b>unknown</b>"));
     else  // The file size is known.  Format it according to the locale.
-        sizeLabelPointer->setText(ki18nc("Download file size.  The bold style should be preserved.", "<b>%1 bytes</b>").subs(downloadItemPointer->totalBytes()).toString());
-
-    // Connect the buttons.
-    connect(saveButtonPointer, SIGNAL(clicked()), this, SLOT(showFileDialog()));
-    connect(dialogButtonBoxPointer, SIGNAL(rejected()), this, SLOT(reject()));
+        sizeLabelPointer->setText(ki18nc("Download file size.  The bold style should be preserved.", "<b>%1 bytes</b>").subs(totalBytes).toString());
 
     // Create the keyboard shortcuts.
     QShortcut *sShortcutPointer = new QShortcut(QKeySequence(i18nc("The save key shortcut.", "s")), this);
     QShortcut *cShortcutPointer = new QShortcut(QKeySequence(i18nc("The close key shortcut.", "c")), this);
 
-    // Connect the shortcuts.
-    connect(sShortcutPointer, SIGNAL(activated()), this, SLOT(showFileDialog()));
+    // Connect the save buttons.
+    if (nativeDownloader)
+    {
+        // Show the file picker for the native download.
+        connect(saveButtonPointer, SIGNAL(clicked()), this, SLOT(showFilePicker()));
+        connect(sShortcutPointer, SIGNAL(activated()), this, SLOT(showFilePicker()));
+    }
+    else
+    {
+        // Use WebEngine's downloader
+        connect(saveButtonPointer, SIGNAL(clicked()), this, SLOT(accept()));
+        connect(sShortcutPointer, SIGNAL(activated()), this, SLOT(accept()));
+    }
+
+    // Connect the cancel button.
+    connect(dialogButtonBoxPointer, SIGNAL(rejected()), this, SLOT(reject()));
     connect(cShortcutPointer, SIGNAL(activated()), this, SLOT(reject()));
 }
 
-void SaveDialog::showFileDialog()
+void SaveDialog::showFilePicker()
 {
     // Show the file picker dialog.
-    emit showSaveFilePickerDialog(downloadUrl, suggestedFileName);
+    emit useNativeDownloader(downloadUrl, suggestedFileName);
 
     // Close the dialog.
     reject();
 }
+