add export functionality for address book / receiving addresses
[novacoin.git] / src / qt / bitcoingui.cpp
1 /*
2  * Qt4 bitcoin GUI.
3  *
4  * W.J. van der Laan 2011
5  */
6 #include "bitcoingui.h"
7 #include "transactiontablemodel.h"
8 #include "addressbookpage.h"
9 #include "sendcoinsdialog.h"
10 #include "optionsdialog.h"
11 #include "aboutdialog.h"
12 #include "clientmodel.h"
13 #include "walletmodel.h"
14 #include "guiutil.h"
15 #include "editaddressdialog.h"
16 #include "optionsmodel.h"
17 #include "transactiondescdialog.h"
18 #include "addresstablemodel.h"
19 #include "transactionview.h"
20 #include "overviewpage.h"
21
22 #include <QApplication>
23 #include <QMainWindow>
24 #include <QMenuBar>
25 #include <QMenu>
26 #include <QIcon>
27 #include <QTabWidget>
28 #include <QVBoxLayout>
29 #include <QToolBar>
30 #include <QStatusBar>
31 #include <QLabel>
32 #include <QLineEdit>
33 #include <QPushButton>
34 #include <QLocale>
35 #include <QMessageBox>
36 #include <QProgressBar>
37 #include <QStackedWidget>
38 #include <QDateTime>
39
40 #include <QDebug>
41
42 #include <iostream>
43
44 BitcoinGUI::BitcoinGUI(QWidget *parent):
45     QMainWindow(parent),
46     clientModel(0),
47     walletModel(0),
48     trayIcon(0)
49 {
50     resize(850, 550);
51     setWindowTitle(tr("Bitcoin Wallet"));
52     setWindowIcon(QIcon(":icons/bitcoin"));
53
54     createActions();
55
56     // Menus
57     QMenu *file = menuBar()->addMenu("&File");
58     file->addAction(sendCoinsAction);
59     file->addAction(receiveCoinsAction);
60     file->addSeparator();
61     file->addAction(quitAction);
62     
63     QMenu *settings = menuBar()->addMenu("&Settings");
64     settings->addAction(optionsAction);
65
66     QMenu *help = menuBar()->addMenu("&Help");
67     help->addAction(aboutAction);
68     
69     // Toolbar
70     QToolBar *toolbar = addToolBar("Main toolbar");
71     toolbar->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
72     toolbar->addAction(overviewAction);
73     toolbar->addAction(sendCoinsAction);
74     toolbar->addAction(receiveCoinsAction);
75     toolbar->addAction(historyAction);
76     toolbar->addAction(addressBookAction);
77
78     QToolBar *toolbar2 = addToolBar("Transactions toolbar");
79     toolbar2->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
80     toolbar2->addAction(exportAction);
81
82     // Overview page
83     overviewPage = new OverviewPage();
84     QVBoxLayout *vbox = new QVBoxLayout();
85
86     transactionView = new TransactionView(this);
87     connect(transactionView, SIGNAL(doubleClicked(const QModelIndex&)), transactionView, SLOT(transactionDetails()));
88     vbox->addWidget(transactionView);
89
90     transactionsPage = new QWidget(this);
91     transactionsPage->setLayout(vbox);
92
93     addressBookPage = new AddressBookPage(AddressBookPage::ForEditing, AddressBookPage::SendingTab);
94
95     receiveCoinsPage = new AddressBookPage(AddressBookPage::ForEditing, AddressBookPage::ReceivingTab);
96
97     sendCoinsPage = new SendCoinsDialog(this);
98
99     centralWidget = new QStackedWidget(this);
100     centralWidget->addWidget(overviewPage);
101     centralWidget->addWidget(transactionsPage);
102     centralWidget->addWidget(addressBookPage);
103     centralWidget->addWidget(receiveCoinsPage);
104     centralWidget->addWidget(sendCoinsPage);
105     setCentralWidget(centralWidget);
106     
107     // Create status bar
108     statusBar();
109
110     labelConnections = new QLabel();
111     labelConnections->setFrameStyle(QFrame::Panel | QFrame::Sunken);
112     labelConnections->setMinimumWidth(150);
113     labelConnections->setMaximumWidth(150);
114     labelConnections->setToolTip(tr("Number of connections to other clients"));
115
116     labelBlocks = new QLabel();
117     labelBlocks->setFrameStyle(QFrame::Panel | QFrame::Sunken);
118     labelBlocks->setMinimumWidth(150);
119     labelBlocks->setMaximumWidth(150);
120     labelBlocks->setToolTip(tr("Number of blocks in the block chain"));
121
122     // Progress bar for blocks download
123     progressBarLabel = new QLabel(tr("Synchronizing with network..."));
124     progressBarLabel->setVisible(false);
125     progressBar = new QProgressBar();
126     progressBar->setToolTip(tr("Block chain synchronization in progress"));
127     progressBar->setVisible(false);
128
129     statusBar()->addWidget(progressBarLabel);
130     statusBar()->addWidget(progressBar);
131     statusBar()->addPermanentWidget(labelConnections);
132     statusBar()->addPermanentWidget(labelBlocks);
133
134     createTrayIcon();
135
136     gotoOverviewPage();
137 }
138
139 void BitcoinGUI::createActions()
140 {
141     QActionGroup *tabGroup = new QActionGroup(this);
142
143     overviewAction = new QAction(QIcon(":/icons/overview"), tr("&Overview"), this);
144     overviewAction->setCheckable(true);
145     tabGroup->addAction(overviewAction);
146
147     historyAction = new QAction(QIcon(":/icons/history"), tr("&Transactions"), this);
148     historyAction->setCheckable(true);
149     tabGroup->addAction(historyAction);
150
151     addressBookAction = new QAction(QIcon(":/icons/address-book"), tr("&Address Book"), this);
152     addressBookAction->setToolTip(tr("Edit the list of stored addresses and labels"));
153     addressBookAction->setCheckable(true);
154     tabGroup->addAction(addressBookAction);
155
156     receiveCoinsAction = new QAction(QIcon(":/icons/receiving_addresses"), tr("&Receive coins"), this);
157     receiveCoinsAction->setToolTip(tr("Show the list of addresses for receiving payments"));
158     receiveCoinsAction->setCheckable(true);
159     tabGroup->addAction(receiveCoinsAction);
160
161     sendCoinsAction = new QAction(QIcon(":/icons/send"), tr("&Send coins"), this);
162     sendCoinsAction->setToolTip(tr("Send coins to a bitcoin address"));
163     sendCoinsAction->setCheckable(true);
164     tabGroup->addAction(sendCoinsAction);
165
166     connect(overviewAction, SIGNAL(triggered()), this, SLOT(gotoOverviewPage()));
167     connect(historyAction, SIGNAL(triggered()), this, SLOT(gotoHistoryPage()));
168     connect(addressBookAction, SIGNAL(triggered()), this, SLOT(gotoAddressBookPage()));
169     connect(receiveCoinsAction, SIGNAL(triggered()), this, SLOT(gotoReceiveCoinsPage()));
170     connect(sendCoinsAction, SIGNAL(triggered()), this, SLOT(gotoSendCoinsPage()));
171
172     quitAction = new QAction(QIcon(":/icons/quit"), tr("&Exit"), this);
173     quitAction->setToolTip(tr("Quit application"));
174     aboutAction = new QAction(QIcon(":/icons/bitcoin"), tr("&About"), this);
175     aboutAction->setToolTip(tr("Show information about Bitcoin"));
176     optionsAction = new QAction(QIcon(":/icons/options"), tr("&Options..."), this);
177     optionsAction->setToolTip(tr("Modify configuration options for bitcoin"));
178     openBitcoinAction = new QAction(QIcon(":/icons/bitcoin"), tr("Open &Bitcoin"), this);
179     openBitcoinAction->setToolTip(tr("Show the Bitcoin window"));
180     exportAction = new QAction(QIcon(":/icons/export"), tr("&Export..."), this);
181     exportAction->setToolTip(tr("Export data in current view to a file"));
182
183     connect(quitAction, SIGNAL(triggered()), qApp, SLOT(quit()));
184     connect(optionsAction, SIGNAL(triggered()), this, SLOT(optionsClicked()));
185     connect(aboutAction, SIGNAL(triggered()), this, SLOT(aboutClicked()));
186     connect(openBitcoinAction, SIGNAL(triggered()), this, SLOT(show()));
187 }
188
189 void BitcoinGUI::setClientModel(ClientModel *clientModel)
190 {
191     this->clientModel = clientModel;
192
193     if(clientModel->isTestNet())
194     {
195         QString title_testnet = windowTitle() + QString(" ") + tr("[testnet]");
196         setWindowTitle(title_testnet);
197         setWindowIcon(QIcon(":icons/bitcoin_testnet"));
198         if(trayIcon)
199         {
200             trayIcon->setToolTip(title_testnet);
201             trayIcon->setIcon(QIcon(":/icons/toolbar_testnet"));
202         }
203     }
204
205     // Keep up to date with client
206     setNumConnections(clientModel->getNumConnections());
207     connect(clientModel, SIGNAL(numConnectionsChanged(int)), this, SLOT(setNumConnections(int)));
208
209     setNumBlocks(clientModel->getNumBlocks());
210     connect(clientModel, SIGNAL(numBlocksChanged(int)), this, SLOT(setNumBlocks(int)));
211
212     // Report errors from network/worker thread
213     connect(clientModel, SIGNAL(error(QString,QString)), this, SLOT(error(QString,QString)));
214 }
215
216 void BitcoinGUI::setWalletModel(WalletModel *walletModel)
217 {
218     this->walletModel = walletModel;
219
220     // Keep up to date with wallet
221     setBalance(walletModel->getBalance());
222     connect(walletModel, SIGNAL(balanceChanged(qint64)), this, SLOT(setBalance(qint64)));
223
224     setNumTransactions(walletModel->getNumTransactions());
225     connect(walletModel, SIGNAL(numTransactionsChanged(int)), this, SLOT(setNumTransactions(int)));
226
227     // Report errors from wallet thread
228     connect(walletModel, SIGNAL(error(QString,QString)), this, SLOT(error(QString,QString)));
229
230     // Put transaction list in tabs
231     transactionView->setModel(walletModel);
232
233     addressBookPage->setModel(walletModel->getAddressTableModel());
234     receiveCoinsPage->setModel(walletModel->getAddressTableModel());
235     sendCoinsPage->setModel(walletModel);
236
237     // Balloon popup for new transaction
238     connect(walletModel->getTransactionTableModel(), SIGNAL(rowsInserted(const QModelIndex &, int, int)),
239             this, SLOT(incomingTransaction(const QModelIndex &, int, int)));
240 }
241
242 void BitcoinGUI::createTrayIcon()
243 {
244     QMenu *trayIconMenu = new QMenu(this);
245     trayIconMenu->addAction(openBitcoinAction);
246     trayIconMenu->addAction(sendCoinsAction);
247     trayIconMenu->addAction(optionsAction);
248     trayIconMenu->addSeparator();
249     trayIconMenu->addAction(quitAction);
250
251     trayIcon = new QSystemTrayIcon(this);
252     trayIcon->setContextMenu(trayIconMenu);
253     trayIcon->setToolTip("Bitcoin client");
254     trayIcon->setIcon(QIcon(":/icons/toolbar"));
255     connect(trayIcon, SIGNAL(activated(QSystemTrayIcon::ActivationReason)),
256             this, SLOT(trayIconActivated(QSystemTrayIcon::ActivationReason)));
257     trayIcon->show();
258 }
259
260 void BitcoinGUI::trayIconActivated(QSystemTrayIcon::ActivationReason reason)
261 {
262     if(reason == QSystemTrayIcon::DoubleClick)
263     {
264         // Doubleclick on system tray icon triggers "open bitcoin"
265         openBitcoinAction->trigger();
266     }
267 }
268
269 void BitcoinGUI::optionsClicked()
270 {
271     OptionsDialog dlg;
272     dlg.setModel(clientModel->getOptionsModel());
273     dlg.exec();
274 }
275
276 void BitcoinGUI::aboutClicked()
277 {
278     AboutDialog dlg;
279     dlg.setModel(clientModel);
280     dlg.exec();
281 }
282
283 void BitcoinGUI::setBalance(qint64 balance)
284 {
285     overviewPage->setBalance(balance);
286 }
287
288 void BitcoinGUI::setNumConnections(int count)
289 {
290     QString icon;
291     switch(count)
292     {
293     case 0: icon = ":/icons/connect_0"; break;
294     case 1: case 2: case 3: icon = ":/icons/connect_1"; break;
295     case 4: case 5: case 6: icon = ":/icons/connect_2"; break;
296     case 7: case 8: case 9: icon = ":/icons/connect_3"; break;
297     default: icon = ":/icons/connect_4"; break;
298     }
299     labelConnections->setTextFormat(Qt::RichText);
300     labelConnections->setText("<img src=\""+icon+"\"> " + tr("%n connection(s)", "", count));
301 }
302
303 void BitcoinGUI::setNumBlocks(int count)
304 {
305     int total = clientModel->getTotalBlocksEstimate();
306     if(count < total)
307     {
308         progressBarLabel->setVisible(true);
309         progressBar->setVisible(true);
310         progressBar->setMaximum(total);
311         progressBar->setValue(count);
312     }
313     else
314     {
315         progressBarLabel->setVisible(false);
316         progressBar->setVisible(false);
317     }
318
319     QDateTime now = QDateTime::currentDateTime();
320     QDateTime lastBlockDate = clientModel->getLastBlockDate();
321     int secs = lastBlockDate.secsTo(now);
322     QString text;
323     QString icon = ":/icons/notsynced";
324
325     // "Up to date" icon, and outdated icon
326     if(secs < 30*60)
327     {
328         text = "Up to date";
329         icon = ":/icons/synced";
330     }
331     else if(secs < 60*60)
332     {
333         text = tr("%n minute(s) ago","",secs/60);
334     }
335     else if(secs < 24*60*60)
336     {
337         text = tr("%n hour(s) ago","",secs/(60*60));
338     }
339     else
340     {
341         text = tr("%n day(s) ago","",secs/(60*60*24));
342     }
343
344     labelBlocks->setText("<img src=\""+icon+"\"> " + text);
345     labelBlocks->setToolTip(tr("%n block(s) in total, last block was generated %1", "", count)
346                             .arg(QLocale::system().toString(lastBlockDate)));
347 }
348
349 void BitcoinGUI::setNumTransactions(int count)
350 {
351     overviewPage->setNumTransactions(count);
352 }
353
354 void BitcoinGUI::error(const QString &title, const QString &message)
355 {
356     // Report errors from network/worker thread
357     if(trayIcon->supportsMessages())
358     {
359         // Show as "balloon" message if possible
360         trayIcon->showMessage(title, message, QSystemTrayIcon::Critical);
361     }
362     else
363     {
364         // Fall back to old fashioned popup dialog if not
365         QMessageBox::critical(this, title,
366             message,
367             QMessageBox::Ok, QMessageBox::Ok);
368     }
369 }
370
371 void BitcoinGUI::changeEvent(QEvent *e)
372 {
373     if (e->type() == QEvent::WindowStateChange)
374     {
375         if(clientModel->getOptionsModel()->getMinimizeToTray())
376         {
377             if (isMinimized())
378             {
379                 hide();
380                 e->ignore();
381             }
382             else
383             {
384                 e->accept();
385             }
386         }
387     }
388     QMainWindow::changeEvent(e);
389 }
390
391 void BitcoinGUI::closeEvent(QCloseEvent *event)
392 {
393     if(!clientModel->getOptionsModel()->getMinimizeToTray() &&
394        !clientModel->getOptionsModel()->getMinimizeOnClose())
395     {
396         qApp->quit();
397     }
398     QMainWindow::closeEvent(event);
399 }
400
401 void BitcoinGUI::askFee(qint64 nFeeRequired, bool *payFee)
402 {
403     QString strMessage =
404         tr("This transaction is over the size limit.  You can still send it for a fee of %1, "
405           "which goes to the nodes that process your transaction and helps to support the network.  "
406           "Do you want to pay the fee?").arg(GUIUtil::formatMoney(nFeeRequired));
407     QMessageBox::StandardButton retval = QMessageBox::question(
408           this, tr("Sending..."), strMessage,
409           QMessageBox::Yes|QMessageBox::Cancel, QMessageBox::Yes);
410     *payFee = (retval == QMessageBox::Yes);
411 }
412
413 void BitcoinGUI::incomingTransaction(const QModelIndex & parent, int start, int end)
414 {
415     TransactionTableModel *ttm = walletModel->getTransactionTableModel();
416     qint64 amount = ttm->index(start, TransactionTableModel::Amount, parent)
417                     .data(Qt::EditRole).toULongLong();
418     if(amount>0 && !clientModel->inInitialBlockDownload())
419     {
420         // On incoming transaction, make an info balloon
421         // Unless the initial block download is in progress, to prevent balloon-spam
422         QString date = ttm->index(start, TransactionTableModel::Date, parent)
423                         .data().toString();
424         QString type = ttm->index(start, TransactionTableModel::Type, parent)
425                         .data().toString();
426         QString address = ttm->index(start, TransactionTableModel::ToAddress, parent)
427                         .data().toString();
428
429         trayIcon->showMessage(tr("Incoming transaction"),
430                               tr("Date: ") + date + "\n" +
431                               tr("Amount: ") + GUIUtil::formatMoney(amount, true) + "\n" +
432                               tr("Type: ") + type + "\n" +
433                               tr("Address: ") + address + "\n",
434                               QSystemTrayIcon::Information);
435     }
436 }
437
438 void BitcoinGUI::gotoOverviewPage()
439 {
440     overviewAction->setChecked(true);
441     centralWidget->setCurrentWidget(overviewPage);
442
443     exportAction->setEnabled(false);
444     disconnect(exportAction, SIGNAL(triggered()), 0, 0);
445 }
446
447 void BitcoinGUI::gotoHistoryPage()
448 {
449     historyAction->setChecked(true);
450     centralWidget->setCurrentWidget(transactionsPage);
451
452     exportAction->setEnabled(true);
453     disconnect(exportAction, SIGNAL(triggered()), 0, 0);
454     connect(exportAction, SIGNAL(triggered()), transactionView, SLOT(exportClicked()));
455 }
456
457 void BitcoinGUI::gotoAddressBookPage()
458 {
459     addressBookAction->setChecked(true);
460     centralWidget->setCurrentWidget(addressBookPage);
461
462     exportAction->setEnabled(true);
463     disconnect(exportAction, SIGNAL(triggered()), 0, 0);
464     connect(exportAction, SIGNAL(triggered()), addressBookPage, SLOT(exportClicked()));
465 }
466
467 void BitcoinGUI::gotoReceiveCoinsPage()
468 {
469     receiveCoinsAction->setChecked(true);
470     centralWidget->setCurrentWidget(receiveCoinsPage);
471
472     exportAction->setEnabled(true);
473     disconnect(exportAction, SIGNAL(triggered()), 0, 0);
474     connect(exportAction, SIGNAL(triggered()), receiveCoinsPage, SLOT(exportClicked()));
475 }
476
477 void BitcoinGUI::gotoSendCoinsPage()
478 {
479     sendCoinsAction->setChecked(true);
480     sendCoinsPage->clear();
481     centralWidget->setCurrentWidget(sendCoinsPage);
482
483     exportAction->setEnabled(false);
484     disconnect(exportAction, SIGNAL(triggered()), 0, 0);
485 }
486