src/qt/editaddressdialog.cpp

   1 // Copyright (c) 2011-2017 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 <qt/editaddressdialog.h>
   6 #include <qt/forms/ui_editaddressdialog.h>
   7
   8 #include <qt/addresstablemodel.h>
   9 #include <qt/guiutil.h>
  10
  11 #include <QDataWidgetMapper>
  12 #include <QMessageBox>
  13
  14 EditAddressDialog::EditAddressDialog(Mode _mode, QWidget *parent) :
  15     QDialog(parent),
  16     ui(new Ui::EditAddressDialog),
  17     mapper(0),
  18     mode(_mode),
  19     model(0)
  20 {
  21     ui->setupUi(this);
  22
  23     GUIUtil::setupAddressWidget(ui->addressEdit, this);
  24
  25     switch(mode)
  26     {
  27     case NewReceivingAddress:
  28         setWindowTitle(tr("New receiving address"));
  29         ui->addressEdit->setEnabled(false);
  30         break;
  31     case NewSendingAddress:
  32         setWindowTitle(tr("New sending address"));
  33         break;
  34     case EditReceivingAddress:
  35         setWindowTitle(tr("Edit receiving address"));
  36         ui->addressEdit->setEnabled(false);
  37         break;
  38     case EditSendingAddress:
  39         setWindowTitle(tr("Edit sending address"));
  40         break;
  41     }
  42
  43     mapper = new QDataWidgetMapper(this);
  44     mapper->setSubmitPolicy(QDataWidgetMapper::ManualSubmit);
  45 }
  46
  47 EditAddressDialog::~EditAddressDialog()
  48 {
  49     delete ui;
  50 }
  51
  52 void EditAddressDialog::setModel(AddressTableModel *_model)
  53 {
  54     this->model = _model;
  55     if(!_model)
  56         return;
  57
  58     mapper->setModel(_model);
  59     mapper->addMapping(ui->labelEdit, AddressTableModel::Label);
  60     mapper->addMapping(ui->addressEdit, AddressTableModel::Address);
  61 }
  62
  63 void EditAddressDialog::loadRow(int row)
  64 {
  65     mapper->setCurrentIndex(row);
  66 }
  67
  68 bool EditAddressDialog::saveCurrentRow()
  69 {
  70     if(!model)
  71         return false;
  72
  73     switch(mode)
  74     {
  75     case NewReceivingAddress:
  76     case NewSendingAddress:
  77         address = model->addRow(
  78                 mode == NewSendingAddress ? AddressTableModel::Send : AddressTableModel::Receive,
  79                 ui->labelEdit->text(),
  80                 ui->addressEdit->text());
  81         break;
  82     case EditReceivingAddress:
  83     case EditSendingAddress:
  84         if(mapper->submit())
  85         {
  86             address = ui->addressEdit->text();
  87         }
  88         break;
  89     }
  90     return !address.isEmpty();
  91 }
  92
  93 void EditAddressDialog::accept()
  94 {
  95     if(!model)
  96         return;
  97
  98     if(!saveCurrentRow())
  99     {
 100         switch(model->getEditStatus())
 101         {
 102         case AddressTableModel::OK:
 103             // Failed with unknown reason. Just reject.
 104             break;
 105         case AddressTableModel::NO_CHANGES:
 106             // No changes were made during edit operation. Just reject.
 107             break;
 108         case AddressTableModel::INVALID_ADDRESS:
 109             QMessageBox::warning(this, windowTitle(),
 110                 tr("The entered address \"%1\" is not a valid Bitcoin address.").arg(ui->addressEdit->text()),
 111                 QMessageBox::Ok, QMessageBox::Ok);
 112             break;
 113         case AddressTableModel::DUPLICATE_ADDRESS:
 114             QMessageBox::warning(this, windowTitle(),
 115                 tr("The entered address \"%1\" is already in the address book.").arg(ui->addressEdit->text()),
 116                 QMessageBox::Ok, QMessageBox::Ok);
 117             break;
 118         case AddressTableModel::WALLET_UNLOCK_FAILURE:
 119             QMessageBox::critical(this, windowTitle(),
 120                 tr("Could not unlock wallet."),
 121                 QMessageBox::Ok, QMessageBox::Ok);
 122             break;
 123         case AddressTableModel::KEY_GENERATION_FAILURE:
 124             QMessageBox::critical(this, windowTitle(),
 125                 tr("New key generation failed."),
 126                 QMessageBox::Ok, QMessageBox::Ok);
 127             break;
 128
 129         }
 130         return;
 131     }
 132     QDialog::accept();
 133 }
 134
 135 QString EditAddressDialog::getAddress() const
 136 {
 137     return address;
 138 }
 139
 140 void EditAddressDialog::setAddress(const QString &_address)
 141 {
 142     this->address = _address;
 143     ui->addressEdit->setText(_address);
 144 }