Wladimir J. van der Laan
14 years ago
10 changed files with 259 additions and 160 deletions
@ -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 @@ |
|||||||
|
#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