src/qt/addressbookpage.cpp

   1 #include "addressbookpage.h"
   2 #include "ui_addressbookpage.h"
   3
   4 #include "addresstablemodel.h"
   5 #include "optionsmodel.h"
   6 #include "bitcoingui.h"
   7 #include "editaddressdialog.h"
   8 #include "csvmodelwriter.h"
   9 #include "guiutil.h"
  10
  11 #include <QSortFilterProxyModel>
  12 #include <QClipboard>
  13 #include <QMessageBox>
  14 #include <QMenu>
  15
  16 #ifdef USE_QRCODE
  17 #include "qrcodedialog.h"
  18 #endif
  19
  20 AddressBookPage::AddressBookPage(Mode mode, Tabs tab, QWidget *parent) :
  21     QDialog(parent),
  22     ui(new Ui::AddressBookPage),
  23     model(0),
  24     optionsModel(0),
  25     mode(mode),
  26     tab(tab)
  27 {
  28     ui->setupUi(this);
  29
  30 #ifdef Q_OS_MAC // Icons on push buttons are very uncommon on Mac
  31     ui->newAddressButton->setIcon(QIcon());
  32     ui->copyToClipboard->setIcon(QIcon());
  33     ui->deleteButton->setIcon(QIcon());
  34 #endif
  35
  36 #ifndef USE_QRCODE
  37     ui->showQRCode->setVisible(false);
  38 #endif
  39
  40     switch(mode)
  41     {
  42     case ForSending:
  43         connect(ui->tableView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(accept()));
  44         ui->tableView->setEditTriggers(QAbstractItemView::NoEditTriggers);
  45         ui->tableView->setFocus();
  46         break;
  47     case ForEditing:
  48         ui->buttonBox->setVisible(false);
  49         break;
  50     }
  51     switch(tab)
  52     {
  53     case SendingTab:
  54         ui->labelExplanation->setVisible(false);
  55         ui->deleteButton->setVisible(true);
  56         ui->signMessage->setVisible(false);
  57         break;
  58     case ReceivingTab:
  59         ui->deleteButton->setVisible(false);
  60         ui->signMessage->setVisible(true);
  61         break;
  62     }
  63
  64     // Context menu actions
  65     QAction *copyLabelAction = new QAction(tr("Copy &Label"), this);
  66     QAction *copyAddressAction = new QAction(ui->copyToClipboard->text(), this);
  67     QAction *editAction = new QAction(tr("&Edit"), this);
  68 #ifdef USE_QRCODE
  69     QAction *showQRCodeAction = new QAction(ui->showQRCode->text(), this);
  70 #endif
  71     QAction *signMessageAction = new QAction(ui->signMessage->text(), this);
  72     QAction *verifyMessageAction = new QAction(ui->verifyMessage->text(), this);
  73     deleteAction = new QAction(ui->deleteButton->text(), this);
  74
  75     // Build context menu
  76     contextMenu = new QMenu();
  77     contextMenu->addAction(copyAddressAction);
  78     contextMenu->addAction(copyLabelAction);
  79     contextMenu->addAction(editAction);
  80     if(tab == SendingTab)
  81         contextMenu->addAction(deleteAction);
  82     contextMenu->addSeparator();
  83 #ifdef USE_QRCODE
  84     contextMenu->addAction(showQRCodeAction);
  85 #endif
  86     if(tab == ReceivingTab)
  87         contextMenu->addAction(signMessageAction);
  88     else if(tab == SendingTab)
  89         contextMenu->addAction(verifyMessageAction);
  90
  91     // Connect signals for context menu actions
  92     connect(copyAddressAction, SIGNAL(triggered()), this, SLOT(on_copyToClipboard_clicked()));
  93     connect(copyLabelAction, SIGNAL(triggered()), this, SLOT(onCopyLabelAction()));
  94     connect(editAction, SIGNAL(triggered()), this, SLOT(onEditAction()));
  95     connect(deleteAction, SIGNAL(triggered()), this, SLOT(on_deleteButton_clicked()));
  96 #ifdef USE_QRCODE
  97     connect(showQRCodeAction, SIGNAL(triggered()), this, SLOT(on_showQRCode_clicked()));
  98 #endif
  99     connect(signMessageAction, SIGNAL(triggered()), this, SLOT(on_signMessage_clicked()));
 100     connect(verifyMessageAction, SIGNAL(triggered()), this, SLOT(on_verifyMessage_clicked()));
 101
 102     connect(ui->tableView, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(contextualMenu(QPoint)));
 103
 104     // Pass through accept action from button box
 105     connect(ui->buttonBox, SIGNAL(accepted()), this, SLOT(accept()));
 106 }
 107
 108 AddressBookPage::~AddressBookPage()
 109 {
 110     delete ui;
 111 }
 112
 113 void AddressBookPage::setModel(AddressTableModel *model)
 114 {
 115     this->model = model;
 116     if(!model)
 117         return;
 118
 119     proxyModel = new QSortFilterProxyModel(this);
 120     proxyModel->setSourceModel(model);
 121     proxyModel->setDynamicSortFilter(true);
 122     proxyModel->setSortCaseSensitivity(Qt::CaseInsensitive);
 123     proxyModel->setFilterCaseSensitivity(Qt::CaseInsensitive);
 124     switch(tab)
 125     {
 126     case ReceivingTab:
 127         // Receive filter
 128         proxyModel->setFilterRole(AddressTableModel::TypeRole);
 129         proxyModel->setFilterFixedString(AddressTableModel::Receive);
 130         break;
 131     case SendingTab:
 132         // Send filter
 133         proxyModel->setFilterRole(AddressTableModel::TypeRole);
 134         proxyModel->setFilterFixedString(AddressTableModel::Send);
 135         break;
 136     }
 137     ui->tableView->setModel(proxyModel);
 138     ui->tableView->sortByColumn(0, Qt::AscendingOrder);
 139
 140     // Set column widths
 141     ui->tableView->horizontalHeader()->resizeSection(
 142             AddressTableModel::Address, 320);
 143 #if QT_VERSION < 0x050000
 144     ui->tableView->horizontalHeader()->setResizeMode(
 145             AddressTableModel::Label, QHeaderView::Stretch);
 146 #else
 147     ui->tableView->horizontalHeader()->setSectionResizeMode(
 148     AddressTableModel::Label, QHeaderView::Stretch);
 149 #endif
 150     connect(ui->tableView->selectionModel(), SIGNAL(selectionChanged(QItemSelection,QItemSelection)),
 151             this, SLOT(selectionChanged()));
 152
 153     // Select row for newly created address
 154     connect(model, SIGNAL(rowsInserted(QModelIndex,int,int)),
 155             this, SLOT(selectNewAddress(QModelIndex,int,int)));
 156
 157     selectionChanged();
 158 }
 159
 160 void AddressBookPage::setOptionsModel(OptionsModel *optionsModel)
 161 {
 162     this->optionsModel = optionsModel;
 163 }
 164
 165 void AddressBookPage::on_copyToClipboard_clicked()
 166 {
 167     GUIUtil::copyEntryData(ui->tableView, AddressTableModel::Address);
 168 }
 169
 170 void AddressBookPage::onCopyLabelAction()
 171 {
 172     GUIUtil::copyEntryData(ui->tableView, AddressTableModel::Label);
 173 }
 174
 175 void AddressBookPage::onEditAction()
 176 {
 177     if(!ui->tableView->selectionModel())
 178         return;
 179     QModelIndexList indexes = ui->tableView->selectionModel()->selectedRows();
 180     if(indexes.isEmpty())
 181         return;
 182
 183     EditAddressDialog dlg(
 184             tab == SendingTab ?
 185             EditAddressDialog::EditSendingAddress :
 186             EditAddressDialog::EditReceivingAddress);
 187     dlg.setModel(model);
 188     QModelIndex origIndex = proxyModel->mapToSource(indexes.at(0));
 189     dlg.loadRow(origIndex.row());
 190     dlg.exec();
 191 }
 192
 193 void AddressBookPage::on_signMessage_clicked()
 194 {
 195     QTableView *table = ui->tableView;
 196     QModelIndexList indexes = table->selectionModel()->selectedRows(AddressTableModel::Address);
 197     QString addr;
 198
 199     foreach (QModelIndex index, indexes)
 200     {
 201         QVariant address = index.data();
 202         addr = address.toString();
 203     }
 204
 205     emit signMessage(addr);
 206 }
 207
 208 void AddressBookPage::on_verifyMessage_clicked()
 209 {
 210     QTableView *table = ui->tableView;
 211     QModelIndexList indexes = table->selectionModel()->selectedRows(AddressTableModel::Address);
 212     QString addr;
 213
 214     foreach (QModelIndex index, indexes)
 215     {
 216         QVariant address = index.data();
 217         addr = address.toString();
 218     }
 219
 220     emit verifyMessage(addr);
 221 }
 222
 223 void AddressBookPage::on_newAddressButton_clicked()
 224 {
 225     if(!model)
 226         return;
 227     EditAddressDialog dlg(
 228             tab == SendingTab ?
 229             EditAddressDialog::NewSendingAddress :
 230             EditAddressDialog::NewReceivingAddress, this);
 231     dlg.setModel(model);
 232     if(dlg.exec())
 233     {
 234         newAddressToSelect = dlg.getAddress();
 235     }
 236 }
 237
 238 void AddressBookPage::on_deleteButton_clicked()
 239 {
 240     QTableView *table = ui->tableView;
 241     if(!table->selectionModel())
 242         return;
 243     QModelIndexList indexes = table->selectionModel()->selectedRows();
 244     if(!indexes.isEmpty())
 245     {
 246         table->model()->removeRow(indexes.at(0).row());
 247     }
 248 }
 249
 250 void AddressBookPage::selectionChanged()
 251 {
 252     // Set button states based on selected tab and selection
 253     QTableView *table = ui->tableView;
 254     if(!table->selectionModel())
 255         return;
 256
 257     if(table->selectionModel()->hasSelection())
 258     {
 259         switch(tab)
 260         {
 261         case SendingTab:
 262             // In sending tab, allow deletion of selection
 263             ui->deleteButton->setEnabled(true);
 264             ui->deleteButton->setVisible(true);
 265             deleteAction->setEnabled(true);
 266             ui->signMessage->setEnabled(false);
 267             ui->signMessage->setVisible(false);
 268             ui->verifyMessage->setEnabled(true);
 269             ui->verifyMessage->setVisible(true);
 270             break;
 271         case ReceivingTab:
 272             // Deleting receiving addresses, however, is not allowed
 273             ui->deleteButton->setEnabled(false);
 274             ui->deleteButton->setVisible(false);
 275             deleteAction->setEnabled(false);
 276             ui->signMessage->setEnabled(true);
 277             ui->signMessage->setVisible(true);
 278             ui->verifyMessage->setEnabled(false);
 279             ui->verifyMessage->setVisible(false);
 280             break;
 281         }
 282         ui->copyToClipboard->setEnabled(true);
 283         ui->showQRCode->setEnabled(true);
 284     }
 285     else
 286     {
 287         ui->deleteButton->setEnabled(false);
 288         ui->showQRCode->setEnabled(false);
 289         ui->copyToClipboard->setEnabled(false);
 290         ui->signMessage->setEnabled(false);
 291         ui->verifyMessage->setEnabled(false);
 292     }
 293 }
 294
 295 void AddressBookPage::done(int retval)
 296 {
 297     QTableView *table = ui->tableView;
 298     if(!table->selectionModel() || !table->model())
 299         return;
 300     // When this is a tab/widget and not a model dialog, ignore "done"
 301     if(mode == ForEditing)
 302         return;
 303
 304     // Figure out which address was selected, and return it
 305     QModelIndexList indexes = table->selectionModel()->selectedRows(AddressTableModel::Address);
 306
 307     foreach (QModelIndex index, indexes)
 308     {
 309         QVariant address = table->model()->data(index);
 310         returnValue = address.toString();
 311     }
 312
 313     if(returnValue.isEmpty())
 314     {
 315         // If no address entry selected, return rejected
 316         retval = Rejected;
 317     }
 318
 319     QDialog::done(retval);
 320 }
 321
 322 void AddressBookPage::exportClicked()
 323 {
 324     // CSV is currently the only supported format
 325     QString filename = GUIUtil::getSaveFileName(
 326             this,
 327             tr("Export Address Book Data"), QString(),
 328             tr("Comma separated file (*.csv)"));
 329
 330     if (filename.isNull()) return;
 331
 332     CSVModelWriter writer(filename);
 333
 334     // name, column, role
 335     writer.setModel(proxyModel);
 336     writer.addColumn("Label", AddressTableModel::Label, Qt::EditRole);
 337     writer.addColumn("Address", AddressTableModel::Address, Qt::EditRole);
 338
 339     if(!writer.write())
 340     {
 341         QMessageBox::critical(this, tr("Error exporting"), tr("Could not write to file %1.").arg(filename),
 342                               QMessageBox::Abort, QMessageBox::Abort);
 343     }
 344 }
 345
 346 void AddressBookPage::on_showQRCode_clicked()
 347 {
 348 #ifdef USE_QRCODE
 349     QTableView *table = ui->tableView;
 350     QModelIndexList indexes = table->selectionModel()->selectedRows(AddressTableModel::Address);
 351
 352     foreach (QModelIndex index, indexes)
 353     {
 354         QString address = index.data().toString(), label = index.sibling(index.row(), 0).data(Qt::EditRole).toString();
 355
 356         QRCodeDialog *dialog = new QRCodeDialog(address, label, tab == ReceivingTab, this);
 357         if(optionsModel)
 358             dialog->setModel(optionsModel);
 359         dialog->setAttribute(Qt::WA_DeleteOnClose);
 360         dialog->show();
 361     }
 362 #endif
 363 }
 364
 365 void AddressBookPage::contextualMenu(const QPoint &point)
 366 {
 367     QModelIndex index = ui->tableView->indexAt(point);
 368     if(index.isValid())
 369     {
 370         contextMenu->exec(QCursor::pos());
 371     }
 372 }
 373
 374 void AddressBookPage::selectNewAddress(const QModelIndex &parent, int begin, int end)
 375 {
 376     QModelIndex idx = proxyModel->mapFromSource(model->index(begin, AddressTableModel::Address, parent));
 377     if(idx.isValid() && (idx.data(Qt::EditRole).toString() == newAddressToSelect))
 378     {
 379         // Select row of newly created address, once
 380         ui->tableView->setFocus();
 381         ui->tableView->selectRow(idx.row());
 382         newAddressToSelect.clear();
 383     }
 384 }