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