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