src/qt/walletview.cpp

   1 // Copyright (c) 2011-2016 The Bitcoin Core developers
   2 // Distributed under the MIT software license, see the accompanying
   3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   4
   5 #include "walletview.h"
   6
   7 #include "addressbookpage.h"
   8 #include "askpassphrasedialog.h"
   9 #include "bitcoingui.h"
  10 #include "clientmodel.h"
  11 #include "guiutil.h"
  12 #include "optionsmodel.h"
  13 #include "overviewpage.h"
  14 #include "platformstyle.h"
  15 #include "receivecoinsdialog.h"
  16 #include "sendcoinsdialog.h"
  17 #include "signverifymessagedialog.h"
  18 #include "transactiontablemodel.h"
  19 #include "transactionview.h"
  20 #include "walletmodel.h"
  21
  22 #include "ui_interface.h"
  23
  24 #include <QAction>
  25 #include <QActionGroup>
  26 #include <QFileDialog>
  27 #include <QHBoxLayout>
  28 #include <QProgressDialog>
  29 #include <QPushButton>
  30 #include <QVBoxLayout>
  31
  32 WalletView::WalletView(const PlatformStyle *_platformStyle, QWidget *parent):
  33     QStackedWidget(parent),
  34     clientModel(0),
  35     walletModel(0),
  36     platformStyle(_platformStyle)
  37 {
  38     // Create tabs
  39     overviewPage = new OverviewPage(platformStyle);
  40
  41     transactionsPage = new QWidget(this);
  42     QVBoxLayout *vbox = new QVBoxLayout();
  43     QHBoxLayout *hbox_buttons = new QHBoxLayout();
  44     transactionView = new TransactionView(platformStyle, this);
  45     vbox->addWidget(transactionView);
  46     QPushButton *exportButton = new QPushButton(tr("&Export"), this);
  47     exportButton->setToolTip(tr("Export the data in the current tab to a file"));
  48     if (platformStyle->getImagesOnButtons()) {
  49         exportButton->setIcon(platformStyle->SingleColorIcon(":/icons/export"));
  50     }
  51     hbox_buttons->addStretch();
  52     hbox_buttons->addWidget(exportButton);
  53     vbox->addLayout(hbox_buttons);
  54     transactionsPage->setLayout(vbox);
  55
  56     receiveCoinsPage = new ReceiveCoinsDialog(platformStyle);
  57     sendCoinsPage = new SendCoinsDialog(platformStyle);
  58
  59     usedSendingAddressesPage = new AddressBookPage(platformStyle, AddressBookPage::ForEditing, AddressBookPage::SendingTab, this);
  60     usedReceivingAddressesPage = new AddressBookPage(platformStyle, AddressBookPage::ForEditing, AddressBookPage::ReceivingTab, this);
  61
  62     addWidget(overviewPage);
  63     addWidget(transactionsPage);
  64     addWidget(receiveCoinsPage);
  65     addWidget(sendCoinsPage);
  66
  67     // Clicking on a transaction on the overview pre-selects the transaction on the transaction history page
  68     connect(overviewPage, SIGNAL(transactionClicked(QModelIndex)), transactionView, SLOT(focusTransaction(QModelIndex)));
  69     connect(overviewPage, SIGNAL(outOfSyncWarningClicked()), this, SLOT(requestedSyncWarningInfo()));
  70
  71     // Double-clicking on a transaction on the transaction history page shows details
  72     connect(transactionView, SIGNAL(doubleClicked(QModelIndex)), transactionView, SLOT(showDetails()));
  73
  74     // Clicking on "Export" allows to export the transaction list
  75     connect(exportButton, SIGNAL(clicked()), transactionView, SLOT(exportClicked()));
  76
  77     // Pass through messages from sendCoinsPage
  78     connect(sendCoinsPage, SIGNAL(message(QString,QString,unsigned int)), this, SIGNAL(message(QString,QString,unsigned int)));
  79     // Pass through messages from transactionView
  80     connect(transactionView, SIGNAL(message(QString,QString,unsigned int)), this, SIGNAL(message(QString,QString,unsigned int)));
  81 }
  82
  83 WalletView::~WalletView()
  84 {
  85 }
  86
  87 void WalletView::setBitcoinGUI(BitcoinGUI *gui)
  88 {
  89     if (gui)
  90     {
  91         // Clicking on a transaction on the overview page simply sends you to transaction history page
  92         connect(overviewPage, SIGNAL(transactionClicked(QModelIndex)), gui, SLOT(gotoHistoryPage()));
  93
  94         // Receive and report messages
  95         connect(this, SIGNAL(message(QString,QString,unsigned int)), gui, SLOT(message(QString,QString,unsigned int)));
  96
  97         // Pass through encryption status changed signals
  98         connect(this, SIGNAL(encryptionStatusChanged(int)), gui, SLOT(setEncryptionStatus(int)));
  99
 100         // Pass through transaction notifications
 101         connect(this, SIGNAL(incomingTransaction(QString,int,CAmount,QString,QString,QString)), gui, SLOT(incomingTransaction(QString,int,CAmount,QString,QString,QString)));
 102
 103         // Connect HD enabled state signal
 104         connect(this, SIGNAL(hdEnabledStatusChanged(int)), gui, SLOT(setHDStatus(int)));
 105     }
 106 }
 107
 108 void WalletView::setClientModel(ClientModel *_clientModel)
 109 {
 110     this->clientModel = _clientModel;
 111
 112     overviewPage->setClientModel(_clientModel);
 113     sendCoinsPage->setClientModel(_clientModel);
 114 }
 115
 116 void WalletView::setWalletModel(WalletModel *_walletModel)
 117 {
 118     this->walletModel = _walletModel;
 119
 120     // Put transaction list in tabs
 121     transactionView->setModel(_walletModel);
 122     overviewPage->setWalletModel(_walletModel);
 123     receiveCoinsPage->setModel(_walletModel);
 124     sendCoinsPage->setModel(_walletModel);
 125     usedReceivingAddressesPage->setModel(_walletModel ? _walletModel->getAddressTableModel() : nullptr);
 126     usedSendingAddressesPage->setModel(_walletModel ? _walletModel->getAddressTableModel() : nullptr);
 127
 128     if (_walletModel)
 129     {
 130         // Receive and pass through messages from wallet model
 131         connect(_walletModel, SIGNAL(message(QString,QString,unsigned int)), this, SIGNAL(message(QString,QString,unsigned int)));
 132
 133         // Handle changes in encryption status
 134         connect(_walletModel, SIGNAL(encryptionStatusChanged(int)), this, SIGNAL(encryptionStatusChanged(int)));
 135         updateEncryptionStatus();
 136
 137         // update HD status
 138         Q_EMIT hdEnabledStatusChanged(_walletModel->hdEnabled());
 139
 140         // Balloon pop-up for new transaction
 141         connect(_walletModel->getTransactionTableModel(), SIGNAL(rowsInserted(QModelIndex,int,int)),
 142                 this, SLOT(processNewTransaction(QModelIndex,int,int)));
 143
 144         // Ask for passphrase if needed
 145         connect(_walletModel, SIGNAL(requireUnlock()), this, SLOT(unlockWallet()));
 146
 147         // Show progress dialog
 148         connect(_walletModel, SIGNAL(showProgress(QString,int)), this, SLOT(showProgress(QString,int)));
 149     }
 150 }
 151
 152 void WalletView::processNewTransaction(const QModelIndex& parent, int start, int /*end*/)
 153 {
 154     // Prevent balloon-spam when initial block download is in progress
 155     if (!walletModel || !clientModel || clientModel->inInitialBlockDownload())
 156         return;
 157
 158     TransactionTableModel *ttm = walletModel->getTransactionTableModel();
 159     if (!ttm || ttm->processingQueuedTransactions())
 160         return;
 161
 162     QString date = ttm->index(start, TransactionTableModel::Date, parent).data().toString();
 163     qint64 amount = ttm->index(start, TransactionTableModel::Amount, parent).data(Qt::EditRole).toULongLong();
 164     QString type = ttm->index(start, TransactionTableModel::Type, parent).data().toString();
 165     QModelIndex index = ttm->index(start, 0, parent);
 166     QString address = ttm->data(index, TransactionTableModel::AddressRole).toString();
 167     QString label = ttm->data(index, TransactionTableModel::LabelRole).toString();
 168
 169     Q_EMIT incomingTransaction(date, walletModel->getOptionsModel()->getDisplayUnit(), amount, type, address, label);
 170 }
 171
 172 void WalletView::gotoOverviewPage()
 173 {
 174     setCurrentWidget(overviewPage);
 175 }
 176
 177 void WalletView::gotoHistoryPage()
 178 {
 179     setCurrentWidget(transactionsPage);
 180 }
 181
 182 void WalletView::gotoReceiveCoinsPage()
 183 {
 184     setCurrentWidget(receiveCoinsPage);
 185 }
 186
 187 void WalletView::gotoSendCoinsPage(QString addr)
 188 {
 189     setCurrentWidget(sendCoinsPage);
 190
 191     if (!addr.isEmpty())
 192         sendCoinsPage->setAddress(addr);
 193 }
 194
 195 void WalletView::gotoSignMessageTab(QString addr)
 196 {
 197     // calls show() in showTab_SM()
 198     SignVerifyMessageDialog *signVerifyMessageDialog = new SignVerifyMessageDialog(platformStyle, this);
 199     signVerifyMessageDialog->setAttribute(Qt::WA_DeleteOnClose);
 200     signVerifyMessageDialog->setModel(walletModel);
 201     signVerifyMessageDialog->showTab_SM(true);
 202
 203     if (!addr.isEmpty())
 204         signVerifyMessageDialog->setAddress_SM(addr);
 205 }
 206
 207 void WalletView::gotoVerifyMessageTab(QString addr)
 208 {
 209     // calls show() in showTab_VM()
 210     SignVerifyMessageDialog *signVerifyMessageDialog = new SignVerifyMessageDialog(platformStyle, this);
 211     signVerifyMessageDialog->setAttribute(Qt::WA_DeleteOnClose);
 212     signVerifyMessageDialog->setModel(walletModel);
 213     signVerifyMessageDialog->showTab_VM(true);
 214
 215     if (!addr.isEmpty())
 216         signVerifyMessageDialog->setAddress_VM(addr);
 217 }
 218
 219 bool WalletView::handlePaymentRequest(const SendCoinsRecipient& recipient)
 220 {
 221     return sendCoinsPage->handlePaymentRequest(recipient);
 222 }
 223
 224 void WalletView::showOutOfSyncWarning(bool fShow)
 225 {
 226     overviewPage->showOutOfSyncWarning(fShow);
 227 }
 228
 229 void WalletView::updateEncryptionStatus()
 230 {
 231     Q_EMIT encryptionStatusChanged(walletModel->getEncryptionStatus());
 232 }
 233
 234 void WalletView::encryptWallet(bool status)
 235 {
 236     if(!walletModel)
 237         return;
 238     AskPassphraseDialog dlg(status ? AskPassphraseDialog::Encrypt : AskPassphraseDialog::Decrypt, this);
 239     dlg.setModel(walletModel);
 240     dlg.exec();
 241
 242     updateEncryptionStatus();
 243 }
 244
 245 void WalletView::backupWallet()
 246 {
 247     QString filename = GUIUtil::getSaveFileName(this,
 248         tr("Backup Wallet"), QString(),
 249         tr("Wallet Data (*.dat)"), nullptr);
 250
 251     if (filename.isEmpty())
 252         return;
 253
 254     if (!walletModel->backupWallet(filename)) {
 255         Q_EMIT message(tr("Backup Failed"), tr("There was an error trying to save the wallet data to %1.").arg(filename),
 256             CClientUIInterface::MSG_ERROR);
 257         }
 258     else {
 259         Q_EMIT message(tr("Backup Successful"), tr("The wallet data was successfully saved to %1.").arg(filename),
 260             CClientUIInterface::MSG_INFORMATION);
 261     }
 262 }
 263
 264 void WalletView::changePassphrase()
 265 {
 266     AskPassphraseDialog dlg(AskPassphraseDialog::ChangePass, this);
 267     dlg.setModel(walletModel);
 268     dlg.exec();
 269 }
 270
 271 void WalletView::unlockWallet()
 272 {
 273     if(!walletModel)
 274         return;
 275     // Unlock wallet when requested by wallet model
 276     if (walletModel->getEncryptionStatus() == WalletModel::Locked)
 277     {
 278         AskPassphraseDialog dlg(AskPassphraseDialog::Unlock, this);
 279         dlg.setModel(walletModel);
 280         dlg.exec();
 281     }
 282 }
 283
 284 void WalletView::usedSendingAddresses()
 285 {
 286     if(!walletModel)
 287         return;
 288
 289     usedSendingAddressesPage->show();
 290     usedSendingAddressesPage->raise();
 291     usedSendingAddressesPage->activateWindow();
 292 }
 293
 294 void WalletView::usedReceivingAddresses()
 295 {
 296     if(!walletModel)
 297         return;
 298
 299     usedReceivingAddressesPage->show();
 300     usedReceivingAddressesPage->raise();
 301     usedReceivingAddressesPage->activateWindow();
 302 }
 303
 304 void WalletView::showProgress(const QString &title, int nProgress)
 305 {
 306     if (nProgress == 0)
 307     {
 308         progressDialog = new QProgressDialog(title, "", 0, 100);
 309         progressDialog->setWindowModality(Qt::ApplicationModal);
 310         progressDialog->setMinimumDuration(0);
 311         progressDialog->setCancelButton(0);
 312         progressDialog->setAutoClose(false);
 313         progressDialog->setValue(0);
 314     }
 315     else if (nProgress == 100)
 316     {
 317         if (progressDialog)
 318         {
 319             progressDialog->close();
 320             progressDialog->deleteLater();
 321         }
 322     }
 323     else if (progressDialog)
 324         progressDialog->setValue(nProgress);
 325 }
 326
 327 void WalletView::requestedSyncWarningInfo()
 328 {
 329     Q_EMIT outOfSyncWarningClicked();
 330 }