fix sorting in address table dialog
[novacoin.git] / src / qt / addressbookdialog.cpp
1 #include "addressbookdialog.h"
2 #include "ui_addressbookdialog.h"
3
4 #include "addresstablemodel.h"
5 #include "editaddressdialog.h"
6
7 #include <QSortFilterProxyModel>
8 #include <QClipboard>
9 #include <QDebug>
10
11 AddressBookDialog::AddressBookDialog(Mode mode, QWidget *parent) :
12     QDialog(parent),
13     ui(new Ui::AddressBookDialog),
14     model(0)
15 {
16     ui->setupUi(this);
17
18     switch(mode)
19     {
20     case ForSending:
21         connect(ui->receiveTableView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(on_buttonBox_accepted()));
22         connect(ui->sendTableView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(on_buttonBox_accepted()));
23         break;
24     case ForEditing:
25         connect(ui->receiveTableView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(on_editButton_clicked()));
26         connect(ui->sendTableView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(on_editButton_clicked()));
27         break;
28     }
29 }
30
31 AddressBookDialog::~AddressBookDialog()
32 {
33     delete ui;
34 }
35
36 void AddressBookDialog::setModel(AddressTableModel *model)
37 {
38     this->model = model;
39     // Refresh list from core
40     model->updateList();
41
42     // Receive filter
43     QSortFilterProxyModel *receive_model = new QSortFilterProxyModel(this);
44     receive_model->setSourceModel(model);
45     receive_model->setDynamicSortFilter(true);
46     receive_model->setFilterRole(AddressTableModel::TypeRole);
47     receive_model->setFilterFixedString(AddressTableModel::Receive);
48     ui->receiveTableView->setModel(receive_model);
49     ui->receiveTableView->sortByColumn(0, Qt::AscendingOrder);
50
51
52     // Send filter
53     QSortFilterProxyModel *send_model = new QSortFilterProxyModel(this);
54     send_model->setSourceModel(model);
55     send_model->setDynamicSortFilter(true);
56     send_model->setFilterRole(AddressTableModel::TypeRole);
57     send_model->setFilterFixedString(AddressTableModel::Send);
58     ui->sendTableView->setModel(send_model);
59     ui->sendTableView->sortByColumn(0, Qt::AscendingOrder);
60
61     // Set column widths
62     ui->receiveTableView->horizontalHeader()->resizeSection(
63             AddressTableModel::Address, 320);
64     ui->receiveTableView->horizontalHeader()->setResizeMode(
65             AddressTableModel::Label, QHeaderView::Stretch);
66     ui->sendTableView->horizontalHeader()->resizeSection(
67             AddressTableModel::Address, 320);
68     ui->sendTableView->horizontalHeader()->setResizeMode(
69             AddressTableModel::Label, QHeaderView::Stretch);
70 }
71
72 void AddressBookDialog::setTab(int tab)
73 {
74     ui->tabWidget->setCurrentIndex(tab);
75     on_tabWidget_currentChanged(tab);
76 }
77
78 QTableView *AddressBookDialog::getCurrentTable()
79 {
80     switch(ui->tabWidget->currentIndex())
81     {
82     case SendingTab:
83         return ui->sendTableView;
84     case ReceivingTab:
85         return ui->receiveTableView;
86     default:
87         return 0;
88     }
89 }
90
91 void AddressBookDialog::on_copyToClipboard_clicked()
92 {
93     // Copy currently selected address to clipboard
94     //   (or nothing, if nothing selected)
95     QTableView *table = getCurrentTable();
96     QModelIndexList indexes = table->selectionModel()->selectedRows(AddressTableModel::Address);
97
98     foreach (QModelIndex index, indexes)
99     {
100         QVariant address = index.data();
101         QApplication::clipboard()->setText(address.toString());
102     }
103 }
104
105 void AddressBookDialog::on_editButton_clicked()
106 {
107     QModelIndexList indexes = getCurrentTable()->selectionModel()->selectedRows();
108     if(indexes.isEmpty())
109     {
110         return;
111     }
112     // Map selected index to source address book model
113     QAbstractProxyModel *proxy_model = static_cast<QAbstractProxyModel*>(getCurrentTable()->model());
114     QModelIndex selected = proxy_model->mapToSource(indexes.at(0));
115
116     // Double click also triggers edit button
117     EditAddressDialog dlg(
118             ui->tabWidget->currentIndex() == SendingTab ?
119             EditAddressDialog::EditSendingAddress :
120             EditAddressDialog::EditReceivingAddress);
121     dlg.setModel(model);
122     dlg.loadRow(selected.row());
123     if(dlg.exec())
124     {
125         dlg.saveCurrentRow();
126     }
127 }
128
129 void AddressBookDialog::on_newAddressButton_clicked()
130 {
131     EditAddressDialog dlg(
132             ui->tabWidget->currentIndex() == SendingTab ?
133             EditAddressDialog::NewSendingAddress :
134             EditAddressDialog::NewReceivingAddress);
135     dlg.setModel(model);
136     if(dlg.exec())
137     {
138         dlg.saveCurrentRow();
139     }
140 }
141
142 void AddressBookDialog::on_tabWidget_currentChanged(int index)
143 {
144     // Enable/disable buttons based on selected tab
145     switch(index)
146     {
147     case SendingTab:
148         ui->deleteButton->setEnabled(true);
149         break;
150     case ReceivingTab:
151         ui->deleteButton->setEnabled(false);
152         break;
153     }
154 }
155
156 void AddressBookDialog::on_deleteButton_clicked()
157 {
158     QTableView *table = getCurrentTable();
159     QModelIndexList indexes = table->selectionModel()->selectedRows();
160     if(!indexes.isEmpty())
161     {
162         table->model()->removeRow(indexes.at(0).row());
163     }
164 }
165
166 void AddressBookDialog::on_buttonBox_accepted()
167 {
168     QTableView *table = getCurrentTable();
169     QModelIndexList indexes = table->selectionModel()->selectedRows(AddressTableModel::Address);
170
171     foreach (QModelIndex index, indexes)
172     {
173         QVariant address = table->model()->data(index);
174         returnValue = address.toString();
175     }
176     if(!returnValue.isEmpty())
177     {
178         accept();
179     }
180     else
181     {
182         reject();
183     }
184 }
185