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 QString BufferItem::toolTip(int column) const {
 286   Q_UNUSED(column);
 287   QStringList toolTip;
 288
 289   switch(bufferType()) {
 290     case BufferInfo::StatusBuffer: {
 291       QString netName = Client::network(bufferInfo().networkId())->networkName();
 292       toolTip.append(QString("<b>Status buffer from %1</b>").arg(netName));
 293       break;
 294     }
 295     case BufferInfo::ChannelBuffer:
 296       toolTip.append(QString("<b>Channel %1</b>").arg(bufferName()));
 297       if(isActive()) {
 298         toolTip.append(QString("Topic: %1").arg(topic()));
 299         toolTip.append(QString("Users: %1").arg(nickCount()));
 300       } else {
 301         toolTip.append(QString("Not active <br /> Double-click to join"));
 302       }
 303       break;
 304     case BufferInfo::QueryBuffer:
 305       toolTip.append(QString("<b>Query with %1</b>").arg(bufferName()));
 306       if(topic() != "") toolTip.append(QString("Away Message: %1").arg(topic()));
 307       break;
 308     default: //this should not happen
 309       toolTip.append(QString("%1 - %2").arg(bufferInfo().bufferId().toInt()).arg(bufferName()));
 310       break;
 311   }
 312
 313   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
 314 }
 315
 316 /*
 317 void BufferItem::setLastMsgInsert(QDateTime msgDate) {
 318   if(msgDate.isValid() && msgDate > _lastMsgInsert)
 319     _lastMsgInsert = msgDate;
 320 }
 321 */
 322 /*
 323 // FIXME emit dataChanged()
 324 bool BufferItem::setLastSeen() {
 325   if(_lastSeen > _lastMsgInsert)
 326     return false;
 327
 328   _lastSeen = _lastMsgInsert;
 329   BufferSettings(bufferInfo().bufferId()).setLastSeen(_lastSeen);
 330   return true;
 331 }
 332
 333 QDateTime BufferItem::lastSeen() {
 334   return _lastSeen;
 335 }
 336 */
 337 /*****************************************
 338 *  Network Items
 339 *****************************************/
 340 NetworkItem::NetworkItem(const NetworkId &netid, AbstractTreeItem *parent)
 341   : PropertyMapItem(QList<QString>() << "networkName" << "currentServer" << "nickCount", parent),
 342     _networkId(netid)
 343 {
 344   setFlags(Qt::ItemIsEnabled);
 345 }
 346
 347 QVariant NetworkItem::data(int column, int role) const {
 348   switch(role) {
 349   case NetworkModel::NetworkIdRole:
 350     return qVariantFromValue(_networkId);
 351   case NetworkModel::ItemTypeRole:
 352     return NetworkModel::NetworkItemType;
 353   case NetworkModel::ItemActiveRole:
 354     return isActive();
 355   default:
 356     return PropertyMapItem::data(column, role);
 357   }
 358 }
 359
 360 quint64 NetworkItem::id() const {
 361   return qHash(_networkId);
 362 }
 363
 364 bool NetworkItem::isActive() const {
 365   if(_network)
 366     return _network->isConnected();
 367   else
 368     return false;
 369 }
 370
 371 QString NetworkItem::networkName() const {
 372   if(_network)
 373     return _network->networkName();
 374   else
 375     return QString();
 376 }
 377
 378 QString NetworkItem::currentServer() const {
 379   if(_network)
 380     return _network->currentServer();
 381   else
 382     return QString();
 383 }
 384
 385 int NetworkItem::nickCount() const {
 386   if(_network)
 387     return _network->ircUsers().count();
 388   else
 389     return 0;
 390 }
 391
 392 void NetworkItem::attachNetwork(Network *network) {
 393   if(!network)
 394     return;
 395
 396   _network = network;
 397
 398   connect(network, SIGNAL(networkNameSet(QString)),
 399           this, SLOT(setNetworkName(QString)));
 400   connect(network, SIGNAL(currentServerSet(QString)),
 401           this, SLOT(setCurrentServer(QString)));
 402   connect(network, SIGNAL(ircChannelAdded(QString)),
 403           this, SLOT(attachIrcChannel(QString)));
 404   connect(network, SIGNAL(connectedSet(bool)),
 405           this, SIGNAL(dataChanged()));
 406
 407   // FIXME: connect this and that...
 408
 409   emit dataChanged();
 410 }
 411
 412 void NetworkItem::attachIrcChannel(const QString &channelName) {
 413   IrcChannel *ircChannel = _network->ircChannel(channelName);
 414   if(!ircChannel) {
 415     qWarning() << "NetworkItem::attachIrcChannel(): unkown Channel" << channelName;
 416     return;
 417   }
 418
 419   BufferItem *bufferItem;
 420   for(int i = 0; i < childCount(); i++) {
 421     bufferItem = qobject_cast<BufferItem *>(child(i));
 422     if(bufferItem->bufferName().toLower() == ircChannel->name().toLower()) {
 423       bufferItem->attachIrcChannel(ircChannel);
 424       break;
 425     }
 426   }
 427 }
 428
 429 void NetworkItem::setNetworkName(const QString &networkName) {
 430   Q_UNUSED(networkName);
 431   emit dataChanged(0);
 432 }
 433
 434 void NetworkItem::setCurrentServer(const QString &serverName) {
 435   Q_UNUSED(serverName);
 436   emit dataChanged(1);
 437 }
 438
 439
 440 QString NetworkItem::toolTip(int column) const {
 441   Q_UNUSED(column);
 442
 443   QStringList toolTip(QString("<b>%1</b>").arg(networkName()));
 444   toolTip.append(QString("Server: %1").arg(currentServer()));
 445   toolTip.append(QString("Users: %1").arg(nickCount()));
 446
 447   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
 448 }
 449
 450
 451 /*****************************************
 452 *  User Category Items (like @vh etc.)
 453 *****************************************/
 454 // we hardcode this even though we have PREFIX in network... but that wouldn't help with mapping modes to
 455 // category strings anyway.
 456 // TODO make this translateable depending on the number of users in a category
 457 //      -> we can't set the real string here, because tr() needs to get the actual number as second param
 458 //      -> tr("%n User(s)", n) needs to be used somewhere where we do know the user number n
 459 const QList<UserCategoryItem::Category> UserCategoryItem::categories = QList<UserCategoryItem::Category>()
 460   << UserCategoryItem::Category('q', tr("Owners"))
 461   << UserCategoryItem::Category('a', tr("Admins"))
 462   << UserCategoryItem::Category('o', tr("Operators"))
 463   << UserCategoryItem::Category('h', tr("Half-Ops"))
 464   << UserCategoryItem::Category('v', tr("Voiced"));
 465
 466 UserCategoryItem::UserCategoryItem(int category, AbstractTreeItem *parent)
 467   : PropertyMapItem(QStringList() << "categoryId", parent),
 468     _category(category)
 469 {
 470 }
 471
 472 QString UserCategoryItem::categoryId() {
 473   if(_category < categories.count())
 474     return categories[_category].displayString;
 475   else
 476     return tr("Users");
 477 }
 478
 479 quint64 UserCategoryItem::id() const {
 480   return qHash(_category);
 481 }
 482
 483 void UserCategoryItem::addUser(IrcUser *ircUser) {
 484   newChild(new IrcUserItem(ircUser, this));
 485 }
 486
 487 bool UserCategoryItem::removeUser(IrcUser *ircUser) {
 488   return removeChildById(qHash(ircUser));
 489 }
 490
 491 int UserCategoryItem::categoryFromModes(const QString &modes) {
 492   for(int i = 0; i < categories.count(); i++) {
 493     if(modes.contains(categories[i].mode))
 494       return i;
 495   }
 496   return categories.count();
 497 }
 498
 499 QVariant UserCategoryItem::data(int column, int role) const {
 500   switch(role) {
 501   case NetworkModel::ItemActiveRole:
 502     return true;
 503   case NetworkModel::ItemTypeRole:
 504     return NetworkModel::UserCategoryItemType;
 505   case NetworkModel::BufferIdRole:
 506     return parent()->data(column, role);
 507   case NetworkModel::NetworkIdRole:
 508     return parent()->data(column, role);
 509   case NetworkModel::BufferInfoRole:
 510     return parent()->data(column, role);
 511   default:
 512     return PropertyMapItem::data(column, role);
 513   }
 514 }
 515
 516
 517 /*****************************************
 518 *  Irc User Items
 519 *****************************************/
 520 IrcUserItem::IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent)
 521   : PropertyMapItem(QStringList() << "nickName", parent),
 522     _ircUser(ircUser),
 523     _id(qHash(ircUser))
 524 {
 525   // we don't need to handle the ircUser's destroyed signal since it's automatically removed
 526   // by the IrcChannel::ircUserParted();
 527
 528   connect(ircUser, SIGNAL(nickSet(QString)),
 529           this, SLOT(setNick(QString)));
 530   connect(ircUser, SIGNAL(awaySet(bool)),
 531           this, SLOT(setAway(bool)));
 532 }
 533
 534 QString IrcUserItem::nickName() const {
 535   if(_ircUser)
 536     return _ircUser->nick();
 537   else
 538     return QString();
 539 }
 540
 541 bool IrcUserItem::isActive() const {
 542   if(_ircUser)
 543     return !_ircUser->isAway();
 544   else
 545     return false;
 546 }
 547
 548 IrcUser *IrcUserItem::ircUser() {
 549   return _ircUser;
 550 }
 551
 552 quint64 IrcUserItem::id() const {
 553   return _id;
 554 }
 555
 556 QVariant IrcUserItem::data(int column, int role) const {
 557   switch(role) {
 558   case NetworkModel::ItemActiveRole:
 559     return isActive();
 560   case NetworkModel::ItemTypeRole:
 561     return NetworkModel::IrcUserItemType;
 562   case NetworkModel::BufferIdRole:
 563     return parent()->data(column, role);
 564   case NetworkModel::NetworkIdRole:
 565     return parent()->data(column, role);
 566   case NetworkModel::BufferInfoRole:
 567     return parent()->data(column, role);
 568   default:
 569     return PropertyMapItem::data(column, role);
 570   }
 571 }
 572
 573 QString IrcUserItem::toolTip(int column) const {
 574   Q_UNUSED(column);
 575   QStringList toolTip(QString("<b>%1</b>").arg(nickName()));
 576   if(_ircUser->isAway()) toolTip[0].append(" is away");
 577   if(!_ircUser->awayMessage().isEmpty()) toolTip[0].append(QString(" (%1)").arg(_ircUser->awayMessage()));
 578   if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
 579   if(!_ircUser->ircOperator().isEmpty()) toolTip.append(_ircUser->ircOperator());
 580   toolTip.append(_ircUser->hostmask());
 581
 582   if(_ircUser->idleTime().isValid()) {
 583     QDateTime now = QDateTime::currentDateTime();
 584     QDateTime idle = _ircUser->idleTime();
 585     int idleTime = idle.secsTo(now);
 586
 587     QList< QPair<int, QString> > timeUnit;
 588     timeUnit.append(qMakePair(365*60*60, tr("year")));
 589     timeUnit.append(qMakePair(24*60*60, tr("day")));
 590     timeUnit.append(qMakePair(60*60, tr("h")));
 591     timeUnit.append(qMakePair(60, tr("min")));
 592     timeUnit.append(qMakePair(1, tr("sec")));
 593
 594     QString idleString(' ');
 595     for(int i=0; i < timeUnit.size(); i++) {
 596       int n = idleTime / timeUnit[i].first;
 597       if(n > 0) {
 598         idleString += QString("%1 %2 ").arg(QString::number(n), timeUnit[i].second);
 599       }
 600       idleTime = idleTime % timeUnit[i].first;
 601     }
 602     toolTip.append(tr("idling since %1").arg(idleString));
 603   }
 604
 605   if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
 606
 607   return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
 608 }
 609
 610 void IrcUserItem::setNick(QString newNick) {
 611   Q_UNUSED(newNick);
 612   emit dataChanged(0);
 613 }
 614
 615 void IrcUserItem::setAway(bool away) {
 616   Q_UNUSED(away);
 617   emit dataChanged(0);
 618 }
 619
 620 /*****************************************
 621  * NetworkModel
 622  *****************************************/
 623 NetworkModel::NetworkModel(QObject *parent)
 624   : TreeModel(NetworkModel::defaultHeader(), parent)
 625 {
 626 }
 627
 628 QList<QVariant >NetworkModel::defaultHeader() {
 629   QList<QVariant> data;
 630   data << tr("Buffer") << tr("Topic") << tr("Nick Count");
 631   return data;
 632 }
 633
 634 bool NetworkModel::isBufferIndex(const QModelIndex &index) const {
 635   return index.data(NetworkModel::ItemTypeRole) == NetworkModel::BufferItemType;
 636 }
 637
 638 /*
 639 Buffer *NetworkModel::getBufferByIndex(const QModelIndex &index) const {
 640   BufferItem *item = static_cast<BufferItem *>(index.internalPointer());
 641   return Client::instance()->buffer(item->id());
 642 }
 643 */
 644
 645
 646 // experimental stuff :)
 647 QModelIndex NetworkModel::networkIndex(NetworkId networkId) {
 648   return indexById(qHash(networkId));
 649 }
 650
 651 NetworkItem *NetworkModel::existsNetworkItem(NetworkId networkId) {
 652   return qobject_cast<NetworkItem *>(rootItem->childById(networkId.toInt()));
 653 }
 654
 655 NetworkItem *NetworkModel::networkItem(NetworkId networkId) {
 656   NetworkItem *netItem = existsNetworkItem(networkId);
 657
 658   if(netItem == 0) {
 659     netItem = new NetworkItem(networkId, rootItem);
 660     rootItem->newChild(netItem);
 661   }
 662
 663   Q_ASSERT(netItem);
 664   return netItem;
 665 }
 666
 667 void NetworkModel::networkRemoved(const NetworkId &networkId) {
 668   rootItem->removeChildById(qHash(networkId));
 669 }
 670
 671 QModelIndex NetworkModel::bufferIndex(BufferId bufferId) {
 672   AbstractTreeItem *netItem, *bufferItem;
 673   for(int i = 0; i < rootItem->childCount(); i++) {
 674     netItem = rootItem->child(i);
 675     if((bufferItem = netItem->childById(qHash(bufferId)))) {
 676       return indexByItem(bufferItem);
 677     }
 678   }
 679   return QModelIndex();
 680 }
 681
 682 BufferItem *NetworkModel::existsBufferItem(const BufferInfo &bufferInfo) {
 683   QModelIndex bufferIdx = bufferIndex(bufferInfo.bufferId());
 684   if(bufferIdx.isValid())
 685     return static_cast<BufferItem *>(bufferIdx.internalPointer());
 686   else
 687     return 0;
 688 }
 689
 690 BufferItem *NetworkModel::bufferItem(const BufferInfo &bufferInfo) {
 691   BufferItem *bufItem = existsBufferItem(bufferInfo);
 692   if(bufItem == 0) {
 693     NetworkItem *netItem = networkItem(bufferInfo.networkId());
 694     bufItem = new BufferItem(bufferInfo, netItem);
 695     netItem->newChild(bufItem);
 696   }
 697
 698   Q_ASSERT(bufItem);
 699   return bufItem;
 700 }
 701
 702 QStringList NetworkModel::mimeTypes() const {
 703   // mimetypes we accept for drops
 704   QStringList types;
 705   // comma separated list of colon separated pairs of networkid and bufferid
 706   // example: 0:1,0:2,1:4
 707   types << "application/Quassel/BufferItemList";
 708   return types;
 709 }
 710
 711 bool NetworkModel::mimeContainsBufferList(const QMimeData *mimeData) {
 712   return mimeData->hasFormat("application/Quassel/BufferItemList");
 713 }
 714
 715 QList< QPair<NetworkId, BufferId> > NetworkModel::mimeDataToBufferList(const QMimeData *mimeData) {
 716   QList< QPair<NetworkId, BufferId> > bufferList;
 717
 718   if(!mimeContainsBufferList(mimeData))
 719     return bufferList;
 720
 721   QStringList rawBufferList = QString::fromAscii(mimeData->data("application/Quassel/BufferItemList")).split(",");
 722   NetworkId networkId;
 723   BufferId bufferUid;
 724   foreach(QString rawBuffer, rawBufferList) {
 725     if(!rawBuffer.contains(":"))
 726       continue;
 727     networkId = rawBuffer.section(":", 0, 0).toInt();
 728     bufferUid = rawBuffer.section(":", 1, 1).toInt();
 729     bufferList.append(qMakePair(networkId, bufferUid));
 730   }
 731   return bufferList;
 732 }
 733
 734
 735 QMimeData *NetworkModel::mimeData(const QModelIndexList &indexes) const {
 736   QMimeData *mimeData = new QMimeData();
 737
 738   QStringList bufferlist;
 739   QString netid, uid, bufferid;
 740   foreach(QModelIndex index, indexes) {
 741     netid = QString::number(index.data(NetworkIdRole).value<NetworkId>().toInt());
 742     uid = QString::number(index.data(BufferIdRole).value<BufferId>().toInt());
 743     bufferid = QString("%1:%2").arg(netid).arg(uid);
 744     if(!bufferlist.contains(bufferid))
 745       bufferlist << bufferid;
 746   }
 747
 748   mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toAscii());
 749
 750   return mimeData;
 751 }
 752
 753 bool NetworkModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) {
 754   Q_UNUSED(action)
 755   Q_UNUSED(row)
 756   Q_UNUSED(column)
 757
 758   if(!mimeContainsBufferList(data))
 759     return false;
 760
 761   // target must be a query
 762   BufferInfo::Type targetType = (BufferInfo::Type)parent.data(NetworkModel::BufferTypeRole).toInt();
 763   if(targetType != BufferInfo::QueryBuffer)
 764     return false;
 765
 766   QList< QPair<NetworkId, BufferId> > bufferList = mimeDataToBufferList(data);
 767
 768   // exactly one buffer has to be dropped
 769   if(bufferList.count() != 1)
 770     return false;
 771
 772   NetworkId netId = bufferList.first().first;
 773   BufferId bufferId = bufferList.first().second;
 774
 775   // no self merges (would kill us)
 776   if(bufferId == parent.data(BufferIdRole).value<BufferId>())
 777     return false;
 778
 779   Q_ASSERT(rootItem->childById(qHash(netId)));
 780   Q_ASSERT(rootItem->childById(qHash(netId))->childById(qHash(bufferId)));
 781
 782   // source must be a query too
 783   BufferInfo::Type sourceType = (BufferInfo::Type)rootItem->childById(qHash(netId))->childById(qHash(bufferId))->data(0, BufferTypeRole).toInt();
 784   if(sourceType != BufferInfo::QueryBuffer)
 785     return false;
 786
 787   // TODO: warn user about buffermerge!
 788   qDebug() << "merging" << bufferId << parent.data(BufferIdRole).value<BufferId>();
 789   removeRow(parent.row(), parent.parent());
 790
 791   return true;
 792 }
 793
 794 void NetworkModel::attachNetwork(Network *net) {
 795   NetworkItem *netItem = networkItem(net->networkId());
 796   netItem->attachNetwork(net);
 797 }
 798
 799 void NetworkModel::bufferUpdated(BufferInfo bufferInfo) {
 800   BufferItem *bufItem = bufferItem(bufferInfo);
 801   QModelIndex itemindex = indexByItem(bufItem);
 802   emit dataChanged(itemindex, itemindex);
 803 }
 804
 805 void NetworkModel::removeBuffer(BufferId bufferId) {
 806   const int numNetworks = rootItem->childCount();
 807   if(numNetworks == 0)
 808     return;
 809
 810   for(int i = 0; i < numNetworks; i++) {
 811     if(rootItem->child(i)->removeChildById(qHash(bufferId)))
 812       break;
 813   }
 814 }
 815
 816 /*
 817 void NetworkModel::updateBufferActivity(const Message &msg) {
 818   BufferItem *buff = bufferItem(msg.bufferInfo());
 819   Q_ASSERT(buff);
 820
 821   buff->setLastMsgInsert(msg.timestamp());
 822
 823   if(buff->lastSeen() >= msg.timestamp())
 824     return;
 825
 826   BufferItem::ActivityLevel level = BufferItem::OtherActivity;
 827   if(msg.type() == Message::Plain || msg.type() == Message::Notice)
 828     level |= BufferItem::NewMessage;
 829
 830   if(msg.flags() & Message::Highlight)
 831       level |= BufferItem::Highlight;
 832
 833   bufferItem(msg.bufferInfo())->updateActivity(level);
 834 }
 835 */
 836
 837 void NetworkModel::setBufferActivity(const BufferInfo &info, Buffer::ActivityLevel level) {
 838   BufferItem *buff = bufferItem(info);
 839   Q_ASSERT(buff);
 840
 841   buff->setActivityLevel(level);
 842 }
 843
 844 const Network *NetworkModel::networkByIndex(const QModelIndex &index) const {
 845   QVariant netVariant = index.data(NetworkIdRole);
 846   if(!netVariant.isValid())
 847     return 0;
 848
 849   NetworkId networkId = netVariant.value<NetworkId>();
 850   return Client::network(networkId);
 851 }
 852