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