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