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