1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
19 ***************************************************************************/
21 #include "messagefilter.h"
23 #include "buffersettings.h"
25 #include "messagemodel.h"
26 #include "networkmodel.h"
28 MessageFilter::MessageFilter(QAbstractItemModel *source, QObject *parent)
29 : QSortFilterProxyModel(parent),
30 _messageTypeFilter(0),
31 _bufferType(BufferInfo::InvalidBuffer)
34 setSourceModel(source);
37 MessageFilter::MessageFilter(MessageModel *source, const QList<BufferId> &buffers, QObject *parent)
38 : QSortFilterProxyModel(parent),
39 _validBuffers(buffers.toSet()),
40 _messageTypeFilter(0),
41 _bufferType(BufferInfo::InvalidBuffer)
44 setSourceModel(source);
47 void MessageFilter::init() {
48 BufferSettings defaultSettings;
49 _messageTypeFilter = defaultSettings.messageFilter();
50 defaultSettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
52 BufferSettings mySettings(idString());
53 if(mySettings.hasFilter())
54 _messageTypeFilter = mySettings.messageFilter();
55 mySettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
58 void MessageFilter::messageTypeFilterChanged() {
60 BufferSettings defaultSettings;
61 newFilter = BufferSettings().messageFilter();
63 BufferSettings mySettings(idString());
64 if(mySettings.hasFilter())
65 newFilter = mySettings.messageFilter();
67 if(_messageTypeFilter != newFilter) {
68 _messageTypeFilter = newFilter;
69 _filteredQuitMsgs.clear();
74 QString MessageFilter::idString() const {
75 if(_validBuffers.isEmpty())
78 QList<BufferId> bufferIds = _validBuffers.toList();;
81 QStringList bufferIdStrings;
82 foreach(BufferId id, bufferIds)
83 bufferIdStrings << QString::number(id.toInt());
85 return bufferIdStrings.join("|");
88 bool MessageFilter::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const {
89 Q_UNUSED(sourceParent);
90 QModelIndex sourceIdx = sourceModel()->index(sourceRow, 2);
91 Message::Type messageType = (Message::Type)sourceModel()->data(sourceIdx, MessageModel::TypeRole).toInt();
93 // apply message type filter
94 if(_messageTypeFilter & messageType)
97 if(_validBuffers.isEmpty())
100 BufferId id = sourceModel()->data(sourceIdx, MessageModel::BufferIdRole).value<BufferId>();
105 if(_validBuffers.contains(id)) {
108 // show Quit messages in Query buffers:
109 if(bufferType() != BufferInfo::QueryBuffer)
111 if(!(messageType & Message::Quit))
114 uint messageTimestamp = sourceModel()->data(sourceIdx, MessageModel::TimestampRole).value<QDateTime>().toTime_t();
115 if(_filteredQuitMsgs.contains(messageTimestamp))
118 QString quiter = sourceModel()->data(sourceIdx, Qt::DisplayRole).toString().section(' ', 0, 0, QString::SectionSkipEmpty);
119 if(quiter.toLower() == bufferName().toLower()) {
120 MessageFilter *that = const_cast<MessageFilter *>(this);
121 that->_filteredQuitMsgs << messageTimestamp;
129 void MessageFilter::requestBacklog() {
130 QSet<BufferId>::const_iterator bufferIdIter = _validBuffers.constBegin();
131 while(bufferIdIter != _validBuffers.constEnd()) {
132 Client::messageModel()->requestBacklog(*bufferIdIter);
137 const QString &MessageFilter::bufferName() const {
138 if(_bufferName.isEmpty()) {
139 MessageFilter *that = const_cast<MessageFilter *>(this);
140 that->_bufferName = Client::networkModel()->bufferName(singleBufferId());
141 return that->_bufferName;
146 BufferInfo::Type MessageFilter::bufferType() const {
147 if(_bufferType == BufferInfo::InvalidBuffer) {
148 MessageFilter *that = const_cast<MessageFilter *>(this);
149 that->_bufferType = Client::networkModel()->bufferType(singleBufferId());
150 return that->_bufferType;