/***************************************************************************
- * Copyright (C) 2005-07 by The Quassel Team *
+ * Copyright (C) 2005-07 by the Quassel IRC Team *
* devel@quassel-irc.org *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
- * (at your option) any later version. *
+ * (at your option) version 3. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
#include <QColor> // FIXME Dependency on QtGui!
-#include "global.h"
#include "buffertreemodel.h"
+#include "mappedselectionmodel.h"
+#include <QAbstractItemView>
+
+#include "bufferinfo.h"
+#include "client.h"
+#include "signalproxy.h"
+#include "networkinfo.h"
+#include "ircchannel.h"
+#include "ircuser.h"
+
/*****************************************
* Fancy Buffer Items
*****************************************/
-BufferTreeItem::BufferTreeItem(Buffer *buffer, TreeItem *parent) : TreeItem(parent) {
- buf = buffer;
- activity = Buffer::NoActivity;
+BufferTreeItem::BufferTreeItem(Buffer *buffer, AbstractTreeItem *parent)
+ : PropertyMapItem(QStringList() << "bufferName" << "topic" << "nickCount", parent),
+ buf(buffer),
+ activity(Buffer::NoActivity)
+{
+ Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
+ if(buf->bufferType() == Buffer::QueryType)
+ flags |= Qt::ItemIsDropEnabled;
+ setFlags(flags);
}
-void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
- activity = level;
+quint64 BufferTreeItem::id() const {
+ return buf->bufferInfo().uid();
}
-QString BufferTreeItem::text(int column) const {
- switch(column) {
- case 0:
- return buf->displayName();
- case 1:
- return buf->networkName();
- default:
- return QString();
- }
+void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
+ activity = level;
}
QColor BufferTreeItem::foreground(int column) const {
+ Q_UNUSED(column)
// for the time beeing we ignore the column :)
if(activity & Buffer::Highlight) {
return QColor(Qt::red);
}
}
-
QVariant BufferTreeItem::data(int column, int role) const {
switch(role) {
- case Qt::DisplayRole:
- return text(column);
- case Qt::ForegroundRole:
- return foreground(column);
- case BufferTreeModel::BufferNameRole:
- return buf->bufferName();
- case BufferTreeModel::BufferTypeRole:
- return buf->bufferType();
- case BufferTreeModel::BufferActiveRole:
- return buf->isActive();
- case BufferTreeModel::BufferIdRole:
- return buf->bufferId().uid();
- default:
- return QVariant();
+ case BufferTreeModel::BufferUidRole:
+ return buf->bufferInfo().uid();
+ case BufferTreeModel::NetworkIdRole:
+ return buf->bufferInfo().networkId();
+ case BufferTreeModel::BufferTypeRole:
+ return int(buf->bufferType());
+ case BufferTreeModel::BufferActiveRole:
+ return buf->isActive();
+ case Qt::ForegroundRole:
+ return foreground(column);
+ default:
+ return PropertyMapItem::data(column, role);
}
}
+void BufferTreeItem::attachIrcChannel(IrcChannel *ircChannel) {
+ if(!ircChannel)
+ return;
+
+ _ircChannel = ircChannel;
+
+ connect(ircChannel, SIGNAL(topicSet(QString)),
+ this, SLOT(setTopic(QString)));
+ connect(ircChannel, SIGNAL(ircUserJoined(IrcUser *)),
+ this, SLOT(join(IrcUser *)));
+ connect(ircChannel, SIGNAL(ircUserParted(IrcUser *)),
+ this, SLOT(part(IrcUser *)));
+}
+
+QString BufferTreeItem::bufferName() const {
+ return buf->name();
+}
+
+QString BufferTreeItem::topic() const {
+ if(_ircChannel)
+ return _ircChannel->topic();
+ else
+ return QString();
+}
+
+int BufferTreeItem::nickCount() const {
+ if(_ircChannel)
+ return _ircChannel->ircUsers().count();
+ else
+ return 0;
+}
+
+void BufferTreeItem::setTopic(const QString &topic) {
+ Q_UNUSED(topic);
+ emit dataChanged(1);
+}
+
+void BufferTreeItem::join(IrcUser *ircUser) {
+ Q_UNUSED(ircUser);
+ emit dataChanged(2);
+}
+
+void BufferTreeItem::part(IrcUser *ircUser) {
+ Q_UNUSED(ircUser);
+ emit dataChanged(2);
+}
+
+/*****************************************
+* Network Items
+*****************************************/
+NetworkTreeItem::NetworkTreeItem(const uint &netid, const QString &network, AbstractTreeItem *parent)
+ : PropertyMapItem(QList<QString>() << "networkName" << "currentServer" << "nickCount", parent),
+ _networkId(netid),
+ _networkName(network)
+{
+ setFlags(Qt::ItemIsEnabled);
+}
+
+QVariant NetworkTreeItem::data(int column, int role) const {
+ switch(role) {
+ case BufferTreeModel::NetworkIdRole:
+ return _networkId;
+ case BufferTreeModel::ItemTypeRole:
+ return BufferTreeModel::NetworkItem;
+ default:
+ return PropertyMapItem::data(column, role);
+ }
+}
+
+quint64 NetworkTreeItem::id() const {
+ return _networkId;
+}
+
+QString NetworkTreeItem::networkName() const {
+ if(_networkInfo)
+ return _networkInfo->networkName();
+ else
+ return _networkName;
+}
+
+QString NetworkTreeItem::currentServer() const {
+ if(_networkInfo)
+ return _networkInfo->currentServer();
+ else
+ return QString();
+}
+
+int NetworkTreeItem::nickCount() const {
+ BufferTreeItem *bufferItem;
+ int count = 0;
+ for(int i = 0; i < childCount(); i++) {
+ bufferItem = qobject_cast<BufferTreeItem *>(child(i));
+ if(!bufferItem)
+ continue;
+ count += bufferItem->nickCount();
+ }
+ return count;
+}
+
+void NetworkTreeItem::attachNetworkInfo(NetworkInfo *networkInfo) {
+ if(!networkInfo)
+ return;
+
+ _networkInfo = networkInfo;
+
+ connect(networkInfo, SIGNAL(networkNameSet(QString)),
+ this, SLOT(setNetworkName(QString)));
+ connect(networkInfo, SIGNAL(currentServerSet(QString)),
+ this, SLOT(setCurrentServer(QString)));
+ connect(networkInfo, SIGNAL(ircChannelAdded(QString)),
+ this, SLOT(attachIrcChannel(QString)));
+ // FIXME: connect this and that...
+}
+
+void NetworkTreeItem::attachIrcChannel(const QString &channelName) {
+ IrcChannel *ircChannel = _networkInfo->ircChannel(channelName);
+ if(!ircChannel) {
+ qWarning() << "NetworkTreeItem::attachIrcChannel(): unkown Channel" << channelName;
+ return;
+ }
+
+ BufferTreeItem *bufferItem;
+ for(int i = 0; i < childCount(); i++) {
+ bufferItem = qobject_cast<BufferTreeItem *>(child(i));
+ if(bufferItem->bufferName() == ircChannel->name()) {
+ bufferItem->attachIrcChannel(ircChannel);
+ break;
+ }
+ }
+}
+
+void NetworkTreeItem::setNetworkName(const QString &networkName) {
+ Q_UNUSED(networkName);
+ emit dataChanged(0);
+}
+
+void NetworkTreeItem::setCurrentServer(const QString &serverName) {
+ Q_UNUSED(serverName);
+ emit dataChanged(1);
+}
+
/*****************************************
* BufferTreeModel
*****************************************/
-BufferTreeModel::BufferTreeModel(QObject *parent) : TreeModel(BufferTreeModel::defaultHeader(), parent) {
- connect(this, SIGNAL(fakeUserInput(BufferId, QString)),
- ClientProxy::instance(), SLOT(gsUserInput(BufferId, QString)));
+BufferTreeModel::BufferTreeModel(QObject *parent)
+ : TreeModel(BufferTreeModel::defaultHeader(), parent),
+ _selectionModelSynchronizer(new SelectionModelSynchronizer(this)),
+ _propertyMapper(new ModelPropertyMapper(this))
+{
+ // initialize the Property Mapper
+ _propertyMapper->setModel(this);
+ delete _propertyMapper->selectionModel();
+ MappedSelectionModel *mappedSelectionModel = new MappedSelectionModel(this);
+ _propertyMapper->setSelectionModel(mappedSelectionModel);
+ synchronizeSelectionModel(mappedSelectionModel);
+
+ connect(_selectionModelSynchronizer, SIGNAL(setCurrentIndex(QModelIndex, QItemSelectionModel::SelectionFlags)),
+ this, SLOT(setCurrentIndex(QModelIndex, QItemSelectionModel::SelectionFlags)));
}
QList<QVariant >BufferTreeModel::defaultHeader() {
QList<QVariant> data;
- data << tr("Buffer") << tr("Network");
+ data << tr("Buffer") << tr("Topic") << tr("Nick Count");
return data;
}
+void BufferTreeModel::synchronizeSelectionModel(MappedSelectionModel *selectionModel) {
+ selectionModelSynchronizer()->addSelectionModel(selectionModel);
+}
-Qt::ItemFlags BufferTreeModel::flags(const QModelIndex &index) const {
- if(!index.isValid())
- return Qt::ItemIsDropEnabled;
- //return 0;
+void BufferTreeModel::synchronizeView(QAbstractItemView *view) {
+ MappedSelectionModel *mappedSelectionModel = new MappedSelectionModel(view->model());
+ selectionModelSynchronizer()->addSelectionModel(mappedSelectionModel);
+ Q_ASSERT(mappedSelectionModel);
+ delete view->selectionModel();
+ view->setSelectionModel(mappedSelectionModel);
+}
- // I think this is pretty ugly..
- if(isBufferIndex(index)) {
- Buffer *buffer = getBufferByIndex(index);
- if(buffer->bufferType() == Buffer::QueryBuffer)
- return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled;
- else
- return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
- } else {
- return Qt::ItemIsEnabled | Qt::ItemIsDropEnabled;
- }
+void BufferTreeModel::mapProperty(int column, int role, QObject *target, const QByteArray &property) {
+ propertyMapper()->addMapping(column, role, target, property);
}
bool BufferTreeModel::isBufferIndex(const QModelIndex &index) const {
+ // not so purdy...
return parent(index) != QModelIndex();
}
Buffer *BufferTreeModel::getBufferByIndex(const QModelIndex &index) const {
BufferTreeItem *item = static_cast<BufferTreeItem *>(index.internalPointer());
+ // FIXME get rid of this
+ Q_ASSERT(item->buffer() == Client::instance()->buffer(item->id()));
return item->buffer();
}
-QModelIndex BufferTreeModel::getOrCreateNetworkItemIndex(Buffer *buffer) {
- QString net = buffer->networkName();
-
- if(networkItem.contains(net)) {
- return index(networkItem[net]->row(), 0);
- } else {
- QList<QVariant> data;
- data << net << "";
-
- int nextRow = rootItem->childCount();
-
- beginInsertRows(QModelIndex(), nextRow, nextRow);
- rootItem->appendChild(new TreeItem(data, rootItem));
- endInsertRows();
-
- networkItem[net] = rootItem->child(nextRow);
- return index(nextRow, 0);
+
+// experimental stuff :)
+QModelIndex BufferTreeModel::networkIndex(uint networkId) {
+ return indexById(networkId);
+}
+
+NetworkTreeItem *BufferTreeModel::network(uint networkId) {
+ return qobject_cast<NetworkTreeItem *>(rootItem->childById(networkId));
+}
+
+NetworkTreeItem *BufferTreeModel::newNetwork(uint networkId, const QString &networkName) {
+ NetworkTreeItem *networkItem = network(networkId);
+
+ if(networkItem == 0) {
+ networkItem = new NetworkTreeItem(networkId, networkName, rootItem);
+ appendChild(rootItem, networkItem);
}
+
+ Q_ASSERT(networkItem);
+ return networkItem;
}
-QModelIndex BufferTreeModel::getOrCreateBufferItemIndex(Buffer *buffer) {
- QModelIndex networkItemIndex = getOrCreateNetworkItemIndex(buffer);
-
- if(bufferItem.contains(buffer)) {
- return index(bufferItem[buffer]->row(), 0, networkItemIndex);
- } else {
- // first we determine the parent of the new Item
- TreeItem *networkItem = static_cast<TreeItem*>(networkItemIndex.internalPointer());
+QModelIndex BufferTreeModel::bufferIndex(BufferInfo bufferInfo) {
+ QModelIndex networkIdx = networkIndex(bufferInfo.networkId());
+ if(!networkIdx.isValid())
+ return QModelIndex();
+ else
+ return indexById(bufferInfo.uid(), networkIdx);
+}
- int nextRow = networkItem->childCount();
+BufferTreeItem *BufferTreeModel::buffer(BufferInfo bufferInfo) {
+ QModelIndex bufferIdx = bufferIndex(bufferInfo);
+ if(bufferIdx.isValid())
+ return static_cast<BufferTreeItem *>(bufferIdx.internalPointer());
+ else
+ return 0;
+}
- beginInsertRows(networkItemIndex, nextRow, nextRow);
- networkItem->appendChild(new BufferTreeItem(buffer, networkItem));
- endInsertRows();
+BufferTreeItem *BufferTreeModel::newBuffer(BufferInfo bufferInfo) {
+ BufferTreeItem *bufferItem = buffer(bufferInfo);
+ if(bufferItem == 0) {
+ NetworkTreeItem *networkItem = newNetwork(bufferInfo.networkId(), bufferInfo.network());
- bufferItem[buffer] = static_cast<BufferTreeItem *>(networkItem->child(nextRow));
- return index(nextRow, 0, networkItemIndex);
+ // FIXME: get rid of the buffer pointer
+ Buffer *buffer = Client::instance()->buffer(bufferInfo.uid());
+ bufferItem = new BufferTreeItem(buffer, networkItem);
+ appendChild(networkItem, bufferItem);
}
+
+ Q_ASSERT(bufferItem);
+ return bufferItem;
}
QStringList BufferTreeModel::mimeTypes() const {
+ // mimetypes we accept for drops
QStringList types;
- types << "application/Quassel/BufferItem/row"
- << "application/Quassel/BufferItem/network"
- << "application/Quassel/BufferItem/bufferId";
+ // comma separated list of colon separated pairs of networkid and bufferid
+ // example: 0:1,0:2,1:4
+ types << "application/Quassel/BufferItemList";
return types;
}
+bool BufferTreeModel::mimeContainsBufferList(const QMimeData *mimeData) {
+ return mimeData->hasFormat("application/Quassel/BufferItemList");
+}
+
+QList< QPair<uint, uint> > BufferTreeModel::mimeDataToBufferList(const QMimeData *mimeData) {
+ QList< QPair<uint, uint> > bufferList;
+
+ if(!mimeContainsBufferList(mimeData))
+ return bufferList;
+
+ QStringList rawBufferList = QString::fromAscii(mimeData->data("application/Quassel/BufferItemList")).split(",");
+ uint networkId, bufferUid;
+ foreach(QString rawBuffer, rawBufferList) {
+ if(!rawBuffer.contains(":"))
+ continue;
+ networkId = rawBuffer.section(":", 0, 0).toUInt();
+ bufferUid = rawBuffer.section(":", 1, 1).toUInt();
+ bufferList.append(qMakePair(networkId, bufferUid));
+ }
+ return bufferList;
+}
+
+
QMimeData *BufferTreeModel::mimeData(const QModelIndexList &indexes) const {
QMimeData *mimeData = new QMimeData();
- QModelIndex index = indexes.first();
-
- mimeData->setData("application/Quassel/BufferItem/row", QByteArray::number(index.row()));
- mimeData->setData("application/Quassel/BufferItem/network", getBufferByIndex(index)->networkName().toUtf8());
- mimeData->setData("application/Quassel/BufferItem/bufferId", QByteArray::number(getBufferByIndex(index)->bufferId().uid()));
+ QStringList bufferlist;
+ QString netid, uid, bufferid;
+ foreach(QModelIndex index, indexes) {
+ netid = QString::number(index.data(NetworkIdRole).toUInt());
+ uid = QString::number(index.data(BufferUidRole).toUInt());
+ bufferid = QString("%1:%2").arg(netid).arg(uid);
+ if(!bufferlist.contains(bufferid))
+ bufferlist << bufferid;
+ }
+
+ mimeData->setData("application/Quassel/BufferItemList", bufferlist.join(",").toAscii());
+
return mimeData;
}
bool BufferTreeModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) {
- if(!(data->hasFormat("application/Quassel/BufferItem/row")
- && data->hasFormat("application/Quassel/BufferItem/network")
- && data->hasFormat("application/Quassel/BufferItem/bufferId")))
- return false; // whatever the drop is... it's not a buffer...
-
- int sourcerow = data->data("application/Quassel/BufferItem/row").toInt();
- QString network = QString::fromUtf8(data->data("application/Quassel/BufferItem/network"));
-
- Q_ASSERT(networkItem.contains(network));
+ Q_UNUSED(action)
+ Q_UNUSED(row)
+ Q_UNUSED(column)
- if(parent == QModelIndex()) // can't be a query...
+ if(!mimeContainsBufferList(data))
return false;
-
- Buffer *sourceBuffer = static_cast<BufferTreeItem *>(networkItem[network]->child(sourcerow))->buffer();
- Buffer *targetBuffer = getBufferByIndex(parent);
- if(!(sourceBuffer->bufferType() & targetBuffer->bufferType() & Buffer::QueryBuffer)) // only queries can be merged
+ // target must be a query
+ Buffer::Type targetType = (Buffer::Type)parent.data(BufferTreeModel::BufferTypeRole).toInt();
+ if(targetType != Buffer::QueryType)
return false;
+
+ QList< QPair<uint, uint> > bufferList = mimeDataToBufferList(data);
+
+ // exactly one buffer has to be dropped
+ if(bufferList.count() != 1)
+ return false;
+
+ uint netId = bufferList.first().first;
+ uint bufferId = bufferList.first().second;
+
+ // no self merges (would kill us)
+ if(bufferId == parent.data(BufferUidRole).toUInt())
+ return false;
- if(sourceBuffer == targetBuffer) // we won't merge with ourself :)
+ Q_ASSERT(rootItem->childById(netId));
+ Q_ASSERT(rootItem->childById(netId)->childById(bufferId));
+
+ // source must be a query too
+ Buffer::Type sourceType = (Buffer::Type)rootItem->childById(netId)->childById(bufferId)->data(0, BufferTypeRole).toInt();
+ if(sourceType != Buffer::QueryType)
return false;
- /*
- if(QMessageBox::warning(static_cast<QWidget *>(QObject::parent()),
- tr("Merge Buffers?"),
- tr("Do you really want to merge the following Buffers?<br />%1.%2<br />%3.%4").arg(sourceBuffer->networkName()).arg(sourceBuffer->bufferName()).arg(targetBuffer->networkName()).arg(targetBuffer->bufferName()),
- QMessageBox::Yes|QMessageBox::No) == QMessageBox::No)
- return false;
-
- */
- qDebug() << "merging" << sourceBuffer->bufferName() << "with" << targetBuffer->bufferName();
- bufferItem.remove(getBufferByIndex(parent));
- removeRow(parent.row(), BufferTreeModel::parent(parent));
+ // TODO: warn user about buffermerge!
+ qDebug() << "merging" << bufferId << parent.data(BufferUidRole).toInt();
+ removeRow(parent.row(), parent.parent());
return true;
}
+void BufferTreeModel::attachNetworkInfo(NetworkInfo *networkInfo) {
+ NetworkTreeItem *networkItem = network(networkInfo->networkId());
+ if(!networkItem) {
+ qWarning() << "BufferTreeModel::attachNetworkInfo(): network is unknown!";
+ return;
+ }
+ networkItem->attachNetworkInfo(networkInfo);
+}
+
void BufferTreeModel::bufferUpdated(Buffer *buffer) {
- QModelIndex itemindex = getOrCreateBufferItemIndex(buffer);
- emit invalidateFilter();
+ BufferTreeItem *bufferItem = newBuffer(buffer->bufferInfo());
+ QModelIndex itemindex = indexByItem(bufferItem);
emit dataChanged(itemindex, itemindex);
}
// This Slot indicates that the user has selected a different buffer in the gui
-void BufferTreeModel::changeCurrent(const QModelIndex ¤t, const QModelIndex &previous) {
- if(isBufferIndex(current)) {
- currentBuffer = getBufferByIndex(current);
+void BufferTreeModel::setCurrentIndex(const QModelIndex &index, QItemSelectionModel::SelectionFlags command) {
+ Q_UNUSED(command)
+ Buffer *newCurrentBuffer;
+ if(isBufferIndex(index) && currentBuffer != (newCurrentBuffer = getBufferByIndex(index))) {
+ currentBuffer = newCurrentBuffer;
bufferActivity(Buffer::NoActivity, currentBuffer);
emit bufferSelected(currentBuffer);
- emit selectionChanged(current);
+ emit selectionChanged(index);
}
}
-// we received a double click on a buffer, so we're going to join it
-void BufferTreeModel::doubleClickReceived(const QModelIndex &clicked) {
- if(isBufferIndex(clicked)) {
- Buffer *buffer = getBufferByIndex(clicked);
- if(!buffer->isStatusBuffer())
- emit fakeUserInput(buffer->bufferId(), QString("/join " + buffer->bufferName()));
+void BufferTreeModel::bufferActivity(Buffer::ActivityLevel level, Buffer *buf) {
+ BufferTreeItem *bufferItem = buffer(buf->bufferInfo());
+ if(!bufferItem) {
+ qWarning() << "BufferTreeModel::bufferActivity(): received Activity Info for uknown Buffer";
+ return;
}
-}
-
-void BufferTreeModel::bufferActivity(Buffer::ActivityLevel level, Buffer *buffer) {
- if(bufferItem.contains(buffer) and buffer != currentBuffer)
- bufferItem[buffer]->setActivity(level);
+
+ if(buf != currentBuffer)
+ bufferItem->setActivity(level);
else
- bufferItem[buffer]->setActivity(Buffer::NoActivity);
- bufferUpdated(buffer);
+ bufferItem->setActivity(Buffer::NoActivity);
+ bufferUpdated(buf);
}
void BufferTreeModel::selectBuffer(Buffer *buffer) {
- QModelIndex index = getOrCreateBufferItemIndex(buffer);
- emit selectionChanged(index);
+ QModelIndex index = bufferIndex(buffer->bufferInfo());
+ if(!index.isValid()) {
+ qWarning() << "BufferTreeModel::selectBuffer(): unknown Buffer has been selected.";
+ return;
+ }
+ // SUPER UGLY!
+ setCurrentIndex(index, 0);
}