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