walletmodel.h

Go to the documentation of this file.

// Copyright (c) 2011-present The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_QT_WALLETMODEL_H
#define BITCOIN_QT_WALLETMODEL_H
 
#include <key.h>
 
#include <qt/walletmodeltransaction.h>
 
#include <interfaces/wallet.h>
#include <primitives/transaction_identifier.h>
#include <support/allocators/secure.h>
 
#include <vector>
 
#include <QObject>
 
enum class OutputType;
 
class AddressTableModel;
class ClientModel;
class OptionsModel;
class PlatformStyle;
class RecentRequestsTableModel;
class SendCoinsRecipient;
class TransactionTableModel;
class WalletModelTransaction;
 
class CKeyID;
class COutPoint;
class CPubKey;
class uint256;
 
namespace interfaces {
class Node;
} // namespace interfaces
namespace wallet {
class CCoinControl;
} // namespace wallet
 
QT_BEGIN_NAMESPACE
class QTimer;
QT_END_NAMESPACE
 
class WalletModel : public QObject
{
    Q_OBJECT
 
public:
    explicit WalletModel(std::unique_ptr<interfaces::Wallet> wallet, ClientModel& client_model, const PlatformStyle *platformStyle, QObject *parent = nullptr);
    ~WalletModel();
 
    enum StatusCode // Returned by sendCoins
    {
        OK,
        InvalidAmount,
        InvalidAddress,
        AmountExceedsBalance,
        DuplicateAddress,
        TransactionCreationFailed, // Error returned when wallet is still locked
        AbsurdFee
    };
 
    enum EncryptionStatus
    {
        NoKeys,       // wallet->IsWalletFlagSet(WALLET_FLAG_DISABLE_PRIVATE_KEYS)
        Unencrypted,  // !wallet->HasEncryptionKeys()
        Locked,       // wallet->HasEncryptionKeys() && wallet->IsLocked()
        Unlocked      // wallet->HasEncryptionKeys() && !wallet->IsLocked()
    };
 
    OptionsModel* getOptionsModel() const;
    AddressTableModel* getAddressTableModel() const;
    TransactionTableModel* getTransactionTableModel() const;
    RecentRequestsTableModel* getRecentRequestsTableModel() const;
 
    EncryptionStatus getEncryptionStatus() const;
 
    // Check address for validity
    bool validateAddress(const QString& address) const;
 
    // Return status record for SendCoins, contains error id + information
    struct SendCoinsReturn
    {
        SendCoinsReturn(StatusCode _status = OK, QString _reasonCommitFailed = "")
            : status(_status),
              reasonCommitFailed(_reasonCommitFailed)
        {
        }
        StatusCode status;
        QString reasonCommitFailed;
    };
 
    // prepare transaction for getting txfee before sending coins
    SendCoinsReturn prepareTransaction(WalletModelTransaction &transaction, const wallet::CCoinControl& coinControl);
 
    // Send coins to a list of recipients
    void sendCoins(WalletModelTransaction& transaction);
 
    // Wallet encryption
    bool setWalletEncrypted(const SecureString& passphrase);
    util::Expected<void, wallet::WalletError> changePassphrase(const SecureString& oldPass, const SecureString& newPass);
 
    // RAII object for unlocking wallet, returned by requestUnlock()
    class UnlockContext
    {
    public:
        UnlockContext(WalletModel *wallet, bool valid, bool relock);
        ~UnlockContext();
 
        bool isValid() const { return valid; }
 
        // Disable unused copy/move constructors/assignments explicitly.
        UnlockContext(const UnlockContext&) = delete;
        UnlockContext(UnlockContext&&) = delete;
        UnlockContext& operator=(const UnlockContext&) = delete;
        UnlockContext& operator=(UnlockContext&&) = delete;
 
    private:
        WalletModel *wallet;
        const bool valid;
        const bool relock;
    };
 
    UnlockContext requestUnlock();
 
    bool bumpFee(Txid hash, Txid& new_hash);
    void displayAddress(std::string sAddress) const;
 
    static bool isWalletEnabled();
 
    interfaces::Node& node() const { return m_node; }
    interfaces::Wallet& wallet() const { return *m_wallet; }
    ClientModel& clientModel() const { return *m_client_model; }
    void setClientModel(ClientModel* client_model);
 
    QString getWalletName() const;
    QString getDisplayName() const;
 
    bool isMultiwallet() const;
 
    void refresh(bool pk_hash_only = false);
 
    uint256 getLastBlockProcessed() const;
 
    // Retrieve the cached wallet balance
    interfaces::WalletBalances getCachedBalance() const;
 
    // If coin control has selected outputs, searches the total amount inside the wallet.
    // Otherwise, uses the wallet's cached available balance.
    CAmount getAvailableBalance(const wallet::CCoinControl* control);
 
private:
    std::unique_ptr<interfaces::Wallet> m_wallet;
    std::unique_ptr<interfaces::Handler> m_handler_unload;
    std::unique_ptr<interfaces::Handler> m_handler_status_changed;
    std::unique_ptr<interfaces::Handler> m_handler_address_book_changed;
    std::unique_ptr<interfaces::Handler> m_handler_transaction_changed;
    std::unique_ptr<interfaces::Handler> m_handler_show_progress;
    std::unique_ptr<interfaces::Handler> m_handler_can_get_addrs_changed;
    ClientModel* m_client_model;
    interfaces::Node& m_node;
 
    bool fForceCheckBalanceChanged{false};
 
    // Wallet has an options model for wallet-specific options
    // (transaction fee, for example)
    OptionsModel *optionsModel;
 
    AddressTableModel* addressTableModel{nullptr};
    TransactionTableModel* transactionTableModel{nullptr};
    RecentRequestsTableModel* recentRequestsTableModel{nullptr};
 
    // Cache some values to be able to detect changes
    interfaces::WalletBalances m_cached_balances;
    EncryptionStatus cachedEncryptionStatus{Unencrypted};
    QTimer* timer;
 
    // Block hash denoting when the last balance update was done.
    uint256 m_cached_last_update_tip{};
 
    void subscribeToCoreSignals();
    void unsubscribeFromCoreSignals();
    void checkBalanceChanged(const interfaces::WalletBalances& new_balances);
 
Q_SIGNALS:
    // Signal that balance in wallet changed
    void balanceChanged(const interfaces::WalletBalances& balances);
 
    // Encryption status of wallet changed
    void encryptionStatusChanged();
 
    // Signal emitted when wallet needs to be unlocked
    // It is valid behaviour for listeners to keep the wallet locked after this signal;
    // this means that the unlocking failed or was cancelled.
    void requireUnlock();
 
    // Fired when a message should be reported to the user
    void message(const QString &title, const QString &message, unsigned int style);
 
    // Coins sent: from wallet, to recipient, in (serialized) transaction:
    void coinsSent(WalletModel* wallet, SendCoinsRecipient recipient, QByteArray transaction);
 
    // Show progress dialog e.g. for rescan
    void showProgress(const QString &title, int nProgress);
 
    // Signal that wallet is about to be removed
    void unload();
 
    // Notify that there are now keys in the keypool
    void canGetAddressesChanged();
 
    void timerTimeout();
 
public Q_SLOTS:
    /* Starts a timer to periodically update the balance */
    void startPollBalance();
 
    /* Wallet status might have changed */
    void updateStatus();
    /* New transaction, or transaction changed status */
    void updateTransaction();
    /* New, updated or removed address book entry */
    void updateAddressBook(const QString &address, const QString &label, bool isMine, wallet::AddressPurpose purpose, int status);
    /* Current, immature or unconfirmed balance might have changed - emit 'balanceChanged' if so */
    void pollBalanceChanged();
};
 
#endif // BITCOIN_QT_WALLETMODEL_H