1 files changed, 156 insertions, 0 deletions
diff --git a/src/qt/clientmodel.cpp b/src/qt/clientmodel.cpp
new file mode 100644
index 0000000000..97391e0938
--- /dev/null
+++ b/src/qt/clientmodel.cpp
@@ -0,0 +1,156 @@
+#include "clientmodel.h"
+#include "main.h"
+#include "guiconstants.h"
+#include "optionsmodel.h"
+#include "addresstablemodel.h"
+#include "transactiontablemodel.h"
+
+#include <QTimer>
+
+ClientModel::ClientModel(QObject *parent) :
+    QObject(parent), 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(this);
+    addressTableModel = new AddressTableModel(this);
+    transactionTableModel = new TransactionTableModel(this);
+}
+
+qint64 ClientModel::getBalance()
+{
+    return GetBalance();
+}
+
+QString ClientModel::getAddress()
+{
+    std::vector<unsigned char> vchPubKey;
+    if (CWalletDB("r").ReadDefaultKey(vchPubKey))
+    {
+        return QString::fromStdString(PubKeyToAddress(vchPubKey));
+    }
+    else
+    {
+        return QString();
+    }
+}
+
+int ClientModel::getNumConnections()
+{
+    return vNodes.size();
+}
+
+int ClientModel::getNumBlocks()
+{
+    return nBestHeight;
+}
+
+int ClientModel::getNumTransactions()
+{
+    int numTransactions = 0;
+    CRITICAL_BLOCK(cs_mapWallet)
+    {
+        numTransactions = mapWallet.size();
+    }
+    return numTransactions;
+}
+
+void ClientModel::update()
+{
+    /* Plainly emit all signals for now. To be precise this should check
+       wether the values actually changed first.
+     */
+    emit balanceChanged(getBalance());
+    emit addressChanged(getAddress());
+    emit numConnectionsChanged(getNumConnections());
+    emit numBlocksChanged(getNumBlocks());
+    emit numTransactionsChanged(getNumTransactions());
+}
+
+void ClientModel::setAddress(const QString &defaultAddress)
+{
+    uint160 hash160;
+    std::string strAddress = defaultAddress.toStdString();
+    if (!AddressToHash160(strAddress, hash160))
+        return;
+    if (!mapPubKeys.count(hash160))
+        return;
+    CWalletDB().WriteDefaultKey(mapPubKeys[hash160]);
+}
+
+ClientModel::StatusCode ClientModel::sendCoins(const QString &payTo, qint64 payAmount)
+{
+    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 = SendMoney(scriptPubKey, payAmount, wtx, true);
+        if (strError == "")
+        {
+            return 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(cs_mapAddressBook)
+        if (!mapAddressBook.count(strAddress))
+            SetAddressBookName(strAddress, "");
+
+    return OK;
+}
+
+OptionsModel *ClientModel::getOptionsModel()
+{
+    return optionsModel;
+}
+
+AddressTableModel *ClientModel::getAddressTableModel()
+{
+    return addressTableModel;
+}
+
+TransactionTableModel *ClientModel::getTransactionTableModel()
+{
+    return transactionTableModel;
+}