src/client/networkmodel.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-08 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  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "networkmodel.h"
  22
  23 #include <QAbstractItemView>
  24
  25 #include "buffermodel.h"
  26 #include "client.h"
  27 #include "signalproxy.h"
  28 #include "network.h"
  29 #include "ircchannel.h"
  30
  31 #include "buffersettings.h"
  32
  33 #include "util.h" // get rid of this (needed for isChannelName)
  34
  35 /*****************************************
  36 *  Network Items
  37 *****************************************/
  38 NetworkItem::NetworkItem(const NetworkId &netid, AbstractTreeItem *parent)
  39   : PropertyMapItem(QList<QString>() << "networkName" << "currentServer" << "nickCount", parent),
  40     _networkId(netid)
  41 {
  42   // DO NOT EMIT dataChanged() DIRECTLY IN NetworkItem
  43   // use networkDataChanged() instead. Otherwise you will end up in a infinite loop
  44   // as we "sync" the dataChanged() signals of NetworkItem and StatusBufferItem
  45   setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
  46   connect(this, SIGNAL(networkDataChanged(int)), this, SIGNAL(dataChanged(int)));
  47 }
  48
  49 QVariant NetworkItem::data(int column, int role) const {
  50   switch(role) {
  51   case NetworkModel::BufferIdRole:
  52   case NetworkModel::BufferInfoRole:
  53   case NetworkModel::BufferTypeRole:
  54   case NetworkModel::BufferActivityRole:
  55     if(childCount())
  56       return child(0)->data(column, role);
  57     else
  58       return QVariant();
  59   case NetworkModel::NetworkIdRole:
  60     return qVariantFromValue(_networkId);
  61   case NetworkModel::ItemTypeRole:
  62     return NetworkModel::NetworkItemType;
  63   case NetworkModel::ItemActiveRole:
  64     return isActive();
  65   default:
  66     return PropertyMapItem::data(column, role);
  67   }
  68 }
  69
  70 BufferItem *NetworkItem::findBufferItem(BufferId bufferId) {
  71   BufferItem *bufferItem = 0;
  72
  73   for(int i = 0; i < childCount(); i++) {
  74     bufferItem = qobject_cast<BufferItem *>(child(i));
  75     if(!bufferItem)
  76       continue;
  77     if(bufferItem->bufferId() == bufferId)
  78       return bufferItem;
  79   }
  80   return 0;
  81 }
  82
  83
  84 BufferItem *NetworkItem::bufferItem(const BufferInfo &bufferInfo) {
  85   BufferItem *bufferItem = findBufferItem(bufferInfo);
  86   if(bufferItem)
  87     return bufferItem;
  88
  89   switch(bufferInfo.type()) {
  90   case BufferInfo::StatusBuffer:
  91     bufferItem = new StatusBufferItem(bufferInfo, this);
  92     disconnect(this, SIGNAL(networkDataChanged(int)), this, SIGNAL(dataChanged(int)));
  93     connect(this, SIGNAL(networkDataChanged(int)), bufferItem, SIGNAL(dataChanged(int)));
  94     connect(bufferItem, SIGNAL(dataChanged(int)), this, SIGNAL(dataChanged(int)));
  95     break;
  96   case BufferInfo::ChannelBuffer:
  97     bufferItem = new ChannelBufferItem(bufferInfo, this);
  98     break;
  99   case BufferInfo::QueryBuffer:
 100     bufferItem = new QueryBufferItem(bufferInfo, this);
 101     break;
 102   default:
 103     bufferItem = new BufferItem(bufferInfo, this);
 104   }
 105
 106   newChild(bufferItem);
 107   return bufferItem;
 108 }
 109
 110 void NetworkItem::attachNetwork(Network *network) {
 111   if(!network)
 112     return;
 113
 114   _network = network;
 115
 116   connect(network, SIGNAL(networkNameSet(QString)),
 117           this, SLOT(setNetworkName(QString)));
 118   connect(network, SIGNAL(currentServerSet(QString)),
 119           this, SLOT(setCurrentServer(QString)));
 120   connect(network, SIGNAL(ircChannelAdded(IrcChannel *)),
 121           this, SLOT(attachIrcChannel(IrcChannel *)));
 122   connect(network, SIGNAL(ircUserAdded(IrcUser *)),
 123           this, SLOT(attachIrcUser(IrcUser *)));
 124   connect(network, SIGNAL(connectedSet(bool)),
 125           this, SIGNAL(networkDataChanged()));
 126   connect(network, SIGNAL(destroyed()),
 127           this, SIGNAL(networkDataChanged()));
 128
 129   emit networkDataChanged();
 130 }
 131
 132 void NetworkItem::attachIrcChannel(IrcChannel *ircChannel) {
 133   ChannelBufferItem *channelItem;
 134   for(int i = 0; i < childCount(); i++) {
 135     channelItem = qobject_cast<ChannelBufferItem *>(child(i));
 136     if(!channelItem)
 137       continue;
 138
 139     if(channelItem->bufferName().toLower() == ircChannel->name().toLower()) {
 140       channelItem->attachIrcChannel(ircChannel);
 141       return;
 142     }
 143   }
 144 }
 145
 146 void NetworkItem::attachIrcUser(IrcUser *ircUser) {
 147   QueryBufferItem *queryItem = 0;
 148   for(int i = 0; i < childCount(); i++) {
 149     queryItem = qobject_cast<QueryBufferItem *>(child(i));
 150     if(!queryItem)
 151       continue;
 152
 153     if(queryItem->bufferName().toLower() == ircUser->nick().toLower()) {
 154       queryItem->setIrcUser(ircUser);
 155       break;
 156     }
 157   }
 158 }
 159
 160 void NetworkItem::setNetworkName(const QString &networkName) {
 161   Q_UNUSED(networkName);
 162   emit networkDataChanged(0);
 163 }
 164
 165 void NetworkItem::setCurrentServer(const QString &serverName) {
 166   Q_UNUSED(serverName);
 167   emit networkDataChanged(1);
 168 }
 169
 170
 171 QString NetworkItem::toolTip(int column) const {
 172   Q_UNUSED(column);
 173
 174   QStringList toolTip(QString("<b>%1</b>").arg(networkName()));
 175   toolTip.append(tr("Server: %1").arg(currentServer()));
 176   toolTip.append(tr("Users: %1").arg(nickCount()));
 177
 178   if(_network) {
 179     toolTip.append(tr("Lag: %1 msecs").arg(_network->latency()));
 180   }
 181
 182   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
 183 }
 184
 185 /*****************************************
 186 *  Fancy Buffer Items
 187 *****************************************/
 188 BufferItem::BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
 189   : PropertyMapItem(QStringList() << "bufferName" << "topic" << "nickCount", parent),
 190     _bufferInfo(bufferInfo),
 191     _activity(BufferInfo::NoActivity)
 192 {
 193   setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled);
 194 }
 195
 196 void BufferItem::setActivityLevel(BufferInfo::ActivityLevel level) {
 197   if(_activity != level) {
 198     _activity = level;
 199     emit dataChanged();
 200   }
 201 }
 202
 203 void BufferItem::clearActivityLevel() {
 204   _activity = BufferInfo::NoActivity;
 205   _lastSeenMarkerMsgId = _lastSeenMsgId;
 206   emit dataChanged();
 207 }
 208
 209 void BufferItem::updateActivityLevel(const Message &msg) {
 210   if(isCurrentBuffer()) {
 211     return;
 212   }
 213
 214   if(msg.flags() & Message::Self)       // don't update activity for our own messages
 215     return;
 216
 217   if(lastSeenMsgId() >= msg.msgId())
 218     return;
 219
 220   BufferInfo::ActivityLevel oldLevel = activityLevel();
 221
 222   _activity |= BufferInfo::OtherActivity;
 223   if(msg.type() & (Message::Plain | Message::Notice | Message::Action))
 224     _activity |= BufferInfo::NewMessage;
 225
 226   if(msg.flags() & Message::Highlight)
 227     _activity |= BufferInfo::Highlight;
 228
 229   if(oldLevel != _activity)
 230     emit dataChanged();
 231 }
 232
 233 QVariant BufferItem::data(int column, int role) const {
 234   switch(role) {
 235   case NetworkModel::ItemTypeRole:
 236     return NetworkModel::BufferItemType;
 237   case NetworkModel::BufferIdRole:
 238     return qVariantFromValue(bufferInfo().bufferId());
 239   case NetworkModel::NetworkIdRole:
 240     return qVariantFromValue(bufferInfo().networkId());
 241   case NetworkModel::BufferInfoRole:
 242     return qVariantFromValue(bufferInfo());
 243   case NetworkModel::BufferTypeRole:
 244     return int(bufferType());
 245   case NetworkModel::ItemActiveRole:
 246     return isActive();
 247   case NetworkModel::BufferActivityRole:
 248     return (int)activityLevel();
 249   default:
 250     return PropertyMapItem::data(column, role);
 251   }
 252 }
 253
 254 bool BufferItem::setData(int column, const QVariant &value, int role) {
 255   switch(role) {
 256   case NetworkModel::BufferActivityRole:
 257     setActivityLevel((BufferInfo::ActivityLevel)value.toInt());
 258     return true;
 259   default:
 260     return PropertyMapItem::setData(column, value, role);
 261   }
 262   return true;
 263 }
 264
 265 void BufferItem::setBufferName(const QString &name) {
 266   _bufferInfo = BufferInfo(_bufferInfo.bufferId(), _bufferInfo.networkId(), _bufferInfo.type(), _bufferInfo.groupId(), name);
 267   emit dataChanged(0);
 268 }
 269
 270 void BufferItem::setLastSeenMsgId(const MsgId &msgId) {
 271   _lastSeenMsgId = msgId;
 272   if(!isCurrentBuffer()) {
 273     _lastSeenMarkerMsgId = msgId;
 274   }
 275   setActivityLevel(BufferInfo::NoActivity);
 276 }
 277
 278 bool BufferItem::isCurrentBuffer() const {
 279   return _bufferInfo.bufferId() == Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
 280 }
 281
 282 QString BufferItem::toolTip(int column) const {
 283   Q_UNUSED(column);
 284   return tr("<p> %1 - %2 </p>").arg(bufferInfo().bufferId().toInt()).arg(bufferName());
 285 }
 286
 287 /*****************************************
 288 *  StatusBufferItem
 289 *****************************************/
 290 StatusBufferItem::StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
 291   : BufferItem(bufferInfo, parent)
 292 {
 293 }
 294
 295 QString StatusBufferItem::toolTip(int column) const {
 296   Q_UNUSED(column);
 297   QStringList toolTip;
 298
 299   QString netName = Client::network(bufferInfo().networkId())->networkName();
 300   toolTip.append(tr("<b>Status buffer of %1</b>").arg(netName));
 301
 302   return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
 303 }
 304
 305 /*****************************************
 306 *  QueryBufferItem
 307 *****************************************/
 308 QueryBufferItem::QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
 309   : BufferItem(bufferInfo, parent),
 310     _ircUser(0)
 311 {
 312   setFlags(flags() | Qt::ItemIsDropEnabled | Qt::ItemIsEditable);
 313
 314   const Network *net = Client::network(bufferInfo.networkId());
 315   if(!net)
 316     return;
 317
 318   IrcUser *ircUser = net->ircUser(bufferInfo.bufferName());
 319   setIrcUser(ircUser);
 320 }
 321
 322 QVariant QueryBufferItem::data(int column, int role) const {
 323   switch(role) {
 324   case NetworkModel::IrcUserRole:
 325     return QVariant::fromValue<QObject *>(_ircUser);
 326   case NetworkModel::UserAwayRole:
 327     return (bool)_ircUser ? _ircUser->isAway() : false;
 328   default:
 329     return BufferItem::data(column, role);
 330   }
 331 }
 332
 333 bool QueryBufferItem::setData(int column, const QVariant &value, int role) {
 334   if(column != 0)
 335     return BufferItem::setData(column, value, role);
 336
 337   switch(role) {
 338   case Qt::EditRole:
 339     {
 340       QString newName = value.toString();
 341       if(!newName.isEmpty()) {
 342         Client::renameBuffer(bufferId(), newName);
 343         return true;
 344       } else {
 345         return false;
 346       }
 347     }
 348     break;
 349   default:
 350     return BufferItem::setData(column, value, role);
 351   }
 352 }
 353
 354 void QueryBufferItem::setBufferName(const QString &name) {
 355   BufferItem::setBufferName(name);
 356   NetworkId netId = data(0, NetworkModel::NetworkIdRole).value<NetworkId>();
 357   const Network *net = Client::network(netId);
 358   if(net)
 359     setIrcUser(net->ircUser(name));
 360 }
 361
 362 QString QueryBufferItem::toolTip(int column) const {
 363   // pretty much code duplication of IrcUserItem::toolTip() but inheritance won't solve this...
 364   Q_UNUSED(column);
 365   QStringList toolTip;
 366
 367   toolTip.append(tr("<b>Query with %1</b>").arg(bufferName()));
 368
 369   if(_ircUser) {
 370     if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
 371     if(_ircUser->isAway()) {
 372       toolTip[0].append(QString(" (away%1)").arg(!_ircUser->awayMessage().isEmpty() ? (QString(" ") + _ircUser->awayMessage()) : QString()));
 373     }
 374     if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
 375     if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(_ircUser->nick()).arg(_ircUser->ircOperator()));
 376     if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
 377     if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
 378
 379     toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
 380
 381     if(_ircUser->idleTime().isValid()) {
 382       QDateTime now = QDateTime::currentDateTime();
 383       QDateTime idle = _ircUser->idleTime();
 384       int idleTime = idle.secsTo(now);
 385       toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
 386     }
 387     if(_ircUser->loginTime().isValid()) {
 388       toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
 389     }
 390
 391     if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
 392   }
 393
 394   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
 395 }
 396
 397 void QueryBufferItem::setIrcUser(IrcUser *ircUser) {
 398   if(_ircUser == ircUser)
 399     return;
 400
 401   if(_ircUser) {
 402     disconnect(_ircUser, 0, this, 0);
 403   }
 404
 405   if(ircUser) {
 406     connect(ircUser, SIGNAL(quited()), this, SLOT(removeIrcUser()));
 407     connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
 408   }
 409
 410   _ircUser = ircUser;
 411   emit dataChanged();
 412 }
 413
 414 void QueryBufferItem::removeIrcUser() {
 415   _ircUser = 0;
 416   emit dataChanged();
 417 }
 418
 419 /*****************************************
 420 *  ChannelBufferItem
 421 *****************************************/
 422 ChannelBufferItem::ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
 423   : BufferItem(bufferInfo, parent),
 424     _ircChannel(0)
 425 {
 426   const Network *net = Client::network(bufferInfo.networkId());
 427   if(!net)
 428     return;
 429
 430   IrcChannel *ircChannel = net->ircChannel(bufferInfo.bufferName());
 431   if(ircChannel)
 432     attachIrcChannel(ircChannel);
 433 }
 434
 435 QVariant ChannelBufferItem::data(int column, int role) const {
 436   switch(role) {
 437     case NetworkModel::IrcChannelRole:
 438       return QVariant::fromValue<QObject *>(_ircChannel);
 439     default:
 440       return BufferItem::data(column, role);
 441   }
 442 }
 443
 444 QString ChannelBufferItem::toolTip(int column) const {
 445   Q_UNUSED(column);
 446   QStringList toolTip;
 447
 448   toolTip.append(tr("<b>Channel %1</b>").arg(bufferName()));
 449   if(isActive()) {
 450     //TODO: add channel modes
 451     toolTip.append(tr("<b>Users:</b> %1").arg(nickCount()));
 452     if(_ircChannel) {
 453       QString channelMode = _ircChannel->channelModeString(); // channelModeString is compiled on the fly -> thus cache the result
 454       if(!channelMode.isEmpty())
 455         toolTip.append(tr("<b>Mode:</b> %1").arg(channelMode));
 456     }
 457
 458     BufferSettings s;
 459     bool showTopic = s.value("DisplayTopicInTooltip", QVariant(false)).toBool();
 460     if(showTopic) {
 461       QString _topic = topic();
 462       if(_topic != "") {
 463         _topic = stripFormatCodes(_topic);
 464         _topic.replace(QString("<"), QString("&lt;"));
 465         _topic.replace(QString(">"), QString("&gt;"));
 466         toolTip.append(QString("<font size='-2'>&nbsp;</font>"));
 467         toolTip.append(tr("<b>Topic:</b> %1").arg(_topic));
 468       }
 469     }
 470   } else {
 471     toolTip.append(tr("Not active <br /> Double-click to join"));
 472   }
 473
 474   return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
 475 }
 476
 477 void ChannelBufferItem::attachIrcChannel(IrcChannel *ircChannel) {
 478   Q_ASSERT(!_ircChannel && ircChannel);
 479
 480   _ircChannel = ircChannel;
 481
 482   connect(ircChannel, SIGNAL(topicSet(QString)),
 483           this, SLOT(setTopic(QString)));
 484   connect(ircChannel, SIGNAL(ircUsersJoined(QList<IrcUser *>)),
 485           this, SLOT(join(QList<IrcUser *>)));
 486   connect(ircChannel, SIGNAL(ircUserParted(IrcUser *)),
 487           this, SLOT(part(IrcUser *)));
 488   connect(ircChannel, SIGNAL(parted()),
 489           this, SLOT(ircChannelParted()));
 490   connect(ircChannel, SIGNAL(ircUserModesSet(IrcUser *, QString)),
 491           this, SLOT(userModeChanged(IrcUser *)));
 492   connect(ircChannel, SIGNAL(ircUserModeAdded(IrcUser *, QString)),
 493           this, SLOT(userModeChanged(IrcUser *)));
 494   connect(ircChannel, SIGNAL(ircUserModeRemoved(IrcUser *, QString)),
 495           this, SLOT(userModeChanged(IrcUser *)));
 496
 497   if(!ircChannel->ircUsers().isEmpty())
 498     join(ircChannel->ircUsers());
 499
 500   emit dataChanged();
 501 }
 502
 503 void ChannelBufferItem::ircChannelParted() {
 504   Q_CHECK_PTR(_ircChannel);
 505   disconnect(_ircChannel, 0, this, 0);
 506   _ircChannel = 0;
 507   emit dataChanged();
 508   removeAllChilds();
 509 }
 510
 511 void ChannelBufferItem::join(const QList<IrcUser *> &ircUsers) {
 512   addUsersToCategory(ircUsers);
 513   emit dataChanged(2);
 514 }
 515
 516 UserCategoryItem *ChannelBufferItem::findCategoryItem(int categoryId) {
 517   UserCategoryItem *categoryItem = 0;
 518
 519   for(int i = 0; i < childCount(); i++) {
 520     categoryItem = qobject_cast<UserCategoryItem *>(child(i));
 521     if(!categoryItem)
 522       continue;
 523     if(categoryItem->categoryId() == categoryId)
 524       return categoryItem;
 525   }
 526   return 0;
 527 }
 528
 529 void ChannelBufferItem::addUserToCategory(IrcUser *ircUser) {
 530   addUsersToCategory(QList<IrcUser *>() << ircUser);
 531 }
 532
 533 void ChannelBufferItem::addUsersToCategory(const QList<IrcUser *> &ircUsers) {
 534   Q_ASSERT(_ircChannel);
 535
 536   QHash<UserCategoryItem *, QList<IrcUser *> > categories;
 537
 538   int categoryId = -1;
 539   UserCategoryItem *categoryItem = 0;
 540
 541   foreach(IrcUser *ircUser, ircUsers) {
 542     categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
 543     categoryItem = findCategoryItem(categoryId);
 544     if(!categoryItem) {
 545       categoryItem = new UserCategoryItem(categoryId, this);
 546       categories[categoryItem] = QList<IrcUser *>();
 547       newChild(categoryItem);
 548     }
 549     categories[categoryItem] << ircUser;
 550   }
 551
 552   QHash<UserCategoryItem *, QList<IrcUser *> >::const_iterator catIter = categories.constBegin();
 553   while(catIter != categories.constEnd()) {
 554     catIter.key()->addUsers(catIter.value());
 555     catIter++;
 556   }
 557 }
 558
 559 void ChannelBufferItem::part(IrcUser *ircUser) {
 560   if(!ircUser) {
 561     qWarning() << bufferName() << "ChannelBufferItem::part(): unknown User" << ircUser;
 562     return;
 563   }
 564
 565   disconnect(ircUser, 0, this, 0);
 566   removeUserFromCategory(ircUser);
 567   emit dataChanged(2);
 568 }
 569
 570 void ChannelBufferItem::removeUserFromCategory(IrcUser *ircUser) {
 571   if(!_ircChannel) {
 572     // If we parted the channel there might still be some ircUsers connected.
 573     // in that case we just ignore the call
 574     Q_ASSERT(childCount() == 0);
 575     return;
 576   }
 577
 578   UserCategoryItem *categoryItem = 0;
 579   for(int i = 0; i < childCount(); i++) {
 580     categoryItem = qobject_cast<UserCategoryItem *>(child(i));
 581     if(categoryItem->removeUser(ircUser)) {
 582       if(categoryItem->childCount() == 0)
 583         removeChild(i);
 584       break;
 585     }
 586   }
 587 }
 588
 589 void ChannelBufferItem::userModeChanged(IrcUser *ircUser) {
 590   Q_ASSERT(_ircChannel);
 591
 592   int categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
 593   UserCategoryItem *categoryItem = findCategoryItem(categoryId);
 594
 595   if(categoryItem) {
 596     if(categoryItem->findIrcUser(ircUser)) {
 597       return; // already in the right category;
 598     }
 599   } else {
 600     categoryItem = new UserCategoryItem(categoryId, this);
 601     newChild(categoryItem);
 602   }
 603
 604   // find the item that needs reparenting
 605   IrcUserItem *ircUserItem = 0;
 606   for(int i = 0; i < childCount(); i++) {
 607     UserCategoryItem *oldCategoryItem = qobject_cast<UserCategoryItem *>(child(i));
 608     Q_ASSERT(oldCategoryItem);
 609     IrcUserItem *userItem = oldCategoryItem->findIrcUser(ircUser);
 610     if(userItem) {
 611       ircUserItem = userItem;
 612       break;
 613     }
 614   }
 615
 616   if(!ircUserItem) {
 617     qWarning() << "ChannelBufferItem::userModeChanged(IrcUser *): unable to determine old category of" << ircUser;
 618     return;
 619   }
 620   ircUserItem->reParent(categoryItem);
 621 }
 622
 623 /*****************************************
 624 *  User Category Items (like @vh etc.)
 625 *****************************************/
 626 // we hardcode this even though we have PREFIX in network... but that wouldn't help with mapping modes to
 627 // category strings anyway.
 628 const QList<QChar> UserCategoryItem::categories = QList<QChar>() << 'q' << 'a' << 'o' << 'h' << 'v';
 629
 630 UserCategoryItem::UserCategoryItem(int category, AbstractTreeItem *parent)
 631   : PropertyMapItem(QStringList() << "categoryName", parent),
 632     _category(category)
 633 {
 634   setFlags(Qt::ItemIsEnabled);
 635   setTreeItemFlags(AbstractTreeItem::DeleteOnLastChildRemoved);
 636   setObjectName(parent->data(0, Qt::DisplayRole).toString() + "/" + QString::number(category));
 637 }
 638
 639 // caching this makes no sense, since we display the user number dynamically
 640 QString UserCategoryItem::categoryName() const {
 641   int n = childCount();
 642   switch(_category) {
 643     case 0: return tr("%n Owner(s)", 0, n);
 644     case 1: return tr("%n Admin(s)", 0, n);
 645     case 2: return tr("%n Operator(s)", 0, n);
 646     case 3: return tr("%n Half-Op(s)", 0, n);
 647     case 4: return tr("%n Voiced", 0, n);
 648     default: return tr("%n User(s)", 0, n);
 649   }
 650 }
 651
 652 IrcUserItem *UserCategoryItem::findIrcUser(IrcUser *ircUser) {
 653   IrcUserItem *userItem = 0;
 654
 655   for(int i = 0; i < childCount(); i++) {
 656     userItem = qobject_cast<IrcUserItem *>(child(i));
 657     if(!userItem)
 658       continue;
 659     if(userItem->ircUser() == ircUser)
 660       return userItem;
 661   }
 662   return 0;
 663 }
 664
 665 void UserCategoryItem::addUsers(const QList<IrcUser *> &ircUsers) {
 666   QList<AbstractTreeItem *> userItems;
 667   foreach(IrcUser *ircUser, ircUsers)
 668     userItems << new IrcUserItem(ircUser, this);
 669   newChilds(userItems);
 670   emit dataChanged(0);
 671 }
 672
 673 bool UserCategoryItem::removeUser(IrcUser *ircUser) {
 674   IrcUserItem *userItem = findIrcUser(ircUser);
 675   bool success = (bool)userItem;
 676   if(success) {
 677     removeChild(userItem);
 678     emit dataChanged(0);
 679   }
 680   return success;
 681 }
 682
 683 int UserCategoryItem::categoryFromModes(const QString &modes) {
 684   for(int i = 0; i < categories.count(); i++) {
 685     if(modes.contains(categories[i]))
 686       return i;
 687   }
 688   return categories.count();
 689 }
 690
 691 QVariant UserCategoryItem::data(int column, int role) const {
 692   switch(role) {
 693   case TreeModel::SortRole:
 694     return _category;
 695   case NetworkModel::ItemActiveRole:
 696     return true;
 697   case NetworkModel::ItemTypeRole:
 698     return NetworkModel::UserCategoryItemType;
 699   case NetworkModel::BufferIdRole:
 700     return parent()->data(column, role);
 701   case NetworkModel::NetworkIdRole:
 702     return parent()->data(column, role);
 703   case NetworkModel::BufferInfoRole:
 704     return parent()->data(column, role);
 705   default:
 706     return PropertyMapItem::data(column, role);
 707   }
 708 }
 709
 710
 711 /*****************************************
 712 *  Irc User Items
 713 *****************************************/
 714 IrcUserItem::IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent)
 715   : PropertyMapItem(QStringList() << "nickName", parent),
 716     _ircUser(ircUser)
 717 {
 718   setObjectName(ircUser->nick());
 719   connect(ircUser, SIGNAL(quited()), this, SLOT(ircUserQuited()));
 720   connect(ircUser, SIGNAL(nickSet(QString)), this, SIGNAL(dataChanged()));
 721   connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
 722 }
 723
 724 QVariant IrcUserItem::data(int column, int role) const {
 725   switch(role) {
 726   case NetworkModel::ItemActiveRole:
 727     return isActive();
 728   case NetworkModel::ItemTypeRole:
 729     return NetworkModel::IrcUserItemType;
 730   case NetworkModel::BufferIdRole:
 731     return parent()->data(column, role);
 732   case NetworkModel::NetworkIdRole:
 733     return parent()->data(column, role);
 734   case NetworkModel::BufferInfoRole:
 735     return parent()->data(column, role);
 736   case NetworkModel::IrcChannelRole:
 737     return parent()->data(column, role);
 738   case NetworkModel::IrcUserRole:
 739     return QVariant::fromValue<QObject *>(_ircUser.data());
 740   case NetworkModel::UserAwayRole:
 741     return (bool)_ircUser ? _ircUser->isAway() : false;
 742   default:
 743     return PropertyMapItem::data(column, role);
 744   }
 745 }
 746
 747 QString IrcUserItem::toolTip(int column) const {
 748   Q_UNUSED(column);
 749   QStringList toolTip(QString("<b>%1</b>").arg(nickName()));
 750   if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
 751   if(_ircUser->isAway()) {
 752     toolTip[0].append(" is away");
 753     if(!_ircUser->awayMessage().isEmpty())
 754       toolTip[0].append(QString(" (%1)").arg(_ircUser->awayMessage()));
 755   }
 756   if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
 757   if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(nickName()).arg(_ircUser->ircOperator()));
 758   if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
 759   if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
 760
 761   toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
 762
 763   if(_ircUser->idleTime().isValid()) {
 764     QDateTime now = QDateTime::currentDateTime();
 765     QDateTime idle = _ircUser->idleTime();
 766     int idleTime = idle.secsTo(now);
 767     toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
 768   }
 769   if(_ircUser->loginTime().isValid()) {
 770     toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
 771   }
 772
 773   if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
 774
 775   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
 776 }
 777
 778 /*****************************************
 779  * NetworkModel
 780  *****************************************/
 781 NetworkModel::NetworkModel(QObject *parent)
 782   : TreeModel(NetworkModel::defaultHeader(), parent)
 783 {
 784   connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),
 785           this, SLOT(checkForNewBuffers(const QModelIndex &, int, int)));
 786   connect(this, SIGNAL(rowsAboutToBeRemoved(const QModelIndex &, int, int)),
 787           this, SLOT(checkForRemovedBuffers(const QModelIndex &, int, int)));
 788 }
 789
 790 QList<QVariant >NetworkModel::defaultHeader() {
 791   QList<QVariant> data;
 792   data << tr("Buffer") << tr("Topic") << tr("Nick Count");
 793   return data;
 794 }
 795
 796 bool NetworkModel::isBufferIndex(const QModelIndex &index) const {
 797   return index.data(NetworkModel::ItemTypeRole) == NetworkModel::BufferItemType;
 798 }
 799
 800 int NetworkModel::networkRow(NetworkId networkId) const {
 801   NetworkItem *netItem = 0;
 802   for(int i = 0; i < rootItem->childCount(); i++) {
 803     netItem = qobject_cast<NetworkItem *>(rootItem->child(i));
 804     if(!netItem)
 805       continue;
 806     if(netItem->networkId() == networkId)
 807       return i;
 808   }
 809   return -1;
 810 }
 811
 812 QModelIndex NetworkModel::networkIndex(NetworkId networkId) {
 813   int netRow = networkRow(networkId);
 814   if(netRow == -1)
 815     return QModelIndex();
 816   else
 817     return indexByItem(qobject_cast<NetworkItem *>(rootItem->child(netRow)));
 818 }
 819
 820 NetworkItem *NetworkModel::findNetworkItem(NetworkId networkId) const {
 821   int netRow = networkRow(networkId);
 822   if(netRow == -1)
 823     return 0;
 824   else
 825     return qobject_cast<NetworkItem *>(rootItem->child(netRow));
 826 }
 827
 828 NetworkItem *NetworkModel::networkItem(NetworkId networkId) {
 829   NetworkItem *netItem = findNetworkItem(networkId);
 830
 831   if(netItem == 0) {
 832     netItem = new NetworkItem(networkId, rootItem);
 833     rootItem->newChild(netItem);
 834   }
 835   return netItem;
 836 }
 837
 838 void NetworkModel::networkRemoved(const NetworkId &networkId) {
 839   int netRow = networkRow(networkId);
 840   if(netRow != -1) {
 841     rootItem->removeChild(netRow);
 842   }
 843 }
 844
 845 QModelIndex NetworkModel::bufferIndex(BufferId bufferId) {
 846   if(!_bufferItemCache.contains(bufferId))
 847     return QModelIndex();
 848
 849   return indexByItem(_bufferItemCache[bufferId]);
 850 }
 851
 852 BufferItem *NetworkModel::findBufferItem(BufferId bufferId) const {
 853   if(_bufferItemCache.contains(bufferId))
 854     return _bufferItemCache[bufferId];
 855   else
 856     return 0;
 857 }
 858
 859 BufferItem *NetworkModel::bufferItem(const BufferInfo &bufferInfo) {
 860   if(_bufferItemCache.contains(bufferInfo.bufferId()))
 861     return _bufferItemCache[bufferInfo.bufferId()];
 862
 863   NetworkItem *netItem = networkItem(bufferInfo.networkId());
 864   return netItem->bufferItem(bufferInfo);
 865 }
 866
 867 QStringList NetworkModel::mimeTypes() const {
 868   // mimetypes we accept for drops
 869   QStringList types;
 870   // comma separated list of colon separated pairs of networkid and bufferid
 871   // example: 0:1,0:2,1:4
 872   types << "application/Quassel/BufferItemList";
 873   return types;
 874 }
 875
 876 bool NetworkModel::mimeContainsBufferList(const QMimeData *mimeData) {
 877   return mimeData->hasFormat("application/Quassel/BufferItemList");
 878 }
 879
 880 QList< QPair<NetworkId, BufferId> > NetworkModel::mimeDataToBufferList(const QMimeData *mimeData) {
 881   QList< QPair<NetworkId, BufferId> > bufferList;
 882
 883   if(!mimeContainsBufferList(mimeData))
 884     return bufferList;
 885
 886   QStringList rawBufferList = QString::fromAscii(mimeData->data("application/Quassel/BufferItemList")).split(",");
 887   NetworkId networkId;
 888   BufferId bufferUid;
 889   foreach(QString rawBuffer, rawBufferList) {
 890     if(!rawBuffer.contains(":"))
 891       continue;
 892     networkId = rawBuffer.section(":", 0, 0).toInt();
 893     bufferUid = rawBuffer.section(":", 1, 1).toInt();
 894     bufferList.append(qMakePair(networkId, bufferUid));
 895   }
 896   return bufferList;
 897 }
 898
 899
 900 QMimeData *NetworkModel::mimeData(const QModelIndexList &indexes) const {
 901   QMimeData *mimeData = new QMimeData();
 902
 903   QStringList bufferlist;
 904   QString netid, uid, bufferid;
 905   foreach(QModelIndex index, indexes) {
 906     netid = QString::number(index.data(NetworkIdRole).value<NetworkId>().toInt());
 907     uid = QString::number(index.data(BufferIdRole).value<BufferId>().toInt());
 908     bufferid = QString("%1:%2").arg(netid).arg(uid);
 909     if(!bufferlist.contains(bufferid))
 910       bufferlist << bufferid;
 911   }
 912
 913   mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toAscii());
 914
 915   return mimeData;
 916 }
 917
 918 void NetworkModel::attachNetwork(Network *net) {
 919   NetworkItem *netItem = networkItem(net->networkId());
 920   netItem->attachNetwork(net);
 921 }
 922
 923 void NetworkModel::bufferUpdated(BufferInfo bufferInfo) {
 924   BufferItem *bufItem = bufferItem(bufferInfo);
 925   QModelIndex itemindex = indexByItem(bufItem);
 926   emit dataChanged(itemindex, itemindex);
 927 }
 928
 929 void NetworkModel::removeBuffer(BufferId bufferId) {
 930   BufferItem *buffItem = findBufferItem(bufferId);
 931   if(!buffItem)
 932     return;
 933
 934   buffItem->parent()->removeChild(buffItem);
 935 }
 936
 937 MsgId NetworkModel::lastSeenMsgId(BufferId bufferId) const {
 938   if(!_bufferItemCache.contains(bufferId))
 939     return MsgId();
 940
 941   return _bufferItemCache[bufferId]->lastSeenMsgId();
 942 }
 943
 944 MsgId NetworkModel::lastSeenMarkerMsgId(BufferId bufferId) const {
 945   if(!_bufferItemCache.contains(bufferId))
 946     return MsgId();
 947
 948   return _bufferItemCache[bufferId]->lastSeenMarkerMsgId();
 949 }
 950
 951 void NetworkModel::setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId) {
 952   BufferItem *bufferItem = findBufferItem(bufferId);
 953   if(!bufferItem) {
 954     qDebug() << "NetworkModel::setLastSeenMsgId(): buffer is unknown:" << bufferId;
 955     return;
 956   }
 957   bufferItem->setLastSeenMsgId(msgId);
 958 }
 959
 960 void NetworkModel::updateBufferActivity(const Message &msg) {
 961   BufferItem *item = bufferItem(msg.bufferInfo());
 962   item->updateActivityLevel(msg);
 963   if(item->isCurrentBuffer())
 964     emit setLastSeenMsg(item->bufferId(), msg.msgId());
 965 }
 966
 967 void NetworkModel::setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel level) {
 968   BufferItem *bufferItem = findBufferItem(bufferId);
 969   if(!bufferItem) {
 970     qDebug() << "NetworkModel::setBufferActivity(): buffer is unknown:" << bufferId;
 971     return;
 972   }
 973   bufferItem->setActivityLevel(level);
 974 }
 975
 976 void NetworkModel::clearBufferActivity(const BufferId &bufferId) {
 977   BufferItem *bufferItem = findBufferItem(bufferId);
 978   if(!bufferItem) {
 979     qDebug() << "NetworkModel::clearBufferActivity(): buffer is unknown:" << bufferId;
 980     return;
 981   }
 982   bufferItem->clearActivityLevel();
 983 }
 984
 985 const Network *NetworkModel::networkByIndex(const QModelIndex &index) const {
 986   QVariant netVariant = index.data(NetworkIdRole);
 987   if(!netVariant.isValid())
 988     return 0;
 989
 990   NetworkId networkId = netVariant.value<NetworkId>();
 991   return Client::network(networkId);
 992 }
 993
 994 void NetworkModel::checkForRemovedBuffers(const QModelIndex &parent, int start, int end) {
 995   if(parent.data(ItemTypeRole) != NetworkItemType)
 996     return;
 997
 998   for(int row = start; row <= end; row++) {
 999     _bufferItemCache.remove(parent.child(row, 0).data(BufferIdRole).value<BufferId>());
1000   }
1001 }
1002
1003 void NetworkModel::checkForNewBuffers(const QModelIndex &parent, int start, int end) {
1004   if(parent.data(ItemTypeRole) != NetworkItemType)
1005     return;
1006
1007   for(int row = start; row <= end; row++) {
1008     QModelIndex child = parent.child(row, 0);
1009     _bufferItemCache[child.data(BufferIdRole).value<BufferId>()] = static_cast<BufferItem *>(child.internalPointer());
1010   }
1011 }
1012
1013 QString NetworkModel::bufferName(BufferId bufferId) const {
1014   if(!_bufferItemCache.contains(bufferId))
1015     return QString();
1016
1017   return _bufferItemCache[bufferId]->bufferName();
1018 }
1019
1020 BufferInfo::Type NetworkModel::bufferType(BufferId bufferId) const {
1021   if(!_bufferItemCache.contains(bufferId))
1022     return BufferInfo::InvalidBuffer;
1023
1024   return _bufferItemCache[bufferId]->bufferType();
1025 }
1026
1027 BufferInfo NetworkModel::bufferInfo(BufferId bufferId) const {
1028   if(!_bufferItemCache.contains(bufferId))
1029     return BufferInfo();
1030
1031   return _bufferItemCache[bufferId]->bufferInfo();
1032 }
1033
1034 NetworkId NetworkModel::networkId(BufferId bufferId) const {
1035   if(!_bufferItemCache.contains(bufferId))
1036     return NetworkId();
1037
1038   NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1039   if(netItem)
1040     return netItem->networkId();
1041   else
1042     return NetworkId();
1043 }
1044
1045 QString NetworkModel::networkName(BufferId bufferId) const {
1046   if(!_bufferItemCache.contains(bufferId))
1047     return QString();
1048
1049   NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1050   if(netItem)
1051     return netItem->networkName();
1052   else
1053     return QString();
1054 }
1055
1056 BufferId NetworkModel::bufferId(NetworkId networkId, const QString &bufferName, Qt::CaseSensitivity cs) const {
1057   const NetworkItem *netItem = findNetworkItem(networkId);
1058   if(!netItem)
1059     return BufferId();
1060
1061   for(int i = 0; i < netItem->childCount(); i++) {
1062     BufferItem *bufferItem = qobject_cast<BufferItem *>(netItem->child(i));
1063     if(bufferItem && !bufferItem->bufferName().compare(bufferName, cs))
1064       return bufferItem->bufferId();
1065   }
1066   return BufferId();
1067 }
1068
1069 void NetworkModel::sortBufferIds(QList<BufferId> &bufferIds) const {
1070   QList<BufferItem *> bufferItems;
1071   foreach(BufferId bufferId, bufferIds) {
1072     if(_bufferItemCache.contains(bufferId))
1073       bufferItems << _bufferItemCache[bufferId];
1074   }
1075
1076   qSort(bufferItems.begin(), bufferItems.end(), bufferItemLessThan);
1077
1078   bufferIds.clear();
1079   foreach(BufferItem *bufferItem, bufferItems) {
1080     bufferIds << bufferItem->bufferId();
1081   }
1082 }
1083
1084 QList<BufferId> NetworkModel::allBufferIdsSorted() const {
1085   QList<BufferId> bufferIds = allBufferIds();
1086   sortBufferIds(bufferIds);
1087   return bufferIds;
1088 }
1089
1090 bool NetworkModel::bufferItemLessThan(const BufferItem *left, const BufferItem *right) {
1091   int leftType = left->bufferType();
1092   int rightType = right->bufferType();
1093
1094   if(leftType != rightType)
1095     return leftType < rightType;
1096   else
1097     return QString::compare(left->bufferName(), right->bufferName(), Qt::CaseInsensitive) < 0;
1098 }
1099