src/client/networkmodel.cpp

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