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 "messagemodel.h"
  26 #include "networkmodel.h"
  27
  28 MessageFilter::MessageFilter(QAbstractItemModel *source, QObject *parent)
  29   : QSortFilterProxyModel(parent),
  30     _messageTypeFilter(0),
  31     _bufferType(BufferInfo::InvalidBuffer)
  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     _bufferType(BufferInfo::InvalidBuffer)
  42 {
  43   init();
  44   setSourceModel(source);
  45 }
  46
  47 void MessageFilter::init() {
  48   BufferSettings defaultSettings;
  49   _messageTypeFilter = defaultSettings.messageFilter();
  50   defaultSettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
  51
  52   BufferSettings mySettings(idString());
  53   if(mySettings.hasFilter())
  54     _messageTypeFilter = mySettings.messageFilter();
  55   mySettings.notify("MessageTypeFilter", this, SLOT(messageTypeFilterChanged()));
  56 }
  57
  58 void MessageFilter::messageTypeFilterChanged() {
  59   int newFilter;
  60   BufferSettings defaultSettings;
  61   newFilter = BufferSettings().messageFilter();
  62
  63   BufferSettings mySettings(idString());
  64   if(mySettings.hasFilter())
  65     newFilter = mySettings.messageFilter();
  66
  67   if(_messageTypeFilter != newFilter) {
  68     _messageTypeFilter = newFilter;
  69     _filteredQuitMsgs.clear();
  70     invalidateFilter();
  71   }
  72 }
  73
  74 QString MessageFilter::idString() const {
  75   if(_validBuffers.isEmpty())
  76     return "*";
  77
  78   QList<BufferId> bufferIds = _validBuffers.toList();;
  79   qSort(bufferIds);
  80
  81   QStringList bufferIdStrings;
  82   foreach(BufferId id, bufferIds)
  83     bufferIdStrings << QString::number(id.toInt());
  84
  85   return bufferIdStrings.join("|");
  86 }
  87
  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();
  92
  93   // apply message type filter
  94   if(_messageTypeFilter & messageType)
  95     return false;
  96
  97   if(_validBuffers.isEmpty())
  98     return true;
  99
 100   BufferId id = sourceModel()->data(sourceIdx, MessageModel::BufferIdRole).value<BufferId>();
 101   if(!id.isValid()) {
 102     return true;
 103   }
 104
 105   if(_validBuffers.contains(id)) {
 106     return true;
 107   } else {
 108     // show Quit messages in Query buffers:
 109     if(bufferType() != BufferInfo::QueryBuffer)
 110       return false;
 111     if(!(messageType & Message::Quit))
 112       return false;
 113
 114     uint messageTimestamp = sourceModel()->data(sourceIdx, MessageModel::TimestampRole).value<QDateTime>().toTime_t();
 115     if(_filteredQuitMsgs.contains(messageTimestamp))
 116       return false;
 117
 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;
 122       return true;
 123     } else {
 124       return false;
 125     }
 126   }
 127 }
 128
 129 void MessageFilter::requestBacklog() {
 130   QSet<BufferId>::const_iterator bufferIdIter = _validBuffers.constBegin();
 131   while(bufferIdIter != _validBuffers.constEnd()) {
 132     Client::messageModel()->requestBacklog(*bufferIdIter);
 133     bufferIdIter++;
 134   }
 135 }
 136
 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;
 142   }
 143   return _bufferName;
 144 }
 145
 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;
 151   }
 152   return _bufferType;
 153 }