Wladimir J. van der Laan
14 years ago
10 changed files with 259 additions and 160 deletions
@ -0,0 +1,124 @@
@@ -0,0 +1,124 @@
|
||||
#include "walletmodel.h" |
||||
#include "guiconstants.h" |
||||
#include "optionsmodel.h" |
||||
#include "addresstablemodel.h" |
||||
#include "transactiontablemodel.h" |
||||
|
||||
#include "headers.h" |
||||
|
||||
#include <QTimer> |
||||
|
||||
WalletModel::WalletModel(CWallet *wallet, QObject *parent) : |
||||
QObject(parent), wallet(wallet), optionsModel(0), addressTableModel(0), |
||||
transactionTableModel(0) |
||||
{ |
||||
// Until signal notifications is built into the bitcoin core,
|
||||
// simply update everything after polling using a timer.
|
||||
QTimer *timer = new QTimer(this); |
||||
connect(timer, SIGNAL(timeout()), this, SLOT(update())); |
||||
timer->start(MODEL_UPDATE_DELAY); |
||||
|
||||
optionsModel = new OptionsModel(wallet, this); |
||||
addressTableModel = new AddressTableModel(wallet, this); |
||||
transactionTableModel = new TransactionTableModel(wallet, this); |
||||
} |
||||
|
||||
qint64 WalletModel::getBalance() const |
||||
{ |
||||
return wallet->GetBalance(); |
||||
} |
||||
|
||||
int WalletModel::getNumTransactions() const |
||||
{ |
||||
int numTransactions = 0; |
||||
CRITICAL_BLOCK(wallet->cs_mapWallet) |
||||
{ |
||||
numTransactions = wallet->mapWallet.size(); |
||||
} |
||||
return numTransactions; |
||||
} |
||||
|
||||
void WalletModel::update() |
||||
{ |
||||
// Plainly emit all signals for now. To be more efficient this should check
|
||||
// whether the values actually changed first, although it'd be even better if these
|
||||
// were events coming in from the bitcoin core.
|
||||
emit balanceChanged(getBalance()); |
||||
emit numTransactionsChanged(getNumTransactions()); |
||||
|
||||
addressTableModel->update(); |
||||
} |
||||
|
||||
WalletModel::StatusCode WalletModel::sendCoins(const QString &payTo, qint64 payAmount, const QString &addToAddressBookAs) |
||||
{ |
||||
uint160 hash160 = 0; |
||||
bool valid = false; |
||||
|
||||
if(!AddressToHash160(payTo.toUtf8().constData(), hash160)) |
||||
{ |
||||
return InvalidAddress; |
||||
} |
||||
|
||||
if(payAmount <= 0) |
||||
{ |
||||
return InvalidAmount; |
||||
} |
||||
|
||||
if(payAmount > getBalance()) |
||||
{ |
||||
return AmountExceedsBalance; |
||||
} |
||||
|
||||
if((payAmount + nTransactionFee) > getBalance()) |
||||
{ |
||||
return AmountWithFeeExceedsBalance; |
||||
} |
||||
|
||||
CRITICAL_BLOCK(cs_main) |
||||
{ |
||||
// Send to bitcoin address
|
||||
CWalletTx wtx; |
||||
CScript scriptPubKey; |
||||
scriptPubKey << OP_DUP << OP_HASH160 << hash160 << OP_EQUALVERIFY << OP_CHECKSIG; |
||||
|
||||
std::string strError = wallet->SendMoney(scriptPubKey, payAmount, wtx, true); |
||||
if (strError == "") |
||||
{ |
||||
// OK
|
||||
} |
||||
else if (strError == "ABORTED") |
||||
{ |
||||
return Aborted; |
||||
} |
||||
else |
||||
{ |
||||
emit error(tr("Sending..."), QString::fromStdString(strError)); |
||||
return MiscError; |
||||
} |
||||
} |
||||
|
||||
// Add addresses that we've sent to to the address book
|
||||
std::string strAddress = payTo.toStdString(); |
||||
CRITICAL_BLOCK(wallet->cs_mapAddressBook) |
||||
{ |
||||
if (!wallet->mapAddressBook.count(strAddress)) |
||||
wallet->SetAddressBookName(strAddress, addToAddressBookAs.toStdString()); |
||||
} |
||||
|
||||
return OK; |
||||
} |
||||
|
||||
OptionsModel *WalletModel::getOptionsModel() |
||||
{ |
||||
return optionsModel; |
||||
} |
||||
|
||||
AddressTableModel *WalletModel::getAddressTableModel() |
||||
{ |
||||
return addressTableModel; |
||||
} |
||||
|
||||
TransactionTableModel *WalletModel::getTransactionTableModel() |
||||
{ |
||||
return transactionTableModel; |
||||
} |
||||
@ -0,0 +1,62 @@
@@ -0,0 +1,62 @@
|
||||
#ifndef WALLETMODEL_H |
||||
#define WALLETMODEL_H |
||||
|
||||
#include <QObject> |
||||
|
||||
class OptionsModel; |
||||
class AddressTableModel; |
||||
class TransactionTableModel; |
||||
class CWallet; |
||||
|
||||
// Interface to a Bitcoin wallet
|
||||
class WalletModel : public QObject |
||||
{ |
||||
Q_OBJECT |
||||
public: |
||||
explicit WalletModel(CWallet *wallet, QObject *parent = 0); |
||||
|
||||
enum StatusCode |
||||
{ |
||||
OK, |
||||
InvalidAmount, |
||||
InvalidAddress, |
||||
AmountExceedsBalance, |
||||
AmountWithFeeExceedsBalance, |
||||
Aborted, |
||||
MiscError |
||||
}; |
||||
|
||||
OptionsModel *getOptionsModel(); |
||||
AddressTableModel *getAddressTableModel(); |
||||
TransactionTableModel *getTransactionTableModel(); |
||||
|
||||
qint64 getBalance() const; |
||||
int getNumTransactions() const; |
||||
|
||||
/* Send coins */ |
||||
StatusCode sendCoins(const QString &payTo, qint64 payAmount, const QString &addToAddressBookAs=QString()); |
||||
private: |
||||
CWallet *wallet; |
||||
|
||||
// Wallet has an options model for wallet-specific options
|
||||
// (transaction fee, for example)
|
||||
OptionsModel *optionsModel; |
||||
|
||||
AddressTableModel *addressTableModel; |
||||
TransactionTableModel *transactionTableModel; |
||||
|
||||
signals: |
||||
void balanceChanged(qint64 balance); |
||||
void numTransactionsChanged(int count); |
||||
|
||||
// Asynchronous error notification
|
||||
void error(const QString &title, const QString &message); |
||||
|
||||
public slots: |
||||
|
||||
private slots: |
||||
void update(); |
||||
}; |
||||
|
||||
|
||||
#endif // WALLETMODEL_H
|
||||
Loading…
Reference in new issue