Fixed highlighting of buffers caused by ignored messages.
[quassel.git] / src / client / networkmodel.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-09 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 "networkmodel.h"
22
23 #include <QAbstractItemView>
24 #include <QTextDocument>        // for Qt::escape()
25
26 #include "buffermodel.h"
27 #include "buffersettings.h"
28 #include "client.h"
29 #include "clientignorelistmanager.h"
30 #include "clientsettings.h"
31 #include "ircchannel.h"
32 #include "network.h"
33 #include "signalproxy.h"
34
35 /*****************************************
36 *  Network Items
37 *****************************************/
38 NetworkItem::NetworkItem(const NetworkId &netid, AbstractTreeItem *parent)
39   : PropertyMapItem(QList<QString>() << "networkName" << "currentServer" << "nickCount", parent),
40     _networkId(netid),
41     _statusBufferItem(0)
42 {
43   // DO NOT EMIT dataChanged() DIRECTLY IN NetworkItem
44   // use networkDataChanged() instead. Otherwise you will end up in a infinite loop
45   // as we "sync" the dataChanged() signals of NetworkItem and StatusBufferItem
46   setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
47   connect(this, SIGNAL(networkDataChanged(int)), this, SIGNAL(dataChanged(int)));
48   connect(this, SIGNAL(beginRemoveChilds(int, int)), this, SLOT(onBeginRemoveChilds(int, int)));
49 }
50
51 QVariant NetworkItem::data(int column, int role) const {
52   switch(role) {
53   case NetworkModel::BufferIdRole:
54   case NetworkModel::BufferInfoRole:
55   case NetworkModel::BufferTypeRole:
56   case NetworkModel::BufferActivityRole:
57     if(_statusBufferItem)
58       return _statusBufferItem->data(column, role);
59     else
60       return QVariant();
61   case NetworkModel::NetworkIdRole:
62     return qVariantFromValue(_networkId);
63   case NetworkModel::ItemTypeRole:
64     return NetworkModel::NetworkItemType;
65   case NetworkModel::ItemActiveRole:
66     return isActive();
67   default:
68     return PropertyMapItem::data(column, role);
69   }
70 }
71
72 // FIXME shouldn't we check the bufferItemCache here?
73 BufferItem *NetworkItem::findBufferItem(BufferId bufferId) {
74   BufferItem *bufferItem = 0;
75
76   for(int i = 0; i < childCount(); i++) {
77     bufferItem = qobject_cast<BufferItem *>(child(i));
78     if(!bufferItem)
79       continue;
80     if(bufferItem->bufferId() == bufferId)
81       return bufferItem;
82   }
83   return 0;
84 }
85
86
87 BufferItem *NetworkItem::bufferItem(const BufferInfo &bufferInfo) {
88   BufferItem *bufferItem = findBufferItem(bufferInfo);
89   if(bufferItem)
90     return bufferItem;
91
92   switch(bufferInfo.type()) {
93   case BufferInfo::StatusBuffer:
94     _statusBufferItem = new StatusBufferItem(bufferInfo, this);
95     bufferItem = _statusBufferItem;
96     disconnect(this, SIGNAL(networkDataChanged(int)), this, SIGNAL(dataChanged(int)));
97     connect(this, SIGNAL(networkDataChanged(int)), bufferItem, SIGNAL(dataChanged(int)));
98     connect(bufferItem, SIGNAL(dataChanged(int)), this, SIGNAL(dataChanged(int)));
99     break;
100   case BufferInfo::ChannelBuffer:
101     bufferItem = new ChannelBufferItem(bufferInfo, this);
102     break;
103   case BufferInfo::QueryBuffer:
104     bufferItem = new QueryBufferItem(bufferInfo, this);
105     break;
106   default:
107     bufferItem = new BufferItem(bufferInfo, this);
108   }
109
110   newChild(bufferItem);
111
112   // postprocess... this is necessary because Qt doesn't seem to like adding children which already have children on their own
113   switch(bufferInfo.type()) {
114   case BufferInfo::ChannelBuffer:
115     {
116       ChannelBufferItem *channelBufferItem = static_cast<ChannelBufferItem *>(bufferItem);
117       if(_network) {
118         IrcChannel *ircChannel = _network->ircChannel(bufferInfo.bufferName());
119         if(ircChannel)
120           channelBufferItem->attachIrcChannel(ircChannel);
121       }
122     }
123     break;
124   default:
125     break;
126   }
127
128   return bufferItem;
129 }
130
131 void NetworkItem::attachNetwork(Network *network) {
132   if(!network)
133     return;
134
135   _network = network;
136
137   connect(network, SIGNAL(networkNameSet(QString)),
138           this, SLOT(setNetworkName(QString)));
139   connect(network, SIGNAL(currentServerSet(QString)),
140           this, SLOT(setCurrentServer(QString)));
141   connect(network, SIGNAL(ircChannelAdded(IrcChannel *)),
142           this, SLOT(attachIrcChannel(IrcChannel *)));
143   connect(network, SIGNAL(ircUserAdded(IrcUser *)),
144           this, SLOT(attachIrcUser(IrcUser *)));
145   connect(network, SIGNAL(connectedSet(bool)),
146           this, SIGNAL(networkDataChanged()));
147   connect(network, SIGNAL(destroyed()),
148           this, SIGNAL(networkDataChanged()));
149
150   emit networkDataChanged();
151 }
152
153 void NetworkItem::attachIrcChannel(IrcChannel *ircChannel) {
154   ChannelBufferItem *channelItem;
155   for(int i = 0; i < childCount(); i++) {
156     channelItem = qobject_cast<ChannelBufferItem *>(child(i));
157     if(!channelItem)
158       continue;
159
160     if(channelItem->bufferName().toLower() == ircChannel->name().toLower()) {
161       channelItem->attachIrcChannel(ircChannel);
162       return;
163     }
164   }
165 }
166
167 void NetworkItem::attachIrcUser(IrcUser *ircUser) {
168   QueryBufferItem *queryItem = 0;
169   for(int i = 0; i < childCount(); i++) {
170     queryItem = qobject_cast<QueryBufferItem *>(child(i));
171     if(!queryItem)
172       continue;
173
174     if(queryItem->bufferName().toLower() == ircUser->nick().toLower()) {
175       queryItem->setIrcUser(ircUser);
176       break;
177     }
178   }
179 }
180
181 void NetworkItem::setNetworkName(const QString &networkName) {
182   Q_UNUSED(networkName);
183   emit networkDataChanged(0);
184 }
185
186 void NetworkItem::setCurrentServer(const QString &serverName) {
187   Q_UNUSED(serverName);
188   emit networkDataChanged(1);
189 }
190
191
192 QString NetworkItem::toolTip(int column) const {
193   Q_UNUSED(column);
194
195   QStringList toolTip(QString("<b>%1</b>").arg(Qt::escape(networkName())));
196   toolTip.append(tr("Server: %1").arg(Qt::escape(currentServer())));
197   toolTip.append(tr("Users: %1").arg(nickCount()));
198
199   if(_network) {
200     toolTip.append(tr("Lag: %1 msecs").arg(_network->latency()));
201   }
202
203   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
204 }
205
206 void NetworkItem::onBeginRemoveChilds(int start, int end) {
207   for(int i = start; i <= end; i++) {
208     StatusBufferItem *statusBufferItem = qobject_cast<StatusBufferItem *>(child(i));
209     if(statusBufferItem) {
210       _statusBufferItem = 0;
211       break;
212     }
213   }
214 }
215
216 /*****************************************
217 *  Fancy Buffer Items
218 *****************************************/
219 BufferItem::BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
220   : PropertyMapItem(QStringList() << "bufferName" << "topic" << "nickCount", parent),
221     _bufferInfo(bufferInfo),
222     _activity(BufferInfo::NoActivity)
223 {
224   setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled);
225 }
226
227 void BufferItem::setActivityLevel(BufferInfo::ActivityLevel level) {
228   if(_activity != level) {
229     _activity = level;
230     emit dataChanged();
231   }
232 }
233
234 void BufferItem::clearActivityLevel() {
235   _activity = BufferInfo::NoActivity;
236   _firstUnreadMsgId = MsgId();
237
238   // FIXME remove with core proto v11
239   if(!(Client::coreFeatures() & Quassel::SynchronizedMarkerLine)) {
240     _markerLineMsgId = _lastSeenMsgId;
241   }
242
243   emit dataChanged();
244 }
245
246 void BufferItem::updateActivityLevel(const Message &msg) {
247   if(isCurrentBuffer()) {
248     return;
249   }
250
251   if(msg.flags() & Message::Self)       // don't update activity for our own messages
252     return;
253
254   if (Client::ignoreListManager()
255       && Client::ignoreListManager()->match(msg, qobject_cast<NetworkItem *>(parent())->networkName()))
256     return;
257
258   if(msg.msgId() <= lastSeenMsgId())
259     return;
260
261   bool stateChanged = false;
262   if(!firstUnreadMsgId().isValid() || msg.msgId() < firstUnreadMsgId()) {
263     stateChanged = true;
264     _firstUnreadMsgId = msg.msgId();
265   }
266
267   BufferInfo::ActivityLevel oldLevel = activityLevel();
268
269   _activity |= BufferInfo::OtherActivity;
270   if(msg.type() & (Message::Plain | Message::Notice | Message::Action))
271     _activity |= BufferInfo::NewMessage;
272
273   if(msg.flags() & Message::Highlight)
274     _activity |= BufferInfo::Highlight;
275
276   stateChanged |= (oldLevel != _activity);
277
278   if(stateChanged)
279     emit dataChanged();
280 }
281
282 QVariant BufferItem::data(int column, int role) const {
283   switch(role) {
284   case NetworkModel::ItemTypeRole:
285     return NetworkModel::BufferItemType;
286   case NetworkModel::BufferIdRole:
287     return qVariantFromValue(bufferInfo().bufferId());
288   case NetworkModel::NetworkIdRole:
289     return qVariantFromValue(bufferInfo().networkId());
290   case NetworkModel::BufferInfoRole:
291     return qVariantFromValue(bufferInfo());
292   case NetworkModel::BufferTypeRole:
293     return int(bufferType());
294   case NetworkModel::ItemActiveRole:
295     return isActive();
296   case NetworkModel::BufferActivityRole:
297     return (int)activityLevel();
298   case NetworkModel::BufferFirstUnreadMsgIdRole:
299     return qVariantFromValue(firstUnreadMsgId());
300   case NetworkModel::MarkerLineMsgIdRole:
301     return qVariantFromValue(markerLineMsgId());
302   default:
303     return PropertyMapItem::data(column, role);
304   }
305 }
306
307 bool BufferItem::setData(int column, const QVariant &value, int role) {
308   switch(role) {
309   case NetworkModel::BufferActivityRole:
310     setActivityLevel((BufferInfo::ActivityLevel)value.toInt());
311     return true;
312   default:
313     return PropertyMapItem::setData(column, value, role);
314   }
315   return true;
316 }
317
318 void BufferItem::setBufferName(const QString &name) {
319   _bufferInfo = BufferInfo(_bufferInfo.bufferId(), _bufferInfo.networkId(), _bufferInfo.type(), _bufferInfo.groupId(), name);
320   emit dataChanged(0);
321 }
322
323 void BufferItem::setLastSeenMsgId(MsgId msgId) {
324   _lastSeenMsgId = msgId;
325
326   // FIXME remove with core protocol v11
327   if(!(Client::coreFeatures() & Quassel::SynchronizedMarkerLine)) {
328     if(!isCurrentBuffer())
329       _markerLineMsgId = msgId;
330   }
331
332   setActivityLevel(BufferInfo::NoActivity);
333 }
334
335 void BufferItem::setMarkerLineMsgId(MsgId msgId) {
336   _markerLineMsgId = msgId;
337   emit dataChanged();
338 }
339
340 bool BufferItem::isCurrentBuffer() const {
341   return _bufferInfo.bufferId() == Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
342 }
343
344 QString BufferItem::toolTip(int column) const {
345   Q_UNUSED(column);
346   return tr("<p> %1 - %2 </p>").arg(bufferInfo().bufferId().toInt()).arg(bufferName());
347 }
348
349 /*****************************************
350 *  StatusBufferItem
351 *****************************************/
352 StatusBufferItem::StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
353   : BufferItem(bufferInfo, parent)
354 {
355 }
356
357 QString StatusBufferItem::toolTip(int column) const {
358   NetworkItem *networkItem = qobject_cast<NetworkItem *>(parent());
359   if(networkItem)
360     return networkItem->toolTip(column);
361   else
362     return QString();
363 }
364
365 /*****************************************
366 *  QueryBufferItem
367 *****************************************/
368 QueryBufferItem::QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
369   : BufferItem(bufferInfo, parent),
370     _ircUser(0)
371 {
372   setFlags(flags() | Qt::ItemIsDropEnabled | Qt::ItemIsEditable);
373
374   const Network *net = Client::network(bufferInfo.networkId());
375   if(!net)
376     return;
377
378   IrcUser *ircUser = net->ircUser(bufferInfo.bufferName());
379   setIrcUser(ircUser);
380 }
381
382 QVariant QueryBufferItem::data(int column, int role) const {
383   switch(role) {
384   case Qt::EditRole:
385     return BufferItem::data(column, Qt::DisplayRole);
386   case NetworkModel::IrcUserRole:
387     return QVariant::fromValue<QObject *>(_ircUser);
388   case NetworkModel::UserAwayRole:
389     return (bool)_ircUser ? _ircUser->isAway() : false;
390   default:
391     return BufferItem::data(column, role);
392   }
393 }
394
395 bool QueryBufferItem::setData(int column, const QVariant &value, int role) {
396   if(column != 0)
397     return BufferItem::setData(column, value, role);
398
399   switch(role) {
400   case Qt::EditRole:
401     {
402       QString newName = value.toString();
403       if(!newName.isEmpty()) {
404         Client::renameBuffer(bufferId(), newName);
405         return true;
406       } else {
407         return false;
408       }
409     }
410     break;
411   default:
412     return BufferItem::setData(column, value, role);
413   }
414 }
415
416 void QueryBufferItem::setBufferName(const QString &name) {
417   BufferItem::setBufferName(name);
418   NetworkId netId = data(0, NetworkModel::NetworkIdRole).value<NetworkId>();
419   const Network *net = Client::network(netId);
420   if(net)
421     setIrcUser(net->ircUser(name));
422 }
423
424 QString QueryBufferItem::toolTip(int column) const {
425   // pretty much code duplication of IrcUserItem::toolTip() but inheritance won't solve this...
426   Q_UNUSED(column);
427   QStringList toolTip;
428
429   toolTip.append(tr("<b>Query with %1</b>").arg(bufferName()));
430
431   if(_ircUser) {
432     if(_ircUser->userModes() != "") toolTip[0].append(QString(" (+%1)").arg(_ircUser->userModes()));
433     if(_ircUser->isAway()) {
434       toolTip[0].append(QString(" (away%1)").arg(!_ircUser->awayMessage().isEmpty() ? (QString(" ") + _ircUser->awayMessage()) : QString()));
435     }
436     if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
437     if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(_ircUser->nick()).arg(_ircUser->ircOperator()));
438     if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
439     if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
440
441     toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
442
443     if(_ircUser->idleTime().isValid()) {
444       QDateTime now = QDateTime::currentDateTime();
445       QDateTime idle = _ircUser->idleTime();
446       int idleTime = idle.secsTo(now);
447       toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
448     }
449     if(_ircUser->loginTime().isValid()) {
450       toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
451     }
452
453     if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
454   }
455
456   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
457 }
458
459 void QueryBufferItem::setIrcUser(IrcUser *ircUser) {
460   if(_ircUser == ircUser)
461     return;
462
463   if(_ircUser) {
464     disconnect(_ircUser, 0, this, 0);
465   }
466
467   if(ircUser) {
468     connect(ircUser, SIGNAL(quited()), this, SLOT(removeIrcUser()));
469     connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
470   }
471
472   _ircUser = ircUser;
473   emit dataChanged();
474 }
475
476 void QueryBufferItem::removeIrcUser() {
477   _ircUser = 0;
478   emit dataChanged();
479 }
480
481 /*****************************************
482 *  ChannelBufferItem
483 *****************************************/
484 ChannelBufferItem::ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
485   : BufferItem(bufferInfo, parent),
486     _ircChannel(0)
487 {
488 }
489
490 QVariant ChannelBufferItem::data(int column, int role) const {
491   switch(role) {
492     case NetworkModel::IrcChannelRole:
493       return QVariant::fromValue<QObject *>(_ircChannel);
494     default:
495       return BufferItem::data(column, role);
496   }
497 }
498
499 QString ChannelBufferItem::toolTip(int column) const {
500   Q_UNUSED(column);
501   QStringList toolTip;
502
503   toolTip.append(tr("<b>Channel %1</b>").arg(Qt::escape(bufferName())));
504   if(isActive()) {
505     //TODO: add channel modes
506     toolTip.append(tr("<b>Users:</b> %1").arg(nickCount()));
507     if(_ircChannel) {
508       QString channelMode = _ircChannel->channelModeString(); // channelModeString is compiled on the fly -> thus cache the result
509       if(!channelMode.isEmpty())
510         toolTip.append(tr("<b>Mode:</b> %1").arg(channelMode));
511     }
512
513     ItemViewSettings s;
514     bool showTopic = s.displayTopicInTooltip();
515     if(showTopic) {
516       QString _topic = topic();
517       if(_topic != "") {
518         _topic = stripFormatCodes(_topic);
519         _topic = Qt::escape(_topic);
520         toolTip.append(QString("<font size='-2'>&nbsp;</font>"));
521         toolTip.append(tr("<b>Topic:</b> %1").arg(_topic));
522       }
523     }
524   } else {
525     toolTip.append(tr("Not active <br /> Double-click to join"));
526   }
527
528   return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
529 }
530
531 void ChannelBufferItem::attachIrcChannel(IrcChannel *ircChannel) {
532   Q_ASSERT(!_ircChannel && ircChannel);
533
534   _ircChannel = ircChannel;
535
536   connect(ircChannel, SIGNAL(topicSet(QString)),
537           this, SLOT(setTopic(QString)));
538   connect(ircChannel, SIGNAL(ircUsersJoined(QList<IrcUser *>)),
539           this, SLOT(join(QList<IrcUser *>)));
540   connect(ircChannel, SIGNAL(ircUserParted(IrcUser *)),
541           this, SLOT(part(IrcUser *)));
542   connect(ircChannel, SIGNAL(parted()),
543           this, SLOT(ircChannelParted()));
544   connect(ircChannel, SIGNAL(ircUserModesSet(IrcUser *, QString)),
545           this, SLOT(userModeChanged(IrcUser *)));
546   connect(ircChannel, SIGNAL(ircUserModeAdded(IrcUser *, QString)),
547           this, SLOT(userModeChanged(IrcUser *)));
548   connect(ircChannel, SIGNAL(ircUserModeRemoved(IrcUser *, QString)),
549           this, SLOT(userModeChanged(IrcUser *)));
550
551   if(!ircChannel->ircUsers().isEmpty())
552     join(ircChannel->ircUsers());
553
554   emit dataChanged();
555 }
556
557 void ChannelBufferItem::ircChannelParted() {
558   Q_CHECK_PTR(_ircChannel);
559   disconnect(_ircChannel, 0, this, 0);
560   _ircChannel = 0;
561   emit dataChanged();
562   removeAllChilds();
563 }
564
565 void ChannelBufferItem::join(const QList<IrcUser *> &ircUsers) {
566   addUsersToCategory(ircUsers);
567   emit dataChanged(2);
568 }
569
570 UserCategoryItem *ChannelBufferItem::findCategoryItem(int categoryId) {
571   UserCategoryItem *categoryItem = 0;
572
573   for(int i = 0; i < childCount(); i++) {
574     categoryItem = qobject_cast<UserCategoryItem *>(child(i));
575     if(!categoryItem)
576       continue;
577     if(categoryItem->categoryId() == categoryId)
578       return categoryItem;
579   }
580   return 0;
581 }
582
583 void ChannelBufferItem::addUserToCategory(IrcUser *ircUser) {
584   addUsersToCategory(QList<IrcUser *>() << ircUser);
585 }
586
587 void ChannelBufferItem::addUsersToCategory(const QList<IrcUser *> &ircUsers) {
588   Q_ASSERT(_ircChannel);
589
590   QHash<UserCategoryItem *, QList<IrcUser *> > categories;
591
592   int categoryId = -1;
593   UserCategoryItem *categoryItem = 0;
594
595   foreach(IrcUser *ircUser, ircUsers) {
596     categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
597     categoryItem = findCategoryItem(categoryId);
598     if(!categoryItem) {
599       categoryItem = new UserCategoryItem(categoryId, this);
600       categories[categoryItem] = QList<IrcUser *>();
601       newChild(categoryItem);
602     }
603     categories[categoryItem] << ircUser;
604   }
605
606   QHash<UserCategoryItem *, QList<IrcUser *> >::const_iterator catIter = categories.constBegin();
607   while(catIter != categories.constEnd()) {
608     catIter.key()->addUsers(catIter.value());
609     catIter++;
610   }
611 }
612
613 void ChannelBufferItem::part(IrcUser *ircUser) {
614   if(!ircUser) {
615     qWarning() << bufferName() << "ChannelBufferItem::part(): unknown User" << ircUser;
616     return;
617   }
618
619   disconnect(ircUser, 0, this, 0);
620   removeUserFromCategory(ircUser);
621   emit dataChanged(2);
622 }
623
624 void ChannelBufferItem::removeUserFromCategory(IrcUser *ircUser) {
625   if(!_ircChannel) {
626     // If we parted the channel there might still be some ircUsers connected.
627     // in that case we just ignore the call
628     Q_ASSERT(childCount() == 0);
629     return;
630   }
631
632   UserCategoryItem *categoryItem = 0;
633   for(int i = 0; i < childCount(); i++) {
634     categoryItem = qobject_cast<UserCategoryItem *>(child(i));
635     if(categoryItem->removeUser(ircUser)) {
636       if(categoryItem->childCount() == 0)
637         removeChild(i);
638       break;
639     }
640   }
641 }
642
643 void ChannelBufferItem::userModeChanged(IrcUser *ircUser) {
644   Q_ASSERT(_ircChannel);
645
646   int categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
647   UserCategoryItem *categoryItem = findCategoryItem(categoryId);
648
649   if(categoryItem) {
650     if(categoryItem->findIrcUser(ircUser)) {
651       return; // already in the right category;
652     }
653   } else {
654     categoryItem = new UserCategoryItem(categoryId, this);
655     newChild(categoryItem);
656   }
657
658   // find the item that needs reparenting
659   IrcUserItem *ircUserItem = 0;
660   for(int i = 0; i < childCount(); i++) {
661     UserCategoryItem *oldCategoryItem = qobject_cast<UserCategoryItem *>(child(i));
662     Q_ASSERT(oldCategoryItem);
663     IrcUserItem *userItem = oldCategoryItem->findIrcUser(ircUser);
664     if(userItem) {
665       ircUserItem = userItem;
666       break;
667     }
668   }
669
670   if(!ircUserItem) {
671     qWarning() << "ChannelBufferItem::userModeChanged(IrcUser *): unable to determine old category of" << ircUser;
672     return;
673   }
674   ircUserItem->reParent(categoryItem);
675 }
676
677 /*****************************************
678 *  User Category Items (like @vh etc.)
679 *****************************************/
680 // we hardcode this even though we have PREFIX in network... but that wouldn't help with mapping modes to
681 // category strings anyway.
682 const QList<QChar> UserCategoryItem::categories = QList<QChar>() << 'q' << 'a' << 'o' << 'h' << 'v';
683
684 UserCategoryItem::UserCategoryItem(int category, AbstractTreeItem *parent)
685   : PropertyMapItem(QStringList() << "categoryName", parent),
686     _category(category)
687 {
688   setFlags(Qt::ItemIsEnabled);
689   setTreeItemFlags(AbstractTreeItem::DeleteOnLastChildRemoved);
690   setObjectName(parent->data(0, Qt::DisplayRole).toString() + "/" + QString::number(category));
691 }
692
693 // caching this makes no sense, since we display the user number dynamically
694 QString UserCategoryItem::categoryName() const {
695   int n = childCount();
696   switch(_category) {
697     case 0: return tr("%n Owner(s)", 0, n);
698     case 1: return tr("%n Admin(s)", 0, n);
699     case 2: return tr("%n Operator(s)", 0, n);
700     case 3: return tr("%n Half-Op(s)", 0, n);
701     case 4: return tr("%n Voiced", 0, n);
702     default: return tr("%n User(s)", 0, n);
703   }
704 }
705
706 IrcUserItem *UserCategoryItem::findIrcUser(IrcUser *ircUser) {
707   IrcUserItem *userItem = 0;
708
709   for(int i = 0; i < childCount(); i++) {
710     userItem = qobject_cast<IrcUserItem *>(child(i));
711     if(!userItem)
712       continue;
713     if(userItem->ircUser() == ircUser)
714       return userItem;
715   }
716   return 0;
717 }
718
719 void UserCategoryItem::addUsers(const QList<IrcUser *> &ircUsers) {
720   QList<AbstractTreeItem *> userItems;
721   foreach(IrcUser *ircUser, ircUsers)
722     userItems << new IrcUserItem(ircUser, this);
723   newChilds(userItems);
724   emit dataChanged(0);
725 }
726
727 bool UserCategoryItem::removeUser(IrcUser *ircUser) {
728   IrcUserItem *userItem = findIrcUser(ircUser);
729   bool success = (bool)userItem;
730   if(success) {
731     removeChild(userItem);
732     emit dataChanged(0);
733   }
734   return success;
735 }
736
737 int UserCategoryItem::categoryFromModes(const QString &modes) {
738   for(int i = 0; i < categories.count(); i++) {
739     if(modes.contains(categories[i]))
740       return i;
741   }
742   return categories.count();
743 }
744
745 QVariant UserCategoryItem::data(int column, int role) const {
746   switch(role) {
747   case TreeModel::SortRole:
748     return _category;
749   case NetworkModel::ItemActiveRole:
750     return true;
751   case NetworkModel::ItemTypeRole:
752     return NetworkModel::UserCategoryItemType;
753   case NetworkModel::BufferIdRole:
754     return parent()->data(column, role);
755   case NetworkModel::NetworkIdRole:
756     return parent()->data(column, role);
757   case NetworkModel::BufferInfoRole:
758     return parent()->data(column, role);
759   default:
760     return PropertyMapItem::data(column, role);
761   }
762 }
763
764
765 /*****************************************
766 *  Irc User Items
767 *****************************************/
768 IrcUserItem::IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent)
769   : PropertyMapItem(QStringList() << "nickName", parent),
770     _ircUser(ircUser)
771 {
772   setObjectName(ircUser->nick());
773   connect(ircUser, SIGNAL(quited()), this, SLOT(ircUserQuited()));
774   connect(ircUser, SIGNAL(nickSet(QString)), this, SIGNAL(dataChanged()));
775   connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
776 }
777
778 QVariant IrcUserItem::data(int column, int role) const {
779   switch(role) {
780   case NetworkModel::ItemActiveRole:
781     return isActive();
782   case NetworkModel::ItemTypeRole:
783     return NetworkModel::IrcUserItemType;
784   case NetworkModel::BufferIdRole:
785     return parent()->data(column, role);
786   case NetworkModel::NetworkIdRole:
787     return parent()->data(column, role);
788   case NetworkModel::BufferInfoRole:
789     return parent()->data(column, role);
790   case NetworkModel::IrcChannelRole:
791     return parent()->data(column, role);
792   case NetworkModel::IrcUserRole:
793     return QVariant::fromValue<QObject *>(_ircUser.data());
794   case NetworkModel::UserAwayRole:
795     return (bool)_ircUser ? _ircUser->isAway() : false;
796   default:
797     return PropertyMapItem::data(column, role);
798   }
799 }
800
801 QString IrcUserItem::toolTip(int column) const {
802   Q_UNUSED(column);
803   QStringList toolTip(QString("<b>%1</b>").arg(nickName()));
804   if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
805   if(_ircUser->isAway()) {
806     toolTip[0].append(tr(" is away"));
807     if(!_ircUser->awayMessage().isEmpty())
808       toolTip[0].append(QString(" (%1)").arg(_ircUser->awayMessage()));
809   }
810   if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
811   if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(nickName()).arg(_ircUser->ircOperator()));
812   if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
813   if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
814
815   toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
816
817   if(_ircUser->idleTime().isValid()) {
818     QDateTime now = QDateTime::currentDateTime();
819     QDateTime idle = _ircUser->idleTime();
820     int idleTime = idle.secsTo(now);
821     toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
822   }
823   if(_ircUser->loginTime().isValid()) {
824     toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
825   }
826
827   if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
828
829   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
830 }
831
832 /*****************************************
833  * NetworkModel
834  *****************************************/
835 NetworkModel::NetworkModel(QObject *parent)
836   : TreeModel(NetworkModel::defaultHeader(), parent)
837 {
838   connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),
839           this, SLOT(checkForNewBuffers(const QModelIndex &, int, int)));
840   connect(this, SIGNAL(rowsAboutToBeRemoved(const QModelIndex &, int, int)),
841           this, SLOT(checkForRemovedBuffers(const QModelIndex &, int, int)));
842
843   BufferSettings defaultSettings;
844   defaultSettings.notify("UserNoticesTarget", this, SLOT(messageRedirectionSettingsChanged()));
845   defaultSettings.notify("ServerNoticesTarget", this, SLOT(messageRedirectionSettingsChanged()));
846   defaultSettings.notify("ErrorMsgsTarget", this, SLOT(messageRedirectionSettingsChanged()));
847   messageRedirectionSettingsChanged();
848 }
849
850 QList<QVariant >NetworkModel::defaultHeader() {
851   QList<QVariant> data;
852   data << tr("Chat") << tr("Topic") << tr("Nick Count");
853   return data;
854 }
855
856 bool NetworkModel::isBufferIndex(const QModelIndex &index) const {
857   return index.data(NetworkModel::ItemTypeRole) == NetworkModel::BufferItemType;
858 }
859
860 int NetworkModel::networkRow(NetworkId networkId) const {
861   NetworkItem *netItem = 0;
862   for(int i = 0; i < rootItem->childCount(); i++) {
863     netItem = qobject_cast<NetworkItem *>(rootItem->child(i));
864     if(!netItem)
865       continue;
866     if(netItem->networkId() == networkId)
867       return i;
868   }
869   return -1;
870 }
871
872 QModelIndex NetworkModel::networkIndex(NetworkId networkId) {
873   int netRow = networkRow(networkId);
874   if(netRow == -1)
875     return QModelIndex();
876   else
877     return indexByItem(qobject_cast<NetworkItem *>(rootItem->child(netRow)));
878 }
879
880 NetworkItem *NetworkModel::findNetworkItem(NetworkId networkId) const {
881   int netRow = networkRow(networkId);
882   if(netRow == -1)
883     return 0;
884   else
885     return qobject_cast<NetworkItem *>(rootItem->child(netRow));
886 }
887
888 NetworkItem *NetworkModel::networkItem(NetworkId networkId) {
889   NetworkItem *netItem = findNetworkItem(networkId);
890
891   if(netItem == 0) {
892     netItem = new NetworkItem(networkId, rootItem);
893     rootItem->newChild(netItem);
894   }
895   return netItem;
896 }
897
898 void NetworkModel::networkRemoved(const NetworkId &networkId) {
899   int netRow = networkRow(networkId);
900   if(netRow != -1) {
901     rootItem->removeChild(netRow);
902   }
903 }
904
905 QModelIndex NetworkModel::bufferIndex(BufferId bufferId) {
906   if(!_bufferItemCache.contains(bufferId))
907     return QModelIndex();
908
909   return indexByItem(_bufferItemCache[bufferId]);
910 }
911
912 BufferItem *NetworkModel::findBufferItem(BufferId bufferId) const {
913   if(_bufferItemCache.contains(bufferId))
914     return _bufferItemCache[bufferId];
915   else
916     return 0;
917 }
918
919 BufferItem *NetworkModel::bufferItem(const BufferInfo &bufferInfo) {
920   if(_bufferItemCache.contains(bufferInfo.bufferId()))
921     return _bufferItemCache[bufferInfo.bufferId()];
922
923   NetworkItem *netItem = networkItem(bufferInfo.networkId());
924   return netItem->bufferItem(bufferInfo);
925 }
926
927 QStringList NetworkModel::mimeTypes() const {
928   // mimetypes we accept for drops
929   QStringList types;
930   // comma separated list of colon separated pairs of networkid and bufferid
931   // example: 0:1,0:2,1:4
932   types << "application/Quassel/BufferItemList";
933   return types;
934 }
935
936 bool NetworkModel::mimeContainsBufferList(const QMimeData *mimeData) {
937   return mimeData->hasFormat("application/Quassel/BufferItemList");
938 }
939
940 QList< QPair<NetworkId, BufferId> > NetworkModel::mimeDataToBufferList(const QMimeData *mimeData) {
941   QList< QPair<NetworkId, BufferId> > bufferList;
942
943   if(!mimeContainsBufferList(mimeData))
944     return bufferList;
945
946   QStringList rawBufferList = QString::fromAscii(mimeData->data("application/Quassel/BufferItemList")).split(",");
947   NetworkId networkId;
948   BufferId bufferUid;
949   foreach(QString rawBuffer, rawBufferList) {
950     if(!rawBuffer.contains(":"))
951       continue;
952     networkId = rawBuffer.section(":", 0, 0).toInt();
953     bufferUid = rawBuffer.section(":", 1, 1).toInt();
954     bufferList.append(qMakePair(networkId, bufferUid));
955   }
956   return bufferList;
957 }
958
959
960 QMimeData *NetworkModel::mimeData(const QModelIndexList &indexes) const {
961   QMimeData *mimeData = new QMimeData();
962
963   QStringList bufferlist;
964   QString netid, uid, bufferid;
965   foreach(QModelIndex index, indexes) {
966     netid = QString::number(index.data(NetworkIdRole).value<NetworkId>().toInt());
967     uid = QString::number(index.data(BufferIdRole).value<BufferId>().toInt());
968     bufferid = QString("%1:%2").arg(netid).arg(uid);
969     if(!bufferlist.contains(bufferid))
970       bufferlist << bufferid;
971   }
972
973   mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toAscii());
974
975   return mimeData;
976 }
977
978 void NetworkModel::attachNetwork(Network *net) {
979   NetworkItem *netItem = networkItem(net->networkId());
980   netItem->attachNetwork(net);
981 }
982
983 void NetworkModel::bufferUpdated(BufferInfo bufferInfo) {
984   BufferItem *bufItem = bufferItem(bufferInfo);
985   QModelIndex itemindex = indexByItem(bufItem);
986   emit dataChanged(itemindex, itemindex);
987 }
988
989 void NetworkModel::removeBuffer(BufferId bufferId) {
990   BufferItem *buffItem = findBufferItem(bufferId);
991   if(!buffItem)
992     return;
993
994   buffItem->parent()->removeChild(buffItem);
995 }
996
997 MsgId NetworkModel::lastSeenMsgId(BufferId bufferId) const {
998   if(!_bufferItemCache.contains(bufferId))
999     return MsgId();
1000
1001   return _bufferItemCache[bufferId]->lastSeenMsgId();
1002 }
1003
1004 MsgId NetworkModel::markerLineMsgId(BufferId bufferId) const {
1005   if(!_bufferItemCache.contains(bufferId))
1006     return MsgId();
1007
1008   return _bufferItemCache[bufferId]->markerLineMsgId();
1009 }
1010
1011 // FIXME we always seem to use this (expensive) non-const version
1012 MsgId NetworkModel::lastSeenMsgId(const BufferId &bufferId) {
1013   BufferItem *bufferItem = findBufferItem(bufferId);
1014   if(!bufferItem) {
1015     qDebug() << "NetworkModel::lastSeenMsgId(): buffer is unknown:" << bufferId;
1016     Client::purgeKnownBufferIds();
1017     return MsgId();
1018   }
1019   return bufferItem->lastSeenMsgId();
1020 }
1021
1022 void NetworkModel::setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId) {
1023   BufferItem *bufferItem = findBufferItem(bufferId);
1024   if(!bufferItem) {
1025     qDebug() << "NetworkModel::setLastSeenMsgId(): buffer is unknown:" << bufferId;
1026     Client::purgeKnownBufferIds();
1027     return;
1028   }
1029   bufferItem->setLastSeenMsgId(msgId);
1030   emit lastSeenMsgSet(bufferId, msgId);
1031 }
1032
1033 void NetworkModel::setMarkerLineMsgId(const BufferId &bufferId, const MsgId &msgId) {
1034   BufferItem *bufferItem = findBufferItem(bufferId);
1035   if(!bufferItem) {
1036     qDebug() << "NetworkModel::setMarkerLineMsgId(): buffer is unknown:" << bufferId;
1037     Client::purgeKnownBufferIds();
1038     return;
1039   }
1040   bufferItem->setMarkerLineMsgId(msgId);
1041   emit markerLineSet(bufferId, msgId);
1042 }
1043
1044 void NetworkModel::updateBufferActivity(Message &msg) {
1045   int redirectionTarget = 0;
1046   switch(msg.type()) {
1047   case Message::Notice:
1048     if(bufferType(msg.bufferId()) != BufferInfo::ChannelBuffer) {
1049       msg.setFlags(msg.flags() | Message::Redirected);
1050       if(msg.flags() & Message::ServerMsg) {
1051         // server notice
1052         redirectionTarget = _serverNoticesTarget;
1053       } else {
1054         redirectionTarget = _userNoticesTarget;
1055       }
1056     }
1057     break;
1058   case Message::Error:
1059     msg.setFlags(msg.flags() | Message::Redirected);
1060     redirectionTarget = _errorMsgsTarget;
1061     break;
1062   // Update IrcUser's last activity
1063   case Message::Plain:
1064   case Message::Action:
1065     if(bufferType(msg.bufferId()) == BufferInfo::ChannelBuffer) {
1066       const Network *net = Client::network(msg.bufferInfo().networkId());
1067       IrcUser *user = net ? net->ircUser(nickFromMask(msg.sender())) : 0;
1068       if(user)
1069         user->setLastChannelActivity(msg.bufferId(), msg.timestamp());
1070     }
1071     break;
1072   default:
1073     break;
1074   }
1075
1076   if(msg.flags() & Message::Redirected) {
1077     if(redirectionTarget & BufferSettings::DefaultBuffer)
1078       updateBufferActivity(bufferItem(msg.bufferInfo()), msg);
1079
1080     if(redirectionTarget & BufferSettings::StatusBuffer) {
1081       const NetworkItem *netItem = findNetworkItem(msg.bufferInfo().networkId());
1082       if(netItem) {
1083         updateBufferActivity(netItem->statusBufferItem(), msg);
1084       }
1085     }
1086   } else {
1087     updateBufferActivity(bufferItem(msg.bufferInfo()), msg);
1088   }
1089 }
1090
1091 void NetworkModel::updateBufferActivity(BufferItem *bufferItem, const Message &msg) {
1092   if(!bufferItem)
1093     return;
1094
1095   bufferItem->updateActivityLevel(msg);
1096   if(bufferItem->isCurrentBuffer())
1097     emit requestSetLastSeenMsg(bufferItem->bufferId(), msg.msgId());
1098 }
1099
1100 void NetworkModel::setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel level) {
1101   BufferItem *bufferItem = findBufferItem(bufferId);
1102   if(!bufferItem) {
1103     qDebug() << "NetworkModel::setBufferActivity(): buffer is unknown:" << bufferId;
1104     return;
1105   }
1106   bufferItem->setActivityLevel(level);
1107 }
1108
1109 void NetworkModel::clearBufferActivity(const BufferId &bufferId) {
1110   BufferItem *bufferItem = findBufferItem(bufferId);
1111   if(!bufferItem) {
1112     qDebug() << "NetworkModel::clearBufferActivity(): buffer is unknown:" << bufferId;
1113     return;
1114   }
1115   bufferItem->clearActivityLevel();
1116 }
1117
1118 const Network *NetworkModel::networkByIndex(const QModelIndex &index) const {
1119   QVariant netVariant = index.data(NetworkIdRole);
1120   if(!netVariant.isValid())
1121     return 0;
1122
1123   NetworkId networkId = netVariant.value<NetworkId>();
1124   return Client::network(networkId);
1125 }
1126
1127 void NetworkModel::checkForRemovedBuffers(const QModelIndex &parent, int start, int end) {
1128   if(parent.data(ItemTypeRole) != NetworkItemType)
1129     return;
1130
1131   for(int row = start; row <= end; row++) {
1132     _bufferItemCache.remove(parent.child(row, 0).data(BufferIdRole).value<BufferId>());
1133   }
1134 }
1135
1136 void NetworkModel::checkForNewBuffers(const QModelIndex &parent, int start, int end) {
1137   if(parent.data(ItemTypeRole) != NetworkItemType)
1138     return;
1139
1140   for(int row = start; row <= end; row++) {
1141     QModelIndex child = parent.child(row, 0);
1142     _bufferItemCache[child.data(BufferIdRole).value<BufferId>()] = static_cast<BufferItem *>(child.internalPointer());
1143   }
1144 }
1145
1146 QString NetworkModel::bufferName(BufferId bufferId) const {
1147   if(!_bufferItemCache.contains(bufferId))
1148     return QString();
1149
1150   return _bufferItemCache[bufferId]->bufferName();
1151 }
1152
1153 BufferInfo::Type NetworkModel::bufferType(BufferId bufferId) const {
1154   if(!_bufferItemCache.contains(bufferId))
1155     return BufferInfo::InvalidBuffer;
1156
1157   return _bufferItemCache[bufferId]->bufferType();
1158 }
1159
1160 BufferInfo NetworkModel::bufferInfo(BufferId bufferId) const {
1161   if(!_bufferItemCache.contains(bufferId))
1162     return BufferInfo();
1163
1164   return _bufferItemCache[bufferId]->bufferInfo();
1165 }
1166
1167 NetworkId NetworkModel::networkId(BufferId bufferId) const {
1168   if(!_bufferItemCache.contains(bufferId))
1169     return NetworkId();
1170
1171   NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1172   if(netItem)
1173     return netItem->networkId();
1174   else
1175     return NetworkId();
1176 }
1177
1178 QString NetworkModel::networkName(BufferId bufferId) const {
1179   if(!_bufferItemCache.contains(bufferId))
1180     return QString();
1181
1182   NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1183   if(netItem)
1184     return netItem->networkName();
1185   else
1186     return QString();
1187 }
1188
1189 BufferId NetworkModel::bufferId(NetworkId networkId, const QString &bufferName, Qt::CaseSensitivity cs) const {
1190   const NetworkItem *netItem = findNetworkItem(networkId);
1191   if(!netItem)
1192     return BufferId();
1193
1194   for(int i = 0; i < netItem->childCount(); i++) {
1195     BufferItem *bufferItem = qobject_cast<BufferItem *>(netItem->child(i));
1196     if(bufferItem && !bufferItem->bufferName().compare(bufferName, cs))
1197       return bufferItem->bufferId();
1198   }
1199   return BufferId();
1200 }
1201
1202 void NetworkModel::sortBufferIds(QList<BufferId> &bufferIds) const {
1203   QList<BufferItem *> bufferItems;
1204   foreach(BufferId bufferId, bufferIds) {
1205     if(_bufferItemCache.contains(bufferId))
1206       bufferItems << _bufferItemCache[bufferId];
1207   }
1208
1209   qSort(bufferItems.begin(), bufferItems.end(), bufferItemLessThan);
1210
1211   bufferIds.clear();
1212   foreach(BufferItem *bufferItem, bufferItems) {
1213     bufferIds << bufferItem->bufferId();
1214   }
1215 }
1216
1217 QList<BufferId> NetworkModel::allBufferIdsSorted() const {
1218   QList<BufferId> bufferIds = allBufferIds();
1219   sortBufferIds(bufferIds);
1220   return bufferIds;
1221 }
1222
1223 bool NetworkModel::bufferItemLessThan(const BufferItem *left, const BufferItem *right) {
1224   int leftType = left->bufferType();
1225   int rightType = right->bufferType();
1226
1227   if(leftType != rightType)
1228     return leftType < rightType;
1229   else
1230     return QString::compare(left->bufferName(), right->bufferName(), Qt::CaseInsensitive) < 0;
1231 }
1232
1233 void NetworkModel::messageRedirectionSettingsChanged() {
1234   BufferSettings bufferSettings;
1235
1236   _userNoticesTarget = bufferSettings.userNoticesTarget();
1237   _serverNoticesTarget = bufferSettings.serverNoticesTarget();
1238   _errorMsgsTarget = bufferSettings.errorMsgsTarget();
1239 }