src/gui/log/logmodel.h

   1 /*
   2  * Bittorrent Client using Qt and libtorrent.
   3  * Copyright (C) 2024  Vladimir Golovnev <glassez@yandex.ru>
   4  * Copyright (C) 2020  Prince Gupta <jagannatharjun11@gmail.com>
   5  * Copyright (C) 2019  sledgehammer999 <hammered999@gmail.com>
   6  *
   7  * This program is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation; either version 2
  10  * of the License, or (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  20  *
  21  * In addition, as a special exception, the copyright holders give permission to
  22  * link this program with the OpenSSL project's "OpenSSL" library (or with
  23  * modified versions of it that use the same license as the "OpenSSL" library),
  24  * and distribute the linked executables. You must obey the GNU General Public
  25  * License in all respects for all of the code used other than "OpenSSL".  If you
  26  * modify file(s), you may extend this exception to your version of the file(s),
  27  * but you are not obligated to do so. If you do not wish to do so, delete this
  28  * exception statement from your version.
  29  */
  30
  31 #pragma once
  32
  33 #include <boost/circular_buffer.hpp>
  34
  35 #include <QAbstractListModel>
  36 #include <QColor>
  37 #include <QHash>
  38
  39 #include "base/logger.h"
  40
  41 class BaseLogModel : public QAbstractListModel
  42 {
  43     Q_DISABLE_COPY_MOVE(BaseLogModel)
  44
  45 public:
  46     enum MessageTypeRole
  47     {
  48         TimeRole = Qt::UserRole,
  49         MessageRole,
  50         TimeForegroundRole,
  51         MessageForegroundRole,
  52         TypeRole
  53     };
  54
  55     explicit BaseLogModel(QObject *parent = nullptr);
  56
  57     int rowCount(const QModelIndex &parent = {}) const override;
  58     int columnCount(const QModelIndex &parent = {}) const override;
  59     QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
  60     void reset();
  61
  62 protected:
  63     class Message
  64     {
  65     public:
  66         Message(const QString &time, const QString &message, Log::MsgType type);
  67
  68         QString time() const;
  69         QString message() const;
  70         Log::MsgType type() const;
  71
  72     private:
  73         QString m_time;
  74         QString m_message;
  75         Log::MsgType m_type;
  76     };
  77
  78     void addNewMessage(const Message &message);
  79     virtual QColor messageForeground(const Message &message) const = 0;
  80     virtual void onUIThemeChanged();
  81
  82 private:
  83     void loadColors();
  84
  85     boost::circular_buffer_space_optimized<Message> m_messages;
  86     QColor m_timeForeground;
  87 };
  88
  89 class LogMessageModel : public BaseLogModel
  90 {
  91     Q_OBJECT
  92     Q_DISABLE_COPY_MOVE(LogMessageModel)
  93
  94 public:
  95     explicit LogMessageModel(QObject *parent = nullptr);
  96
  97 private slots:
  98     void handleNewMessage(const Log::Msg &message);
  99
 100 private:
 101     QColor messageForeground(const Message &message) const override;
 102     void onUIThemeChanged() override;
 103     void loadColors();
 104
 105     QHash<int, QColor> m_foregroundForMessageTypes;
 106 };
 107
 108 class LogPeerModel : public BaseLogModel
 109 {
 110     Q_OBJECT
 111     Q_DISABLE_COPY_MOVE(LogPeerModel)
 112
 113 public:
 114     explicit LogPeerModel(QObject *parent = nullptr);
 115
 116 private slots:
 117     void handleNewMessage(const Log::Peer &peer);
 118
 119 private:
 120     QColor messageForeground(const Message &message) const override;
 121     void onUIThemeChanged() override;
 122     void loadColors();
 123
 124     QColor m_bannedPeerForeground;
 125 };