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