1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 ***************************************************************************/
21 #include "networkmodel.h"
23 #include <QAbstractItemView>
25 #include "buffermodel.h"
27 #include "signalproxy.h"
29 #include "ircchannel.h"
31 #include "buffersettings.h"
33 #include "util.h" // get rid of this (needed for isChannelName)
35 /*****************************************
37 *****************************************/
38 NetworkItem::NetworkItem(const NetworkId &netid, AbstractTreeItem *parent)
39 : PropertyMapItem(QList<QString>() << "networkName" << "currentServer" << "nickCount", parent),
42 setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
45 QVariant NetworkItem::data(int column, int role) const {
47 case NetworkModel::BufferIdRole:
49 return child(0)->data(column, role);
52 case NetworkModel::NetworkIdRole:
53 return qVariantFromValue(_networkId);
54 case NetworkModel::ItemTypeRole:
55 return NetworkModel::NetworkItemType;
56 case NetworkModel::ItemActiveRole:
59 return PropertyMapItem::data(column, role);
63 BufferItem *NetworkItem::findBufferItem(BufferId bufferId) {
64 BufferItem *bufferItem = 0;
66 for(int i = 0; i < childCount(); i++) {
67 bufferItem = qobject_cast<BufferItem *>(child(i));
70 if(bufferItem->bufferId() == bufferId)
77 BufferItem *NetworkItem::bufferItem(const BufferInfo &bufferInfo) {
78 BufferItem *bufferItem = findBufferItem(bufferInfo);
82 switch(bufferInfo.type()) {
83 case BufferInfo::StatusBuffer:
84 bufferItem = new StatusBufferItem(bufferInfo, this);
86 case BufferInfo::ChannelBuffer:
87 bufferItem = new ChannelBufferItem(bufferInfo, this);
89 case BufferInfo::QueryBuffer:
90 bufferItem = new QueryBufferItem(bufferInfo, this);
93 bufferItem = new BufferItem(bufferInfo, this);
100 void NetworkItem::attachNetwork(Network *network) {
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()));
122 void NetworkItem::attachIrcChannel(IrcChannel *ircChannel) {
123 ChannelBufferItem *channelItem;
124 for(int i = 0; i < childCount(); i++) {
125 channelItem = qobject_cast<ChannelBufferItem *>(child(i));
129 if(channelItem->bufferName().toLower() == ircChannel->name().toLower()) {
130 channelItem->attachIrcChannel(ircChannel);
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));
143 if(queryItem->bufferName().toLower() == ircUser->nick().toLower()) {
144 queryItem->attachIrcUser(ircUser);
150 void NetworkItem::setNetworkName(const QString &networkName) {
151 Q_UNUSED(networkName);
155 void NetworkItem::setCurrentServer(const QString &serverName) {
156 Q_UNUSED(serverName);
161 QString NetworkItem::toolTip(int column) const {
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()));
169 toolTip.append(tr("Lag: %1 msecs").arg(_network->latency()));
172 return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
175 /*****************************************
177 *****************************************/
178 BufferItem::BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
179 : PropertyMapItem(QStringList() << "bufferName" << "topic" << "nickCount", parent),
180 _bufferInfo(bufferInfo),
181 _activity(BufferInfo::NoActivity)
183 setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled);
186 void BufferItem::setActivityLevel(BufferInfo::ActivityLevel level) {
187 if(_activity != level) {
193 void BufferItem::clearActivityLevel() {
194 _activity = BufferInfo::NoActivity;
195 _lastSeenMarkerMsgId = _lastSeenMsgId;
199 void BufferItem::updateActivityLevel(const Message &msg) {
200 if(isCurrentBuffer()) {
204 if(msg.flags() & Message::Self) // don't update activity for our own messages
207 if(lastSeenMsgId() >= msg.msgId())
210 BufferInfo::ActivityLevel oldLevel = activityLevel();
212 _activity |= BufferInfo::OtherActivity;
213 if(msg.type() & (Message::Plain | Message::Notice | Message::Action))
214 _activity |= BufferInfo::NewMessage;
216 if(msg.flags() & Message::Highlight)
217 _activity |= BufferInfo::Highlight;
219 if(oldLevel != _activity)
223 QVariant BufferItem::data(int column, int role) const {
225 case NetworkModel::ItemTypeRole:
226 return NetworkModel::BufferItemType;
227 case NetworkModel::BufferIdRole:
228 return qVariantFromValue(bufferInfo().bufferId());
229 case NetworkModel::NetworkIdRole:
230 return qVariantFromValue(bufferInfo().networkId());
231 case NetworkModel::BufferInfoRole:
232 return qVariantFromValue(bufferInfo());
233 case NetworkModel::BufferTypeRole:
234 return int(bufferType());
235 case NetworkModel::ItemActiveRole:
237 case NetworkModel::BufferActivityRole:
238 return (int)activityLevel();
240 return PropertyMapItem::data(column, role);
244 bool BufferItem::setData(int column, const QVariant &value, int role) {
245 qDebug() << "BufferItem::setData(int column, const QVariant &value, int role):" << this << column << value << role;
247 case NetworkModel::BufferActivityRole:
248 setActivityLevel((BufferInfo::ActivityLevel)value.toInt());
251 return PropertyMapItem::setData(column, value, role);
256 void BufferItem::setBufferName(const QString &name) {
257 _bufferInfo = BufferInfo(_bufferInfo.bufferId(), _bufferInfo.networkId(), _bufferInfo.type(), _bufferInfo.groupId(), name);
261 void BufferItem::setLastSeenMsgId(const MsgId &msgId) {
262 _lastSeenMsgId = msgId;
263 if(!isCurrentBuffer()) {
264 _lastSeenMarkerMsgId = msgId;
266 setActivityLevel(BufferInfo::NoActivity);
269 bool BufferItem::isCurrentBuffer() const {
270 return _bufferInfo.bufferId() == Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
273 QString BufferItem::toolTip(int column) const {
275 return tr("<p> %1 - %2 </p>").arg(bufferInfo().bufferId().toInt()).arg(bufferName());
278 /*****************************************
280 *****************************************/
281 StatusBufferItem::StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
282 : BufferItem(bufferInfo, parent)
285 connect(parent, SIGNAL(dataChanged()), this, SIGNAL(dataChanged()));
288 QString StatusBufferItem::toolTip(int column) const {
292 QString netName = Client::network(bufferInfo().networkId())->networkName();
293 toolTip.append(tr("<b>Status buffer of %1</b>").arg(netName));
295 return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
298 /*****************************************
300 *****************************************/
301 QueryBufferItem::QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
302 : BufferItem(bufferInfo, parent),
305 setFlags(flags() | Qt::ItemIsDropEnabled);
307 const Network *net = Client::network(bufferInfo.networkId());
311 IrcUser *ircUser = net->ircUser(bufferInfo.bufferName());
313 attachIrcUser(ircUser);
316 QVariant QueryBufferItem::data(int column, int role) const {
318 case NetworkModel::UserAwayRole:
319 return (bool)_ircUser ? _ircUser->isAway() : false;
321 return BufferItem::data(column, role);
325 QString QueryBufferItem::toolTip(int column) const {
326 // pretty much code duplication of IrcUserItem::toolTip() but inheritance won't solve this...
330 toolTip.append(tr("<b>Query with %1</b>").arg(bufferName()));
333 if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
334 if(_ircUser->isAway()) {
335 toolTip[0].append(QString(" (away%1)").arg(!_ircUser->awayMessage().isEmpty() ? (QString(" ") + _ircUser->awayMessage()) : QString()));
337 if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
338 if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(_ircUser->nick()).arg(_ircUser->ircOperator()));
339 if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
340 if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
342 toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
344 if(_ircUser->idleTime().isValid()) {
345 QDateTime now = QDateTime::currentDateTime();
346 QDateTime idle = _ircUser->idleTime();
347 int idleTime = idle.secsTo(now);
348 toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
350 if(_ircUser->loginTime().isValid()) {
351 toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
354 if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
357 return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
360 void QueryBufferItem::attachIrcUser(IrcUser *ircUser) {
362 connect(_ircUser, SIGNAL(destroyed()), this, SLOT(ircUserDestroyed()));
363 connect(_ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
367 void QueryBufferItem::ircUserDestroyed() {
372 /*****************************************
374 *****************************************/
375 ChannelBufferItem::ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
376 : BufferItem(bufferInfo, parent),
379 const Network *net = Client::network(bufferInfo.networkId());
383 IrcChannel *ircChannel = net->ircChannel(bufferInfo.bufferName());
385 attachIrcChannel(ircChannel);
388 QString ChannelBufferItem::toolTip(int column) const {
392 toolTip.append(tr("<b>Channel %1</b>").arg(bufferName()));
394 //TODO: add channel modes
395 toolTip.append(tr("<b>Users:</b> %1").arg(nickCount()));
397 QString channelMode = _ircChannel->channelModeString(); // channelModeString is compiled on the fly -> thus cache the result
398 if(!channelMode.isEmpty())
399 toolTip.append(tr("<b>Mode:</b> %1").arg(channelMode));
403 bool showTopic = s.value("DisplayTopicInTooltip", QVariant(false)).toBool();
405 QString _topic = topic();
407 _topic = stripFormatCodes(_topic);
408 _topic.replace(QString("<"), QString("<"));
409 _topic.replace(QString(">"), QString(">"));
410 toolTip.append(QString("<font size='-2'> </font>"));
411 toolTip.append(tr("<b>Topic:</b> %1").arg(_topic));
415 toolTip.append(tr("Not active <br /> Double-click to join"));
418 return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
421 void ChannelBufferItem::attachIrcChannel(IrcChannel *ircChannel) {
422 Q_ASSERT(!_ircChannel && ircChannel);
424 _ircChannel = ircChannel;
426 connect(ircChannel, SIGNAL(topicSet(QString)),
427 this, SLOT(setTopic(QString)));
428 connect(ircChannel, SIGNAL(ircUsersJoined(QList<IrcUser *>)),
429 this, SLOT(join(QList<IrcUser *>)));
430 connect(ircChannel, SIGNAL(ircUserParted(IrcUser *)),
431 this, SLOT(part(IrcUser *)));
432 connect(ircChannel, SIGNAL(destroyed()),
433 this, SLOT(ircChannelDestroyed()));
434 connect(ircChannel, SIGNAL(ircUserModesSet(IrcUser *, QString)),
435 this, SLOT(userModeChanged(IrcUser *)));
436 connect(ircChannel, SIGNAL(ircUserModeAdded(IrcUser *, QString)),
437 this, SLOT(userModeChanged(IrcUser *)));
438 connect(ircChannel, SIGNAL(ircUserModeRemoved(IrcUser *, QString)),
439 this, SLOT(userModeChanged(IrcUser *)));
441 if(!ircChannel->ircUsers().isEmpty())
442 join(ircChannel->ircUsers());
447 void ChannelBufferItem::ircChannelDestroyed() {
448 Q_CHECK_PTR(_ircChannel);
449 disconnect(_ircChannel, 0, this, 0);
455 void ChannelBufferItem::join(const QList<IrcUser *> &ircUsers) {
456 addUsersToCategory(ircUsers);
460 UserCategoryItem *ChannelBufferItem::findCategoryItem(int categoryId) {
461 UserCategoryItem *categoryItem = 0;
463 for(int i = 0; i < childCount(); i++) {
464 categoryItem = qobject_cast<UserCategoryItem *>(child(i));
467 if(categoryItem->categoryId() == categoryId)
473 void ChannelBufferItem::addUserToCategory(IrcUser *ircUser) {
474 addUsersToCategory(QList<IrcUser *>() << ircUser);
477 void ChannelBufferItem::addUsersToCategory(const QList<IrcUser *> &ircUsers) {
478 Q_ASSERT(_ircChannel);
480 QHash<UserCategoryItem *, QList<IrcUser *> > categories;
483 UserCategoryItem *categoryItem = 0;
485 foreach(IrcUser *ircUser, ircUsers) {
486 categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
487 categoryItem = findCategoryItem(categoryId);
489 categoryItem = new UserCategoryItem(categoryId, this);
490 categories[categoryItem] = QList<IrcUser *>();
491 newChild(categoryItem);
493 categories[categoryItem] << ircUser;
496 QHash<UserCategoryItem *, QList<IrcUser *> >::const_iterator catIter = categories.constBegin();
497 while(catIter != categories.constEnd()) {
498 catIter.key()->addUsers(catIter.value());
503 void ChannelBufferItem::part(IrcUser *ircUser) {
505 qWarning() << bufferName() << "ChannelBufferItem::part(): unknown User" << ircUser;
509 disconnect(ircUser, 0, this, 0);
510 removeUserFromCategory(ircUser);
514 void ChannelBufferItem::removeUserFromCategory(IrcUser *ircUser) {
516 // If we parted the channel there might still be some ircUsers connected.
517 // in that case we just ignore the call
518 Q_ASSERT(childCount() == 0);
522 UserCategoryItem *categoryItem = 0;
523 for(int i = 0; i < childCount(); i++) {
524 categoryItem = qobject_cast<UserCategoryItem *>(child(i));
525 if(categoryItem->removeUser(ircUser)) {
526 if(categoryItem->childCount() == 0)
533 void ChannelBufferItem::userModeChanged(IrcUser *ircUser) {
534 Q_ASSERT(_ircChannel);
536 int categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
537 UserCategoryItem *categoryItem = findCategoryItem(categoryId);
540 if(categoryItem->findIrcUser(ircUser)) {
541 return; // already in the right category;
544 categoryItem = new UserCategoryItem(categoryId, this);
545 newChild(categoryItem);
548 // find the item that needs reparenting
549 IrcUserItem *ircUserItem = 0;
550 for(int i = 0; i < childCount(); i++) {
551 UserCategoryItem *oldCategoryItem = qobject_cast<UserCategoryItem *>(child(i));
552 Q_ASSERT(oldCategoryItem);
553 IrcUserItem *userItem = oldCategoryItem->findIrcUser(ircUser);
555 ircUserItem = userItem;
561 qWarning() << "ChannelBufferItem::userModeChanged(IrcUser *): unable to determine old category of" << ircUser;
564 ircUserItem->reParent(categoryItem);
567 /*****************************************
568 * User Category Items (like @vh etc.)
569 *****************************************/
570 // we hardcode this even though we have PREFIX in network... but that wouldn't help with mapping modes to
571 // category strings anyway.
572 const QList<QChar> UserCategoryItem::categories = QList<QChar>() << 'q' << 'a' << 'o' << 'h' << 'v';
574 UserCategoryItem::UserCategoryItem(int category, AbstractTreeItem *parent)
575 : PropertyMapItem(QStringList() << "categoryName", parent),
578 setTreeItemFlags(AbstractTreeItem::DeleteOnLastChildRemoved);
579 setObjectName(parent->data(0, Qt::DisplayRole).toString() + "/" + QString::number(category));
582 // caching this makes no sense, since we display the user number dynamically
583 QString UserCategoryItem::categoryName() const {
584 int n = childCount();
586 case 0: return tr("%n Owner(s)", 0, n);
587 case 1: return tr("%n Admin(s)", 0, n);
588 case 2: return tr("%n Operator(s)", 0, n);
589 case 3: return tr("%n Half-Op(s)", 0, n);
590 case 4: return tr("%n Voiced", 0, n);
591 default: return tr("%n User(s)", 0, n);
595 IrcUserItem *UserCategoryItem::findIrcUser(IrcUser *ircUser) {
596 IrcUserItem *userItem = 0;
598 for(int i = 0; i < childCount(); i++) {
599 userItem = qobject_cast<IrcUserItem *>(child(i));
602 if(userItem->ircUser() == ircUser)
608 void UserCategoryItem::addUsers(const QList<IrcUser *> &ircUsers) {
609 QList<AbstractTreeItem *> userItems;
610 foreach(IrcUser *ircUser, ircUsers)
611 userItems << new IrcUserItem(ircUser, this);
612 newChilds(userItems);
616 bool UserCategoryItem::removeUser(IrcUser *ircUser) {
617 IrcUserItem *userItem = findIrcUser(ircUser);
618 bool success = (bool)userItem;
620 removeChild(userItem);
626 int UserCategoryItem::categoryFromModes(const QString &modes) {
627 for(int i = 0; i < categories.count(); i++) {
628 if(modes.contains(categories[i]))
631 return categories.count();
634 QVariant UserCategoryItem::data(int column, int role) const {
636 case TreeModel::SortRole:
638 case NetworkModel::ItemActiveRole:
640 case NetworkModel::ItemTypeRole:
641 return NetworkModel::UserCategoryItemType;
642 case NetworkModel::BufferIdRole:
643 return parent()->data(column, role);
644 case NetworkModel::NetworkIdRole:
645 return parent()->data(column, role);
646 case NetworkModel::BufferInfoRole:
647 return parent()->data(column, role);
649 return PropertyMapItem::data(column, role);
654 /*****************************************
656 *****************************************/
657 IrcUserItem::IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent)
658 : PropertyMapItem(QStringList() << "nickName", parent),
661 setObjectName(ircUser->nick());
662 connect(ircUser, SIGNAL(destroyed()), this, SLOT(ircUserDestroyed()));
663 connect(ircUser, SIGNAL(nickSet(QString)), this, SIGNAL(dataChanged()));
664 connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
667 QVariant IrcUserItem::data(int column, int role) const {
669 case NetworkModel::ItemActiveRole:
671 case NetworkModel::ItemTypeRole:
672 return NetworkModel::IrcUserItemType;
673 case NetworkModel::BufferIdRole:
674 return parent()->data(column, role);
675 case NetworkModel::NetworkIdRole:
676 return parent()->data(column, role);
677 case NetworkModel::BufferInfoRole:
678 return parent()->data(column, role);
680 return PropertyMapItem::data(column, role);
684 QString IrcUserItem::toolTip(int column) const {
686 QStringList toolTip(QString("<b>%1</b>").arg(nickName()));
687 if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
688 if(_ircUser->isAway()) {
689 toolTip[0].append(" is away");
690 if(!_ircUser->awayMessage().isEmpty())
691 toolTip[0].append(QString(" (%1)").arg(_ircUser->awayMessage()));
693 if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
694 if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(nickName()).arg(_ircUser->ircOperator()));
695 if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
696 if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
698 toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
700 if(_ircUser->idleTime().isValid()) {
701 QDateTime now = QDateTime::currentDateTime();
702 QDateTime idle = _ircUser->idleTime();
703 int idleTime = idle.secsTo(now);
704 toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
706 if(_ircUser->loginTime().isValid()) {
707 toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
710 if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
712 return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
715 // void IrcUserItem::ircUserDestroyed() {
716 // parent()->removeChild(this);
717 // if(parent()->childCount() == 0)
718 // parent()->parent()->removeChild(parent());
721 /*****************************************
723 *****************************************/
724 NetworkModel::NetworkModel(QObject *parent)
725 : TreeModel(NetworkModel::defaultHeader(), parent)
727 connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),
728 this, SLOT(checkForNewBuffers(const QModelIndex &, int, int)));
729 connect(this, SIGNAL(rowsAboutToBeRemoved(const QModelIndex &, int, int)),
730 this, SLOT(checkForRemovedBuffers(const QModelIndex &, int, int)));
733 QList<QVariant >NetworkModel::defaultHeader() {
734 QList<QVariant> data;
735 data << tr("Buffer") << tr("Topic") << tr("Nick Count");
739 bool NetworkModel::isBufferIndex(const QModelIndex &index) const {
740 return index.data(NetworkModel::ItemTypeRole) == NetworkModel::BufferItemType;
743 int NetworkModel::networkRow(NetworkId networkId) {
744 NetworkItem *netItem = 0;
745 for(int i = 0; i < rootItem->childCount(); i++) {
746 netItem = qobject_cast<NetworkItem *>(rootItem->child(i));
749 if(netItem->networkId() == networkId)
755 QModelIndex NetworkModel::networkIndex(NetworkId networkId) {
756 int netRow = networkRow(networkId);
758 return QModelIndex();
760 return indexByItem(qobject_cast<NetworkItem *>(rootItem->child(netRow)));
763 NetworkItem *NetworkModel::findNetworkItem(NetworkId networkId) {
764 int netRow = networkRow(networkId);
768 return qobject_cast<NetworkItem *>(rootItem->child(netRow));
771 NetworkItem *NetworkModel::networkItem(NetworkId networkId) {
772 NetworkItem *netItem = findNetworkItem(networkId);
775 netItem = new NetworkItem(networkId, rootItem);
776 rootItem->newChild(netItem);
781 void NetworkModel::networkRemoved(const NetworkId &networkId) {
782 int netRow = networkRow(networkId);
784 rootItem->removeChild(netRow);
788 QModelIndex NetworkModel::bufferIndex(BufferId bufferId) {
789 if(!_bufferItemCache.contains(bufferId))
790 return QModelIndex();
792 return indexByItem(_bufferItemCache[bufferId]);
795 BufferItem *NetworkModel::findBufferItem(BufferId bufferId) {
796 if(_bufferItemCache.contains(bufferId))
797 return _bufferItemCache[bufferId];
802 BufferItem *NetworkModel::bufferItem(const BufferInfo &bufferInfo) {
803 if(_bufferItemCache.contains(bufferInfo.bufferId()))
804 return _bufferItemCache[bufferInfo.bufferId()];
806 NetworkItem *netItem = networkItem(bufferInfo.networkId());
807 return netItem->bufferItem(bufferInfo);
810 QStringList NetworkModel::mimeTypes() const {
811 // mimetypes we accept for drops
813 // comma separated list of colon separated pairs of networkid and bufferid
814 // example: 0:1,0:2,1:4
815 types << "application/Quassel/BufferItemList";
819 bool NetworkModel::mimeContainsBufferList(const QMimeData *mimeData) {
820 return mimeData->hasFormat("application/Quassel/BufferItemList");
823 QList< QPair<NetworkId, BufferId> > NetworkModel::mimeDataToBufferList(const QMimeData *mimeData) {
824 QList< QPair<NetworkId, BufferId> > bufferList;
826 if(!mimeContainsBufferList(mimeData))
829 QStringList rawBufferList = QString::fromAscii(mimeData->data("application/Quassel/BufferItemList")).split(",");
832 foreach(QString rawBuffer, rawBufferList) {
833 if(!rawBuffer.contains(":"))
835 networkId = rawBuffer.section(":", 0, 0).toInt();
836 bufferUid = rawBuffer.section(":", 1, 1).toInt();
837 bufferList.append(qMakePair(networkId, bufferUid));
843 QMimeData *NetworkModel::mimeData(const QModelIndexList &indexes) const {
844 QMimeData *mimeData = new QMimeData();
846 QStringList bufferlist;
847 QString netid, uid, bufferid;
848 foreach(QModelIndex index, indexes) {
849 netid = QString::number(index.data(NetworkIdRole).value<NetworkId>().toInt());
850 uid = QString::number(index.data(BufferIdRole).value<BufferId>().toInt());
851 bufferid = QString("%1:%2").arg(netid).arg(uid);
852 if(!bufferlist.contains(bufferid))
853 bufferlist << bufferid;
856 mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toAscii());
861 bool NetworkModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) {
866 if(!mimeContainsBufferList(data))
869 // target must be a query
870 BufferInfo::Type targetType = (BufferInfo::Type)parent.data(NetworkModel::BufferTypeRole).toInt();
871 if(targetType != BufferInfo::QueryBuffer)
874 QList< QPair<NetworkId, BufferId> > bufferList = mimeDataToBufferList(data);
876 // exactly one buffer has to be dropped
877 if(bufferList.count() != 1)
880 NetworkId netId = bufferList.first().first;
881 BufferId bufferId = bufferList.first().second;
883 // no self merges (would kill us)
884 if(bufferId == parent.data(BufferIdRole).value<BufferId>())
887 NetworkItem *netItem = findNetworkItem(netId);
890 BufferItem *bufferItem = netItem->findBufferItem(bufferId);
891 Q_ASSERT(bufferItem);
893 // source must be a query too
894 if(bufferItem->bufferType() != BufferInfo::QueryBuffer)
897 // TODO: warn user about buffermerge!
898 qDebug() << "merging" << bufferId << parent.data(BufferIdRole).value<BufferId>();
899 removeRow(parent.row(), parent.parent());
904 void NetworkModel::attachNetwork(Network *net) {
905 NetworkItem *netItem = networkItem(net->networkId());
906 netItem->attachNetwork(net);
909 void NetworkModel::bufferUpdated(BufferInfo bufferInfo) {
910 BufferItem *bufItem = bufferItem(bufferInfo);
911 QModelIndex itemindex = indexByItem(bufItem);
912 emit dataChanged(itemindex, itemindex);
915 void NetworkModel::removeBuffer(BufferId bufferId) {
916 BufferItem *buffItem = findBufferItem(bufferId);
920 buffItem->parent()->removeChild(buffItem);
923 MsgId NetworkModel::lastSeenMsgId(BufferId bufferId) {
924 if(!_bufferItemCache.contains(bufferId))
927 return _bufferItemCache[bufferId]->lastSeenMsgId();
930 MsgId NetworkModel::lastSeenMarkerMsgId(BufferId bufferId) {
931 if(!_bufferItemCache.contains(bufferId))
934 return _bufferItemCache[bufferId]->lastSeenMarkerMsgId();
937 void NetworkModel::setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId) {
938 BufferItem *bufferItem = findBufferItem(bufferId);
940 qDebug() << "NetworkModel::setLastSeenMsgId(): buffer is unknown:" << bufferId;
943 bufferItem->setLastSeenMsgId(msgId);
946 void NetworkModel::updateBufferActivity(const Message &msg) {
947 BufferItem *item = bufferItem(msg.bufferInfo());
948 item->updateActivityLevel(msg);
949 if(item->isCurrentBuffer())
950 emit setLastSeenMsg(item->bufferId(), msg.msgId());
953 void NetworkModel::setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel level) {
954 BufferItem *bufferItem = findBufferItem(bufferId);
956 qDebug() << "NetworkModel::setBufferActivity(): buffer is unknown:" << bufferId;
959 bufferItem->setActivityLevel(level);
962 void NetworkModel::clearBufferActivity(const BufferId &bufferId) {
963 BufferItem *bufferItem = findBufferItem(bufferId);
965 qDebug() << "NetworkModel::clearBufferActivity(): buffer is unknown:" << bufferId;
968 bufferItem->clearActivityLevel();
971 const Network *NetworkModel::networkByIndex(const QModelIndex &index) const {
972 QVariant netVariant = index.data(NetworkIdRole);
973 if(!netVariant.isValid())
976 NetworkId networkId = netVariant.value<NetworkId>();
977 return Client::network(networkId);
980 void NetworkModel::checkForRemovedBuffers(const QModelIndex &parent, int start, int end) {
981 if(parent.data(ItemTypeRole) != NetworkItemType)
984 for(int row = start; row <= end; row++) {
985 _bufferItemCache.remove(parent.child(row, 0).data(BufferIdRole).value<BufferId>());
989 void NetworkModel::checkForNewBuffers(const QModelIndex &parent, int start, int end) {
990 if(parent.data(ItemTypeRole) != NetworkItemType)
993 for(int row = start; row <= end; row++) {
994 QModelIndex child = parent.child(row, 0);
995 _bufferItemCache[child.data(BufferIdRole).value<BufferId>()] = static_cast<BufferItem *>(child.internalPointer());
999 QString NetworkModel::bufferName(BufferId bufferId) {
1000 if(!_bufferItemCache.contains(bufferId))
1003 return _bufferItemCache[bufferId]->bufferName();
1006 BufferInfo::Type NetworkModel::bufferType(BufferId bufferId) {
1007 if(!_bufferItemCache.contains(bufferId))
1008 return BufferInfo::InvalidBuffer;
1010 return _bufferItemCache[bufferId]->bufferType();
1013 BufferInfo NetworkModel::bufferInfo(BufferId bufferId) {
1014 if(!_bufferItemCache.contains(bufferId))
1015 return BufferInfo();
1017 return _bufferItemCache[bufferId]->bufferInfo();
1020 NetworkId NetworkModel::networkId(BufferId bufferId) {
1021 if(!_bufferItemCache.contains(bufferId))
1024 NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1026 return netItem->networkId();
1031 QString NetworkModel::networkName(BufferId bufferId) {
1032 if(!_bufferItemCache.contains(bufferId))
1035 NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1037 return netItem->networkName();