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 "buffermodel.h"
26 #include "messagemodel.h"
27 #include "networkmodel.h"
29 MessageFilter::MessageFilter(QAbstractItemModel *source, QObject *parent)
30 : QSortFilterProxyModel(parent),
34 setSourceModel(source);
37 MessageFilter::MessageFilter(MessageModel *source, const QList<BufferId> &buffers, QObject *parent)
38 : QSortFilterProxyModel(parent),
39 _validBuffers(buffers.toSet()),
43 setSourceModel(source);
46 void MessageFilter::init() {
47 BufferSettings defaultSettings;
48 _messageTypeFilter = defaultSettings.messageFilter();
49 defaultSettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
50 defaultSettings.notify("UserNoticesInDefaultBuffer", this, SLOT(messageRedirectionChanged()));
51 defaultSettings.notify("UserNoticesInStatusBuffer", this, SLOT(messageRedirectionChanged()));
52 defaultSettings.notify("UserNoticesInCurrentBuffer", this, SLOT(messageRedirectionChanged()));
54 defaultSettings.notify("serverNoticesInDefaultBuffer", this, SLOT(messageRedirectionChanged()));
55 defaultSettings.notify("serverNoticesInStatusBuffer", this, SLOT(messageRedirectionChanged()));
56 defaultSettings.notify("serverNoticesInCurrentBuffer", this, SLOT(messageRedirectionChanged()));
58 defaultSettings.notify("ErrorMsgsInDefaultBuffer", this, SLOT(messageRedirectionChanged()));
59 defaultSettings.notify("ErrorMsgsInStatusBuffer", this, SLOT(messageRedirectionChanged()));
60 defaultSettings.notify("ErrorMsgsInCurrentBuffer", this, SLOT(messageRedirectionChanged()));
61 messageRedirectionChanged();
63 BufferSettings mySettings(idString());
64 if(mySettings.hasFilter())
65 _messageTypeFilter = mySettings.messageFilter();
66 mySettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
69 void MessageFilter::messageTypeFilterChanged() {
71 BufferSettings defaultSettings;
72 newFilter = BufferSettings().messageFilter();
74 BufferSettings mySettings(idString());
75 if(mySettings.hasFilter())
76 newFilter = mySettings.messageFilter();
78 if(_messageTypeFilter != newFilter) {
79 _messageTypeFilter = newFilter;
80 _filteredQuitMsgs.clear();
85 void MessageFilter::messageRedirectionChanged() {
86 BufferSettings bufferSettings;
87 _userNoticesInDefaultBuffer = bufferSettings.value("UserNoticesInDefaultBuffer", QVariant(true)).toBool();
88 _userNoticesInStatusBuffer = bufferSettings.value("UserNoticesInStatusBuffer", QVariant(false)).toBool();
89 _userNoticesInCurrentBuffer = bufferSettings.value("UserNoticesInCurrentBuffer", QVariant(false)).toBool();
91 _serverNoticesInDefaultBuffer = bufferSettings.value("ServerNoticesInDefaultBuffer", QVariant(false)).toBool();
92 _serverNoticesInStatusBuffer = bufferSettings.value("ServerNoticesInStatusBuffer", QVariant(true)).toBool();
93 _serverNoticesInCurrentBuffer = bufferSettings.value("ServerNoticesInCurrentBuffer", QVariant(false)).toBool();
95 _errorMsgsInDefaultBuffer = bufferSettings.value("ErrorMsgsInDefaultBuffer", QVariant(true)).toBool();
96 _errorMsgsInStatusBuffer = bufferSettings.value("ErrorMsgsInStatusBuffer", QVariant(false)).toBool();
97 _errorMsgsInCurrentBuffer = bufferSettings.value("ErrorMsgsInCurrentBuffer", QVariant(false)).toBool();
102 QString MessageFilter::idString() const {
103 if(_validBuffers.isEmpty())
106 QList<BufferId> bufferIds = _validBuffers.toList();
109 QStringList bufferIdStrings;
110 foreach(BufferId id, bufferIds)
111 bufferIdStrings << QString::number(id.toInt());
113 return bufferIdStrings.join("|");
116 bool MessageFilter::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const {
117 Q_UNUSED(sourceParent);
118 QModelIndex sourceIdx = sourceModel()->index(sourceRow, 2);
119 Message::Type messageType = (Message::Type)sourceModel()->data(sourceIdx, MessageModel::TypeRole).toInt();
121 // apply message type filter
122 if(_messageTypeFilter & messageType)
125 if(_validBuffers.isEmpty())
128 BufferId bufferId = sourceModel()->data(sourceIdx, MessageModel::BufferIdRole).value<BufferId>();
129 if(!bufferId.isValid()) {
133 MsgId msgId = sourceModel()->data(sourceIdx, MessageModel::MsgIdRole).value<MsgId>();
134 Message::Flags flags = (Message::Flags)sourceModel()->data(sourceIdx, MessageModel::FlagsRole).toInt();
136 NetworkId myNetworkId = networkId();
137 NetworkId msgNetworkId = Client::networkModel()->networkId(bufferId);
138 if(myNetworkId != msgNetworkId)
141 bool redirect = false;
142 bool inDefaultBuffer;
144 bool inCurrentBuffer;
146 switch(messageType) {
147 case Message::Notice:
148 if(Client::networkModel()->bufferType(bufferId) != BufferInfo::ChannelBuffer) {
150 if(flags & Message::ServerMsg) {
152 inDefaultBuffer = _serverNoticesInDefaultBuffer;
153 inStatusBuffer = _serverNoticesInStatusBuffer;
154 inCurrentBuffer = _serverNoticesInCurrentBuffer;
156 inDefaultBuffer = _userNoticesInDefaultBuffer;
157 inStatusBuffer = _userNoticesInStatusBuffer;
158 inCurrentBuffer = _userNoticesInCurrentBuffer;
164 inDefaultBuffer = _errorMsgsInDefaultBuffer;
165 inStatusBuffer = _errorMsgsInStatusBuffer;
166 inCurrentBuffer = _errorMsgsInCurrentBuffer;
173 if(_redirectedMsgs.contains(msgId))
176 if(inDefaultBuffer && _validBuffers.contains(bufferId))
179 if(inCurrentBuffer && !(flags & Message::Backlog) && _validBuffers.contains(Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>())) {
180 BufferId redirectedTo = sourceModel()->data(sourceIdx, MessageModel::RedirectedToRole).value<BufferId>();
181 if(!redirectedTo.isValid()) {
182 sourceModel()->setData(sourceIdx, QVariant::fromValue<BufferId>(singleBufferId()), MessageModel::RedirectedToRole);
183 _redirectedMsgs << msgId;
185 } else if(_validBuffers.contains(redirectedTo)) {
190 QSet<BufferId>::const_iterator idIter = _validBuffers.constBegin();
191 while(idIter != _validBuffers.constEnd()) {
192 if(inStatusBuffer && Client::networkModel()->bufferType(*idIter) == BufferInfo::StatusBuffer)
201 if(_validBuffers.contains(bufferId)) {
204 // show Quit messages in Query buffers:
205 if(bufferType() != BufferInfo::QueryBuffer)
207 if(!(messageType & Message::Quit))
210 if(myNetworkId != msgNetworkId)
213 uint messageTimestamp = sourceModel()->data(sourceIdx, MessageModel::TimestampRole).value<QDateTime>().toTime_t();
214 QString quiter = sourceModel()->data(sourceIdx, Qt::DisplayRole).toString().section(' ', 0, 0, QString::SectionSkipEmpty).toLower();
215 if(quiter != bufferName().toLower())
218 if(_filteredQuitMsgs.contains(quiter, messageTimestamp))
221 MessageFilter *that = const_cast<MessageFilter *>(this);
222 that->_filteredQuitMsgs.insert(quiter, messageTimestamp);
227 void MessageFilter::requestBacklog() {
228 QSet<BufferId>::const_iterator bufferIdIter = _validBuffers.constBegin();
229 while(bufferIdIter != _validBuffers.constEnd()) {
230 Client::messageModel()->requestBacklog(*bufferIdIter);