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