text: // Copyright (c) 2017-2021 The Merit Foundation
  // Copyright (c) 2011-2016 The Bitcoin Core developers
  // Distributed under the MIT software license, see the accompanying
  // file COPYING or http://www.opensource.org/licenses/mit-license.php.
  
  #include "walletview.h"
  
  #include "addressbookpage.h"
  #include "askpassphrasedialog.h"
  #include "meritgui.h"
  #include "clientmodel.h"
  #include "guiutil.h"
  #include "optionsmodel.h"
  #include "overviewpage.h"
  #include "platformstyle.h"
  #include "receivecoinsdialog.h"
  #include "sendcoinsdialog.h"
  #include "signverifymessagedialog.h"
  #include "transactiontablemodel.h"
  #include "transactionview.h"
  #include "walletmodel.h"
  
  #include "ui_interface.h"
  
  #include <QAction>
  #include <QActionGroup>
  #include <QFileDialog>
  #include <QHBoxLayout>
  #include <QProgressDialog>
  #include <QPushButton>
  #include <QVBoxLayout>
  
  WalletView::WalletView(const PlatformStyle *_platformStyle, QWidget *parent):
      QStackedWidget(parent),
      clientModel(0),
      walletModel(0),
      platformStyle(_platformStyle)
  {
      // Create tabs
      overviewPage = new OverviewPage(platformStyle);
  
      transactionsPage = new QWidget(this);
      QVBoxLayout *vbox = new QVBoxLayout();
      QHBoxLayout *hbox_buttons = new QHBoxLayout();
      transactionView = new TransactionView(platformStyle, this);
      vbox->addWidget(transactionView);
      QPushButton *exportButton = new QPushButton(tr("&Export"), this);
      exportButton->setToolTip(tr("Export the data in the current tab to a file"));
      if (platformStyle->getImagesOnButtons()) {
          exportButton->setIcon(platformStyle->SingleColorIcon(":/icons/export"));
      }
      hbox_buttons->addWidget(exportButton);
      hbox_buttons->addStretch();
      vbox->addLayout(hbox_buttons);
      transactionsPage->setLayout(vbox);
  
      receiveCoinsPage = new ReceiveCoinsDialog(platformStyle);
      sendCoinsPage = new SendCoinsDialog(platformStyle);
  
      addWidget(overviewPage);
      addWidget(transactionsPage);
      addWidget(receiveCoinsPage);
      addWidget(sendCoinsPage);
  
      // Clicking on a transaction on the overview pre-selects the transaction on the transaction history page
      connect(overviewPage, SIGNAL(transactionClicked(QModelIndex)), transactionView, SLOT(focusTransaction(QModelIndex)));
      connect(overviewPage, SIGNAL(outOfSyncWarningClicked()), this, SLOT(requestedSyncWarningInfo()));
      connect(this, SIGNAL(DoUpdateOverviewPage()), overviewPage, SLOT(Update()));
  
      // Double-clicking on a transaction on the transaction history page shows details
      connect(transactionView, SIGNAL(doubleClicked(QModelIndex)), transactionView, SLOT(showDetails()));
  
      // Clicking on "Export" allows to export the transaction list
      connect(exportButton, SIGNAL(clicked()), transactionView, SLOT(exportClicked()));
  
      // Pass through messages from sendCoinsPage
      connect(sendCoinsPage, SIGNAL(message(QString,QString,unsigned int)), this, SIGNAL(message(QString,QString,unsigned int)));
      // Pass through messages from transactionView
      connect(transactionView, SIGNAL(message(QString,QString,unsigned int)), this, SIGNAL(message(QString,QString,unsigned int)));
  }
  
  WalletView::~WalletView()
  {
  }
  
  void WalletView::setMeritGUI(MeritGUI *gui)
  {
      if (gui)
      {
          // Clicking on a transaction on the overview page simply sends you to transaction history page
          connect(overviewPage, SIGNAL(transactionClicked(QModelIndex)), gui, SLOT(gotoHistoryPage()));
  
          // Receive and report messages
          connect(this, SIGNAL(message(QString,QString,unsigned int)), gui, SLOT(message(QString,QString,unsigned int)));
  
          // Pass through encryption status changed signals
          connect(this, SIGNAL(encryptionStatusChanged(int)), gui, SLOT(setEncryptionStatus(int)));
  
          // Pass through transaction notifications
          connect(
                  this, SIGNAL(incomingTransaction(QString,int,CAmount,QString,QString,QString,QString)),
                  gui, SLOT(incomingTransaction(QString,int,CAmount,QString,QString,QString,QString)));
  
          // Connect HD enabled state signal
          connect(this, SIGNAL(hdEnabledStatusChanged(int)), gui, SLOT(setHDStatus(int)));
      }
  }
  
  void WalletView::setClientModel(ClientModel *_clientModel)
  {
      this->clientModel = _clientModel;
  
      overviewPage->setClientModel(_clientModel);
      sendCoinsPage->setClientModel(_clientModel);
  }
  
  void WalletView::setWalletModel(WalletModel *_walletModel)
  {
      this->walletModel = _walletModel;
  
      // Put transaction list in tabs
      transactionView->setModel(_walletModel);
      overviewPage->setWalletModel(_walletModel);
      receiveCoinsPage->setModel(_walletModel);
      sendCoinsPage->setModel(_walletModel);
  
      usedSendingAddressesPage = new AddressBookPage(platformStyle, AddressBookPage::ForEditing, AddressBookPage::SendingTab, this, walletModel);
      usedReceivingAddressesPage = new AddressBookPage(platformStyle, AddressBookPage::ForEditing, AddressBookPage::ReceivingTab, this, walletModel);
  
      usedReceivingAddressesPage->setModel(_walletModel->getAddressTableModel());
      usedSendingAddressesPage->setModel(_walletModel->getAddressTableModel());
  
      if (_walletModel)
      {
          // Receive and pass through messages from wallet model
          connect(_walletModel, SIGNAL(message(QString,QString,unsigned int)), this, SIGNAL(message(QString,QString,unsigned int)));
  
          // Handle changes in encryption status
          connect(_walletModel, SIGNAL(encryptionStatusChanged(int)), this, SIGNAL(encryptionStatusChanged(int)));
          updateEncryptionStatus();
  
          // update HD status
          Q_EMIT hdEnabledStatusChanged(_walletModel->hdEnabled());
  
          // Balloon pop-up for new transaction
          connect(_walletModel->getTransactionTableModel(), SIGNAL(rowsInserted(QModelIndex,int,int)),
                  this, SLOT(processNewTransaction(QModelIndex,int,int)));
  
          // Ask for passphrase if needed
          connect(_walletModel, SIGNAL(requireUnlock()), this, SLOT(unlockWallet()));
  
          // Show progress dialog
          connect(_walletModel, SIGNAL(showProgress(QString,int)), this, SLOT(showProgress(QString,int)));
  
          QTimer::singleShot(3000, this, SLOT(CheckChangePassphrase()));
      }
  }
  
  void WalletView::processNewTransaction(const QModelIndex& parent, int start, int /*end*/)
  {
      // Prevent balloon-spam when initial block download is in progress
      if (!walletModel || !clientModel || clientModel->inInitialBlockDownload())
          return;
  
      TransactionTableModel *ttm = walletModel->getTransactionTableModel();
      if (!ttm || ttm->processingQueuedTransactions())
          return;
  
      QString date = ttm->index(start, TransactionTableModel::Date, parent).data().toString();
      QString type = ttm->index(start, TransactionTableModel::Type, parent).data().toString();
  
      qint64 amount;
      if (type.contains("invite"))
          amount = ttm->index(start, TransactionTableModel::Invite, parent).data(Qt::EditRole).toULongLong();
      else
          amount = ttm->index(start, TransactionTableModel::Amount, parent).data(Qt::EditRole).toULongLong();
  
      QModelIndex index = ttm->index(start, 0, parent);
      QString from = ttm->data(index, TransactionTableModel::FromRole).toString();
      QString to = ttm->data(index, TransactionTableModel::ToRole).toString();
      QString label = ttm->data(index, TransactionTableModel::LabelRole).toString();
  
      Q_EMIT incomingTransaction(
              date,
              walletModel->getOptionsModel()->getDisplayUnit(),
              amount,
              type,
              from,
              to,
              label);
  }
  
  void WalletView::gotoOverviewPage()
  {
      setCurrentWidget(overviewPage);
  }
  
  void WalletView::gotoHistoryPage()
  {
      setCurrentWidget(transactionsPage);
  }
  
  void WalletView::gotoReceiveCoinsPage()
  {
      setCurrentWidget(receiveCoinsPage);
  }
  
  void WalletView::gotoSendCoinsPage(QString addr)
  {
      setCurrentWidget(sendCoinsPage);
  
      if (!addr.isEmpty())
          sendCoinsPage->setAddress(addr);
  }
  
  void WalletView::gotoSignMessageTab(QString addr)
  {
      // calls show() in showTab_SM()
      SignVerifyMessageDialog *signVerifyMessageDialog = new SignVerifyMessageDialog(platformStyle, this);
      signVerifyMessageDialog->setAttribute(Qt::WA_DeleteOnClose);
      signVerifyMessageDialog->setModel(walletModel);
      signVerifyMessageDialog->showTab_SM(true);
  
      if (!addr.isEmpty())
          signVerifyMessageDialog->setAddress_SM(addr);
  }
  
  void WalletView::gotoVerifyMessageTab(QString addr)
  {
      // calls show() in showTab_VM()
      SignVerifyMessageDialog *signVerifyMessageDialog = new SignVerifyMessageDialog(platformStyle, this);
      signVerifyMessageDialog->setAttribute(Qt::WA_DeleteOnClose);
      signVerifyMessageDialog->setModel(walletModel);
      signVerifyMessageDialog->showTab_VM(true);
  
      if (!addr.isEmpty())
          signVerifyMessageDialog->setAddress_VM(addr);
  }
  
  bool WalletView::handlePaymentRequest(const SendCoinsRecipient& recipient)
  {
      return sendCoinsPage->handlePaymentRequest(recipient);
  }
  
  void WalletView::showOutOfSyncWarning(bool fShow)
  {
      overviewPage->showOutOfSyncWarning(fShow);
  }
  
  void WalletView::updateEncryptionStatus()
  {
      Q_EMIT encryptionStatusChanged(walletModel->getEncryptionStatus());
  }
  
  void WalletView::encryptWallet(bool status)
  {
      if(!walletModel)
          return;
      AskPassphraseDialog dlg(status ? AskPassphraseDialog::Encrypt : AskPassphraseDialog::Decrypt, this);
      dlg.setModel(walletModel);
      dlg.exec();
  
      updateEncryptionStatus();
  }
  
  void WalletView::backupWallet()
  {
      QString filename = GUIUtil::getSaveFileName(this,
          tr("Backup Wallet"), QString(),
          tr("Wallet Data (*.dat)"), nullptr);
  
      if (filename.isEmpty())
          return;
  
      if (!walletModel->backupWallet(filename)) {
          Q_EMIT message(tr("Backup Failed"), tr("There was an error trying to save the wallet data to %1.").arg(filename),
              CClientUIInterface::MSG_ERROR);
          }
      else {
          Q_EMIT message(tr("Backup Successful"), tr("The wallet data was successfully saved to %1.").arg(filename),
              CClientUIInterface::MSG_INFORMATION);
      }
  }
  
  void WalletView::changePassphrase()
  {
      AskPassphraseDialog dlg(AskPassphraseDialog::ChangePass, this);
      dlg.setModel(walletModel);
      dlg.exec();
  }
  
  void WalletView::CheckChangePassphrase()
  {
      assert(walletModel);
      if(!walletModel->IsReferred()) {
          return;
      }
  
      if(walletModel->CryptedWalletNeedsNewPassphrase()) {
          QMessageBox msgBox{QMessageBox::Question,
              "Insecure Mnemonic", 
              "Versions prior to 0.5.2 did not properly encrypt the mnemonic. "
              "It looks like your wallet needs to be upgraded. "
              "We recommend you change your passphrase to secure your wallet. "
              "Do you want to change your passphrase now?",
              QMessageBox::Yes | QMessageBox::No,
              this};
          msgBox.setStyleSheet(QString("QMessageBox { background-color: white; }"));
          auto ret = msgBox.exec();
          
          if(ret == QMessageBox::Yes) {
              changePassphrase();
          }
      }
  }
  
  void WalletView::unlockWallet()
  {
      if(!walletModel)
          return;
      // Unlock wallet when requested by wallet model
      if (walletModel->getEncryptionStatus() == WalletModel::Locked)
      {
          AskPassphraseDialog dlg(AskPassphraseDialog::Unlock, this);
          dlg.setModel(walletModel);
          dlg.exec();
      }
  }
  
  void WalletView::usedSendingAddresses()
  {
      if(!walletModel)
          return;
  
      usedSendingAddressesPage->show();
      usedSendingAddressesPage->raise();
      usedSendingAddressesPage->activateWindow();
  }
  
  void WalletView::usedReceivingAddresses()
  {
      if(!walletModel)
          return;
  
      usedReceivingAddressesPage->show();
      usedReceivingAddressesPage->raise();
      usedReceivingAddressesPage->activateWindow();
  }
  
  void WalletView::showProgress(const QString &title, int nProgress)
  {
      if (nProgress == 0)
      {
          progressDialog = new QProgressDialog(title, "", 0, 100);
          progressDialog->setWindowModality(Qt::ApplicationModal);
          progressDialog->setMinimumDuration(0);
          progressDialog->setCancelButton(0);
          progressDialog->setAutoClose(false);
          progressDialog->setValue(0);
      }
      else if (nProgress == 100)
      {
          if (progressDialog)
          {
              progressDialog->close();
              progressDialog->deleteLater();
          }
      }
      else if (progressDialog)
          progressDialog->setValue(nProgress);
  }
  
  void WalletView::requestedSyncWarningInfo()
  {
      Q_EMIT outOfSyncWarningClicked();
  }
  
  void WalletView::UpdateOverviewPage()
  {
      Q_EMIT DoUpdateOverviewPage();
  }
  