Improve ChatMonitorFilter, only show plain/notice/action msgs, and display network...
[quassel.git] / src / qtui / chatmonitorfilter.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 "chatmonitorfilter.h"
22
23 #include "buffer.h"
24 #include "client.h"
25 #include "chatlinemodel.h"
26 #include "networkmodel.h"
27
28 ChatMonitorFilter::ChatMonitorFilter(MessageModel *model, QObject *parent)
29 : MessageFilter(model, QList<BufferId>(), parent)
30 {
31   _initTime = QDateTime::currentDateTime();
32
33 }
34
35 bool ChatMonitorFilter::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const {
36   Q_UNUSED(sourceParent)
37   Message::Type type = (Message::Type)sourceModel()->data(sourceModel()->index(sourceRow, 0), MessageModel::TypeRole).toInt();
38   Message::Flags flags = (Message::Flags)sourceModel()->data(sourceModel()->index(sourceRow, 0), MessageModel::FlagsRole).toInt();
39   if(!((type & (Message::Plain | Message::Notice | Message::Action)) || flags & Message::Self))
40     return false;
41   QDateTime msgTime = sourceModel()->data(sourceModel()->index(sourceRow, 0), MessageModel::TimestampRole).toDateTime();
42   return msgTime > _initTime;
43 }
44
45 QString ChatMonitorFilter::idString() const {
46   return "ChatMonitor";
47 }
48
49 // override this to inject display of network and channel
50 QVariant ChatMonitorFilter::data(const QModelIndex &index, int role) const {
51   if(index.column() != ChatLineModel::SenderColumn) return MessageFilter::data(index, role);
52   if(role == ChatLineModel::DisplayRole) {
53     BufferId bufid = data(index, ChatLineModel::BufferIdRole).value<BufferId>();
54     if(bufid.isValid()) {
55       Buffer *buf = Client::buffer(bufid);
56       if(!buf) {
57         qDebug() << "invalid buffer!";
58         return QVariant();
59       }
60       const Network *net = Client::networkModel()->networkByIndex(Client::networkModel()->bufferIndex(bufid));
61       if(!net) {
62         qDebug() << "invalid net!";
63         return QVariant();
64       }
65       QString result = QString("<%1:%2:%3").arg(net->networkName())
66                                             .arg(buf->bufferInfo().bufferName())
67                                             .arg(MessageFilter::data(index, role).toString().mid(1));
68       return result;
69     }
70
71   }
72   return MessageFilter::data(index, role);
73
74 }