From 429f66951364f21cf3ba520d23db6a3c6c5cfba7 Mon Sep 17 00:00:00 2001
From: Altair Wei <altair_wei@outlook.com>
Date: Sat, 13 Apr 2024 21:51:24 +0800
Subject: [PATCH] DEV: fix FEPageExport to fit new logic

---
 src/WizMainWindow.cpp            |  8 ++------
 src/widgets/FileExportWizard.cpp | 29 ++++++++++++-----------------
 2 files changed, 14 insertions(+), 23 deletions(-)

diff --git a/src/WizMainWindow.cpp b/src/WizMainWindow.cpp
index 69295302..55a7f9f9 100755
--- a/src/WizMainWindow.cpp
+++ b/src/WizMainWindow.cpp
@@ -601,11 +601,7 @@ void WizMainWindow::resizeEvent(QResizeEvent *event)
     update();
 }
 
-/**
- * @brief Show upgrade notification.
- * 
- * @param bUpgradeAvaliable 
- */
+/** Show WizNotePlus upgrade notification dialog. */
 void WizMainWindow::on_checkUpgrade_finished(QJsonObject latestStable, QJsonObject latestTest)
 {
     if (latestStable.isEmpty() && latestTest.isEmpty())
@@ -623,7 +619,7 @@ void WizMainWindow::on_checkUpgrade_finished(QJsonObject latestStable, QJsonObje
         return;
     }
 
-    WizUpgradeNotifyDialog notifyDialog(this);
+    WizUpgradeNotifyDialog notifyDialog;
     notifyDialog.showMarkdownContent(strMarkdown);
     if (QDialog::Accepted == notifyDialog.exec()) {
         QDesktopServices::openUrl(QUrl(strUrl));
diff --git a/src/widgets/FileExportWizard.cpp b/src/widgets/FileExportWizard.cpp
index 61d49af6..40fb0b19 100644
--- a/src/widgets/FileExportWizard.cpp
+++ b/src/widgets/FileExportWizard.cpp
@@ -711,8 +711,9 @@ void FEPageExport::insertLog(const QString& text)
 
 void FEPageExport::handleExportFile()
 {
-    QStringList notes = field("documents*").toStringList();
-    auto outputFormats = field("outputFormats*").value<QMap<QString, QString> >();
+    auto notes = field("documents").toList();
+
+    auto outputFormats = field("outputFormats").value<QMap<QString, QString> >();
     QString outputFolder = field("outputFolder").toString();
     bool keepFolder = field("keepFolder").toBool();
 
@@ -725,24 +726,18 @@ void FEPageExport::handleExportFile()
     m_progress->setRange(0, notes.size());
     m_progress->setValue(0);
 
-    // TODO: add databse option
-    WizDatabase& db = m_dbMgr.db();
-    foreach (auto &note, notes) {
+    foreach (const QVariant &note, notes) {
         m_progress->setValue(m_progress->value() + 1);
         qApp->processEvents();
         if (m_cancel) return;
 
-        WIZDOCUMENTDATA data;
-        if (!db.documentFromGuid(note, data)) {
-            ASKCONTINUE(
-                tr("Do you want to continue?"),
-                tr("Can't find document for GUID: %1").arg(note)
-            );
-        }
+        auto data = note.value<const WIZDOCUMENTDATA*>();
+        insertLog(QString("Exporting %1\n").arg(data->strTitle));
 
-        insertLog(QString("Exporting %1\n").arg(data.strTitle));
+        // Get the corresponding databaset for a document
+        WizDatabase& db = m_dbMgr.db(data->strKbGUID);
 
-        if (data.nProtected == 1) {
+        if (data->nProtected == 1) {
             if (!db.loadUserCert()) {
                 ASKCONTINUE(
                     tr("Do you want to continue?"),
@@ -775,14 +770,14 @@ void FEPageExport::handleExportFile()
 
         QString destFolder = outputFolder;
         if (keepFolder)
-            destFolder = outputFolder + data.strLocation;
+            destFolder = outputFolder + data->strLocation;
 
         auto format = WizFileExporter::HTML;
-        if (WizIsMarkdownNote(data))
+        if (WizIsMarkdownNote(*data))
             format = WizFileExporter::Markdown;
 
         bool ok = m_exporter->exportNote(
-            data, destFolder, format);
+            *data, destFolder, format);
         if (!ok) {
             ASKCONTINUE(
                 tr("Do you want to continue?"),