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