From bbe461a515ff55cd6733ec0392b0a4f664499ecf Mon Sep 17 00:00:00 2001
From: Chris Xiong <chirs241097@gmail.com>
Date: Mon, 19 Sep 2022 12:40:21 -0400
Subject: Implement database save & load.

---
 qdeduper/mingui.cpp | 40 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

(limited to 'qdeduper/mingui.cpp')

diff --git a/qdeduper/mingui.cpp b/qdeduper/mingui.cpp
index 2f21e5e..845bc70 100644
--- a/qdeduper/mingui.cpp
+++ b/qdeduper/mingui.cpp
@@ -56,6 +56,15 @@ QString fsstr_to_qstring(const fs::path::string_type &s)
 #endif
 }
 
+fs::path qstring_to_path(const QString &s)
+{
+#if PATH_VALSIZE == 2 //the degenerate platform
+    return fs::path(s.toStdWString());
+#else
+    return fs::path(s.toStdString());
+#endif
+}
+
 Q_DECLARE_METATYPE(fs::path)
 
 DeduperMainWindow::DeduperMainWindow()
@@ -174,8 +183,33 @@ void DeduperMainWindow::setup_menu()
     QAction *create_db = file->addAction("Create Database...");
     QObject::connect(create_db, &QAction::triggered, this, &DeduperMainWindow::create_new);
     menuact["create_db"] = create_db;
-    file->addAction("Load Database...");
-    file->addAction("Save Database...");
+    QAction *load_db = file->addAction("Load Database...");
+    load_db->setIcon(this->style()->standardIcon(QStyle::StandardPixmap::SP_DialogOpenButton));
+    QObject::connect(load_db, &QAction::triggered, [this] {
+        QString dbpath = QFileDialog::getOpenFileName(this, "Load Database", QString(), "Signature database (*.sigdb)");
+        if (!dbpath.isNull())
+        {
+            this->sdb = new SignatureDB(qstring_to_path(dbpath));
+            if (!this->sdb->valid()) {
+                delete this->sdb;
+                this->sdb = nullptr;
+                QMessageBox::critical(this, "Error", "Error loading database.");
+                return;
+            }
+            curgroup = 0;
+            show_group(0);
+        }
+    });
+    menuact["load_db"] = load_db;
+
+    QAction *save_db = file->addAction("Save Database...");
+    save_db->setIcon(this->style()->standardIcon(QStyle::StandardPixmap::SP_DialogSaveButton));
+    QObject::connect(save_db, &QAction::triggered, [this] {
+        QString dbpath = QFileDialog::getSaveFileName(this, "Save Database", QString(), "Signature database (*.sigdb)");
+        if (!dbpath.isNull() && this->sdb)
+            this->sdb->save(qstring_to_path(dbpath));
+    });
+    menuact["save_db"] = save_db;
     file->addSeparator();
     file->addAction("Export Marked Images List...");
     file->addAction("Import Marked Images List...");
@@ -249,11 +283,13 @@ void DeduperMainWindow::update_actions()
         menuact["next_group"]->setEnabled(false);
         menuact["prev_group"]->setEnabled(false);
         menuact["skip_group"]->setEnabled(false);
+        menuact["save_db"]->setEnabled(false);
         return;
     }
     menuact["skip_group"]->setEnabled(true);
     menuact["prev_group"]->setEnabled(curgroup > 0);
     menuact["next_group"]->setEnabled(curgroup + 1 < sdb->num_groups());
+    menuact["save_db"]->setEnabled(true);
 }
 
 void DeduperMainWindow::show_images(const std::vector<fs::path> &fns)
-- 
cgit v1.2.3