cd9291ac60da4087596151af6d3bf8cf5ea9934c
[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 "addressbookdialog.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
21 #include <QApplication>
22 #include <QMainWindow>
23 #include <QMenuBar>
24 #include <QMenu>
25 #include <QIcon>
26 #include <QTabWidget>
27 #include <QVBoxLayout>
28 #include <QToolBar>
29 #include <QStatusBar>
30 #include <QLabel>
31 #include <QLineEdit>
32 #include <QPushButton>
33 #include <QLocale>
34 #include <QMessageBox>
35 #include <QProgressBar>
36
37 #include <QDebug>
38
39 #include <iostream>
40
41 BitcoinGUI::BitcoinGUI(QWidget *parent):
42     QMainWindow(parent),
43     clientModel(0),
44     walletModel(0),
45     trayIcon(0)
46 {
47     resize(850, 550);
48     setWindowTitle(tr("Bitcoin"));
49     setWindowIcon(QIcon(":icons/bitcoin"));
50
51     createActions();
52
53     // Menus
54     QMenu *file = menuBar()->addMenu("&File");
55     file->addAction(sendCoins);
56     file->addAction(receiveCoins);
57     file->addSeparator();
58     file->addAction(quit);
59     
60     QMenu *settings = menuBar()->addMenu("&Settings");
61     settings->addAction(options);
62
63     QMenu *help = menuBar()->addMenu("&Help");
64     help->addAction(about);
65     
66     // Toolbar
67     QToolBar *toolbar = addToolBar("Main toolbar");
68     toolbar->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
69     toolbar->addAction(sendCoins);
70     toolbar->addAction(receiveCoins);
71     toolbar->addAction(addressbook);
72
73     // Balance: <balance>
74     QHBoxLayout *hbox_balance = new QHBoxLayout();
75     hbox_balance->addWidget(new QLabel(tr("Balance:")));
76     hbox_balance->addSpacing(5);/* Add some spacing between the label and the text */
77
78     labelBalance = new QLabel();
79     labelBalance->setFont(QFont("Monospace", -1, QFont::Bold));
80     labelBalance->setToolTip(tr("Your current balance"));
81     labelBalance->setTextInteractionFlags(Qt::TextSelectableByMouse|Qt::TextSelectableByKeyboard);
82     hbox_balance->addWidget(labelBalance);
83     hbox_balance->addStretch(1);
84     
85     QVBoxLayout *vbox = new QVBoxLayout();
86     vbox->addLayout(hbox_balance);
87
88     transactionView = new TransactionView(this);
89     connect(transactionView, SIGNAL(doubleClicked(const QModelIndex&)), this, SLOT(transactionDetails(const QModelIndex&)));
90     vbox->addWidget(transactionView);
91
92     QWidget *centralwidget = new QWidget(this);
93     centralwidget->setLayout(vbox);
94     setCentralWidget(centralwidget);
95     
96     // Create status bar
97     statusBar();
98
99     labelConnections = new QLabel();
100     labelConnections->setFrameStyle(QFrame::Panel | QFrame::Sunken);
101     labelConnections->setMinimumWidth(150);
102     labelConnections->setToolTip(tr("Number of connections to other clients"));
103
104     labelBlocks = new QLabel();
105     labelBlocks->setFrameStyle(QFrame::Panel | QFrame::Sunken);
106     labelBlocks->setMinimumWidth(130);
107     labelBlocks->setToolTip(tr("Number of blocks in the block chain"));
108
109     labelTransactions = new QLabel();
110     labelTransactions->setFrameStyle(QFrame::Panel | QFrame::Sunken);
111     labelTransactions->setMinimumWidth(130);
112     labelTransactions->setToolTip(tr("Number of transactions in your wallet"));
113
114     // Progress bar for blocks download
115     progressBarLabel = new QLabel(tr("Synchronizing with network..."));
116     progressBarLabel->setVisible(false);
117     progressBar = new QProgressBar();
118     progressBar->setToolTip(tr("Block chain synchronization in progress"));
119     progressBar->setVisible(false);
120
121     statusBar()->addWidget(progressBarLabel);
122     statusBar()->addWidget(progressBar);
123     statusBar()->addPermanentWidget(labelConnections);
124     statusBar()->addPermanentWidget(labelBlocks);
125     statusBar()->addPermanentWidget(labelTransactions);
126
127     createTrayIcon();
128 }
129
130 void BitcoinGUI::createActions()
131 {
132     quit = new QAction(QIcon(":/icons/quit"), tr("&Exit"), this);
133     quit->setToolTip(tr("Quit application"));
134     sendCoins = new QAction(QIcon(":/icons/send"), tr("&Send coins"), this);
135     sendCoins->setToolTip(tr("Send coins to a bitcoin address"));
136     addressbook = new QAction(QIcon(":/icons/address-book"), tr("&Address Book"), this);
137     addressbook->setToolTip(tr("Edit the list of stored addresses and labels"));
138     about = new QAction(QIcon(":/icons/bitcoin"), tr("&About"), this);
139     about->setToolTip(tr("Show information about Bitcoin"));
140     receiveCoins = new QAction(QIcon(":/icons/receiving_addresses"), tr("&Receive coins"), this);
141     receiveCoins->setToolTip(tr("Show the list of addresses for receiving payments"));
142     options = new QAction(QIcon(":/icons/options"), tr("&Options..."), this);
143     options->setToolTip(tr("Modify configuration options for bitcoin"));
144     openBitcoin = new QAction(QIcon(":/icons/bitcoin"), "Open &Bitcoin", this);
145     openBitcoin->setToolTip(tr("Show the Bitcoin window"));
146
147     connect(quit, SIGNAL(triggered()), qApp, SLOT(quit()));
148     connect(sendCoins, SIGNAL(triggered()), this, SLOT(sendCoinsClicked()));
149     connect(addressbook, SIGNAL(triggered()), this, SLOT(addressbookClicked()));
150     connect(receiveCoins, SIGNAL(triggered()), this, SLOT(receiveCoinsClicked()));
151     connect(options, SIGNAL(triggered()), this, SLOT(optionsClicked()));
152     connect(about, SIGNAL(triggered()), this, SLOT(aboutClicked()));
153     connect(openBitcoin, SIGNAL(triggered()), this, SLOT(show()));
154 }
155
156 void BitcoinGUI::setClientModel(ClientModel *clientModel)
157 {
158     this->clientModel = clientModel;
159
160     if(clientModel->isTestNet())
161     {
162         setWindowTitle(tr("Bitcoin [testnet]"));
163         setWindowIcon(QIcon(":icons/bitcoin_testnet"));
164         if(trayIcon)
165         {
166             trayIcon->setToolTip(tr("Bitcoin [testnet]"));
167             trayIcon->setIcon(QIcon(":/icons/toolbar_testnet"));
168         }
169     }
170
171     // Keep up to date with client
172     setNumConnections(clientModel->getNumConnections());
173     connect(clientModel, SIGNAL(numConnectionsChanged(int)), this, SLOT(setNumConnections(int)));
174
175     setNumBlocks(clientModel->getNumBlocks());
176     connect(clientModel, SIGNAL(numBlocksChanged(int)), this, SLOT(setNumBlocks(int)));
177
178     // Report errors from network/worker thread
179     connect(clientModel, SIGNAL(error(QString,QString)), this, SLOT(error(QString,QString)));
180 }
181
182 void BitcoinGUI::setWalletModel(WalletModel *walletModel)
183 {
184     this->walletModel = walletModel;
185
186     // Keep up to date with wallet
187     setBalance(walletModel->getBalance());
188     connect(walletModel, SIGNAL(balanceChanged(qint64)), this, SLOT(setBalance(qint64)));
189
190     setNumTransactions(walletModel->getNumTransactions());
191     connect(walletModel, SIGNAL(numTransactionsChanged(int)), this, SLOT(setNumTransactions(int)));
192
193     // Report errors from wallet thread
194     connect(walletModel, SIGNAL(error(QString,QString)), this, SLOT(error(QString,QString)));
195
196     // Put transaction list in tabs
197     transactionView->setModel(walletModel->getTransactionTableModel());
198
199     // Balloon popup for new transaction
200     connect(walletModel->getTransactionTableModel(), SIGNAL(rowsInserted(const QModelIndex &, int, int)),
201             this, SLOT(incomingTransaction(const QModelIndex &, int, int)));
202 }
203
204 void BitcoinGUI::createTrayIcon()
205 {
206     QMenu *trayIconMenu = new QMenu(this);
207     trayIconMenu->addAction(openBitcoin);
208     trayIconMenu->addAction(sendCoins);
209     trayIconMenu->addAction(options);
210     trayIconMenu->addSeparator();
211     trayIconMenu->addAction(quit);
212
213     trayIcon = new QSystemTrayIcon(this);
214     trayIcon->setContextMenu(trayIconMenu);
215     trayIcon->setToolTip("Bitcoin client");
216     trayIcon->setIcon(QIcon(":/icons/toolbar"));
217     connect(trayIcon, SIGNAL(activated(QSystemTrayIcon::ActivationReason)),
218             this, SLOT(trayIconActivated(QSystemTrayIcon::ActivationReason)));
219     trayIcon->show();
220 }
221
222 void BitcoinGUI::trayIconActivated(QSystemTrayIcon::ActivationReason reason)
223 {
224     if(reason == QSystemTrayIcon::DoubleClick)
225     {
226         // Doubleclick on system tray icon triggers "open bitcoin"
227         openBitcoin->trigger();
228     }
229 }
230
231 void BitcoinGUI::sendCoinsClicked()
232 {
233     SendCoinsDialog dlg;
234     dlg.setModel(walletModel);
235     dlg.exec();
236 }
237
238 void BitcoinGUI::addressbookClicked()
239 {
240     AddressBookDialog dlg(AddressBookDialog::ForEditing);
241     dlg.setModel(walletModel->getAddressTableModel());
242     dlg.setTab(AddressBookDialog::SendingTab);
243     dlg.exec();
244 }
245
246 void BitcoinGUI::receiveCoinsClicked()
247 {
248     AddressBookDialog dlg(AddressBookDialog::ForEditing);
249     dlg.setModel(walletModel->getAddressTableModel());
250     dlg.setTab(AddressBookDialog::ReceivingTab);
251     dlg.exec();
252 }
253
254 void BitcoinGUI::optionsClicked()
255 {
256     OptionsDialog dlg;
257     dlg.setModel(clientModel->getOptionsModel());
258     dlg.exec();
259 }
260
261 void BitcoinGUI::aboutClicked()
262 {
263     AboutDialog dlg;
264     dlg.setModel(clientModel);
265     dlg.exec();
266 }
267
268 void BitcoinGUI::setBalance(qint64 balance)
269 {
270     labelBalance->setText(GUIUtil::formatMoney(balance) + QString(" BTC"));
271 }
272
273 void BitcoinGUI::setNumConnections(int count)
274 {
275     QString icon;
276     switch(count)
277     {
278     case 0: icon = ":/icons/connect_0"; break;
279     case 1: case 2: case 3: icon = ":/icons/connect_1"; break;
280     case 4: case 5: case 6: icon = ":/icons/connect_2"; break;
281     case 7: case 8: case 9: icon = ":/icons/connect_3"; break;
282     default: icon = ":/icons/connect_4"; break;
283     }
284     labelConnections->setTextFormat(Qt::RichText);
285     labelConnections->setText("<img src=\""+icon+"\"> " + tr("%n connection(s)", "", count));
286 }
287
288 void BitcoinGUI::setNumBlocks(int count)
289 {
290     int total = clientModel->getTotalBlocksEstimate();
291     if(count < total)
292     {
293         progressBarLabel->setVisible(true);
294         progressBar->setVisible(true);
295         progressBar->setMaximum(total);
296         progressBar->setValue(count);
297     }
298     else
299     {
300         progressBarLabel->setVisible(false);
301         progressBar->setVisible(false);
302     }
303
304     labelBlocks->setText(tr("%n block(s)", "", count));
305 }
306
307 void BitcoinGUI::setNumTransactions(int count)
308 {
309     labelTransactions->setText(tr("%n transaction(s)", "", count));
310 }
311
312 void BitcoinGUI::error(const QString &title, const QString &message)
313 {
314     // Report errors from network/worker thread
315     if(trayIcon->supportsMessages())
316     {
317         // Show as "balloon" message if possible
318         trayIcon->showMessage(title, message, QSystemTrayIcon::Critical);
319     }
320     else
321     {
322         // Fall back to old fashioned popup dialog if not
323         QMessageBox::critical(this, title,
324             message,
325             QMessageBox::Ok, QMessageBox::Ok);
326     }
327 }
328
329 void BitcoinGUI::changeEvent(QEvent *e)
330 {
331     if (e->type() == QEvent::WindowStateChange)
332     {
333         if(clientModel->getOptionsModel()->getMinimizeToTray())
334         {
335             if (isMinimized())
336             {
337                 hide();
338                 e->ignore();
339             }
340             else
341             {
342                 e->accept();
343             }
344         }
345     }
346     QMainWindow::changeEvent(e);
347 }
348
349 void BitcoinGUI::closeEvent(QCloseEvent *event)
350 {
351     if(!clientModel->getOptionsModel()->getMinimizeToTray() &&
352        !clientModel->getOptionsModel()->getMinimizeOnClose())
353     {
354         qApp->quit();
355     }
356     QMainWindow::closeEvent(event);
357 }
358
359 void BitcoinGUI::askFee(qint64 nFeeRequired, bool *payFee)
360 {
361     QString strMessage =
362         tr("This transaction is over the size limit.  You can still send it for a fee of %1, "
363           "which goes to the nodes that process your transaction and helps to support the network.  "
364           "Do you want to pay the fee?").arg(GUIUtil::formatMoney(nFeeRequired));
365     QMessageBox::StandardButton retval = QMessageBox::question(
366           this, tr("Sending..."), strMessage,
367           QMessageBox::Yes|QMessageBox::Cancel, QMessageBox::Yes);
368     *payFee = (retval == QMessageBox::Yes);
369 }
370
371 void BitcoinGUI::transactionDetails(const QModelIndex& idx)
372 {
373     // A transaction is doubleclicked
374     TransactionDescDialog dlg(idx);
375     dlg.exec();
376 }
377
378 void BitcoinGUI::incomingTransaction(const QModelIndex & parent, int start, int end)
379 {
380     TransactionTableModel *ttm = walletModel->getTransactionTableModel();
381     qint64 amount = ttm->index(start, TransactionTableModel::Amount, parent)
382                     .data(Qt::EditRole).toULongLong();
383     if(amount>0 && !clientModel->inInitialBlockDownload())
384     {
385         // On incoming transaction, make an info balloon
386         // Unless the initial block download is in progress, to prevent balloon-spam
387         QString date = ttm->index(start, TransactionTableModel::Date, parent)
388                         .data().toString();
389         QString type = ttm->index(start, TransactionTableModel::Type, parent)
390                         .data().toString();
391         QString address = ttm->index(start, TransactionTableModel::ToAddress, parent)
392                         .data().toString();
393
394         trayIcon->showMessage(tr("Incoming transaction"),
395                               tr("Date: ") + date + "\n" +
396                               tr("Amount: ") + GUIUtil::formatMoney(amount, true) + "\n" +
397                               tr("Type: ") + type + "\n" +
398                               tr("Address: ") + address + "\n",
399                               QSystemTrayIcon::Information);
400     }
401 }