src/client/networkmodel.cpp

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