1 /***************************************************************************
2 * Copyright (C) 2005-07 by The Quassel Team *
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) any later version. *
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 <QColor> // FIXME Dependency on QtGui!
23 #include "buffertreemodel.h"
25 #include "bufferinfo.h"
27 #include "signalproxy.h"
29 /*****************************************
31 *****************************************/
32 BufferTreeItem::BufferTreeItem(Buffer *buffer, TreeItem *parent) : TreeItem(parent) {
34 activity = Buffer::NoActivity;
37 uint BufferTreeItem::id() const {
38 return buf->bufferInfo().uid();
41 void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
45 QString BufferTreeItem::text(int column) const {
48 return buf->displayName();
50 return buf->networkName();
56 QColor BufferTreeItem::foreground(int /*column*/) const {
57 // for the time beeing we ignore the column :)
58 if(activity & Buffer::Highlight) {
59 return QColor(Qt::red);
60 } else if(activity & Buffer::NewMessage) {
61 return QColor(Qt::darkYellow);
62 } else if(activity & Buffer::OtherActivity) {
63 return QColor(Qt::darkGreen);
66 return QColor(Qt::black);
68 return QColor(Qt::gray);
73 QVariant BufferTreeItem::data(int column, int role) const {
77 case Qt::ForegroundRole:
78 return foreground(column);
79 case BufferTreeModel::BufferNameRole:
80 return buf->bufferName();
81 case BufferTreeModel::BufferTypeRole:
82 return buf->bufferType();
83 case BufferTreeModel::BufferActiveRole:
84 return buf->isActive();
85 case BufferTreeModel::BufferInfoRole:
86 return buf->bufferInfo().uid();
92 Qt::ItemFlags BufferTreeItem::flags() const {
93 Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
94 if(buf->bufferType() == Buffer::QueryBuffer)
95 flags |= Qt::ItemIsDropEnabled;
100 /*****************************************
102 *****************************************/
103 NetworkTreeItem::NetworkTreeItem(const QString &network, TreeItem *parent) : TreeItem(parent) {
105 itemData << net << "";
108 uint NetworkTreeItem::id() const {
112 Qt::ItemFlags NetworkTreeItem::flags() const {
113 return Qt::ItemIsEnabled | Qt::ItemIsDropEnabled;
116 /*****************************************
118 *****************************************/
119 BufferTreeModel::BufferTreeModel(QObject *parent)
120 : TreeModel(BufferTreeModel::defaultHeader(), parent)
122 Client::signalProxy()->attachSignal(this, SIGNAL(fakeUserInput(BufferInfo, QString)), SIGNAL(sendInput(BufferInfo, QString)));
125 QList<QVariant >BufferTreeModel::defaultHeader() {
126 QList<QVariant> data;
127 data << tr("Buffer") << tr("Network");
131 bool BufferTreeModel::isBufferIndex(const QModelIndex &index) const {
133 return parent(index) != QModelIndex();
136 Buffer *BufferTreeModel::getBufferByIndex(const QModelIndex &index) const {
137 BufferTreeItem *item = static_cast<BufferTreeItem *>(index.internalPointer());
138 return item->buffer();
141 QModelIndex BufferTreeModel::getOrCreateNetworkItemIndex(Buffer *buffer) {
142 QString net = buffer->networkName();
143 TreeItem *networkItem;
145 if(!(networkItem = rootItem->childById(qHash(net)))) {
146 int nextRow = rootItem->childCount();
147 networkItem = new NetworkTreeItem(net, rootItem);
149 beginInsertRows(QModelIndex(), nextRow, nextRow);
150 rootItem->appendChild(networkItem);
154 Q_ASSERT(networkItem);
155 return index(networkItem->row(), 0);
158 QModelIndex BufferTreeModel::getOrCreateBufferItemIndex(Buffer *buffer) {
159 QModelIndex networkItemIndex = getOrCreateNetworkItemIndex(buffer);
160 NetworkTreeItem *networkItem = static_cast<NetworkTreeItem*>(networkItemIndex.internalPointer());
161 TreeItem *bufferItem;
163 if(!(bufferItem = networkItem->childById(buffer->bufferInfo().uid()))) {
164 int nextRow = networkItem->childCount();
165 bufferItem = new BufferTreeItem(buffer, networkItem);
167 beginInsertRows(networkItemIndex, nextRow, nextRow);
168 networkItem->appendChild(bufferItem);
172 Q_ASSERT(bufferItem);
173 return index(bufferItem->row(), 0, networkItemIndex);
176 QStringList BufferTreeModel::mimeTypes() const {
178 types << "application/Quassel/BufferItem/row"
179 << "application/Quassel/BufferItem/network"
180 << "application/Quassel/BufferItem/bufferInfo";
184 QMimeData *BufferTreeModel::mimeData(const QModelIndexList &indexes) const {
185 QMimeData *mimeData = new QMimeData();
187 QModelIndex index = indexes.first();
189 mimeData->setData("application/Quassel/BufferItem/row", QByteArray::number(index.row()));
190 mimeData->setData("application/Quassel/BufferItem/network", getBufferByIndex(index)->networkName().toUtf8());
191 mimeData->setData("application/Quassel/BufferItem/bufferInfo", QByteArray::number(getBufferByIndex(index)->bufferInfo().uid()));
195 bool BufferTreeModel::dropMimeData(const QMimeData *data, Qt::DropAction /*action*/, int /*row*/, int /*column*/, const QModelIndex &parent) {
196 foreach(QString mimeType, mimeTypes()) {
197 if(!(data->hasFormat(mimeType)))
198 return false; // whatever the drop is... it's not a buffer...
201 int sourcerow = data->data("application/Quassel/BufferItem/row").toInt();
202 QString network = QString::fromUtf8(data->data("application/Quassel/BufferItem/network"));
204 Q_ASSERT(rootItem->childById(qHash(network)));
206 if(parent == QModelIndex()) // can't be a query...
209 Buffer *sourceBuffer = static_cast<BufferTreeItem *>(rootItem->childById(qHash(network))->child(sourcerow))->buffer();
210 Buffer *targetBuffer = getBufferByIndex(parent);
212 if(!(sourceBuffer->bufferType() & targetBuffer->bufferType() & Buffer::QueryBuffer)) // only queries can be merged
215 if(sourceBuffer == targetBuffer) // we won't merge with ourself :)
218 // TODO: warn user about buffermerge!
219 qDebug() << "merging" << sourceBuffer->bufferName() << "with" << targetBuffer->bufferName();
220 removeRow(parent.row(), BufferTreeModel::parent(parent));
225 void BufferTreeModel::bufferUpdated(Buffer *buffer) {
226 QModelIndex itemindex = getOrCreateBufferItemIndex(buffer);
227 emit invalidateFilter();
228 emit dataChanged(itemindex, itemindex);
231 // This Slot indicates that the user has selected a different buffer in the gui
232 void BufferTreeModel::changeCurrent(const QModelIndex ¤t, const QModelIndex &/*previous*/) {
233 if(isBufferIndex(current)) {
234 currentBuffer = getBufferByIndex(current);
235 bufferActivity(Buffer::NoActivity, currentBuffer);
236 emit bufferSelected(currentBuffer);
237 emit selectionChanged(current);
241 // we received a double click on a buffer, so we're going to join it
242 void BufferTreeModel::doubleClickReceived(const QModelIndex &clicked) {
243 if(isBufferIndex(clicked)) {
244 Buffer *buffer = getBufferByIndex(clicked);
245 if(!buffer->isStatusBuffer())
246 emit fakeUserInput(buffer->bufferInfo(), QString("/join " + buffer->bufferName()));
250 void BufferTreeModel::bufferActivity(Buffer::ActivityLevel level, Buffer *buffer) {
251 BufferTreeItem *bufferItem = static_cast<BufferTreeItem*>(getOrCreateBufferItemIndex(buffer).internalPointer());
252 if(buffer != currentBuffer)
253 bufferItem->setActivity(level);
255 bufferItem->setActivity(Buffer::NoActivity);
256 bufferUpdated(buffer);
259 void BufferTreeModel::selectBuffer(Buffer *buffer) {
260 QModelIndex index = getOrCreateBufferItemIndex(buffer);
261 //emit selectionChanged(index);
262 changeCurrent(index, QModelIndex());