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