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