1 /***************************************************************************
2 * Copyright (C) 2005-09 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"
28 #include "clientignorelistmanager.h"
30 MessageFilter::MessageFilter(QAbstractItemModel *source, QObject *parent)
31 : QSortFilterProxyModel(parent),
35 setSourceModel(source);
38 MessageFilter::MessageFilter(MessageModel *source, const QList<BufferId> &buffers, QObject *parent)
39 : QSortFilterProxyModel(parent),
40 _validBuffers(buffers.toSet()),
44 setSourceModel(source);
47 void MessageFilter::init() {
48 setDynamicSortFilter(true);
50 _userNoticesTarget = _serverNoticesTarget = _errorMsgsTarget = -1;
52 BufferSettings defaultSettings;
53 defaultSettings.notify("UserNoticesTarget", this, SLOT(messageRedirectionChanged()));
54 defaultSettings.notify("ServerNoticesTarget", this, SLOT(messageRedirectionChanged()));
55 defaultSettings.notify("ErrorMsgsTarget", this, SLOT(messageRedirectionChanged()));
56 messageRedirectionChanged();
58 _messageTypeFilter = defaultSettings.messageFilter();
59 defaultSettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
61 BufferSettings mySettings(idString());
62 if(mySettings.hasFilter())
63 _messageTypeFilter = mySettings.messageFilter();
64 mySettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
65 mySettings.notify("hasMessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
68 void MessageFilter::messageTypeFilterChanged() {
70 BufferSettings defaultSettings;
71 newFilter = BufferSettings().messageFilter();
73 BufferSettings mySettings(idString());
74 if(mySettings.hasFilter())
75 newFilter = mySettings.messageFilter();
77 if(_messageTypeFilter != newFilter) {
78 _messageTypeFilter = newFilter;
79 _filteredQuitMsgs.clear();
84 void MessageFilter::messageRedirectionChanged() {
85 BufferSettings bufferSettings;
88 if(_userNoticesTarget != bufferSettings.userNoticesTarget()) {
89 _userNoticesTarget = bufferSettings.userNoticesTarget();
93 if(_serverNoticesTarget != bufferSettings.serverNoticesTarget()) {
94 _serverNoticesTarget = bufferSettings.serverNoticesTarget();
98 if(_errorMsgsTarget != bufferSettings.errorMsgsTarget()) {
99 _errorMsgsTarget = bufferSettings.errorMsgsTarget();
107 QString MessageFilter::idString() const {
108 if(_validBuffers.isEmpty())
111 QList<BufferId> bufferIds = _validBuffers.toList();
114 QStringList bufferIdStrings;
115 foreach(BufferId id, bufferIds)
116 bufferIdStrings << QString::number(id.toInt());
118 return bufferIdStrings.join("|");
121 bool MessageFilter::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const {
122 Q_UNUSED(sourceParent);
123 QModelIndex sourceIdx = sourceModel()->index(sourceRow, 2);
124 Message::Type messageType = (Message::Type)sourceModel()->data(sourceIdx, MessageModel::TypeRole).toInt();
126 // apply message type filter
127 if(_messageTypeFilter & messageType)
130 if(_validBuffers.isEmpty())
133 BufferId bufferId = sourceModel()->data(sourceIdx, MessageModel::BufferIdRole).value<BufferId>();
134 if(!bufferId.isValid()) {
138 MsgId msgId = sourceModel()->data(sourceIdx, MessageModel::MsgIdRole).value<MsgId>();
139 Message::Flags flags = (Message::Flags)sourceModel()->data(sourceIdx, MessageModel::FlagsRole).toInt();
141 NetworkId myNetworkId = networkId();
142 NetworkId msgNetworkId = Client::networkModel()->networkId(bufferId);
143 if(myNetworkId != msgNetworkId)
146 // ignorelist handling
147 const MessageModelItem *item = const_cast<const MessageModel*>(static_cast<MessageModel*>(sourceModel()))->messageItemAt(sourceRow);
148 // only match if message is not flagged as server msg
149 if(!(item->message().flags() & Message::ServerMsg) &&
150 Client::ignoreListManager()->match(item->message(), Client::networkModel()->networkName(item->bufferId())))
153 if(flags & Message::Redirected) {
154 int redirectionTarget = 0;
155 switch(messageType) {
156 case Message::Notice:
157 if(Client::networkModel()->bufferType(bufferId) != BufferInfo::ChannelBuffer) {
158 if(flags & Message::ServerMsg) {
160 redirectionTarget = _serverNoticesTarget;
162 redirectionTarget = _userNoticesTarget;
167 redirectionTarget = _errorMsgsTarget;
173 if(redirectionTarget & BufferSettings::DefaultBuffer && _validBuffers.contains(bufferId))
176 if(redirectionTarget & BufferSettings::CurrentBuffer && !(flags & Message::Backlog)) {
177 BufferId redirectedTo = sourceModel()->data(sourceIdx, MessageModel::RedirectedToRole).value<BufferId>();
178 if(!redirectedTo.isValid()) {
179 BufferId redirectedTo = Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
180 if(redirectedTo.isValid())
181 sourceModel()->setData(sourceIdx, QVariant::fromValue<BufferId>(redirectedTo), MessageModel::RedirectedToRole);
184 if(_validBuffers.contains(redirectedTo))
188 if(redirectionTarget & BufferSettings::StatusBuffer) {
189 QSet<BufferId>::const_iterator idIter = _validBuffers.constBegin();
190 while(idIter != _validBuffers.constEnd()) {
191 if(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);