Add "last month" filter
[novacoin.git] / src / qt / transactionview.cpp
1 #include "transactionview.h"
2
3 // Temp includes for filtering prototype
4 // Move to TransactionFilterRow class
5 #include "transactionfilterproxy.h"
6 #include "transactionrecord.h"
7 #include "transactiontablemodel.h"
8 #include "guiutil.h"
9
10 #include <QScrollBar>
11 #include <QComboBox>
12 #include <QDoubleValidator>
13 #include <QHBoxLayout>
14 #include <QVBoxLayout>
15 #include <QLineEdit>
16 #include <QTableView>
17 #include <QHeaderView>
18
19 #include <QDebug>
20
21 TransactionView::TransactionView(QWidget *parent) :
22     QWidget(parent), model(0), transactionProxyModel(0),
23     transactionView(0)
24 {
25     // Build filter row
26     QHBoxLayout *hlayout = new QHBoxLayout();
27     hlayout->setContentsMargins(QMargins(0,0,0,0));
28     hlayout->setSpacing(0);
29
30     hlayout->addSpacing(23);
31
32     dateWidget = new QComboBox(this);
33     dateWidget->setMaximumWidth(120);
34     dateWidget->setMinimumWidth(120);
35     dateWidget->addItem(tr("All"), All);
36     dateWidget->addItem(tr("Today"), Today);
37     dateWidget->addItem(tr("This week"), ThisWeek);
38     dateWidget->addItem(tr("This month"), ThisMonth);
39     dateWidget->addItem(tr("Last month"), LastMonth);
40     dateWidget->addItem(tr("This year"), ThisYear);
41     dateWidget->addItem(tr("Range..."), Range);
42     hlayout->addWidget(dateWidget);
43
44     typeWidget = new QComboBox(this);
45     typeWidget->setMaximumWidth(120);
46     typeWidget->setMinimumWidth(120);
47
48     typeWidget->addItem(tr("All"), TransactionFilterProxy::ALL_TYPES);
49     typeWidget->addItem(tr("Received with"), TransactionFilterProxy::TYPE(TransactionRecord::RecvWithAddress) |
50                                         TransactionFilterProxy::TYPE(TransactionRecord::RecvFromIP));
51     typeWidget->addItem(tr("Sent to"), TransactionFilterProxy::TYPE(TransactionRecord::SendToAddress) |
52                                   TransactionFilterProxy::TYPE(TransactionRecord::SendToIP));
53     typeWidget->addItem(tr("To yourself"), TransactionFilterProxy::TYPE(TransactionRecord::SendToSelf));
54     typeWidget->addItem(tr("Generated"), TransactionFilterProxy::TYPE(TransactionRecord::Generated));
55     typeWidget->addItem(tr("Other"), TransactionFilterProxy::TYPE(TransactionRecord::Other));
56
57     hlayout->addWidget(typeWidget);
58
59     addressWidget = new QLineEdit(this);
60     addressWidget->setPlaceholderText("Enter address or label to search");
61     hlayout->addWidget(addressWidget);
62
63     amountWidget = new QLineEdit(this);
64     amountWidget->setPlaceholderText("Min amount");
65     amountWidget->setMaximumWidth(100);
66     amountWidget->setMinimumWidth(100);
67     amountWidget->setValidator(new QDoubleValidator(0, 1e20, 8, this));
68     hlayout->addWidget(amountWidget);
69
70     QVBoxLayout *vlayout = new QVBoxLayout(this);
71
72     QTableView *view = new QTableView(this);
73     vlayout->addLayout(hlayout);
74     vlayout->addWidget(view);
75     vlayout->setSpacing(0);
76     int width = view->verticalScrollBar()->sizeHint().width();
77     // Cover scroll bar width with spacing
78     hlayout->addSpacing(width);
79     // Always show scroll bar
80     view->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOn);
81
82     transactionView = view;
83
84     connect(dateWidget, SIGNAL(activated(int)), this, SLOT(chooseDate(int)));
85     connect(typeWidget, SIGNAL(activated(int)), this, SLOT(chooseType(int)));
86     connect(addressWidget, SIGNAL(textChanged(const QString&)), this, SLOT(changedPrefix(const QString&)));
87     connect(amountWidget, SIGNAL(textChanged(const QString&)), this, SLOT(changedAmount(const QString&)));
88
89     connect(view, SIGNAL(doubleClicked(const QModelIndex&)), this, SIGNAL(doubleClicked(const QModelIndex&)));
90 }
91
92 void TransactionView::setModel(TransactionTableModel *model)
93 {
94     this->model = model;
95
96     transactionProxyModel = new TransactionFilterProxy(this);
97     transactionProxyModel->setSourceModel(model);
98     transactionProxyModel->setDynamicSortFilter(true);
99
100     transactionProxyModel->setSortRole(Qt::EditRole);
101
102     transactionView->setModel(transactionProxyModel);
103     transactionView->setAlternatingRowColors(true);
104     transactionView->setSelectionBehavior(QAbstractItemView::SelectRows);
105     transactionView->setSelectionMode(QAbstractItemView::ExtendedSelection);
106     transactionView->setSortingEnabled(true);
107     transactionView->sortByColumn(TransactionTableModel::Status, Qt::DescendingOrder);
108     transactionView->verticalHeader()->hide();
109
110     transactionView->horizontalHeader()->resizeSection(
111             TransactionTableModel::Status, 23);
112     transactionView->horizontalHeader()->resizeSection(
113             TransactionTableModel::Date, 120);
114     transactionView->horizontalHeader()->resizeSection(
115             TransactionTableModel::Type, 120);
116     transactionView->horizontalHeader()->setResizeMode(
117             TransactionTableModel::ToAddress, QHeaderView::Stretch);
118     transactionView->horizontalHeader()->resizeSection(
119             TransactionTableModel::Amount, 100);
120
121 }
122
123 void TransactionView::chooseDate(int idx)
124 {
125     QDate current = QDate::currentDate();
126     switch(dateWidget->itemData(idx).toInt())
127     {
128     case All:
129         transactionProxyModel->setDateRange(
130                 TransactionFilterProxy::MIN_DATE,
131                 TransactionFilterProxy::MAX_DATE);
132         break;
133     case Today:
134         transactionProxyModel->setDateRange(
135                 QDateTime(current),
136                 TransactionFilterProxy::MAX_DATE);
137         break;
138     case ThisWeek: {
139         // Find last monday
140         QDate startOfWeek = current.addDays(-(current.dayOfWeek()-1));
141         transactionProxyModel->setDateRange(
142                 QDateTime(startOfWeek),
143                 TransactionFilterProxy::MAX_DATE);
144
145         } break;
146     case ThisMonth:
147         transactionProxyModel->setDateRange(
148                 QDateTime(QDate(current.year(), current.month(), 1)),
149                 TransactionFilterProxy::MAX_DATE);
150         break;
151     case LastMonth:
152         transactionProxyModel->setDateRange(
153                 QDateTime(QDate(current.year(), current.month()-1, 1)),
154                 QDateTime(QDate(current.year(), current.month(), 1)));
155         break;
156     case ThisYear:
157         transactionProxyModel->setDateRange(
158                 QDateTime(QDate(current.year(), 1, 1)),
159                 TransactionFilterProxy::MAX_DATE);
160         break;
161     case Range:
162         // TODO ask specific range
163         break;
164     }
165
166 }
167
168 void TransactionView::chooseType(int idx)
169 {
170     transactionProxyModel->setTypeFilter(
171         typeWidget->itemData(idx).toInt());
172 }
173
174 void TransactionView::changedPrefix(const QString &prefix)
175 {
176     transactionProxyModel->setAddressPrefix(prefix);
177 }
178
179 void TransactionView::changedAmount(const QString &amount)
180 {
181     qint64 amount_parsed = 0;
182     if(GUIUtil::parseMoney(amount, &amount_parsed))
183     {
184         transactionProxyModel->setMinAmount(amount_parsed);
185     }
186     else
187     {
188         transactionProxyModel->setMinAmount(0);
189     }
190 }