src/client/messagefilter.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-08 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  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "messagefilter.h"
  22
  23 #include "buffersettings.h"
  24 #include "client.h"
  25 #include "buffermodel.h"
  26 #include "messagemodel.h"
  27 #include "networkmodel.h"
  28
  29 MessageFilter::MessageFilter(QAbstractItemModel *source, QObject *parent)
  30   : QSortFilterProxyModel(parent),
  31     _messageTypeFilter(0)
  32 {
  33   init();
  34   setSourceModel(source);
  35 }
  36
  37 MessageFilter::MessageFilter(MessageModel *source, const QList<BufferId> &buffers, QObject *parent)
  38   : QSortFilterProxyModel(parent),
  39     _validBuffers(buffers.toSet()),
  40     _messageTypeFilter(0)
  41 {
  42   init();
  43   setSourceModel(source);
  44 }
  45
  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()));
  53
  54   defaultSettings.notify("serverNoticesInDefaultBuffer", this, SLOT(messageRedirectionChanged()));
  55   defaultSettings.notify("serverNoticesInStatusBuffer", this, SLOT(messageRedirectionChanged()));
  56   defaultSettings.notify("serverNoticesInCurrentBuffer", this, SLOT(messageRedirectionChanged()));
  57
  58   defaultSettings.notify("ErrorMsgsInDefaultBuffer", this, SLOT(messageRedirectionChanged()));
  59   defaultSettings.notify("ErrorMsgsInStatusBuffer", this, SLOT(messageRedirectionChanged()));
  60   defaultSettings.notify("ErrorMsgsInCurrentBuffer", this, SLOT(messageRedirectionChanged()));
  61   messageRedirectionChanged();
  62
  63   BufferSettings mySettings(idString());
  64   if(mySettings.hasFilter())
  65     _messageTypeFilter = mySettings.messageFilter();
  66   mySettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
  67 }
  68
  69 void MessageFilter::messageTypeFilterChanged() {
  70   int newFilter;
  71   BufferSettings defaultSettings;
  72   newFilter = BufferSettings().messageFilter();
  73
  74   BufferSettings mySettings(idString());
  75   if(mySettings.hasFilter())
  76     newFilter = mySettings.messageFilter();
  77
  78   if(_messageTypeFilter != newFilter) {
  79     _messageTypeFilter = newFilter;
  80     _filteredQuitMsgs.clear();
  81     invalidateFilter();
  82   }
  83 }
  84
  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();
  90
  91   _serverNoticesInDefaultBuffer = bufferSettings.value("ServerNoticesInDefaultBuffer", QVariant(false)).toBool();
  92   _serverNoticesInStatusBuffer = bufferSettings.value("ServerNoticesInStatusBuffer", QVariant(true)).toBool();
  93   _serverNoticesInCurrentBuffer = bufferSettings.value("ServerNoticesInCurrentBuffer", QVariant(false)).toBool();
  94
  95   _errorMsgsInDefaultBuffer = bufferSettings.value("ErrorMsgsInDefaultBuffer", QVariant(true)).toBool();
  96   _errorMsgsInStatusBuffer = bufferSettings.value("ErrorMsgsInStatusBuffer", QVariant(false)).toBool();
  97   _errorMsgsInCurrentBuffer = bufferSettings.value("ErrorMsgsInCurrentBuffer", QVariant(false)).toBool();
  98
  99   invalidateFilter();
 100 }
 101
 102 QString MessageFilter::idString() const {
 103   if(_validBuffers.isEmpty())
 104     return "*";
 105
 106   QList<BufferId> bufferIds = _validBuffers.toList();
 107   qSort(bufferIds);
 108
 109   QStringList bufferIdStrings;
 110   foreach(BufferId id, bufferIds)
 111     bufferIdStrings << QString::number(id.toInt());
 112
 113   return bufferIdStrings.join("|");
 114 }
 115
 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();
 120
 121   // apply message type filter
 122   if(_messageTypeFilter & messageType)
 123     return false;
 124
 125   if(_validBuffers.isEmpty())
 126     return true;
 127
 128   BufferId bufferId = sourceModel()->data(sourceIdx, MessageModel::BufferIdRole).value<BufferId>();
 129   if(!bufferId.isValid()) {
 130     return true;
 131   }
 132
 133   MsgId msgId = sourceModel()->data(sourceIdx, MessageModel::MsgIdRole).value<MsgId>();
 134   Message::Flags flags = (Message::Flags)sourceModel()->data(sourceIdx, MessageModel::FlagsRole).toInt();
 135
 136   NetworkId myNetworkId = networkId();
 137   NetworkId msgNetworkId = Client::networkModel()->networkId(bufferId);
 138   if(myNetworkId != msgNetworkId)
 139     return false;
 140
 141   bool redirect = false;
 142   bool inDefaultBuffer;
 143   bool inStatusBuffer;
 144   bool inCurrentBuffer;
 145
 146   switch(messageType) {
 147   case Message::Notice:
 148     if(Client::networkModel()->bufferType(bufferId) != BufferInfo::ChannelBuffer) {
 149       redirect = true;
 150       if(flags & Message::ServerMsg) {
 151         // server notice
 152         inDefaultBuffer = _serverNoticesInDefaultBuffer;
 153         inStatusBuffer = _serverNoticesInStatusBuffer;
 154         inCurrentBuffer = _serverNoticesInCurrentBuffer;
 155       } else {
 156         inDefaultBuffer = _userNoticesInDefaultBuffer;
 157         inStatusBuffer = _userNoticesInStatusBuffer;
 158         inCurrentBuffer = _userNoticesInCurrentBuffer;
 159       }
 160     }
 161     break;
 162   case Message::Error:
 163     redirect = true;
 164     inDefaultBuffer = _errorMsgsInDefaultBuffer;
 165     inStatusBuffer = _errorMsgsInStatusBuffer;
 166     inCurrentBuffer = _errorMsgsInCurrentBuffer;
 167     break;
 168   default:
 169     break;
 170   }
 171
 172   if(redirect) {
 173     if(_redirectedMsgs.contains(msgId))
 174       return true;
 175
 176     if(inDefaultBuffer && _validBuffers.contains(bufferId))
 177       return true;
 178
 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;
 184         return true;
 185       } else if(_validBuffers.contains(redirectedTo)) {
 186         return true;
 187       }
 188     }
 189
 190     QSet<BufferId>::const_iterator idIter = _validBuffers.constBegin();
 191     while(idIter != _validBuffers.constEnd()) {
 192       if(inStatusBuffer && Client::networkModel()->bufferType(*idIter) == BufferInfo::StatusBuffer)
 193         return true;
 194       idIter++;
 195     }
 196
 197     return false;
 198   }
 199
 200
 201   if(_validBuffers.contains(bufferId)) {
 202     return true;
 203   } else {
 204     // show Quit messages in Query buffers:
 205     if(bufferType() != BufferInfo::QueryBuffer)
 206       return false;
 207     if(!(messageType & Message::Quit))
 208       return false;
 209
 210     if(myNetworkId != msgNetworkId)
 211       return false;
 212
 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())
 216       return false;
 217
 218     if(_filteredQuitMsgs.contains(quiter, messageTimestamp))
 219       return false;
 220
 221     MessageFilter *that = const_cast<MessageFilter *>(this);
 222     that->_filteredQuitMsgs.insert(quiter,  messageTimestamp);
 223     return true;
 224   }
 225 }
 226
 227 void MessageFilter::requestBacklog() {
 228   QSet<BufferId>::const_iterator bufferIdIter = _validBuffers.constBegin();
 229   while(bufferIdIter != _validBuffers.constEnd()) {
 230     Client::messageModel()->requestBacklog(*bufferIdIter);
 231     bufferIdIter++;
 232   }
 233 }