clientmodel.h

Go to the documentation of this file.

// Copyright (c) 2011-2014 The Bitcoin developers
// Copyright (c) 2014-2015 The Dash developers
// Copyright (c) 2015-2022 The PIVX Core developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef PIVX_QT_CLIENTMODEL_H
#define PIVX_QT_CLIENTMODEL_H
 
#include "uint256.h"
#include "chain.h"
 
#include <QObject>
#include <QDateTime>
 
#include <atomic>
#include <memory>
 
class AddressTableModel;
class BanTableModel;
class OptionsModel;
class PeerTableModel;
class TransactionTableModel;
 
namespace interfaces {
    class Handler;
}
 
QT_BEGIN_NAMESPACE
class QDateTime;
class QTimer;
QT_END_NAMESPACE
 
enum BlockSource {
    BLOCK_SOURCE_NONE,
    BLOCK_SOURCE_REINDEX,
    BLOCK_SOURCE_DISK,
    BLOCK_SOURCE_NETWORK
};
 
enum NumConnections {
    CONNECTIONS_NONE = 0,
    CONNECTIONS_IN = (1U << 0),
    CONNECTIONS_OUT = (1U << 1),
    CONNECTIONS_ALL = (CONNECTIONS_IN | CONNECTIONS_OUT),
};
 
class ClientModel : public QObject
{
    Q_OBJECT
 
public:
    explicit ClientModel(OptionsModel* optionsModel, QObject* parent = 0);
    ~ClientModel();
 
    OptionsModel* getOptionsModel();
    PeerTableModel* getPeerTableModel();
    BanTableModel *getBanTableModel();
 
    int getNumConnections(unsigned int flags = CONNECTIONS_ALL) const;
    int getNumBlocksAtStartup();
 
    // from cached block index
    int getNumBlocks();
    QDateTime getLastBlockDate() const;
    QString getLastBlockHash() const;
    uint256 getLastBlockProcessed() const;
    int getLastBlockProcessedHeight() const;
    int64_t getLastBlockProcessedTime() const;
    double getVerificationProgress() const;
    bool isTipCached() const;
 
    quint64 getTotalBytesRecv() const;
    quint64 getTotalBytesSent() const;
 
    bool inInitialBlockDownload() const;
    enum BlockSource getBlockSource() const;
    bool getNetworkActive() const;
    void setNetworkActive(bool active);
    QString getStatusBarWarnings() const;
 
    QString formatFullVersion() const;
    bool isReleaseVersion() const;
    QString clientName() const;
    QString formatClientStartupTime() const;
    QString dataDir() const;
 
    void setCacheTip(const CBlockIndex* const tip) { cacheTip = tip; }
    void setCacheReindexing(bool reindex) { cachedReindexing = reindex; }
    void setCacheImporting(bool import) { cachedImporting = import; }
    void setCacheInitialSync(bool _initialSync) { cachedInitialSync = _initialSync; }
 
    bool getTorInfo(std::string& ip_port) const;
 
    static void mapPort(bool use_upnp, bool use_natpmp);
 
    // Start/Stop the masternode polling timer
    void startMasternodesTimer();
    void stopMasternodesTimer();
    // Force a MN count update calling mnmanager directly locking its internal mutex.
    // Future todo: implement an event based update and remove the lock requirement.
    QString getMasternodesCountString();
    int getMasternodesCount() const { return m_cached_masternodes_count; }
 
private:
    // Listeners
    std::unique_ptr<interfaces::Handler> m_handler_show_progress;
    std::unique_ptr<interfaces::Handler> m_handler_notify_num_connections_changed;
    std::unique_ptr<interfaces::Handler> m_handler_notify_net_activity_changed;
    std::unique_ptr<interfaces::Handler> m_handler_notify_alert_changed;
    std::unique_ptr<interfaces::Handler> m_handler_banned_list_changed;
    std::unique_ptr<interfaces::Handler> m_handler_notify_block_tip;
 
    QString getMasternodeCountString();
    OptionsModel* optionsModel;
    PeerTableModel* peerTableModel;
    BanTableModel *banTableModel;
 
    const CBlockIndex* cacheTip{nullptr};
    QString cachedMasternodeCountString;
    bool cachedReindexing;
    bool cachedImporting;
    std::atomic<bool> cachedInitialSync{false};
 
    int numBlocksAtStartup;
 
    QTimer* pollTimer;
    QTimer* pollMnTimer;
 
    std::atomic_int m_cached_masternodes_count{0};
 
    void subscribeToCoreSignals();
    void unsubscribeFromCoreSignals();
 
Q_SIGNALS:
    void numConnectionsChanged(int count);
    void numBlocksChanged(int count);
    void networkActiveChanged(bool networkActive);
    void strMasternodesChanged(const QString& strMasternodes);
    void alertsChanged(const QString& warnings);
    void bytesChanged(quint64 totalBytesIn, quint64 totalBytesOut);
 
    void message(const QString& title, const QString& message, unsigned int style, bool* ret = nullptr);
 
    // Show progress dialog e.g. for verifychain
    void showProgress(const QString& title, int nProgress);
 
public Q_SLOTS:
    void updateTimer();
    void updateMnTimer();
    void updateNumConnections(int numConnections);
    void updateNetworkActive(bool networkActive);
    void updateAlert();
    void updateBanlist();
};
 
#endif // PIVX_QT_CLIENTMODEL_H