Separate highlight handling
[quassel.git] / src / client / networkmodel.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-2016 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  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
19  ***************************************************************************/
20
21 #include "networkmodel.h"
22
23 #include <QAbstractItemView>
24 #include <QMimeData>
25 #if QT_VERSION < 0x050000
26 #include <QTextDocument>        // for Qt::escape()
27 #endif
28
29 #include "buffermodel.h"
30 #include "buffersettings.h"
31 #include "client.h"
32 #include "clientignorelistmanager.h"
33 #include "clientsettings.h"
34 #include "ircchannel.h"
35 #include "network.h"
36 #include "signalproxy.h"
37
38 /*****************************************
39 *  Network Items
40 *****************************************/
41 NetworkItem::NetworkItem(const NetworkId &netid, AbstractTreeItem *parent)
42     : PropertyMapItem(QList<QString>() << "networkName" << "currentServer" << "nickCount", parent),
43     _networkId(netid),
44     _statusBufferItem(0)
45 {
46     // DO NOT EMIT dataChanged() DIRECTLY IN NetworkItem
47     // use networkDataChanged() instead. Otherwise you will end up in a infinite loop
48     // as we "sync" the dataChanged() signals of NetworkItem and StatusBufferItem
49     setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
50     connect(this, SIGNAL(networkDataChanged(int)), this, SIGNAL(dataChanged(int)));
51     connect(this, SIGNAL(beginRemoveChilds(int, int)), this, SLOT(onBeginRemoveChilds(int, int)));
52 }
53
54
55 QVariant NetworkItem::data(int column, int role) const
56 {
57     switch (role) {
58     case NetworkModel::BufferIdRole:
59     case NetworkModel::BufferInfoRole:
60     case NetworkModel::BufferTypeRole:
61     case NetworkModel::BufferActivityRole:
62         if (_statusBufferItem)
63             return _statusBufferItem->data(column, role);
64         else
65             return QVariant();
66     case NetworkModel::NetworkIdRole:
67         return qVariantFromValue(_networkId);
68     case NetworkModel::ItemTypeRole:
69         return NetworkModel::NetworkItemType;
70     case NetworkModel::ItemActiveRole:
71         return isActive();
72     default:
73         return PropertyMapItem::data(column, role);
74     }
75 }
76
77 QString NetworkItem::escapeHTML(const QString &string, bool useNonbreakingSpaces)
78 {
79     // QString.replace() doesn't guarantee the source string will remain constant.
80     // Use a local variable to avoid compiler errors.
81 #if QT_VERSION < 0x050000
82     QString formattedString = Qt::escape(string);
83 #else
84     QString formattedString = string.toHtmlEscaped();
85 #endif
86     return (useNonbreakingSpaces ? formattedString.replace(" ", "&nbsp;") : formattedString);
87 }
88
89
90 // FIXME shouldn't we check the bufferItemCache here?
91 BufferItem *NetworkItem::findBufferItem(BufferId bufferId)
92 {
93     BufferItem *bufferItem = 0;
94
95     for (int i = 0; i < childCount(); i++) {
96         bufferItem = qobject_cast<BufferItem *>(child(i));
97         if (!bufferItem)
98             continue;
99         if (bufferItem->bufferId() == bufferId)
100             return bufferItem;
101     }
102     return 0;
103 }
104
105
106 BufferItem *NetworkItem::bufferItem(const BufferInfo &bufferInfo)
107 {
108     BufferItem *bufferItem = findBufferItem(bufferInfo);
109     if (bufferItem)
110         return bufferItem;
111
112     switch (bufferInfo.type()) {
113     case BufferInfo::StatusBuffer:
114         _statusBufferItem = new StatusBufferItem(bufferInfo, this);
115         bufferItem = _statusBufferItem;
116         disconnect(this, SIGNAL(networkDataChanged(int)), this, SIGNAL(dataChanged(int)));
117         connect(this, SIGNAL(networkDataChanged(int)), bufferItem, SIGNAL(dataChanged(int)));
118         connect(bufferItem, SIGNAL(dataChanged(int)), this, SIGNAL(dataChanged(int)));
119         break;
120     case BufferInfo::ChannelBuffer:
121         bufferItem = new ChannelBufferItem(bufferInfo, this);
122         break;
123     case BufferInfo::QueryBuffer:
124         bufferItem = new QueryBufferItem(bufferInfo, this);
125         break;
126     default:
127         bufferItem = new BufferItem(bufferInfo, this);
128     }
129
130     newChild(bufferItem);
131
132     // postprocess... this is necessary because Qt doesn't seem to like adding children which already have children on their own
133     switch (bufferInfo.type()) {
134     case BufferInfo::ChannelBuffer:
135     {
136         ChannelBufferItem *channelBufferItem = static_cast<ChannelBufferItem *>(bufferItem);
137         if (_network) {
138             IrcChannel *ircChannel = _network->ircChannel(bufferInfo.bufferName());
139             if (ircChannel)
140                 channelBufferItem->attachIrcChannel(ircChannel);
141         }
142     }
143     break;
144     default:
145         break;
146     }
147
148     return bufferItem;
149 }
150
151
152 void NetworkItem::attachNetwork(Network *network)
153 {
154     if (!network)
155         return;
156
157     _network = network;
158
159     connect(network, SIGNAL(networkNameSet(QString)),
160         this, SLOT(setNetworkName(QString)));
161     connect(network, SIGNAL(currentServerSet(QString)),
162         this, SLOT(setCurrentServer(QString)));
163     connect(network, SIGNAL(ircChannelAdded(IrcChannel *)),
164         this, SLOT(attachIrcChannel(IrcChannel *)));
165     connect(network, SIGNAL(ircUserAdded(IrcUser *)),
166         this, SLOT(attachIrcUser(IrcUser *)));
167     connect(network, SIGNAL(connectedSet(bool)),
168         this, SIGNAL(networkDataChanged()));
169     connect(network, SIGNAL(destroyed()),
170         this, SLOT(onNetworkDestroyed()));
171
172     emit networkDataChanged();
173 }
174
175
176 void NetworkItem::attachIrcChannel(IrcChannel *ircChannel)
177 {
178     ChannelBufferItem *channelItem;
179     for (int i = 0; i < childCount(); i++) {
180         channelItem = qobject_cast<ChannelBufferItem *>(child(i));
181         if (!channelItem)
182             continue;
183
184         if (channelItem->bufferName().toLower() == ircChannel->name().toLower()) {
185             channelItem->attachIrcChannel(ircChannel);
186             return;
187         }
188     }
189 }
190
191
192 void NetworkItem::attachIrcUser(IrcUser *ircUser)
193 {
194     QueryBufferItem *queryItem = 0;
195     for (int i = 0; i < childCount(); i++) {
196         queryItem = qobject_cast<QueryBufferItem *>(child(i));
197         if (!queryItem)
198             continue;
199
200         if (queryItem->bufferName().toLower() == ircUser->nick().toLower()) {
201             queryItem->setIrcUser(ircUser);
202             break;
203         }
204     }
205 }
206
207
208 void NetworkItem::setNetworkName(const QString &networkName)
209 {
210     Q_UNUSED(networkName);
211     emit networkDataChanged(0);
212 }
213
214
215 void NetworkItem::setCurrentServer(const QString &serverName)
216 {
217     Q_UNUSED(serverName);
218     emit networkDataChanged(1);
219 }
220
221
222 QString NetworkItem::toolTip(int column) const
223 {
224     Q_UNUSED(column);
225     QString strTooltip;
226     QTextStream tooltip( &strTooltip, QIODevice::WriteOnly );
227     tooltip << "<qt><style>.bold { font-weight: bold; } .italic { font-style: italic; }</style>";
228
229     // Function to add a row to the tooltip table
230     auto addRow = [&](const QString& key, const QString& value, bool condition) {
231         if (condition) {
232             tooltip << "<tr><td class='bold' align='right'>" << key << "</td><td>" << value << "</td></tr>";
233         }
234     };
235
236     tooltip << "<p class='bold' align='center'>" << NetworkItem::escapeHTML(networkName(), true) << "</p>";
237     if (isActive()) {
238         tooltip << "<table cellspacing='5' cellpadding='0'>";
239         addRow(tr("Server"), NetworkItem::escapeHTML(currentServer(), true), !currentServer().isEmpty());
240         addRow(tr("Users"), QString::number(nickCount()), true);
241         if (_network)
242             addRow(tr("Lag"), NetworkItem::escapeHTML(tr("%1 msecs").arg(_network->latency()), true), true);
243
244         tooltip << "</table>";
245     } else {
246         tooltip << "<p class='italic' align='center'>" << tr("Not connected") << "</p>";
247     }
248     tooltip << "</qt>";
249     return strTooltip;
250 }
251
252
253 void NetworkItem::onBeginRemoveChilds(int start, int end)
254 {
255     for (int i = start; i <= end; i++) {
256         StatusBufferItem *statusBufferItem = qobject_cast<StatusBufferItem *>(child(i));
257         if (statusBufferItem) {
258             _statusBufferItem = 0;
259             break;
260         }
261     }
262 }
263
264
265 void NetworkItem::onNetworkDestroyed()
266 {
267     _network = 0;
268     emit networkDataChanged();
269     removeAllChilds();
270 }
271
272
273 /*****************************************
274 *  Fancy Buffer Items
275 *****************************************/
276 BufferItem::BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
277     : PropertyMapItem(QStringList() << "bufferName" << "topic" << "nickCount", parent),
278     _bufferInfo(bufferInfo),
279     _activity(BufferInfo::NoActivity)
280 {
281     setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled);
282 }
283
284
285 void BufferItem::setActivityLevel(BufferInfo::ActivityLevel level)
286 {
287     if (_activity != level) {
288         _activity = level;
289         emit dataChanged();
290     }
291 }
292
293
294 void BufferItem::clearActivityLevel()
295 {
296     _activity = BufferInfo::NoActivity;
297     _firstUnreadMsgId = MsgId();
298
299     // FIXME remove with core proto v11
300     if (!(Client::coreFeatures() & Quassel::SynchronizedMarkerLine)) {
301         _markerLineMsgId = _lastSeenMsgId;
302     }
303
304     emit dataChanged();
305 }
306
307
308 void BufferItem::updateActivityLevel(const Message &msg)
309 {
310     if (Client::coreFeatures().testFlag(Quassel::Feature::BufferActivitySync)) {
311         if (msg.flags().testFlag(Message::Highlight)) {
312             if (addActivity(Message::Types(msg.type()), msg.flags().testFlag(Message::Highlight))) {
313                 emit dataChanged();
314             }
315         }
316         return;
317     }
318
319     if (isCurrentBuffer()) {
320         return;
321     }
322
323     if (msg.flags() & Message::Self)    // don't update activity for our own messages
324         return;
325
326     if (Client::ignoreListManager()
327         && Client::ignoreListManager()->match(msg, qobject_cast<NetworkItem *>(parent())->networkName()))
328         return;
329
330     if (msg.msgId() <= lastSeenMsgId())
331         return;
332
333     bool stateChanged = false;
334     if (!firstUnreadMsgId().isValid() || msg.msgId() < firstUnreadMsgId()) {
335         stateChanged = true;
336         _firstUnreadMsgId = msg.msgId();
337     }
338
339     if (addActivity(Message::Types(msg.type()), msg.flags().testFlag(Message::Highlight)) || stateChanged) {
340         emit dataChanged();
341     }
342 }
343
344 void BufferItem::setActivity(Message::Types type, bool highlight) {
345     BufferInfo::ActivityLevel oldLevel = activityLevel();
346
347     _activity &= BufferInfo::Highlight;
348     addActivity(type, highlight);
349
350     if (_activity != oldLevel) {
351         emit dataChanged();
352     }
353 }
354
355 bool BufferItem::addActivity(Message::Types type, bool highlight) {
356     auto oldActivity = activityLevel();
357
358     if (type != 0)
359         _activity |= BufferInfo::OtherActivity;
360
361     if (type.testFlag(Message::Plain) || type.testFlag(Message::Notice) || type.testFlag(Message::Action))
362         _activity |= BufferInfo::NewMessage;
363
364     if (highlight)
365         _activity |= BufferInfo::Highlight;
366
367     return oldActivity != _activity;
368 }
369
370
371 QVariant BufferItem::data(int column, int role) const
372 {
373     switch (role) {
374     case NetworkModel::ItemTypeRole:
375         return NetworkModel::BufferItemType;
376     case NetworkModel::BufferIdRole:
377         return qVariantFromValue(bufferInfo().bufferId());
378     case NetworkModel::NetworkIdRole:
379         return qVariantFromValue(bufferInfo().networkId());
380     case NetworkModel::BufferInfoRole:
381         return qVariantFromValue(bufferInfo());
382     case NetworkModel::BufferTypeRole:
383         return int(bufferType());
384     case NetworkModel::ItemActiveRole:
385         return isActive();
386     case NetworkModel::BufferActivityRole:
387         return (int)activityLevel();
388     case NetworkModel::BufferFirstUnreadMsgIdRole:
389         return qVariantFromValue(firstUnreadMsgId());
390     case NetworkModel::MarkerLineMsgIdRole:
391         return qVariantFromValue(markerLineMsgId());
392     default:
393         return PropertyMapItem::data(column, role);
394     }
395 }
396
397
398 bool BufferItem::setData(int column, const QVariant &value, int role)
399 {
400     switch (role) {
401     case NetworkModel::BufferActivityRole:
402         setActivityLevel((BufferInfo::ActivityLevel)value.toInt());
403         return true;
404     default:
405         return PropertyMapItem::setData(column, value, role);
406     }
407     return true;
408 }
409
410
411 void BufferItem::setBufferName(const QString &name)
412 {
413     _bufferInfo = BufferInfo(_bufferInfo.bufferId(), _bufferInfo.networkId(), _bufferInfo.type(), _bufferInfo.groupId(), name);
414     emit dataChanged(0);
415 }
416
417
418 void BufferItem::setLastSeenMsgId(MsgId msgId)
419 {
420     _lastSeenMsgId = msgId;
421
422     // FIXME remove with core protocol v11
423     if (!(Client::coreFeatures() & Quassel::SynchronizedMarkerLine)) {
424         if (!isCurrentBuffer())
425             _markerLineMsgId = msgId;
426     }
427
428     setActivityLevel(BufferInfo::NoActivity);
429 }
430
431
432 void BufferItem::setMarkerLineMsgId(MsgId msgId)
433 {
434     _markerLineMsgId = msgId;
435     emit dataChanged();
436 }
437
438
439 bool BufferItem::isCurrentBuffer() const
440 {
441     return _bufferInfo.bufferId() == Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
442 }
443
444
445 QString BufferItem::toolTip(int column) const
446 {
447     Q_UNUSED(column);
448     return tr("<p> %1 - %2 </p>").arg(bufferInfo().bufferId().toInt()).arg(bufferName());
449 }
450
451
452 /*****************************************
453 *  StatusBufferItem
454 *****************************************/
455 StatusBufferItem::StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
456     : BufferItem(bufferInfo, parent)
457 {
458 }
459
460
461 QString StatusBufferItem::toolTip(int column) const
462 {
463     NetworkItem *networkItem = qobject_cast<NetworkItem *>(parent());
464     if (networkItem)
465         return networkItem->toolTip(column);
466     else
467         return QString();
468 }
469
470
471 /*****************************************
472 *  QueryBufferItem
473 *****************************************/
474 QueryBufferItem::QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
475     : BufferItem(bufferInfo, parent),
476     _ircUser(0)
477 {
478     setFlags(flags() | Qt::ItemIsDropEnabled | Qt::ItemIsEditable);
479
480     const Network *net = Client::network(bufferInfo.networkId());
481     if (!net)
482         return;
483
484     IrcUser *ircUser = net->ircUser(bufferInfo.bufferName());
485     setIrcUser(ircUser);
486 }
487
488
489 QVariant QueryBufferItem::data(int column, int role) const
490 {
491     switch (role) {
492     case Qt::EditRole:
493         return BufferItem::data(column, Qt::DisplayRole);
494     case NetworkModel::IrcUserRole:
495         return QVariant::fromValue<QObject *>(_ircUser);
496     case NetworkModel::UserAwayRole:
497         return (bool)_ircUser ? _ircUser->isAway() : false;
498     default:
499         return BufferItem::data(column, role);
500     }
501 }
502
503
504 bool QueryBufferItem::setData(int column, const QVariant &value, int role)
505 {
506     if (column != 0)
507         return BufferItem::setData(column, value, role);
508
509     switch (role) {
510     case Qt::EditRole:
511     {
512         QString newName = value.toString();
513
514         // Sanity check - buffer names must not contain newlines!
515         int nlpos = newName.indexOf('\n');
516         if (nlpos >= 0)
517             newName = newName.left(nlpos);
518
519         if (!newName.isEmpty()) {
520             Client::renameBuffer(bufferId(), newName);
521             return true;
522         }
523         else {
524             return false;
525         }
526     }
527     break;
528     default:
529         return BufferItem::setData(column, value, role);
530     }
531 }
532
533
534 void QueryBufferItem::setBufferName(const QString &name)
535 {
536     BufferItem::setBufferName(name);
537     NetworkId netId = data(0, NetworkModel::NetworkIdRole).value<NetworkId>();
538     const Network *net = Client::network(netId);
539     if (net)
540         setIrcUser(net->ircUser(name));
541 }
542
543
544 QString QueryBufferItem::toolTip(int column) const
545 {
546     // pretty much code duplication of IrcUserItem::toolTip() but inheritance won't solve this...
547     Q_UNUSED(column);
548     QString strTooltip;
549     QTextStream tooltip( &strTooltip, QIODevice::WriteOnly );
550     tooltip << "<qt><style>.bold { font-weight: bold; } .italic { font-style: italic; }</style>";
551
552     // Keep track of whether or not information has been added
553     bool infoAdded = false;
554
555     // Use bufferName() for QueryBufferItem, nickName() for IrcUserItem
556     tooltip << "<p class='bold' align='center'>";
557     tooltip << tr("Query with %1").arg(NetworkItem::escapeHTML(bufferName(), true));
558     if (!_ircUser) {
559         // User seems to be offline, let the no information message be added below
560         tooltip << "</p>";
561     } else {
562         // Function to add a row to the tooltip table
563         auto addRow = [&](const QString& key, const QString& value, bool condition) {
564             if (condition) {
565                 tooltip << "<tr><td class='bold' align='right'>" << key << "</td><td>" << value << "</td></tr>";
566                 infoAdded = true;
567             }
568         };
569
570         // User information is available
571         if (_ircUser->userModes() != "") {
572             //TODO Translate user Modes and add them to the table below and in IrcUserItem::toolTip
573             tooltip << " (" << _ircUser->userModes() << ")";
574         }
575         tooltip << "</p>";
576
577         tooltip << "<table cellspacing='5' cellpadding='0'>";
578         if (_ircUser->isAway()) {
579             QString awayMessageHTML = QString("<p class='italic'>%1</p>").arg(tr("Unknown"));
580
581             // If away message is known, replace with the escaped message.
582             if (!_ircUser->awayMessage().isEmpty()) {
583                 awayMessageHTML = NetworkItem::escapeHTML(_ircUser->awayMessage());
584             }
585             addRow(NetworkItem::escapeHTML(tr("Away message"), true), awayMessageHTML, true);
586         }
587         addRow(tr("Realname"),
588                NetworkItem::escapeHTML(_ircUser->realName()),
589                !_ircUser->realName().isEmpty());
590         // suserHost may return "<nick> is available for help", which should be translated.
591         // See https://www.alien.net.au/irc/irc2numerics.html
592         if(_ircUser->suserHost().endsWith("available for help")) {
593             addRow(NetworkItem::escapeHTML(tr("Help status"), true),
594                    NetworkItem::escapeHTML(tr("Available for help")),
595                    true);
596         } else {
597             addRow(NetworkItem::escapeHTML(tr("Service status"), true),
598                    NetworkItem::escapeHTML(_ircUser->suserHost()),
599                    !_ircUser->suserHost().isEmpty());
600         }
601
602         // Keep track of whether or not the account information's been added.  Don't show it twice.
603         bool accountAdded = false;
604         if(!_ircUser->account().isEmpty()) {
605             // IRCv3 account-notify is supported by the core and IRC server.
606             // Assume logged out (seems to be more common)
607             QString accountHTML = QString("<p class='italic'>%1</p>").arg(tr("Not logged in"));
608
609             // If account is logged in, replace with the escaped account name.
610             if (_ircUser->account() != "*") {
611                 accountHTML = NetworkItem::escapeHTML(_ircUser->account());
612             }
613             addRow(NetworkItem::escapeHTML(tr("Account"), true),
614                    accountHTML,
615                    true);
616             // Mark the row as added
617             accountAdded = true;
618         }
619         // whoisServiceReply may return "<nick> is identified for this nick", which should be translated.
620         // See https://www.alien.net.au/irc/irc2numerics.html
621         if(_ircUser->whoisServiceReply().endsWith("identified for this nick")) {
622             addRow(NetworkItem::escapeHTML(tr("Account"), true),
623                    NetworkItem::escapeHTML(tr("Identified for this nick")),
624                    !accountAdded);
625             // Don't add the account row again if information's already added via account-notify
626             // Mark the row as added
627             accountAdded = true;
628         } else {
629             addRow(NetworkItem::escapeHTML(tr("Service Reply"), true),
630                    NetworkItem::escapeHTML(_ircUser->whoisServiceReply()),
631                    !_ircUser->whoisServiceReply().isEmpty());
632         }
633         addRow(tr("Hostmask"),
634                NetworkItem::escapeHTML(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!") + 1)),
635                !(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!") + 1) == "@"));
636         // ircOperator may contain "is an" or "is a", which should be removed.
637         addRow(tr("Operator"),
638                NetworkItem::escapeHTML(_ircUser->ircOperator().replace("is an ", "").replace("is a ", "")),
639                !_ircUser->ircOperator().isEmpty());
640
641         if (_ircUser->idleTime().isValid()) {
642             QDateTime now = QDateTime::currentDateTime();
643             QDateTime idle = _ircUser->idleTime();
644             int idleTime = idle.secsTo(now);
645             addRow(NetworkItem::escapeHTML(tr("Idling since"), true), secondsToString(idleTime), true);
646         }
647
648         if (_ircUser->loginTime().isValid()) {
649             addRow(NetworkItem::escapeHTML(tr("Login time"), true), _ircUser->loginTime().toString(), true);
650         }
651
652         addRow(tr("Server"), NetworkItem::escapeHTML(_ircUser->server()), !_ircUser->server().isEmpty());
653         tooltip << "</table>";
654     }
655
656     // If no further information found, offer an explanatory message
657     if (!infoAdded)
658         tooltip << "<p class='italic' align='center'>" << tr("No information available") << "</p>";
659
660     tooltip << "</qt>";
661     return strTooltip;
662 }
663
664
665 void QueryBufferItem::setIrcUser(IrcUser *ircUser)
666 {
667     if (_ircUser == ircUser)
668         return;
669
670     if (_ircUser) {
671         disconnect(_ircUser, 0, this, 0);
672     }
673
674     if (ircUser) {
675         connect(ircUser, SIGNAL(destroyed(QObject*)), SLOT(removeIrcUser()));
676         connect(ircUser, SIGNAL(quited()), this, SLOT(removeIrcUser()));
677         connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
678         connect(ircUser, SIGNAL(encryptedSet(bool)), this, SLOT(setEncrypted(bool)));
679     }
680
681     _ircUser = ircUser;
682     emit dataChanged();
683 }
684
685
686 void QueryBufferItem::removeIrcUser()
687 {
688     if (_ircUser) {
689         // Disconnect the active IrcUser before removing it, otherwise it will fire removeIrcUser()
690         // a second time when the object's destroyed due to QueryBufferItem::setIrcUser() connecting
691         // SIGNAL destroyed(QObject*) to SLOT removeIrcUser().
692         // This fixes removing an active IrcUser if the user had quit then rejoined in a nonstandard
693         // manner (e.g. updateNickFromHost calling newIrcUser, triggered by an away-notify message).
694         disconnect(_ircUser, 0, this, 0);
695
696         // Clear IrcUser (only set to 0 if not already 0)
697         _ircUser = 0;
698
699         // Only emit dataChanged() if data actually changed.  This might serve as a small
700         // optimization, but it can be moved outside the if statement if other behavior depends on
701         // it always being called.
702         emit dataChanged();
703     }
704 }
705
706
707 /*****************************************
708 *  ChannelBufferItem
709 *****************************************/
710 ChannelBufferItem::ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
711     : BufferItem(bufferInfo, parent),
712     _ircChannel(0)
713 {
714     setFlags(flags() | Qt::ItemIsDropEnabled);
715 }
716
717
718 QVariant ChannelBufferItem::data(int column, int role) const
719 {
720     switch (role) {
721     case NetworkModel::IrcChannelRole:
722         return QVariant::fromValue<QObject *>(_ircChannel);
723     default:
724         return BufferItem::data(column, role);
725     }
726 }
727
728
729 QString ChannelBufferItem::toolTip(int column) const
730 {
731     Q_UNUSED(column);
732     QString strTooltip;
733     QTextStream tooltip( &strTooltip, QIODevice::WriteOnly );
734     tooltip << "<qt><style>.bold { font-weight: bold; } .italic { font-style: italic; }</style>";
735
736     // Function to add a row to the tooltip table
737     auto addRow = [&](const QString& key, const QString& value, bool condition) {
738         if (condition) {
739             tooltip << "<tr><td class='bold' align='right'>" << key << "</td><td>" << value << "</td></tr>";
740         }
741     };
742
743     tooltip << "<p class='bold' align='center'>";
744     tooltip << NetworkItem::escapeHTML(tr("Channel %1").arg(bufferName()), true) << "</p>";
745
746     if (isActive()) {
747         tooltip << "<table cellspacing='5' cellpadding='0'>";
748         addRow(tr("Users"), QString::number(nickCount()), true);
749
750         if (_ircChannel) {
751             QString channelMode = _ircChannel->channelModeString(); // channelModeString is compiled on the fly -> thus cache the result
752             if (!channelMode.isEmpty())
753                 addRow(tr("Mode"), channelMode, true);
754         }
755
756         ItemViewSettings s;
757         bool showTopic = s.displayTopicInTooltip();
758         if (showTopic) {
759             QString _topic = topic();
760             if (_topic != "") {
761                 _topic = stripFormatCodes(_topic);
762                 _topic = NetworkItem::escapeHTML(_topic);
763                 addRow(tr("Topic"), _topic, true);
764             }
765         }
766
767         tooltip << "</table>";
768     } else {
769         tooltip << "<p class='italic' align='center'>" << tr("Not active, double-click to join") << "</p>";
770     }
771
772     tooltip << "</qt>";
773     return strTooltip;
774 }
775
776
777 void ChannelBufferItem::attachIrcChannel(IrcChannel *ircChannel)
778 {
779     if (_ircChannel) {
780         qWarning() << Q_FUNC_INFO << "IrcChannel already set; cleanup failed!?";
781         disconnect(_ircChannel, 0, this, 0);
782     }
783
784     _ircChannel = ircChannel;
785
786     connect(ircChannel, SIGNAL(destroyed(QObject*)),
787         this, SLOT(ircChannelDestroyed()));
788     connect(ircChannel, SIGNAL(topicSet(QString)),
789         this, SLOT(setTopic(QString)));
790     connect(ircChannel, SIGNAL(encryptedSet(bool)),
791         this, SLOT(setEncrypted(bool)));
792     connect(ircChannel, SIGNAL(ircUsersJoined(QList<IrcUser *> )),
793         this, SLOT(join(QList<IrcUser *> )));
794     connect(ircChannel, SIGNAL(ircUserParted(IrcUser *)),
795         this, SLOT(part(IrcUser *)));
796     connect(ircChannel, SIGNAL(parted()),
797         this, SLOT(ircChannelParted()));
798     connect(ircChannel, SIGNAL(ircUserModesSet(IrcUser *, QString)),
799         this, SLOT(userModeChanged(IrcUser *)));
800     connect(ircChannel, SIGNAL(ircUserModeAdded(IrcUser *, QString)),
801         this, SLOT(userModeChanged(IrcUser *)));
802     connect(ircChannel, SIGNAL(ircUserModeRemoved(IrcUser *, QString)),
803         this, SLOT(userModeChanged(IrcUser *)));
804
805     if (!ircChannel->ircUsers().isEmpty())
806         join(ircChannel->ircUsers());
807
808     emit dataChanged();
809 }
810
811 QString ChannelBufferItem::nickChannelModes(const QString &nick) const
812 {
813     if (!_ircChannel) {
814         qDebug() << Q_FUNC_INFO << "IrcChannel not set, can't get user modes";
815         return QString();
816     }
817
818     return _ircChannel->userModes(nick);
819 }
820
821
822 void ChannelBufferItem::ircChannelParted()
823 {
824     Q_CHECK_PTR(_ircChannel);
825     disconnect(_ircChannel, 0, this, 0);
826     _ircChannel = 0;
827     emit dataChanged();
828     removeAllChilds();
829 }
830
831
832 void ChannelBufferItem::ircChannelDestroyed()
833 {
834     if (_ircChannel) {
835         _ircChannel = 0;
836         emit dataChanged();
837         removeAllChilds();
838     }
839 }
840
841
842 void ChannelBufferItem::join(const QList<IrcUser *> &ircUsers)
843 {
844     addUsersToCategory(ircUsers);
845     emit dataChanged(2);
846 }
847
848
849 UserCategoryItem *ChannelBufferItem::findCategoryItem(int categoryId)
850 {
851     UserCategoryItem *categoryItem = 0;
852
853     for (int i = 0; i < childCount(); i++) {
854         categoryItem = qobject_cast<UserCategoryItem *>(child(i));
855         if (!categoryItem)
856             continue;
857         if (categoryItem->categoryId() == categoryId)
858             return categoryItem;
859     }
860     return 0;
861 }
862
863
864 void ChannelBufferItem::addUserToCategory(IrcUser *ircUser)
865 {
866     addUsersToCategory(QList<IrcUser *>() << ircUser);
867 }
868
869
870 void ChannelBufferItem::addUsersToCategory(const QList<IrcUser *> &ircUsers)
871 {
872     Q_ASSERT(_ircChannel);
873
874     QHash<UserCategoryItem *, QList<IrcUser *> > categories;
875
876     int categoryId = -1;
877     UserCategoryItem *categoryItem = 0;
878
879     foreach(IrcUser *ircUser, ircUsers) {
880         categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
881         categoryItem = findCategoryItem(categoryId);
882         if (!categoryItem) {
883             categoryItem = new UserCategoryItem(categoryId, this);
884             categories[categoryItem] = QList<IrcUser *>();
885             newChild(categoryItem);
886         }
887         categories[categoryItem] << ircUser;
888     }
889
890     QHash<UserCategoryItem *, QList<IrcUser *> >::const_iterator catIter = categories.constBegin();
891     while (catIter != categories.constEnd()) {
892         catIter.key()->addUsers(catIter.value());
893         ++catIter;
894     }
895 }
896
897
898 void ChannelBufferItem::part(IrcUser *ircUser)
899 {
900     if (!ircUser) {
901         qWarning() << bufferName() << "ChannelBufferItem::part(): unknown User" << ircUser;
902         return;
903     }
904
905     disconnect(ircUser, 0, this, 0);
906     removeUserFromCategory(ircUser);
907     emit dataChanged(2);
908 }
909
910
911 void ChannelBufferItem::removeUserFromCategory(IrcUser *ircUser)
912 {
913     if (!_ircChannel) {
914         // If we parted the channel there might still be some ircUsers connected.
915         // in that case we just ignore the call
916         Q_ASSERT(childCount() == 0);
917         return;
918     }
919
920     UserCategoryItem *categoryItem = 0;
921     for (int i = 0; i < childCount(); i++) {
922         categoryItem = qobject_cast<UserCategoryItem *>(child(i));
923         if (categoryItem->removeUser(ircUser)) {
924             if (categoryItem->childCount() == 0)
925                 removeChild(i);
926             break;
927         }
928     }
929 }
930
931
932 void ChannelBufferItem::userModeChanged(IrcUser *ircUser)
933 {
934     Q_ASSERT(_ircChannel);
935
936     int categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
937     UserCategoryItem *categoryItem = findCategoryItem(categoryId);
938
939     if (categoryItem) {
940         if (categoryItem->findIrcUser(ircUser)) {
941             return; // already in the right category;
942         }
943     }
944     else {
945         categoryItem = new UserCategoryItem(categoryId, this);
946         newChild(categoryItem);
947     }
948
949     // find the item that needs reparenting
950     IrcUserItem *ircUserItem = 0;
951     for (int i = 0; i < childCount(); i++) {
952         UserCategoryItem *oldCategoryItem = qobject_cast<UserCategoryItem *>(child(i));
953         Q_ASSERT(oldCategoryItem);
954         IrcUserItem *userItem = oldCategoryItem->findIrcUser(ircUser);
955         if (userItem) {
956             ircUserItem = userItem;
957             break;
958         }
959     }
960
961     if (!ircUserItem) {
962         qWarning() << "ChannelBufferItem::userModeChanged(IrcUser *): unable to determine old category of" << ircUser;
963         return;
964     }
965     ircUserItem->reParent(categoryItem);
966 }
967
968
969 /*****************************************
970 *  User Category Items (like @vh etc.)
971 *****************************************/
972 // we hardcode this even though we have PREFIX in network... but that wouldn't help with mapping modes to
973 // category strings anyway.
974 const QList<QChar> UserCategoryItem::categories = QList<QChar>() << 'q' << 'a' << 'o' << 'h' << 'v';
975
976 UserCategoryItem::UserCategoryItem(int category, AbstractTreeItem *parent)
977     : PropertyMapItem(QStringList() << "categoryName", parent),
978     _category(category)
979 {
980     setFlags(Qt::ItemIsEnabled);
981     setTreeItemFlags(AbstractTreeItem::DeleteOnLastChildRemoved);
982     setObjectName(parent->data(0, Qt::DisplayRole).toString() + "/" + QString::number(category));
983 }
984
985
986 // caching this makes no sense, since we display the user number dynamically
987 QString UserCategoryItem::categoryName() const
988 {
989     int n = childCount();
990     switch (_category) {
991     case 0:
992         return tr("%n Owner(s)", 0, n);
993     case 1:
994         return tr("%n Admin(s)", 0, n);
995     case 2:
996         return tr("%n Operator(s)", 0, n);
997     case 3:
998         return tr("%n Half-Op(s)", 0, n);
999     case 4:
1000         return tr("%n Voiced", 0, n);
1001     default:
1002         return tr("%n User(s)", 0, n);
1003     }
1004 }
1005
1006
1007 IrcUserItem *UserCategoryItem::findIrcUser(IrcUser *ircUser)
1008 {
1009     IrcUserItem *userItem = 0;
1010
1011     for (int i = 0; i < childCount(); i++) {
1012         userItem = qobject_cast<IrcUserItem *>(child(i));
1013         if (!userItem)
1014             continue;
1015         if (userItem->ircUser() == ircUser)
1016             return userItem;
1017     }
1018     return 0;
1019 }
1020
1021
1022 void UserCategoryItem::addUsers(const QList<IrcUser *> &ircUsers)
1023 {
1024     QList<AbstractTreeItem *> userItems;
1025     foreach(IrcUser *ircUser, ircUsers)
1026     userItems << new IrcUserItem(ircUser, this);
1027     newChilds(userItems);
1028     emit dataChanged(0);
1029 }
1030
1031
1032 bool UserCategoryItem::removeUser(IrcUser *ircUser)
1033 {
1034     IrcUserItem *userItem = findIrcUser(ircUser);
1035     bool success = (bool)userItem;
1036     if (success) {
1037         removeChild(userItem);
1038         emit dataChanged(0);
1039     }
1040     return success;
1041 }
1042
1043
1044 int UserCategoryItem::categoryFromModes(const QString &modes)
1045 {
1046     for (int i = 0; i < categories.count(); i++) {
1047         if (modes.contains(categories[i]))
1048             return i;
1049     }
1050     return categories.count();
1051 }
1052
1053
1054 QVariant UserCategoryItem::data(int column, int role) const
1055 {
1056     switch (role) {
1057     case TreeModel::SortRole:
1058         return _category;
1059     case NetworkModel::ItemActiveRole:
1060         return true;
1061     case NetworkModel::ItemTypeRole:
1062         return NetworkModel::UserCategoryItemType;
1063     case NetworkModel::BufferIdRole:
1064         return parent()->data(column, role);
1065     case NetworkModel::NetworkIdRole:
1066         return parent()->data(column, role);
1067     case NetworkModel::BufferInfoRole:
1068         return parent()->data(column, role);
1069     default:
1070         return PropertyMapItem::data(column, role);
1071     }
1072 }
1073
1074
1075 /*****************************************
1076 *  Irc User Items
1077 *****************************************/
1078 IrcUserItem::IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent)
1079     : PropertyMapItem(QStringList() << "nickName", parent),
1080     _ircUser(ircUser)
1081 {
1082     setObjectName(ircUser->nick());
1083     connect(ircUser, SIGNAL(quited()), this, SLOT(ircUserQuited()));
1084     connect(ircUser, SIGNAL(nickSet(QString)), this, SIGNAL(dataChanged()));
1085     connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
1086 }
1087
1088
1089 QVariant IrcUserItem::data(int column, int role) const
1090 {
1091     switch (role) {
1092     case NetworkModel::ItemActiveRole:
1093         return isActive();
1094     case NetworkModel::ItemTypeRole:
1095         return NetworkModel::IrcUserItemType;
1096     case NetworkModel::BufferIdRole:
1097         return parent()->data(column, role);
1098     case NetworkModel::NetworkIdRole:
1099         return parent()->data(column, role);
1100     case NetworkModel::BufferInfoRole:
1101         return parent()->data(column, role);
1102     case NetworkModel::IrcChannelRole:
1103         return parent()->data(column, role);
1104     case NetworkModel::IrcUserRole:
1105         return QVariant::fromValue<QObject *>(_ircUser.data());
1106     case NetworkModel::UserAwayRole:
1107         return (bool)_ircUser ? _ircUser->isAway() : false;
1108     default:
1109         return PropertyMapItem::data(column, role);
1110     }
1111 }
1112
1113
1114 QString IrcUserItem::toolTip(int column) const
1115 {
1116     Q_UNUSED(column);
1117     QString strTooltip;
1118     QTextStream tooltip( &strTooltip, QIODevice::WriteOnly );
1119     tooltip << "<qt><style>.bold { font-weight: bold; } .italic { font-style: italic; }</style>";
1120
1121     // Keep track of whether or not information has been added
1122     bool infoAdded = false;
1123
1124     // Use bufferName() for QueryBufferItem, nickName() for IrcUserItem
1125     tooltip << "<p class='bold' align='center'>" << NetworkItem::escapeHTML(nickName(), true);
1126     if (_ircUser->userModes() != "") {
1127         //TODO: Translate user Modes and add them to the table below and in QueryBufferItem::toolTip
1128         tooltip << " (" << _ircUser->userModes() << ")";
1129     }
1130     tooltip << "</p>";
1131
1132     auto addRow = [&](const QString& key, const QString& value, bool condition) {
1133         if (condition)
1134         {
1135             tooltip << "<tr><td class='bold' align='right'>" << key << "</td><td>" << value << "</td></tr>";
1136             infoAdded = true;
1137         }
1138     };
1139
1140     tooltip << "<table cellspacing='5' cellpadding='0'>";
1141     addRow(tr("Modes"),
1142            NetworkItem::escapeHTML(channelModes()),
1143            !channelModes().isEmpty());
1144     if (_ircUser->isAway()) {
1145         QString awayMessageHTML = QString("<p class='italic'>%1</p>").arg(tr("Unknown"));
1146
1147         // If away message is known, replace with the escaped message.
1148         if (!_ircUser->awayMessage().isEmpty()) {
1149             awayMessageHTML = NetworkItem::escapeHTML(_ircUser->awayMessage());
1150         }
1151         addRow(NetworkItem::escapeHTML(tr("Away message"), true), awayMessageHTML, true);
1152     }
1153     addRow(tr("Realname"),
1154            NetworkItem::escapeHTML(_ircUser->realName()),
1155            !_ircUser->realName().isEmpty());
1156
1157     // suserHost may return "<nick> is available for help", which should be translated.
1158     // See https://www.alien.net.au/irc/irc2numerics.html
1159     if(_ircUser->suserHost().endsWith("available for help")) {
1160         addRow(NetworkItem::escapeHTML(tr("Help status"), true),
1161                NetworkItem::escapeHTML(tr("Available for help")),
1162                true);
1163     } else {
1164         addRow(NetworkItem::escapeHTML(tr("Service status"), true),
1165                NetworkItem::escapeHTML(_ircUser->suserHost()),
1166                !_ircUser->suserHost().isEmpty());
1167     }
1168
1169     // Keep track of whether or not the account information's been added.  Don't show it twice.
1170     bool accountAdded = false;
1171     if(!_ircUser->account().isEmpty()) {
1172         // IRCv3 account-notify is supported by the core and IRC server.
1173         // Assume logged out (seems to be more common)
1174         QString accountHTML = QString("<p class='italic'>%1</p>").arg(tr("Not logged in"));
1175
1176         // If account is logged in, replace with the escaped account name.
1177         if (_ircUser->account() != "*") {
1178             accountHTML = NetworkItem::escapeHTML(_ircUser->account());
1179         }
1180         addRow(NetworkItem::escapeHTML(tr("Account"), true),
1181                accountHTML,
1182                true);
1183         // Mark the row as added
1184         accountAdded = true;
1185     }
1186     // whoisServiceReply may return "<nick> is identified for this nick", which should be translated.
1187     // See https://www.alien.net.au/irc/irc2numerics.html
1188     if(_ircUser->whoisServiceReply().endsWith("identified for this nick")) {
1189         addRow(NetworkItem::escapeHTML(tr("Account"), true),
1190                NetworkItem::escapeHTML(tr("Identified for this nick")),
1191                !accountAdded);
1192         // Don't add the account row again if information's already added via account-notify
1193         // Mark the row as added
1194         accountAdded = true;
1195     } else {
1196         addRow(NetworkItem::escapeHTML(tr("Service Reply"), true),
1197                NetworkItem::escapeHTML(_ircUser->whoisServiceReply()),
1198                !_ircUser->whoisServiceReply().isEmpty());
1199     }
1200     addRow(tr("Hostmask"),
1201            NetworkItem::escapeHTML(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!") + 1)),
1202            !(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!") + 1) == "@"));
1203     // ircOperator may contain "is an" or "is a", which should be removed.
1204     addRow(tr("Operator"),
1205            NetworkItem::escapeHTML(_ircUser->ircOperator().replace("is an ", "").replace("is a ", "")),
1206            !_ircUser->ircOperator().isEmpty());
1207
1208     if (_ircUser->idleTime().isValid()) {
1209         QDateTime now = QDateTime::currentDateTime();
1210         QDateTime idle = _ircUser->idleTime();
1211         int idleTime = idle.secsTo(now);
1212         addRow(NetworkItem::escapeHTML(tr("Idling since"), true), secondsToString(idleTime), true);
1213     }
1214
1215     if (_ircUser->loginTime().isValid()) {
1216         addRow(NetworkItem::escapeHTML(tr("Login time"), true), _ircUser->loginTime().toString(), true);
1217     }
1218
1219     addRow(tr("Server"), NetworkItem::escapeHTML(_ircUser->server()), !_ircUser->server().isEmpty());
1220     tooltip << "</table>";
1221
1222     // If no further information found, offer an explanatory message
1223     if (!infoAdded)
1224         tooltip << "<p class='italic' align='center'>" << tr("No information available") << "</p>";
1225
1226     tooltip << "</qt>";
1227     return strTooltip;
1228 }
1229
1230 QString IrcUserItem::channelModes() const
1231 {
1232     // IrcUserItems are parented to UserCategoryItem, which are parented to ChannelBufferItem.
1233     // We want the channel buffer item in order to get the channel-specific user modes.
1234     UserCategoryItem *category = qobject_cast<UserCategoryItem *>(parent());
1235     if (!category)
1236         return QString();
1237
1238     ChannelBufferItem *channel = qobject_cast<ChannelBufferItem *>(category->parent());
1239     if (!channel)
1240         return QString();
1241
1242     return channel->nickChannelModes(nickName());
1243 }
1244
1245
1246 /*****************************************
1247  * NetworkModel
1248  *****************************************/
1249 NetworkModel::NetworkModel(QObject *parent)
1250     : TreeModel(NetworkModel::defaultHeader(), parent)
1251 {
1252     connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),
1253         this, SLOT(checkForNewBuffers(const QModelIndex &, int, int)));
1254     connect(this, SIGNAL(rowsAboutToBeRemoved(const QModelIndex &, int, int)),
1255         this, SLOT(checkForRemovedBuffers(const QModelIndex &, int, int)));
1256
1257     BufferSettings defaultSettings;
1258     defaultSettings.notify("UserNoticesTarget", this, SLOT(messageRedirectionSettingsChanged()));
1259     defaultSettings.notify("ServerNoticesTarget", this, SLOT(messageRedirectionSettingsChanged()));
1260     defaultSettings.notify("ErrorMsgsTarget", this, SLOT(messageRedirectionSettingsChanged()));
1261     messageRedirectionSettingsChanged();
1262 }
1263
1264
1265 QList<QVariant> NetworkModel::defaultHeader()
1266 {
1267     QList<QVariant> data;
1268     data << tr("Chat") << tr("Topic") << tr("Nick Count");
1269     return data;
1270 }
1271
1272
1273 bool NetworkModel::isBufferIndex(const QModelIndex &index) const
1274 {
1275     return index.data(NetworkModel::ItemTypeRole) == NetworkModel::BufferItemType;
1276 }
1277
1278
1279 int NetworkModel::networkRow(NetworkId networkId) const
1280 {
1281     NetworkItem *netItem = 0;
1282     for (int i = 0; i < rootItem->childCount(); i++) {
1283         netItem = qobject_cast<NetworkItem *>(rootItem->child(i));
1284         if (!netItem)
1285             continue;
1286         if (netItem->networkId() == networkId)
1287             return i;
1288     }
1289     return -1;
1290 }
1291
1292
1293 QModelIndex NetworkModel::networkIndex(NetworkId networkId)
1294 {
1295     int netRow = networkRow(networkId);
1296     if (netRow == -1)
1297         return QModelIndex();
1298     else
1299         return indexByItem(qobject_cast<NetworkItem *>(rootItem->child(netRow)));
1300 }
1301
1302
1303 NetworkItem *NetworkModel::findNetworkItem(NetworkId networkId) const
1304 {
1305     int netRow = networkRow(networkId);
1306     if (netRow == -1)
1307         return 0;
1308     else
1309         return qobject_cast<NetworkItem *>(rootItem->child(netRow));
1310 }
1311
1312
1313 NetworkItem *NetworkModel::networkItem(NetworkId networkId)
1314 {
1315     NetworkItem *netItem = findNetworkItem(networkId);
1316
1317     if (netItem == 0) {
1318         netItem = new NetworkItem(networkId, rootItem);
1319         rootItem->newChild(netItem);
1320     }
1321     return netItem;
1322 }
1323
1324
1325 void NetworkModel::networkRemoved(const NetworkId &networkId)
1326 {
1327     int netRow = networkRow(networkId);
1328     if (netRow != -1) {
1329         rootItem->removeChild(netRow);
1330     }
1331 }
1332
1333
1334 QModelIndex NetworkModel::bufferIndex(BufferId bufferId)
1335 {
1336     if (!_bufferItemCache.contains(bufferId))
1337         return QModelIndex();
1338
1339     return indexByItem(_bufferItemCache[bufferId]);
1340 }
1341
1342
1343 BufferItem *NetworkModel::findBufferItem(BufferId bufferId) const
1344 {
1345     if (_bufferItemCache.contains(bufferId))
1346         return _bufferItemCache[bufferId];
1347     else
1348         return 0;
1349 }
1350
1351
1352 BufferItem *NetworkModel::bufferItem(const BufferInfo &bufferInfo)
1353 {
1354     if (_bufferItemCache.contains(bufferInfo.bufferId()))
1355         return _bufferItemCache[bufferInfo.bufferId()];
1356
1357     NetworkItem *netItem = networkItem(bufferInfo.networkId());
1358     return netItem->bufferItem(bufferInfo);
1359 }
1360
1361
1362 QStringList NetworkModel::mimeTypes() const
1363 {
1364     // mimetypes we accept for drops
1365     QStringList types;
1366     // comma separated list of colon separated pairs of networkid and bufferid
1367     // example: 0:1,0:2,1:4
1368     types << "application/Quassel/BufferItemList";
1369     return types;
1370 }
1371
1372
1373 bool NetworkModel::mimeContainsBufferList(const QMimeData *mimeData)
1374 {
1375     return mimeData->hasFormat("application/Quassel/BufferItemList");
1376 }
1377
1378
1379 QList<QPair<NetworkId, BufferId> > NetworkModel::mimeDataToBufferList(const QMimeData *mimeData)
1380 {
1381     QList<QPair<NetworkId, BufferId> > bufferList;
1382
1383     if (!mimeContainsBufferList(mimeData))
1384         return bufferList;
1385
1386     QStringList rawBufferList = QString::fromLatin1(mimeData->data("application/Quassel/BufferItemList")).split(",");
1387     NetworkId networkId;
1388     BufferId bufferUid;
1389     foreach(QString rawBuffer, rawBufferList) {
1390         if (!rawBuffer.contains(":"))
1391             continue;
1392         networkId = rawBuffer.section(":", 0, 0).toInt();
1393         bufferUid = rawBuffer.section(":", 1, 1).toInt();
1394         bufferList.append(qMakePair(networkId, bufferUid));
1395     }
1396     return bufferList;
1397 }
1398
1399
1400 QMimeData *NetworkModel::mimeData(const QModelIndexList &indexes) const
1401 {
1402     QMimeData *mimeData = new QMimeData();
1403
1404     QStringList bufferlist;
1405     QString netid, uid, bufferid;
1406     foreach(QModelIndex index, indexes) {
1407         netid = QString::number(index.data(NetworkIdRole).value<NetworkId>().toInt());
1408         uid = QString::number(index.data(BufferIdRole).value<BufferId>().toInt());
1409         bufferid = QString("%1:%2").arg(netid).arg(uid);
1410         if (!bufferlist.contains(bufferid))
1411             bufferlist << bufferid;
1412     }
1413
1414     mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toLatin1());
1415
1416     return mimeData;
1417 }
1418
1419
1420 void NetworkModel::attachNetwork(Network *net)
1421 {
1422     NetworkItem *netItem = networkItem(net->networkId());
1423     netItem->attachNetwork(net);
1424 }
1425
1426
1427 void NetworkModel::bufferUpdated(BufferInfo bufferInfo)
1428 {
1429     BufferItem *bufItem = bufferItem(bufferInfo);
1430     QModelIndex itemindex = indexByItem(bufItem);
1431     emit dataChanged(itemindex, itemindex);
1432 }
1433
1434
1435 void NetworkModel::removeBuffer(BufferId bufferId)
1436 {
1437     BufferItem *buffItem = findBufferItem(bufferId);
1438     if (!buffItem)
1439         return;
1440
1441     buffItem->parent()->removeChild(buffItem);
1442 }
1443
1444
1445 MsgId NetworkModel::lastSeenMsgId(BufferId bufferId) const
1446 {
1447     if (!_bufferItemCache.contains(bufferId))
1448         return MsgId();
1449
1450     return _bufferItemCache[bufferId]->lastSeenMsgId();
1451 }
1452
1453
1454 MsgId NetworkModel::markerLineMsgId(BufferId bufferId) const
1455 {
1456     if (!_bufferItemCache.contains(bufferId))
1457         return MsgId();
1458
1459     return _bufferItemCache[bufferId]->markerLineMsgId();
1460 }
1461
1462
1463 // FIXME we always seem to use this (expensive) non-const version
1464 MsgId NetworkModel::lastSeenMsgId(const BufferId &bufferId)
1465 {
1466     BufferItem *bufferItem = findBufferItem(bufferId);
1467     if (!bufferItem) {
1468         qDebug() << "NetworkModel::lastSeenMsgId(): buffer is unknown:" << bufferId;
1469         Client::purgeKnownBufferIds();
1470         return MsgId();
1471     }
1472     return bufferItem->lastSeenMsgId();
1473 }
1474
1475
1476 void NetworkModel::setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId)
1477 {
1478     BufferItem *bufferItem = findBufferItem(bufferId);
1479     if (!bufferItem) {
1480         qDebug() << "NetworkModel::setLastSeenMsgId(): buffer is unknown:" << bufferId;
1481         Client::purgeKnownBufferIds();
1482         return;
1483     }
1484     bufferItem->setLastSeenMsgId(msgId);
1485     emit lastSeenMsgSet(bufferId, msgId);
1486 }
1487
1488
1489 void NetworkModel::setMarkerLineMsgId(const BufferId &bufferId, const MsgId &msgId)
1490 {
1491     BufferItem *bufferItem = findBufferItem(bufferId);
1492     if (!bufferItem) {
1493         qDebug() << "NetworkModel::setMarkerLineMsgId(): buffer is unknown:" << bufferId;
1494         Client::purgeKnownBufferIds();
1495         return;
1496     }
1497     bufferItem->setMarkerLineMsgId(msgId);
1498     emit markerLineSet(bufferId, msgId);
1499 }
1500
1501
1502 void NetworkModel::updateBufferActivity(Message &msg)
1503 {
1504     int redirectionTarget = 0;
1505     switch (msg.type()) {
1506     case Message::Notice:
1507         if (bufferType(msg.bufferId()) != BufferInfo::ChannelBuffer) {
1508             msg.setFlags(msg.flags() | Message::Redirected);
1509             if (msg.flags() & Message::ServerMsg) {
1510                 // server notice
1511                 redirectionTarget = _serverNoticesTarget;
1512             }
1513             else {
1514                 redirectionTarget = _userNoticesTarget;
1515             }
1516         }
1517         break;
1518     case Message::Error:
1519         msg.setFlags(msg.flags() | Message::Redirected);
1520         redirectionTarget = _errorMsgsTarget;
1521         break;
1522     // Update IrcUser's last activity
1523     case Message::Plain:
1524     case Message::Action:
1525         if (bufferType(msg.bufferId()) == BufferInfo::ChannelBuffer) {
1526             const Network *net = Client::network(msg.bufferInfo().networkId());
1527             IrcUser *user = net ? net->ircUser(nickFromMask(msg.sender())) : 0;
1528             if (user)
1529                 user->setLastChannelActivity(msg.bufferId(), msg.timestamp());
1530         }
1531         break;
1532     default:
1533         break;
1534     }
1535
1536     if (msg.flags() & Message::Redirected) {
1537         if (redirectionTarget & BufferSettings::DefaultBuffer)
1538             updateBufferActivity(bufferItem(msg.bufferInfo()), msg);
1539
1540         if (redirectionTarget & BufferSettings::StatusBuffer) {
1541             const NetworkItem *netItem = findNetworkItem(msg.bufferInfo().networkId());
1542             if (netItem) {
1543                 updateBufferActivity(netItem->statusBufferItem(), msg);
1544             }
1545         }
1546     }
1547     else {
1548         if ((BufferSettings(msg.bufferId()).messageFilter() & msg.type()) != msg.type())
1549             updateBufferActivity(bufferItem(msg.bufferInfo()), msg);
1550     }
1551 }
1552
1553
1554 void NetworkModel::updateBufferActivity(BufferItem *bufferItem, const Message &msg)
1555 {
1556     if (!bufferItem)
1557         return;
1558
1559     bufferItem->updateActivityLevel(msg);
1560     if (bufferItem->isCurrentBuffer())
1561         emit requestSetLastSeenMsg(bufferItem->bufferId(), msg.msgId());
1562 }
1563
1564
1565 void NetworkModel::setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel level)
1566 {
1567     BufferItem *bufferItem = findBufferItem(bufferId);
1568     if (!bufferItem) {
1569         qDebug() << "NetworkModel::setBufferActivity(): buffer is unknown:" << bufferId;
1570         return;
1571     }
1572     bufferItem->setActivityLevel(level);
1573 }
1574
1575
1576 void NetworkModel::clearBufferActivity(const BufferId &bufferId)
1577 {
1578     BufferItem *bufferItem = findBufferItem(bufferId);
1579     if (!bufferItem) {
1580         qDebug() << "NetworkModel::clearBufferActivity(): buffer is unknown:" << bufferId;
1581         return;
1582     }
1583     bufferItem->clearActivityLevel();
1584 }
1585
1586
1587 const Network *NetworkModel::networkByIndex(const QModelIndex &index) const
1588 {
1589     QVariant netVariant = index.data(NetworkIdRole);
1590     if (!netVariant.isValid())
1591         return 0;
1592
1593     NetworkId networkId = netVariant.value<NetworkId>();
1594     return Client::network(networkId);
1595 }
1596
1597
1598 void NetworkModel::checkForRemovedBuffers(const QModelIndex &parent, int start, int end)
1599 {
1600     if (parent.data(ItemTypeRole) != NetworkItemType)
1601         return;
1602
1603     for (int row = start; row <= end; row++) {
1604         _bufferItemCache.remove(parent.child(row, 0).data(BufferIdRole).value<BufferId>());
1605     }
1606 }
1607
1608
1609 void NetworkModel::checkForNewBuffers(const QModelIndex &parent, int start, int end)
1610 {
1611     if (parent.data(ItemTypeRole) != NetworkItemType)
1612         return;
1613
1614     for (int row = start; row <= end; row++) {
1615         QModelIndex child = parent.child(row, 0);
1616         _bufferItemCache[child.data(BufferIdRole).value < BufferId > ()] = static_cast<BufferItem *>(child.internalPointer());
1617     }
1618 }
1619
1620
1621 QString NetworkModel::bufferName(BufferId bufferId) const
1622 {
1623     if (!_bufferItemCache.contains(bufferId))
1624         return QString();
1625
1626     return _bufferItemCache[bufferId]->bufferName();
1627 }
1628
1629
1630 BufferInfo::Type NetworkModel::bufferType(BufferId bufferId) const
1631 {
1632     if (!_bufferItemCache.contains(bufferId))
1633         return BufferInfo::InvalidBuffer;
1634
1635     return _bufferItemCache[bufferId]->bufferType();
1636 }
1637
1638
1639 BufferInfo NetworkModel::bufferInfo(BufferId bufferId) const
1640 {
1641     if (!_bufferItemCache.contains(bufferId))
1642         return BufferInfo();
1643
1644     return _bufferItemCache[bufferId]->bufferInfo();
1645 }
1646
1647
1648 NetworkId NetworkModel::networkId(BufferId bufferId) const
1649 {
1650     if (!_bufferItemCache.contains(bufferId))
1651         return NetworkId();
1652
1653     NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1654     if (netItem)
1655         return netItem->networkId();
1656     else
1657         return NetworkId();
1658 }
1659
1660
1661 QString NetworkModel::networkName(BufferId bufferId) const
1662 {
1663     if (!_bufferItemCache.contains(bufferId))
1664         return QString();
1665
1666     NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1667     if (netItem)
1668         return netItem->networkName();
1669     else
1670         return QString();
1671 }
1672
1673
1674 BufferId NetworkModel::bufferId(NetworkId networkId, const QString &bufferName, Qt::CaseSensitivity cs) const
1675 {
1676     const NetworkItem *netItem = findNetworkItem(networkId);
1677     if (!netItem)
1678         return BufferId();
1679
1680     for (int i = 0; i < netItem->childCount(); i++) {
1681         BufferItem *bufferItem = qobject_cast<BufferItem *>(netItem->child(i));
1682         if (bufferItem && !bufferItem->bufferName().compare(bufferName, cs))
1683             return bufferItem->bufferId();
1684     }
1685     return BufferId();
1686 }
1687
1688
1689 void NetworkModel::sortBufferIds(QList<BufferId> &bufferIds) const
1690 {
1691     QList<BufferItem *> bufferItems;
1692     foreach(BufferId bufferId, bufferIds) {
1693         if (_bufferItemCache.contains(bufferId))
1694             bufferItems << _bufferItemCache[bufferId];
1695     }
1696
1697     qSort(bufferItems.begin(), bufferItems.end(), bufferItemLessThan);
1698
1699     bufferIds.clear();
1700     foreach(BufferItem *bufferItem, bufferItems) {
1701         bufferIds << bufferItem->bufferId();
1702     }
1703 }
1704
1705
1706 QList<BufferId> NetworkModel::allBufferIdsSorted() const
1707 {
1708     QList<BufferId> bufferIds = allBufferIds();
1709     sortBufferIds(bufferIds);
1710     return bufferIds;
1711 }
1712
1713
1714 bool NetworkModel::bufferItemLessThan(const BufferItem *left, const BufferItem *right)
1715 {
1716     int leftType = left->bufferType();
1717     int rightType = right->bufferType();
1718
1719     if (leftType != rightType)
1720         return leftType < rightType;
1721     else
1722         return QString::compare(left->bufferName(), right->bufferName(), Qt::CaseInsensitive) < 0;
1723 }
1724
1725
1726 void NetworkModel::messageRedirectionSettingsChanged()
1727 {
1728     BufferSettings bufferSettings;
1729
1730     _userNoticesTarget = bufferSettings.userNoticesTarget();
1731     _serverNoticesTarget = bufferSettings.serverNoticesTarget();
1732     _errorMsgsTarget = bufferSettings.errorMsgsTarget();
1733 }
1734
1735 void NetworkModel::bufferActivityChanged(BufferId bufferId, const Message::Types activity) {
1736     auto bufferItem = findBufferItem(bufferId);
1737     if (!bufferItem) {
1738         qDebug() << "NetworkModel::clearBufferActivity(): buffer is unknown:" << bufferId;
1739         return;
1740     }
1741     auto hiddenTypes = BufferSettings(bufferId).messageFilter();
1742     auto visibleTypes = ~hiddenTypes;
1743     auto activityVisibleTypesIntersection = activity & visibleTypes;
1744     bufferItem->setActivity(activityVisibleTypesIntersection, false);
1745 }