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