From 68c493b736cf8560948a29abd847756c4701a3dc Mon Sep 17 00:00:00 2001
From: tobtoht <thotbot@protonmail.com>
Date: Wed, 23 Dec 2020 03:57:01 +0100
Subject: [PATCH] Wizard: autoselect open if wallets available & refresh
 wallets when wizard opened

---
 src/wizard/menu.cpp         | 19 ++++++++++++++-----
 src/wizard/menu.h           |  5 +++--
 src/wizard/menu.ui          | 16 ++++++++++++++--
 src/wizard/openwallet.cpp   | 21 +++++++++++----------
 src/wizard/openwallet.h     |  7 +++----
 src/wizard/walletwizard.cpp | 14 ++++++++++----
 src/wizard/walletwizard.h   |  2 ++
 7 files changed, 57 insertions(+), 27 deletions(-)

diff --git a/src/wizard/menu.cpp b/src/wizard/menu.cpp
index 89a5556..339fda5 100644
--- a/src/wizard/menu.cpp
+++ b/src/wizard/menu.cpp
@@ -11,13 +11,22 @@
 
 #include "libwalletqt/WalletManager.h"
 
-MenuPage::MenuPage(AppContext *ctx, QWidget *parent) :
-        QWizardPage(parent),
-        ui(new Ui::MenuPage),
-        m_ctx(ctx) {
+MenuPage::MenuPage(AppContext *ctx, WalletKeysFilesModel *wallets, QWidget *parent)
+        : QWizardPage(parent)
+        , ui(new Ui::MenuPage)
+        , m_ctx(ctx)
+        , m_walletKeysFilesModel(wallets)
+{
     ui->setupUi(this);
     this->setButtonText(QWizard::FinishButton, "Open recent wallet");
-    ui->radioCreate->setChecked(true);
+}
+
+void MenuPage::initializePage() {
+    if (m_walletKeysFilesModel->rowCount() > 0) {
+        ui->radioOpen->setChecked(true);
+    } else {
+        ui->radioCreate->setChecked(true);
+    }
 }
 
 int MenuPage::nextId() const {
diff --git a/src/wizard/menu.h b/src/wizard/menu.h
index 3b1c7bb..2fb24a2 100644
--- a/src/wizard/menu.h
+++ b/src/wizard/menu.h
@@ -19,13 +19,14 @@ class MenuPage : public QWizardPage
     Q_OBJECT
 
 public:
-    explicit MenuPage(AppContext *ctx, QWidget *parent = nullptr);
+    explicit MenuPage(AppContext *ctx, WalletKeysFilesModel *wallets, QWidget *parent = nullptr);
+    void initializePage() override;
     bool validatePage() override;
     int nextId() const override;
 
 private:
     AppContext *m_ctx;
-    QLabel *topLabel;
+    WalletKeysFilesModel *m_walletKeysFilesModel;
     Ui::MenuPage *ui;
 };
 
diff --git a/src/wizard/menu.ui b/src/wizard/menu.ui
index c022518..a45d1ee 100644
--- a/src/wizard/menu.ui
+++ b/src/wizard/menu.ui
@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>424</width>
-    <height>245</height>
+    <width>617</width>
+    <height>463</height>
    </rect>
   </property>
   <property name="windowTitle">
@@ -29,6 +29,9 @@
    </item>
    <item>
     <widget class="QRadioButton" name="radioCreate">
+     <property name="focusPolicy">
+      <enum>Qt::ClickFocus</enum>
+     </property>
      <property name="text">
       <string>Create new wallet</string>
      </property>
@@ -36,6 +39,9 @@
    </item>
    <item>
     <widget class="QRadioButton" name="radioOpen">
+     <property name="focusPolicy">
+      <enum>Qt::ClickFocus</enum>
+     </property>
      <property name="text">
       <string>Open wallet file</string>
      </property>
@@ -43,6 +49,9 @@
    </item>
    <item>
     <widget class="QRadioButton" name="radioSeed">
+     <property name="focusPolicy">
+      <enum>Qt::ClickFocus</enum>
+     </property>
      <property name="text">
       <string>Restore wallet from seed</string>
      </property>
@@ -50,6 +59,9 @@
    </item>
    <item>
     <widget class="QRadioButton" name="radioViewOnly">
+     <property name="focusPolicy">
+      <enum>Qt::ClickFocus</enum>
+     </property>
      <property name="text">
       <string>Import from keys</string>
      </property>
diff --git a/src/wizard/openwallet.cpp b/src/wizard/openwallet.cpp
index 6ecab7a..74facc6 100644
--- a/src/wizard/openwallet.cpp
+++ b/src/wizard/openwallet.cpp
@@ -10,12 +10,12 @@
 #include <QFileDialog>
 #include <QMessageBox>
 
-// @TODO: rescan wallet dir on wizard open
-
-OpenWalletPage::OpenWalletPage(AppContext *ctx, QWidget *parent) :
-        QWizardPage(parent),
-        ui(new Ui::OpenWalletPage),
-        m_ctx(ctx) {
+OpenWalletPage::OpenWalletPage(AppContext *ctx, WalletKeysFilesModel *wallets, QWidget *parent)
+        : QWizardPage(parent)
+        , ui(new Ui::OpenWalletPage)
+        , m_ctx(ctx)
+        , m_walletKeysFilesModel(wallets)
+{
     ui->setupUi(this);
 
     connect(ui->btnBrowse, &QPushButton::clicked, [=]{
@@ -46,11 +46,8 @@ OpenWalletPage::OpenWalletPage(AppContext *ctx, QWidget *parent) :
     ui->walletTable->setSelectionBehavior(QAbstractItemView::SelectRows);
     ui->walletTable->setContextMenuPolicy(Qt::CustomContextMenu);
 
-    this->walletKeysFilesModel = new WalletKeysFilesModel(m_ctx);
-    this->walletKeysFilesModel->refresh();
-
     m_keysProxy = new WalletKeysFilesProxyModel(this, m_ctx->networkType);
-    m_keysProxy->setSourceModel(this->walletKeysFilesModel);
+    m_keysProxy->setSourceModel(m_walletKeysFilesModel);
     m_keysProxy->setSortRole(Qt::UserRole);
 
     ui->walletTable->setModel(m_keysProxy);
@@ -67,6 +64,10 @@ OpenWalletPage::OpenWalletPage(AppContext *ctx, QWidget *parent) :
     });
 }
 
+void OpenWalletPage::initializePage() {
+    m_walletKeysFilesModel->refresh();
+}
+
 void OpenWalletPage::updatePath() {
     QModelIndex index = ui->walletTable->currentIndex();
     if (!index.isValid()) {
diff --git a/src/wizard/openwallet.h b/src/wizard/openwallet.h
index 558dcf6..a6d0b78 100644
--- a/src/wizard/openwallet.h
+++ b/src/wizard/openwallet.h
@@ -20,7 +20,8 @@ class OpenWalletPage : public QWizardPage
     Q_OBJECT
 
 public:
-    explicit OpenWalletPage(AppContext *ctx, QWidget *parent = nullptr);
+    explicit OpenWalletPage(AppContext *ctx, WalletKeysFilesModel *wallets, QWidget *parent = nullptr);
+    void initializePage() override;
     bool validatePage() override;
     int nextId() const override;
 
@@ -31,10 +32,8 @@ private:
     void updatePath();
 
     AppContext *m_ctx;
-    WalletKeysFilesModel *walletKeysFilesModel;
+    WalletKeysFilesModel *m_walletKeysFilesModel;
     WalletKeysFilesProxyModel *m_keysProxy;
-    QSortFilterProxyModel *ll;
-    QLabel *topLabel;
     Ui::OpenWalletPage *ui;
     QStandardItemModel *m_model;
 };
diff --git a/src/wizard/walletwizard.cpp b/src/wizard/walletwizard.cpp
index 01ce572..f20d3d3 100644
--- a/src/wizard/walletwizard.cpp
+++ b/src/wizard/walletwizard.cpp
@@ -18,14 +18,20 @@
 #include <QScreen>
 #include <QApplication>
 
-WalletWizard::WalletWizard(AppContext *ctx, WalletWizard::Page startPage, QWidget *parent) : QWizard(parent),
-        m_ctx(ctx) {
+WalletWizard::WalletWizard(AppContext *ctx, WalletWizard::Page startPage, QWidget *parent)
+        : QWizard(parent)
+        , m_ctx(ctx)
+{
     this->setWindowTitle("Welcome to Feather Wallet");
     this->setWindowIcon(QIcon(":/assets/images/appicons/64x64.png"));
-    auto openWalletPage = new OpenWalletPage(m_ctx, this);
+
+    m_walletKeysFilesModel = new WalletKeysFilesModel(m_ctx, this);
+    m_walletKeysFilesModel->refresh();
+
+    auto openWalletPage = new OpenWalletPage(m_ctx, m_walletKeysFilesModel, this);
     auto createWallet = new CreateWalletPage(m_ctx, this);
     auto createWalletSeed = new CreateWalletSeedPage(m_ctx, this);
-    setPage(Page_Menu, new MenuPage(m_ctx, this));
+    setPage(Page_Menu, new MenuPage(m_ctx, m_walletKeysFilesModel, this));
     setPage(Page_CreateWallet, createWallet);
     setPage(Page_OpenWallet, openWalletPage);
     setPage(Page_CreateWalletSeed, createWalletSeed);
diff --git a/src/wizard/walletwizard.h b/src/wizard/walletwizard.h
index 589ac99..acffac5 100644
--- a/src/wizard/walletwizard.h
+++ b/src/wizard/walletwizard.h
@@ -25,6 +25,8 @@ signals:
 
 private:
     AppContext *m_ctx;
+    WalletKeysFilesModel *m_walletKeysFilesModel;
+
     void createWallet();
 };