client: Fix duplicated query quit messages
[quassel.git] / src / client / messagefilter.h
1 /***************************************************************************
2  *   Copyright (C) 2005-2018 by the Quassel Project                        *
3  *   devel@quassel-irc.org                                                 *
4  *                                                                         *
5  *   This program is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU General Public License as published by  *
7  *   the Free Software Foundation; either version 2 of the License, or     *
8  *   (at your option) version 3.                                           *
9  *                                                                         *
10  *   This program is distributed in the hope that it will be useful,       *
11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
13  *   GNU General Public License for more details.                          *
14  *                                                                         *
15  *   You should have received a copy of the GNU General Public License     *
16  *   along with this program; if not, write to the                         *
17  *   Free Software Foundation, Inc.,                                       *
18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
19  ***************************************************************************/
20
21 #ifndef MESSAGEFILTER_H_
22 #define MESSAGEFILTER_H_
23
24 #include <QSortFilterProxyModel>
25 #include <set>
26
27 #include "bufferinfo.h"
28 #include "client.h"
29 #include "messagemodel.h"
30 #include "networkmodel.h"
31 #include "types.h"
32
33 class MessageFilter : public QSortFilterProxyModel
34 {
35     Q_OBJECT
36
37 protected:
38     MessageFilter(QAbstractItemModel *source, QObject *parent = 0);
39
40 public:
41     MessageFilter(MessageModel *, const QList<BufferId> &buffers = QList<BufferId>(), QObject *parent = 0);
42
43     virtual bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const;
44     virtual QString idString() const;
45     inline bool isSingleBufferFilter() const { return _validBuffers.count() == 1; }
46     BufferId singleBufferId() const { return *(_validBuffers.constBegin()); }
47     inline bool containsBuffer(const BufferId &id) const { return _validBuffers.contains(id); }
48     inline QSet<BufferId> containedBuffers() const { return _validBuffers; }
49
50 public slots:
51     void messageTypeFilterChanged();
52     void messageRedirectionChanged();
53     void requestBacklog();
54     // redefined as public slot
55     void invalidateFilter() { QSortFilterProxyModel::invalidateFilter(); }
56
57 protected:
58     QString bufferName() const { return Client::networkModel()->bufferName(singleBufferId()); }
59     BufferInfo::Type bufferType() const { return Client::networkModel()->bufferType(singleBufferId()); }
60     NetworkId networkId() const { return Client::networkModel()->networkId(singleBufferId()); }
61
62 private:
63     void init();
64
65     QSet<BufferId> _validBuffers;
66     std::set<qint64> _filteredQuitMsgTime; ///< Timestamps (ms) of already forwarded quit messages
67     int _messageTypeFilter;
68
69     int _userNoticesTarget;
70     int _serverNoticesTarget;
71     int _errorMsgsTarget;
72 };
73
74
75 #endif