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:
48 case NetworkModel::BufferInfoRole:
49 case NetworkModel::BufferTypeRole:
51 return child(0)->data(column, role);
54 case NetworkModel::NetworkIdRole:
55 return qVariantFromValue(_networkId);
56 case NetworkModel::ItemTypeRole:
57 return NetworkModel::NetworkItemType;
58 case NetworkModel::ItemActiveRole:
61 return PropertyMapItem::data(column, role);
65 BufferItem *NetworkItem::findBufferItem(BufferId bufferId) {
66 BufferItem *bufferItem = 0;
68 for(int i = 0; i < childCount(); i++) {
69 bufferItem = qobject_cast<BufferItem *>(child(i));
72 if(bufferItem->bufferId() == bufferId)
79 BufferItem *NetworkItem::bufferItem(const BufferInfo &bufferInfo) {
80 BufferItem *bufferItem = findBufferItem(bufferInfo);
84 switch(bufferInfo.type()) {
85 case BufferInfo::StatusBuffer:
86 bufferItem = new StatusBufferItem(bufferInfo, this);
88 case BufferInfo::ChannelBuffer:
89 bufferItem = new ChannelBufferItem(bufferInfo, this);
91 case BufferInfo::QueryBuffer:
92 bufferItem = new QueryBufferItem(bufferInfo, this);
95 bufferItem = new BufferItem(bufferInfo, this);
102 void NetworkItem::attachNetwork(Network *network) {
108 connect(network, SIGNAL(networkNameSet(QString)),
109 this, SLOT(setNetworkName(QString)));
110 connect(network, SIGNAL(currentServerSet(QString)),
111 this, SLOT(setCurrentServer(QString)));
112 connect(network, SIGNAL(ircChannelAdded(IrcChannel *)),
113 this, SLOT(attachIrcChannel(IrcChannel *)));
114 connect(network, SIGNAL(ircUserAdded(IrcUser *)),
115 this, SLOT(attachIrcUser(IrcUser *)));
116 connect(network, SIGNAL(connectedSet(bool)),
117 this, SIGNAL(dataChanged()));
118 connect(network, SIGNAL(destroyed()),
119 this, SIGNAL(dataChanged()));
124 void NetworkItem::attachIrcChannel(IrcChannel *ircChannel) {
125 ChannelBufferItem *channelItem;
126 for(int i = 0; i < childCount(); i++) {
127 channelItem = qobject_cast<ChannelBufferItem *>(child(i));
131 if(channelItem->bufferName().toLower() == ircChannel->name().toLower()) {
132 channelItem->attachIrcChannel(ircChannel);
138 void NetworkItem::attachIrcUser(IrcUser *ircUser) {
139 QueryBufferItem *queryItem = 0;
140 for(int i = 0; i < childCount(); i++) {
141 queryItem = qobject_cast<QueryBufferItem *>(child(i));
145 if(queryItem->bufferName().toLower() == ircUser->nick().toLower()) {
146 queryItem->setIrcUser(ircUser);
152 void NetworkItem::setNetworkName(const QString &networkName) {
153 Q_UNUSED(networkName);
157 void NetworkItem::setCurrentServer(const QString &serverName) {
158 Q_UNUSED(serverName);
163 QString NetworkItem::toolTip(int column) const {
166 QStringList toolTip(QString("<b>%1</b>").arg(networkName()));
167 toolTip.append(tr("Server: %1").arg(currentServer()));
168 toolTip.append(tr("Users: %1").arg(nickCount()));
171 toolTip.append(tr("Lag: %1 msecs").arg(_network->latency()));
174 return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
177 /*****************************************
179 *****************************************/
180 BufferItem::BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
181 : PropertyMapItem(QStringList() << "bufferName" << "topic" << "nickCount", parent),
182 _bufferInfo(bufferInfo),
183 _activity(BufferInfo::NoActivity)
185 setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled);
188 void BufferItem::setActivityLevel(BufferInfo::ActivityLevel level) {
189 if(_activity != level) {
195 void BufferItem::clearActivityLevel() {
196 _activity = BufferInfo::NoActivity;
197 _lastSeenMarkerMsgId = _lastSeenMsgId;
201 void BufferItem::updateActivityLevel(const Message &msg) {
202 if(isCurrentBuffer()) {
206 if(msg.flags() & Message::Self) // don't update activity for our own messages
209 if(lastSeenMsgId() >= msg.msgId())
212 BufferInfo::ActivityLevel oldLevel = activityLevel();
214 _activity |= BufferInfo::OtherActivity;
215 if(msg.type() & (Message::Plain | Message::Notice | Message::Action))
216 _activity |= BufferInfo::NewMessage;
218 if(msg.flags() & Message::Highlight)
219 _activity |= BufferInfo::Highlight;
221 if(oldLevel != _activity)
225 QVariant BufferItem::data(int column, int role) const {
227 case NetworkModel::ItemTypeRole:
228 return NetworkModel::BufferItemType;
229 case NetworkModel::BufferIdRole:
230 return qVariantFromValue(bufferInfo().bufferId());
231 case NetworkModel::NetworkIdRole:
232 return qVariantFromValue(bufferInfo().networkId());
233 case NetworkModel::BufferInfoRole:
234 return qVariantFromValue(bufferInfo());
235 case NetworkModel::BufferTypeRole:
236 return int(bufferType());
237 case NetworkModel::ItemActiveRole:
239 case NetworkModel::BufferActivityRole:
240 return (int)activityLevel();
242 return PropertyMapItem::data(column, role);
246 bool BufferItem::setData(int column, const QVariant &value, int role) {
248 case NetworkModel::BufferActivityRole:
249 setActivityLevel((BufferInfo::ActivityLevel)value.toInt());
252 return PropertyMapItem::setData(column, value, role);
257 void BufferItem::setBufferName(const QString &name) {
258 _bufferInfo = BufferInfo(_bufferInfo.bufferId(), _bufferInfo.networkId(), _bufferInfo.type(), _bufferInfo.groupId(), name);
262 void BufferItem::setLastSeenMsgId(const MsgId &msgId) {
263 _lastSeenMsgId = msgId;
264 if(!isCurrentBuffer()) {
265 _lastSeenMarkerMsgId = msgId;
267 setActivityLevel(BufferInfo::NoActivity);
270 bool BufferItem::isCurrentBuffer() const {
271 return _bufferInfo.bufferId() == Client::bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
274 QString BufferItem::toolTip(int column) const {
276 return tr("<p> %1 - %2 </p>").arg(bufferInfo().bufferId().toInt()).arg(bufferName());
279 /*****************************************
281 *****************************************/
282 StatusBufferItem::StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
283 : BufferItem(bufferInfo, parent)
286 connect(parent, SIGNAL(dataChanged()), this, SIGNAL(dataChanged()));
289 QString StatusBufferItem::toolTip(int column) const {
293 QString netName = Client::network(bufferInfo().networkId())->networkName();
294 toolTip.append(tr("<b>Status buffer of %1</b>").arg(netName));
296 return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
299 /*****************************************
301 *****************************************/
302 QueryBufferItem::QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent)
303 : BufferItem(bufferInfo, parent),
306 setFlags(flags() | Qt::ItemIsDropEnabled | Qt::ItemIsEditable);
308 const Network *net = Client::network(bufferInfo.networkId());
312 IrcUser *ircUser = net->ircUser(bufferInfo.bufferName());
316 QVariant QueryBufferItem::data(int column, int role) const {
318 case NetworkModel::IrcUserRole:
319 return QVariant::fromValue<QObject *>(_ircUser);
320 case NetworkModel::UserAwayRole:
321 return (bool)_ircUser ? _ircUser->isAway() : false;
323 return BufferItem::data(column, role);
327 bool QueryBufferItem::setData(int column, const QVariant &value, int role) {
329 return BufferItem::setData(column, value, role);
334 QString newName = value.toString();
335 if(!newName.isEmpty()) {
336 Client::renameBuffer(bufferId(), newName);
344 return BufferItem::setData(column, value, role);
348 void QueryBufferItem::setBufferName(const QString &name) {
349 BufferItem::setBufferName(name);
350 NetworkId netId = data(0, NetworkModel::NetworkIdRole).value<NetworkId>();
351 const Network *net = Client::network(netId);
353 setIrcUser(net->ircUser(name));
356 QString QueryBufferItem::toolTip(int column) const {
357 // pretty much code duplication of IrcUserItem::toolTip() but inheritance won't solve this...
361 toolTip.append(tr("<b>Query with %1</b>").arg(bufferName()));
364 if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
365 if(_ircUser->isAway()) {
366 toolTip[0].append(QString(" (away%1)").arg(!_ircUser->awayMessage().isEmpty() ? (QString(" ") + _ircUser->awayMessage()) : QString()));
368 if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
369 if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(_ircUser->nick()).arg(_ircUser->ircOperator()));
370 if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
371 if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
373 toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
375 if(_ircUser->idleTime().isValid()) {
376 QDateTime now = QDateTime::currentDateTime();
377 QDateTime idle = _ircUser->idleTime();
378 int idleTime = idle.secsTo(now);
379 toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
381 if(_ircUser->loginTime().isValid()) {
382 toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
385 if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
388 return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
391 void QueryBufferItem::setIrcUser(IrcUser *ircUser) {
392 if(_ircUser == ircUser)
396 disconnect(_ircUser, 0, this, 0);
400 connect(ircUser, SIGNAL(quited()), this, SLOT(removeIrcUser()));
401 connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
408 void QueryBufferItem::removeIrcUser() {
413 /*****************************************
415 *****************************************/
416 ChannelBufferItem::ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent)
417 : BufferItem(bufferInfo, parent),
420 const Network *net = Client::network(bufferInfo.networkId());
424 IrcChannel *ircChannel = net->ircChannel(bufferInfo.bufferName());
426 attachIrcChannel(ircChannel);
429 QVariant ChannelBufferItem::data(int column, int role) const {
431 case NetworkModel::IrcChannelRole:
432 return QVariant::fromValue<QObject *>(_ircChannel);
434 return BufferItem::data(column, role);
438 QString ChannelBufferItem::toolTip(int column) const {
442 toolTip.append(tr("<b>Channel %1</b>").arg(bufferName()));
444 //TODO: add channel modes
445 toolTip.append(tr("<b>Users:</b> %1").arg(nickCount()));
447 QString channelMode = _ircChannel->channelModeString(); // channelModeString is compiled on the fly -> thus cache the result
448 if(!channelMode.isEmpty())
449 toolTip.append(tr("<b>Mode:</b> %1").arg(channelMode));
453 bool showTopic = s.value("DisplayTopicInTooltip", QVariant(false)).toBool();
455 QString _topic = topic();
457 _topic = stripFormatCodes(_topic);
458 _topic.replace(QString("<"), QString("<"));
459 _topic.replace(QString(">"), QString(">"));
460 toolTip.append(QString("<font size='-2'> </font>"));
461 toolTip.append(tr("<b>Topic:</b> %1").arg(_topic));
465 toolTip.append(tr("Not active <br /> Double-click to join"));
468 return tr("<p> %1 </p>").arg(toolTip.join("<br />"));
471 void ChannelBufferItem::attachIrcChannel(IrcChannel *ircChannel) {
472 Q_ASSERT(!_ircChannel && ircChannel);
474 _ircChannel = ircChannel;
476 connect(ircChannel, SIGNAL(topicSet(QString)),
477 this, SLOT(setTopic(QString)));
478 connect(ircChannel, SIGNAL(ircUsersJoined(QList<IrcUser *>)),
479 this, SLOT(join(QList<IrcUser *>)));
480 connect(ircChannel, SIGNAL(ircUserParted(IrcUser *)),
481 this, SLOT(part(IrcUser *)));
482 connect(ircChannel, SIGNAL(parted()),
483 this, SLOT(ircChannelParted()));
484 connect(ircChannel, SIGNAL(ircUserModesSet(IrcUser *, QString)),
485 this, SLOT(userModeChanged(IrcUser *)));
486 connect(ircChannel, SIGNAL(ircUserModeAdded(IrcUser *, QString)),
487 this, SLOT(userModeChanged(IrcUser *)));
488 connect(ircChannel, SIGNAL(ircUserModeRemoved(IrcUser *, QString)),
489 this, SLOT(userModeChanged(IrcUser *)));
491 if(!ircChannel->ircUsers().isEmpty())
492 join(ircChannel->ircUsers());
497 void ChannelBufferItem::ircChannelParted() {
498 Q_CHECK_PTR(_ircChannel);
499 disconnect(_ircChannel, 0, this, 0);
505 void ChannelBufferItem::join(const QList<IrcUser *> &ircUsers) {
506 addUsersToCategory(ircUsers);
510 UserCategoryItem *ChannelBufferItem::findCategoryItem(int categoryId) {
511 UserCategoryItem *categoryItem = 0;
513 for(int i = 0; i < childCount(); i++) {
514 categoryItem = qobject_cast<UserCategoryItem *>(child(i));
517 if(categoryItem->categoryId() == categoryId)
523 void ChannelBufferItem::addUserToCategory(IrcUser *ircUser) {
524 addUsersToCategory(QList<IrcUser *>() << ircUser);
527 void ChannelBufferItem::addUsersToCategory(const QList<IrcUser *> &ircUsers) {
528 Q_ASSERT(_ircChannel);
530 QHash<UserCategoryItem *, QList<IrcUser *> > categories;
533 UserCategoryItem *categoryItem = 0;
535 foreach(IrcUser *ircUser, ircUsers) {
536 categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
537 categoryItem = findCategoryItem(categoryId);
539 categoryItem = new UserCategoryItem(categoryId, this);
540 categories[categoryItem] = QList<IrcUser *>();
541 newChild(categoryItem);
543 categories[categoryItem] << ircUser;
546 QHash<UserCategoryItem *, QList<IrcUser *> >::const_iterator catIter = categories.constBegin();
547 while(catIter != categories.constEnd()) {
548 catIter.key()->addUsers(catIter.value());
553 void ChannelBufferItem::part(IrcUser *ircUser) {
555 qWarning() << bufferName() << "ChannelBufferItem::part(): unknown User" << ircUser;
559 disconnect(ircUser, 0, this, 0);
560 removeUserFromCategory(ircUser);
564 void ChannelBufferItem::removeUserFromCategory(IrcUser *ircUser) {
566 // If we parted the channel there might still be some ircUsers connected.
567 // in that case we just ignore the call
568 Q_ASSERT(childCount() == 0);
572 UserCategoryItem *categoryItem = 0;
573 for(int i = 0; i < childCount(); i++) {
574 categoryItem = qobject_cast<UserCategoryItem *>(child(i));
575 if(categoryItem->removeUser(ircUser)) {
576 if(categoryItem->childCount() == 0)
583 void ChannelBufferItem::userModeChanged(IrcUser *ircUser) {
584 Q_ASSERT(_ircChannel);
586 int categoryId = UserCategoryItem::categoryFromModes(_ircChannel->userModes(ircUser));
587 UserCategoryItem *categoryItem = findCategoryItem(categoryId);
590 if(categoryItem->findIrcUser(ircUser)) {
591 return; // already in the right category;
594 categoryItem = new UserCategoryItem(categoryId, this);
595 newChild(categoryItem);
598 // find the item that needs reparenting
599 IrcUserItem *ircUserItem = 0;
600 for(int i = 0; i < childCount(); i++) {
601 UserCategoryItem *oldCategoryItem = qobject_cast<UserCategoryItem *>(child(i));
602 Q_ASSERT(oldCategoryItem);
603 IrcUserItem *userItem = oldCategoryItem->findIrcUser(ircUser);
605 ircUserItem = userItem;
611 qWarning() << "ChannelBufferItem::userModeChanged(IrcUser *): unable to determine old category of" << ircUser;
614 ircUserItem->reParent(categoryItem);
617 /*****************************************
618 * User Category Items (like @vh etc.)
619 *****************************************/
620 // we hardcode this even though we have PREFIX in network... but that wouldn't help with mapping modes to
621 // category strings anyway.
622 const QList<QChar> UserCategoryItem::categories = QList<QChar>() << 'q' << 'a' << 'o' << 'h' << 'v';
624 UserCategoryItem::UserCategoryItem(int category, AbstractTreeItem *parent)
625 : PropertyMapItem(QStringList() << "categoryName", parent),
628 setFlags(Qt::ItemIsEnabled);
629 setTreeItemFlags(AbstractTreeItem::DeleteOnLastChildRemoved);
630 setObjectName(parent->data(0, Qt::DisplayRole).toString() + "/" + QString::number(category));
633 // caching this makes no sense, since we display the user number dynamically
634 QString UserCategoryItem::categoryName() const {
635 int n = childCount();
637 case 0: return tr("%n Owner(s)", 0, n);
638 case 1: return tr("%n Admin(s)", 0, n);
639 case 2: return tr("%n Operator(s)", 0, n);
640 case 3: return tr("%n Half-Op(s)", 0, n);
641 case 4: return tr("%n Voiced", 0, n);
642 default: return tr("%n User(s)", 0, n);
646 IrcUserItem *UserCategoryItem::findIrcUser(IrcUser *ircUser) {
647 IrcUserItem *userItem = 0;
649 for(int i = 0; i < childCount(); i++) {
650 userItem = qobject_cast<IrcUserItem *>(child(i));
653 if(userItem->ircUser() == ircUser)
659 void UserCategoryItem::addUsers(const QList<IrcUser *> &ircUsers) {
660 QList<AbstractTreeItem *> userItems;
661 foreach(IrcUser *ircUser, ircUsers)
662 userItems << new IrcUserItem(ircUser, this);
663 newChilds(userItems);
667 bool UserCategoryItem::removeUser(IrcUser *ircUser) {
668 IrcUserItem *userItem = findIrcUser(ircUser);
669 bool success = (bool)userItem;
671 removeChild(userItem);
677 int UserCategoryItem::categoryFromModes(const QString &modes) {
678 for(int i = 0; i < categories.count(); i++) {
679 if(modes.contains(categories[i]))
682 return categories.count();
685 QVariant UserCategoryItem::data(int column, int role) const {
687 case TreeModel::SortRole:
689 case NetworkModel::ItemActiveRole:
691 case NetworkModel::ItemTypeRole:
692 return NetworkModel::UserCategoryItemType;
693 case NetworkModel::BufferIdRole:
694 return parent()->data(column, role);
695 case NetworkModel::NetworkIdRole:
696 return parent()->data(column, role);
697 case NetworkModel::BufferInfoRole:
698 return parent()->data(column, role);
700 return PropertyMapItem::data(column, role);
705 /*****************************************
707 *****************************************/
708 IrcUserItem::IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent)
709 : PropertyMapItem(QStringList() << "nickName", parent),
712 setObjectName(ircUser->nick());
713 connect(ircUser, SIGNAL(quited()), this, SLOT(ircUserQuited()));
714 connect(ircUser, SIGNAL(nickSet(QString)), this, SIGNAL(dataChanged()));
715 connect(ircUser, SIGNAL(awaySet(bool)), this, SIGNAL(dataChanged()));
718 QVariant IrcUserItem::data(int column, int role) const {
720 case NetworkModel::ItemActiveRole:
722 case NetworkModel::ItemTypeRole:
723 return NetworkModel::IrcUserItemType;
724 case NetworkModel::BufferIdRole:
725 return parent()->data(column, role);
726 case NetworkModel::NetworkIdRole:
727 return parent()->data(column, role);
728 case NetworkModel::BufferInfoRole:
729 return parent()->data(column, role);
730 case NetworkModel::IrcChannelRole:
731 return parent()->data(column, role);
732 case NetworkModel::IrcUserRole:
733 return QVariant::fromValue<QObject *>(_ircUser.data());
734 case NetworkModel::UserAwayRole:
735 return (bool)_ircUser ? _ircUser->isAway() : false;
737 return PropertyMapItem::data(column, role);
741 QString IrcUserItem::toolTip(int column) const {
743 QStringList toolTip(QString("<b>%1</b>").arg(nickName()));
744 if(_ircUser->userModes() != "") toolTip[0].append(QString(" (%1)").arg(_ircUser->userModes()));
745 if(_ircUser->isAway()) {
746 toolTip[0].append(" is away");
747 if(!_ircUser->awayMessage().isEmpty())
748 toolTip[0].append(QString(" (%1)").arg(_ircUser->awayMessage()));
750 if(!_ircUser->realName().isEmpty()) toolTip.append(_ircUser->realName());
751 if(!_ircUser->ircOperator().isEmpty()) toolTip.append(QString("%1 %2").arg(nickName()).arg(_ircUser->ircOperator()));
752 if(!_ircUser->suserHost().isEmpty()) toolTip.append(_ircUser->suserHost());
753 if(!_ircUser->whoisServiceReply().isEmpty()) toolTip.append(_ircUser->whoisServiceReply());
755 toolTip.append(_ircUser->hostmask().remove(0, _ircUser->hostmask().indexOf("!")+1));
757 if(_ircUser->idleTime().isValid()) {
758 QDateTime now = QDateTime::currentDateTime();
759 QDateTime idle = _ircUser->idleTime();
760 int idleTime = idle.secsTo(now);
761 toolTip.append(tr("idling since %1").arg(secondsToString(idleTime)));
763 if(_ircUser->loginTime().isValid()) {
764 toolTip.append(tr("login time: %1").arg(_ircUser->loginTime().toString()));
767 if(!_ircUser->server().isEmpty()) toolTip.append(tr("server: %1").arg(_ircUser->server()));
769 return QString("<p> %1 </p>").arg(toolTip.join("<br />"));
772 /*****************************************
774 *****************************************/
775 NetworkModel::NetworkModel(QObject *parent)
776 : TreeModel(NetworkModel::defaultHeader(), parent)
778 connect(this, SIGNAL(rowsInserted(const QModelIndex &, int, int)),
779 this, SLOT(checkForNewBuffers(const QModelIndex &, int, int)));
780 connect(this, SIGNAL(rowsAboutToBeRemoved(const QModelIndex &, int, int)),
781 this, SLOT(checkForRemovedBuffers(const QModelIndex &, int, int)));
784 QList<QVariant >NetworkModel::defaultHeader() {
785 QList<QVariant> data;
786 data << tr("Buffer") << tr("Topic") << tr("Nick Count");
790 bool NetworkModel::isBufferIndex(const QModelIndex &index) const {
791 return index.data(NetworkModel::ItemTypeRole) == NetworkModel::BufferItemType;
794 int NetworkModel::networkRow(NetworkId networkId) const {
795 NetworkItem *netItem = 0;
796 for(int i = 0; i < rootItem->childCount(); i++) {
797 netItem = qobject_cast<NetworkItem *>(rootItem->child(i));
800 if(netItem->networkId() == networkId)
806 QModelIndex NetworkModel::networkIndex(NetworkId networkId) {
807 int netRow = networkRow(networkId);
809 return QModelIndex();
811 return indexByItem(qobject_cast<NetworkItem *>(rootItem->child(netRow)));
814 NetworkItem *NetworkModel::findNetworkItem(NetworkId networkId) const {
815 int netRow = networkRow(networkId);
819 return qobject_cast<NetworkItem *>(rootItem->child(netRow));
822 NetworkItem *NetworkModel::networkItem(NetworkId networkId) {
823 NetworkItem *netItem = findNetworkItem(networkId);
826 netItem = new NetworkItem(networkId, rootItem);
827 rootItem->newChild(netItem);
832 void NetworkModel::networkRemoved(const NetworkId &networkId) {
833 int netRow = networkRow(networkId);
835 rootItem->removeChild(netRow);
839 QModelIndex NetworkModel::bufferIndex(BufferId bufferId) {
840 if(!_bufferItemCache.contains(bufferId))
841 return QModelIndex();
843 return indexByItem(_bufferItemCache[bufferId]);
846 BufferItem *NetworkModel::findBufferItem(BufferId bufferId) const {
847 if(_bufferItemCache.contains(bufferId))
848 return _bufferItemCache[bufferId];
853 BufferItem *NetworkModel::bufferItem(const BufferInfo &bufferInfo) {
854 if(_bufferItemCache.contains(bufferInfo.bufferId()))
855 return _bufferItemCache[bufferInfo.bufferId()];
857 NetworkItem *netItem = networkItem(bufferInfo.networkId());
858 return netItem->bufferItem(bufferInfo);
861 QStringList NetworkModel::mimeTypes() const {
862 // mimetypes we accept for drops
864 // comma separated list of colon separated pairs of networkid and bufferid
865 // example: 0:1,0:2,1:4
866 types << "application/Quassel/BufferItemList";
870 bool NetworkModel::mimeContainsBufferList(const QMimeData *mimeData) {
871 return mimeData->hasFormat("application/Quassel/BufferItemList");
874 QList< QPair<NetworkId, BufferId> > NetworkModel::mimeDataToBufferList(const QMimeData *mimeData) {
875 QList< QPair<NetworkId, BufferId> > bufferList;
877 if(!mimeContainsBufferList(mimeData))
880 QStringList rawBufferList = QString::fromAscii(mimeData->data("application/Quassel/BufferItemList")).split(",");
883 foreach(QString rawBuffer, rawBufferList) {
884 if(!rawBuffer.contains(":"))
886 networkId = rawBuffer.section(":", 0, 0).toInt();
887 bufferUid = rawBuffer.section(":", 1, 1).toInt();
888 bufferList.append(qMakePair(networkId, bufferUid));
894 QMimeData *NetworkModel::mimeData(const QModelIndexList &indexes) const {
895 QMimeData *mimeData = new QMimeData();
897 QStringList bufferlist;
898 QString netid, uid, bufferid;
899 foreach(QModelIndex index, indexes) {
900 netid = QString::number(index.data(NetworkIdRole).value<NetworkId>().toInt());
901 uid = QString::number(index.data(BufferIdRole).value<BufferId>().toInt());
902 bufferid = QString("%1:%2").arg(netid).arg(uid);
903 if(!bufferlist.contains(bufferid))
904 bufferlist << bufferid;
907 mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toAscii());
912 void NetworkModel::attachNetwork(Network *net) {
913 NetworkItem *netItem = networkItem(net->networkId());
914 netItem->attachNetwork(net);
917 void NetworkModel::bufferUpdated(BufferInfo bufferInfo) {
918 BufferItem *bufItem = bufferItem(bufferInfo);
919 QModelIndex itemindex = indexByItem(bufItem);
920 emit dataChanged(itemindex, itemindex);
923 void NetworkModel::removeBuffer(BufferId bufferId) {
924 BufferItem *buffItem = findBufferItem(bufferId);
928 buffItem->parent()->removeChild(buffItem);
931 MsgId NetworkModel::lastSeenMsgId(BufferId bufferId) const {
932 if(!_bufferItemCache.contains(bufferId))
935 return _bufferItemCache[bufferId]->lastSeenMsgId();
938 MsgId NetworkModel::lastSeenMarkerMsgId(BufferId bufferId) const {
939 if(!_bufferItemCache.contains(bufferId))
942 return _bufferItemCache[bufferId]->lastSeenMarkerMsgId();
945 void NetworkModel::setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId) {
946 BufferItem *bufferItem = findBufferItem(bufferId);
948 qDebug() << "NetworkModel::setLastSeenMsgId(): buffer is unknown:" << bufferId;
951 bufferItem->setLastSeenMsgId(msgId);
954 void NetworkModel::updateBufferActivity(const Message &msg) {
955 BufferItem *item = bufferItem(msg.bufferInfo());
956 item->updateActivityLevel(msg);
957 if(item->isCurrentBuffer())
958 emit setLastSeenMsg(item->bufferId(), msg.msgId());
961 void NetworkModel::setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel level) {
962 BufferItem *bufferItem = findBufferItem(bufferId);
964 qDebug() << "NetworkModel::setBufferActivity(): buffer is unknown:" << bufferId;
967 bufferItem->setActivityLevel(level);
970 void NetworkModel::clearBufferActivity(const BufferId &bufferId) {
971 BufferItem *bufferItem = findBufferItem(bufferId);
973 qDebug() << "NetworkModel::clearBufferActivity(): buffer is unknown:" << bufferId;
976 bufferItem->clearActivityLevel();
979 const Network *NetworkModel::networkByIndex(const QModelIndex &index) const {
980 QVariant netVariant = index.data(NetworkIdRole);
981 if(!netVariant.isValid())
984 NetworkId networkId = netVariant.value<NetworkId>();
985 return Client::network(networkId);
988 void NetworkModel::checkForRemovedBuffers(const QModelIndex &parent, int start, int end) {
989 if(parent.data(ItemTypeRole) != NetworkItemType)
992 for(int row = start; row <= end; row++) {
993 _bufferItemCache.remove(parent.child(row, 0).data(BufferIdRole).value<BufferId>());
997 void NetworkModel::checkForNewBuffers(const QModelIndex &parent, int start, int end) {
998 if(parent.data(ItemTypeRole) != NetworkItemType)
1001 for(int row = start; row <= end; row++) {
1002 QModelIndex child = parent.child(row, 0);
1003 _bufferItemCache[child.data(BufferIdRole).value<BufferId>()] = static_cast<BufferItem *>(child.internalPointer());
1007 QString NetworkModel::bufferName(BufferId bufferId) const {
1008 if(!_bufferItemCache.contains(bufferId))
1011 return _bufferItemCache[bufferId]->bufferName();
1014 BufferInfo::Type NetworkModel::bufferType(BufferId bufferId) const {
1015 if(!_bufferItemCache.contains(bufferId))
1016 return BufferInfo::InvalidBuffer;
1018 return _bufferItemCache[bufferId]->bufferType();
1021 BufferInfo NetworkModel::bufferInfo(BufferId bufferId) const {
1022 if(!_bufferItemCache.contains(bufferId))
1023 return BufferInfo();
1025 return _bufferItemCache[bufferId]->bufferInfo();
1028 NetworkId NetworkModel::networkId(BufferId bufferId) const {
1029 if(!_bufferItemCache.contains(bufferId))
1032 NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1034 return netItem->networkId();
1039 QString NetworkModel::networkName(BufferId bufferId) const {
1040 if(!_bufferItemCache.contains(bufferId))
1043 NetworkItem *netItem = qobject_cast<NetworkItem *>(_bufferItemCache[bufferId]->parent());
1045 return netItem->networkName();
1050 BufferId NetworkModel::bufferId(NetworkId networkId, const QString &bufferName, Qt::CaseSensitivity cs) const {
1051 const NetworkItem *netItem = findNetworkItem(networkId);
1055 for(int i = 0; i < netItem->childCount(); i++) {
1056 BufferItem *bufferItem = qobject_cast<BufferItem *>(netItem->child(i));
1057 if(bufferItem && !bufferItem->bufferName().compare(bufferName, cs))
1058 return bufferItem->bufferId();
1063 void NetworkModel::sortBufferIds(QList<BufferId> &bufferIds) const {
1064 QList<BufferItem *> bufferItems;
1065 foreach(BufferId bufferId, bufferIds) {
1066 if(_bufferItemCache.contains(bufferId))
1067 bufferItems << _bufferItemCache[bufferId];
1070 qSort(bufferItems.begin(), bufferItems.end(), bufferItemLessThan);
1073 foreach(BufferItem *bufferItem, bufferItems) {
1074 bufferIds << bufferItem->bufferId();
1078 QList<BufferId> NetworkModel::allBufferIdsSorted() const {
1079 QList<BufferId> bufferIds = allBufferIds();
1080 sortBufferIds(bufferIds);
1084 bool NetworkModel::bufferItemLessThan(const BufferItem *left, const BufferItem *right) {
1085 int leftType = left->bufferType();
1086 int rightType = right->bufferType();
1088 if(leftType != rightType)
1089 return leftType < rightType;
1091 return QString::compare(left->bufferName(), right->bufferName(), Qt::CaseInsensitive) < 0;