gui: toggle chain via menu

The selected chain is stored in the QSettings for mainnet.
bitcoin-core · Sep 1, 2021 · 0391893 · 0391893
1 parent 7be143a
commit 0391893
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 0 deletions.
diff --git a/src/qt/bitcoin.cpp b/src/qt/bitcoin.cpp
@@ -559,6 +559,10 @@ int GuiMain(int argc, char* argv[])
     // - QSettings() will use the new application name after this, resulting in network-specific settings
     // - Needs to be done before createOptionsModel
 
+    QSettings settings;
+    const std::string chain = settings.value("chain", "").toString().toStdString();
+    if (chain != "") gArgs.SoftSetArg("-chain", chain);
+
     // Check for chain settings (Params() calls are only valid after this clause)
     try {
         SelectParams(gArgs.GetChainName());

diff --git a/src/qt/bitcoingui.cpp b/src/qt/bitcoingui.cpp
@@ -441,6 +441,41 @@ void BitcoinGUI::createActions()
     }
 #endif // ENABLE_WALLET
 
+    m_chain_selection_action = new QAction(tr("Switch chain"), this);
+    m_chain_selection_action->setStatusTip(tr("Restart application using a different (test) network"));
+    m_chain_selection_menu = new QMenu(this);
+
+    connect(m_chain_selection_menu, &QMenu::aboutToShow, [this] {
+        m_chain_selection_menu->clear();
+        const std::vector<std::pair<QString, QString>> chains = {{"main", "Bitcoin"}, {"test", "Testnet"}, {"regtest", "Regtest"}, {"signet", "Signet"}};
+        for (auto chain : chains) {
+            QAction* action = m_chain_selection_menu->addAction(chain.second);
+
+            if (gArgs.GetChainName() == chain.first.toStdString()) {
+              action->setEnabled(false);
+            }
+
+            connect(action, &QAction::triggered, [this, chain] {
+              QMessageBox::StandardButton btnRetVal = QMessageBox::question(this, tr("Confirm chain"),
+                  tr("Client restart required to switch chain.") + "<br><br>" + tr("Client will be shut down. Do you want to proceed?"),
+                  QMessageBox::Yes | QMessageBox::Cancel, QMessageBox::Cancel);
+
+              if(btnRetVal == QMessageBox::Cancel) return;
+
+              // QSettings are stored seperately for each network. Switch application name
+              // to mainnet before storing the selected chain.
+              QScopedPointer<const NetworkStyle> networkStyle(NetworkStyle::instantiate("main"));
+              assert(!networkStyle.isNull());
+              QApplication::setApplicationName(networkStyle->getAppName());
+
+              QSettings settings;
+              settings.setValue("chain", chain.first);
+              QApplication::quit();
+            });
+        }
+    });
+    m_chain_selection_action->setMenu(m_chain_selection_menu);
+
     connect(new QShortcut(QKeySequence(Qt::CTRL + Qt::SHIFT + Qt::Key_C), this), &QShortcut::activated, this, &BitcoinGUI::showDebugWindowActivateConsole);
     connect(new QShortcut(QKeySequence(Qt::CTRL + Qt::SHIFT + Qt::Key_D), this), &QShortcut::activated, this, &BitcoinGUI::showDebugWindow);
 }
@@ -472,6 +507,8 @@ void BitcoinGUI::createMenuBar()
         file->addAction(m_load_psbt_clipboard_action);
         file->addSeparator();
     }
+    file->addAction(m_chain_selection_action);
+    file->addSeparator();
     file->addAction(quitAction);
 
     QMenu *settings = appMenuBar->addMenu(tr("&Settings"));

diff --git a/src/qt/bitcoingui.h b/src/qt/bitcoingui.h
@@ -165,6 +165,9 @@ class BitcoinGUI : public QMainWindow
     QAction* m_wallet_selector_action = nullptr;
     QAction* m_mask_values_action{nullptr};
 
+    QAction* m_chain_selection_action{nullptr};
+    QMenu* m_chain_selection_menu{nullptr};
+
     QLabel *m_wallet_selector_label = nullptr;
     QComboBox* m_wallet_selector = nullptr;