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