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