Merge pull request #491 from zcoinofficial/remint-unlock

Add unlock prompt to Remint page
firoorg · Jul 5, 2019 · 27a1eec · 27a1eec
2 parents f72f454 + 372e432
commit 27a1eec
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/src/qt/walletview.cpp b/src/qt/walletview.cpp
@@ -263,6 +263,7 @@ void WalletView::setWalletModel(WalletModel *walletModel)
     znodeListPage->setWalletModel(walletModel);
     sendZcoinView->setModel(walletModel);
     exoAssetsPage->setWalletModel(walletModel);
+    zc2SigmaPage->setWalletModel(walletModel);
 
     if (exodusTransactionsView) {
         exodusTransactionsView->setWalletModel(walletModel);

diff --git a/src/qt/zc2sigmapage.cpp b/src/qt/zc2sigmapage.cpp
@@ -18,6 +18,7 @@
 #include "zc2sigmamodel.h"
 #include "znode-sync.h"
 #include "clientmodel.h"
+#include "walletmodel.h"
 
 #include "../wallet/wallet.h"
 #include "main.h"
@@ -90,7 +91,16 @@ void Zc2SigmaPage::setClientModel(ClientModel *clientModel_) {
     clientModel = clientModel_;
 }
 
+void Zc2SigmaPage::setWalletModel(WalletModel *walletModel_) {
+    walletModel = walletModel_;
+}
+
 void Zc2SigmaPage::on_remintButton_clicked() {
+
+    WalletModel::UnlockContext ctx(walletModel->requestUnlock());
+    if (!ctx.isValid()) {
+        return;
+    }
     QItemSelectionModel * select = ui->availMintsTable->selectionModel();
 
     if(!select->hasSelection())

diff --git a/src/qt/zc2sigmapage.h b/src/qt/zc2sigmapage.h
@@ -14,6 +14,7 @@ class OptionsModel;
 class PlatformStyle;
 class Zc2SigmaModel;
 class ClientModel;
+class WalletModel;
 
 namespace Ui {
     class Zc2SigmaPage;
@@ -35,13 +36,15 @@ class Zc2SigmaPage : public QWidget
     ~Zc2SigmaPage();
     void createModel();
     void setClientModel(ClientModel *clientModel_);
+    void setWalletModel(WalletModel *walletModel_);
 
     static bool showZc2SigmaPage();
 
 private:
     Ui::Zc2SigmaPage *ui;
     std::shared_ptr<Zc2SigmaModel> model;
     ClientModel *clientModel;
+    WalletModel *walletModel;
 
 protected:
     void showEvent(QShowEvent* event);