src/qt/bantablemodel.h

   1 // Copyright (c) 2011-2017 The Bitcoin Core developers
   2 // Distributed under the MIT software license, see the accompanying
   3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   4
   5 #ifndef BITCOIN_QT_BANTABLEMODEL_H
   6 #define BITCOIN_QT_BANTABLEMODEL_H
   7
   8 #include <net.h>
   9
  10 #include <QAbstractTableModel>
  11 #include <QStringList>
  12
  13 class ClientModel;
  14 class BanTablePriv;
  15
  16 struct CCombinedBan {
  17     CSubNet subnet;
  18     CBanEntry banEntry;
  19 };
  20
  21 class BannedNodeLessThan
  22 {
  23 public:
  24     BannedNodeLessThan(int nColumn, Qt::SortOrder fOrder) :
  25         column(nColumn), order(fOrder) {}
  26     bool operator()(const CCombinedBan& left, const CCombinedBan& right) const;
  27
  28 private:
  29     int column;
  30     Qt::SortOrder order;
  31 };
  32
  33 /**
  34    Qt model providing information about connected peers, similar to the
  35    "getpeerinfo" RPC call. Used by the rpc console UI.
  36  */
  37 class BanTableModel : public QAbstractTableModel
  38 {
  39     Q_OBJECT
  40
  41 public:
  42     explicit BanTableModel(ClientModel *parent = 0);
  43     ~BanTableModel();
  44     void startAutoRefresh();
  45     void stopAutoRefresh();
  46
  47     enum ColumnIndex {
  48         Address = 0,
  49         Bantime = 1
  50     };
  51
  52     /** @name Methods overridden from QAbstractTableModel
  53         @{*/
  54     int rowCount(const QModelIndex &parent) const;
  55     int columnCount(const QModelIndex &parent) const;
  56     QVariant data(const QModelIndex &index, int role) const;
  57     QVariant headerData(int section, Qt::Orientation orientation, int role) const;
  58     QModelIndex index(int row, int column, const QModelIndex &parent) const;
  59     Qt::ItemFlags flags(const QModelIndex &index) const;
  60     void sort(int column, Qt::SortOrder order);
  61     bool shouldShow();
  62     /*@}*/
  63
  64 public Q_SLOTS:
  65     void refresh();
  66
  67 private:
  68     ClientModel *clientModel;
  69     QStringList columns;
  70     std::unique_ptr<BanTablePriv> priv;
  71 };
  72
  73 #endif // BITCOIN_QT_BANTABLEMODEL_H