gui: Add openWallet and getWalletsAvailableToOpen to WalletController

guiomie · Feb 4, 2019 · 32a8c6a · 32a8c6a
1 parent ab288b4
commit 32a8c6a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/qt/walletcontroller.cpp b/src/qt/walletcontroller.cpp
@@ -9,6 +9,7 @@
 
 #include <algorithm>
 
+#include <QMessageBox>
 #include <QMutexLocker>
 #include <QThread>
 
@@ -37,6 +38,30 @@ std::vector<WalletModel*> WalletController::getWallets() const
     return m_wallets;
 }
 
+std::vector<std::string> WalletController::getWalletsAvailableToOpen() const
+{
+    QMutexLocker locker(&m_mutex);
+    std::vector<std::string> wallets = m_node.listWalletDir();
+    for (WalletModel* wallet_model : m_wallets) {
+        auto it = std::remove(wallets.begin(), wallets.end(), wallet_model->wallet().getWalletName());
+        if (it != wallets.end()) wallets.erase(it);
+    }
+    return wallets;
+}
+
+WalletModel* WalletController::openWallet(const std::string& name, QWidget* parent)
+{
+    std::string error, warning;
+    WalletModel* wallet_model = getOrCreateWallet(m_node.loadWallet(name, error, warning));
+    if (!wallet_model) {
+        QMessageBox::warning(parent, tr("Open Wallet"), QString::fromStdString(error));
+    }
+    if (!warning.empty()) {
+        QMessageBox::information(parent, tr("Open Wallet"), QString::fromStdString(warning));
+    }
+    return wallet_model;
+}
+
 WalletModel* WalletController::getOrCreateWallet(std::unique_ptr<interfaces::Wallet> wallet)
 {
     QMutexLocker locker(&m_mutex);

diff --git a/src/qt/walletcontroller.h b/src/qt/walletcontroller.h
@@ -37,6 +37,9 @@ class WalletController : public QObject
     ~WalletController();
 
     std::vector<WalletModel*> getWallets() const;
+    std::vector<std::string> getWalletsAvailableToOpen() const;
+
+    WalletModel* openWallet(const std::string& name, QWidget* parent = nullptr);
 
 private Q_SLOTS:
     void addWallet(WalletModel* wallet_model);