Pre-0.4.8 update
[novacoin.git] / src / qt / transactionview.cpp
1 #include "transactionview.h"
2
3 #include "transactionfilterproxy.h"
4 #include "transactionrecord.h"
5 #include "walletmodel.h"
6 #include "addresstablemodel.h"
7 #include "transactiontablemodel.h"
8 #include "bitcoinunits.h"
9 #include "csvmodelwriter.h"
10 #include "transactiondescdialog.h"
11 #include "editaddressdialog.h"
12 #include "optionsmodel.h"
13 #include "guiutil.h"
14
15 #include <QScrollBar>
16 #include <QComboBox>
17 #include <QDoubleValidator>
18 #include <QHBoxLayout>
19 #include <QVBoxLayout>
20 #include <QLineEdit>
21 #include <QTableView>
22 #include <QHeaderView>
23 #include <QPushButton>
24 #include <QMessageBox>
25 #include <QPoint>
26 #include <QMenu>
27 #include <QApplication>
28 #include <QClipboard>
29 #include <QLabel>
30 #include <QDateTimeEdit>
31
32 TransactionView::TransactionView(QWidget *parent) :
33     QWidget(parent), model(0), transactionProxyModel(0),
34     transactionView(0)
35 {
36     // Build filter row
37     setContentsMargins(0,0,0,0);
38
39     QHBoxLayout *hlayout = new QHBoxLayout();
40     hlayout->setContentsMargins(0,0,0,0);
41 #ifdef Q_OS_MAC
42     hlayout->setSpacing(5);
43     hlayout->addSpacing(26);
44 #else
45     hlayout->setSpacing(0);
46     hlayout->addSpacing(23);
47 #endif
48
49     dateWidget = new QComboBox(this);
50 #ifdef Q_OS_MAC
51     dateWidget->setFixedWidth(121);
52 #else
53     dateWidget->setFixedWidth(120);
54 #endif
55     dateWidget->addItem(tr("All"), All);
56     dateWidget->addItem(tr("Today"), Today);
57     dateWidget->addItem(tr("This week"), ThisWeek);
58     dateWidget->addItem(tr("This month"), ThisMonth);
59     dateWidget->addItem(tr("Last month"), LastMonth);
60     dateWidget->addItem(tr("This year"), ThisYear);
61     dateWidget->addItem(tr("Range..."), Range);
62     hlayout->addWidget(dateWidget);
63
64     typeWidget = new QComboBox(this);
65 #ifdef Q_OS_MAC
66     typeWidget->setFixedWidth(121);
67 #else
68     typeWidget->setFixedWidth(120);
69 #endif
70
71     typeWidget->addItem(tr("All"), TransactionFilterProxy::ALL_TYPES);
72     typeWidget->addItem(tr("Received with"), TransactionFilterProxy::TYPE(TransactionRecord::RecvWithAddress) |
73                                         TransactionFilterProxy::TYPE(TransactionRecord::RecvFromOther));
74     typeWidget->addItem(tr("Sent to"), TransactionFilterProxy::TYPE(TransactionRecord::SendToAddress) |
75                                   TransactionFilterProxy::TYPE(TransactionRecord::SendToOther));
76     typeWidget->addItem(tr("To yourself"), TransactionFilterProxy::TYPE(TransactionRecord::SendToSelf));
77     typeWidget->addItem(tr("Mined"), TransactionFilterProxy::TYPE(TransactionRecord::Generated));
78     typeWidget->addItem(tr("Other"), TransactionFilterProxy::TYPE(TransactionRecord::Other));
79
80     hlayout->addWidget(typeWidget);
81
82     addressWidget = new QLineEdit(this);
83 #if QT_VERSION >= 0x040700
84     /* Do not move this to the XML file, Qt before 4.7 will choke on it */
85     addressWidget->setPlaceholderText(tr("Enter address or label to search"));
86 #endif
87     hlayout->addWidget(addressWidget);
88
89     amountWidget = new QLineEdit(this);
90 #if QT_VERSION >= 0x040700
91     /* Do not move this to the XML file, Qt before 4.7 will choke on it */
92     amountWidget->setPlaceholderText(tr("Min amount"));
93 #endif
94 #ifdef Q_OS_MAC
95     amountWidget->setFixedWidth(97);
96 #else
97     amountWidget->setFixedWidth(100);
98 #endif
99     amountWidget->setValidator(new QDoubleValidator(0, 1e20, 8, this));
100     hlayout->addWidget(amountWidget);
101
102     QVBoxLayout *vlayout = new QVBoxLayout(this);
103     vlayout->setContentsMargins(0,0,0,0);
104     vlayout->setSpacing(0);
105
106     QTableView *view = new QTableView(this);
107     vlayout->addLayout(hlayout);
108     vlayout->addWidget(createDateRangeWidget());
109     vlayout->addWidget(view);
110     vlayout->setSpacing(0);
111     int width = view->verticalScrollBar()->sizeHint().width();
112     // Cover scroll bar width with spacing
113 #ifdef Q_OS_MAC
114     hlayout->addSpacing(width+2);
115 #else
116     hlayout->addSpacing(width);
117 #endif
118     // Always show scroll bar
119     view->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOn);
120     view->setTabKeyNavigation(false);
121     view->setContextMenuPolicy(Qt::CustomContextMenu);
122
123     transactionView = view;
124
125     // Actions
126     QAction *copyAddressAction = new QAction(tr("Copy address"), this);
127     QAction *copyLabelAction = new QAction(tr("Copy label"), this);
128     QAction *copyAmountAction = new QAction(tr("Copy amount"), this);
129     QAction *copyTxIDAction = new QAction(tr("Copy transaction ID"), this);
130     QAction *editLabelAction = new QAction(tr("Edit label"), this);
131     QAction *showDetailsAction = new QAction(tr("Show transaction details"), this);
132
133     contextMenu = new QMenu();
134     contextMenu->addAction(copyAddressAction);
135     contextMenu->addAction(copyLabelAction);
136     contextMenu->addAction(copyAmountAction);
137     contextMenu->addAction(copyTxIDAction);
138     contextMenu->addAction(editLabelAction);
139     contextMenu->addAction(showDetailsAction);
140
141     // Connect actions
142     connect(dateWidget, SIGNAL(activated(int)), this, SLOT(chooseDate(int)));
143     connect(typeWidget, SIGNAL(activated(int)), this, SLOT(chooseType(int)));
144     connect(addressWidget, SIGNAL(textChanged(QString)), this, SLOT(changedPrefix(QString)));
145     connect(amountWidget, SIGNAL(textChanged(QString)), this, SLOT(changedAmount(QString)));
146
147     connect(view, SIGNAL(doubleClicked(QModelIndex)), this, SIGNAL(doubleClicked(QModelIndex)));
148     connect(view, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(contextualMenu(QPoint)));
149
150     connect(copyAddressAction, SIGNAL(triggered()), this, SLOT(copyAddress()));
151     connect(copyLabelAction, SIGNAL(triggered()), this, SLOT(copyLabel()));
152     connect(copyAmountAction, SIGNAL(triggered()), this, SLOT(copyAmount()));
153     connect(copyTxIDAction, SIGNAL(triggered()), this, SLOT(copyTxID()));
154     connect(editLabelAction, SIGNAL(triggered()), this, SLOT(editLabel()));
155     connect(showDetailsAction, SIGNAL(triggered()), this, SLOT(showDetails()));
156 }
157
158 void TransactionView::setModel(WalletModel *model)
159 {
160     this->model = model;
161     if(model)
162     {
163         transactionProxyModel = new TransactionFilterProxy(this);
164         transactionProxyModel->setSourceModel(model->getTransactionTableModel());
165         transactionProxyModel->setDynamicSortFilter(true);
166         transactionProxyModel->setSortCaseSensitivity(Qt::CaseInsensitive);
167         transactionProxyModel->setFilterCaseSensitivity(Qt::CaseInsensitive);
168
169         transactionProxyModel->setSortRole(Qt::EditRole);
170
171         transactionView->setModel(transactionProxyModel);
172         transactionView->setAlternatingRowColors(true);
173         transactionView->setSelectionBehavior(QAbstractItemView::SelectRows);
174         transactionView->setSelectionMode(QAbstractItemView::ExtendedSelection);
175         transactionView->setSortingEnabled(true);
176         transactionView->sortByColumn(TransactionTableModel::Status, Qt::DescendingOrder);
177         transactionView->verticalHeader()->hide();
178
179         transactionView->horizontalHeader()->resizeSection(
180                 TransactionTableModel::Status, 23);
181         transactionView->horizontalHeader()->resizeSection(
182                 TransactionTableModel::Date, 120);
183         transactionView->horizontalHeader()->resizeSection(
184                 TransactionTableModel::Type, 120);
185         transactionView->horizontalHeader()->setResizeMode(
186                 TransactionTableModel::ToAddress, QHeaderView::Stretch);
187         transactionView->horizontalHeader()->resizeSection(
188                 TransactionTableModel::Amount, 100);
189     }
190 }
191
192 void TransactionView::chooseDate(int idx)
193 {
194     if(!transactionProxyModel)
195         return;
196     QDate current = QDate::currentDate();
197     dateRangeWidget->setVisible(false);
198     switch(dateWidget->itemData(idx).toInt())
199     {
200     case All:
201         transactionProxyModel->setDateRange(
202                 TransactionFilterProxy::MIN_DATE,
203                 TransactionFilterProxy::MAX_DATE);
204         break;
205     case Today:
206         transactionProxyModel->setDateRange(
207                 QDateTime(current),
208                 TransactionFilterProxy::MAX_DATE);
209         break;
210     case ThisWeek: {
211         // Find last Monday
212         QDate startOfWeek = current.addDays(-(current.dayOfWeek()-1));
213         transactionProxyModel->setDateRange(
214                 QDateTime(startOfWeek),
215                 TransactionFilterProxy::MAX_DATE);
216
217         } break;
218     case ThisMonth:
219         transactionProxyModel->setDateRange(
220                 QDateTime(QDate(current.year(), current.month(), 1)),
221                 TransactionFilterProxy::MAX_DATE);
222         break;
223     case LastMonth:
224         transactionProxyModel->setDateRange(
225                 QDateTime(QDate(current.year(), current.month()-1, 1)),
226                 QDateTime(QDate(current.year(), current.month(), 1)));
227         break;
228     case ThisYear:
229         transactionProxyModel->setDateRange(
230                 QDateTime(QDate(current.year(), 1, 1)),
231                 TransactionFilterProxy::MAX_DATE);
232         break;
233     case Range:
234         dateRangeWidget->setVisible(true);
235         dateRangeChanged();
236         break;
237     }
238 }
239
240 void TransactionView::chooseType(int idx)
241 {
242     if(!transactionProxyModel)
243         return;
244     transactionProxyModel->setTypeFilter(
245         typeWidget->itemData(idx).toInt());
246 }
247
248 void TransactionView::changedPrefix(const QString &prefix)
249 {
250     if(!transactionProxyModel)
251         return;
252     transactionProxyModel->setAddressPrefix(prefix);
253 }
254
255 void TransactionView::changedAmount(const QString &amount)
256 {
257     if(!transactionProxyModel)
258         return;
259     qint64 amount_parsed = 0;
260     if(BitcoinUnits::parse(model->getOptionsModel()->getDisplayUnit(), amount, &amount_parsed))
261     {
262         transactionProxyModel->setMinAmount(amount_parsed);
263     }
264     else
265     {
266         transactionProxyModel->setMinAmount(0);
267     }
268 }
269
270 void TransactionView::exportClicked()
271 {
272     // CSV is currently the only supported format
273     QString filename = GUIUtil::getSaveFileName(
274             this,
275             tr("Export Transaction Data"), QString(),
276             tr("Comma separated file (*.csv)"));
277
278     if (filename.isNull()) return;
279
280     CSVModelWriter writer(filename);
281
282     // name, column, role
283     writer.setModel(transactionProxyModel);
284     writer.addColumn(tr("Confirmed"), 0, TransactionTableModel::ConfirmedRole);
285     writer.addColumn(tr("Date"), 0, TransactionTableModel::DateRole);
286     writer.addColumn(tr("Type"), TransactionTableModel::Type, Qt::EditRole);
287     writer.addColumn(tr("Label"), 0, TransactionTableModel::LabelRole);
288     writer.addColumn(tr("Address"), 0, TransactionTableModel::AddressRole);
289     writer.addColumn(tr("Amount"), 0, TransactionTableModel::FormattedAmountRole);
290     writer.addColumn(tr("ID"), 0, TransactionTableModel::TxIDRole);
291
292     if(!writer.write())
293     {
294         QMessageBox::critical(this, tr("Error exporting"), tr("Could not write to file %1.").arg(filename),
295                               QMessageBox::Abort, QMessageBox::Abort);
296     }
297 }
298
299 void TransactionView::contextualMenu(const QPoint &point)
300 {
301     QModelIndex index = transactionView->indexAt(point);
302     if(index.isValid())
303     {
304         contextMenu->exec(QCursor::pos());
305     }
306 }
307
308 void TransactionView::copyAddress()
309 {
310     GUIUtil::copyEntryData(transactionView, 0, TransactionTableModel::AddressRole);
311 }
312
313 void TransactionView::copyLabel()
314 {
315     GUIUtil::copyEntryData(transactionView, 0, TransactionTableModel::LabelRole);
316 }
317
318 void TransactionView::copyAmount()
319 {
320     GUIUtil::copyEntryData(transactionView, 0, TransactionTableModel::FormattedAmountRole);
321 }
322
323 void TransactionView::copyTxID()
324 {
325     GUIUtil::copyEntryData(transactionView, 0, TransactionTableModel::TxIDRole);
326 }
327
328 void TransactionView::editLabel()
329 {
330     if(!transactionView->selectionModel() ||!model)
331         return;
332     QModelIndexList selection = transactionView->selectionModel()->selectedRows();
333     if(!selection.isEmpty())
334     {
335         AddressTableModel *addressBook = model->getAddressTableModel();
336         if(!addressBook)
337             return;
338         QString address = selection.at(0).data(TransactionTableModel::AddressRole).toString();
339         if(address.isEmpty())
340         {
341             // If this transaction has no associated address, exit
342             return;
343         }
344         // Is address in address book? Address book can miss address when a transaction is
345         // sent from outside the UI.
346         int idx = addressBook->lookupAddress(address);
347         if(idx != -1)
348         {
349             // Edit sending / receiving address
350             QModelIndex modelIdx = addressBook->index(idx, 0, QModelIndex());
351             // Determine type of address, launch appropriate editor dialog type
352             QString type = modelIdx.data(AddressTableModel::TypeRole).toString();
353
354             EditAddressDialog dlg(type==AddressTableModel::Receive
355                                          ? EditAddressDialog::EditReceivingAddress
356                                          : EditAddressDialog::EditSendingAddress,
357                                   this);
358             dlg.setModel(addressBook);
359             dlg.loadRow(idx);
360             dlg.exec();
361         }
362         else
363         {
364             // Add sending address
365             EditAddressDialog dlg(EditAddressDialog::NewSendingAddress,
366                                   this);
367             dlg.setModel(addressBook);
368             dlg.setAddress(address);
369             dlg.exec();
370         }
371     }
372 }
373
374 void TransactionView::showDetails()
375 {
376     if(!transactionView->selectionModel())
377         return;
378     QModelIndexList selection = transactionView->selectionModel()->selectedRows();
379     if(!selection.isEmpty())
380     {
381         TransactionDescDialog dlg(selection.at(0));
382         dlg.exec();
383     }
384 }
385
386 QWidget *TransactionView::createDateRangeWidget()
387 {
388     dateRangeWidget = new QFrame();
389     dateRangeWidget->setFrameStyle(QFrame::Panel | QFrame::Raised);
390     dateRangeWidget->setContentsMargins(1,1,1,1);
391     QHBoxLayout *layout = new QHBoxLayout(dateRangeWidget);
392     layout->setContentsMargins(0,0,0,0);
393     layout->addSpacing(23);
394     layout->addWidget(new QLabel(tr("Range:")));
395
396     dateFrom = new QDateTimeEdit(this);
397     dateFrom->setDisplayFormat("dd/MM/yy");
398     dateFrom->setCalendarPopup(true);
399     dateFrom->setMinimumWidth(100);
400     dateFrom->setDate(QDate::currentDate().addDays(-7));
401     layout->addWidget(dateFrom);
402     layout->addWidget(new QLabel(tr("to")));
403
404     dateTo = new QDateTimeEdit(this);
405     dateTo->setDisplayFormat("dd/MM/yy");
406     dateTo->setCalendarPopup(true);
407     dateTo->setMinimumWidth(100);
408     dateTo->setDate(QDate::currentDate());
409     layout->addWidget(dateTo);
410     layout->addStretch();
411
412     // Hide by default
413     dateRangeWidget->setVisible(false);
414
415     // Notify on change
416     connect(dateFrom, SIGNAL(dateChanged(QDate)), this, SLOT(dateRangeChanged()));
417     connect(dateTo, SIGNAL(dateChanged(QDate)), this, SLOT(dateRangeChanged()));
418
419     return dateRangeWidget;
420 }
421
422 void TransactionView::dateRangeChanged()
423 {
424     if(!transactionProxyModel)
425         return;
426     transactionProxyModel->setDateRange(
427             QDateTime(dateFrom->date()),
428             QDateTime(dateTo->date()).addDays(1));
429 }
430
431 void TransactionView::focusTransaction(const QModelIndex &idx)
432 {
433     if(!transactionProxyModel)
434         return;
435     QModelIndex targetIdx = transactionProxyModel->mapFromSource(idx);
436     transactionView->scrollTo(targetIdx);
437     transactionView->setCurrentIndex(targetIdx);
438     transactionView->setFocus();
439 }