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 "clientproxy.h"
24 #include "buffertreemodel.h"
26 /*****************************************
28 *****************************************/
29 BufferTreeItem::BufferTreeItem(Buffer *buffer, TreeItem *parent) : TreeItem(parent) {
31 activity = Buffer::NoActivity;
34 uint BufferTreeItem::id() const {
35 return buf->bufferId().uid();
38 void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
42 QString BufferTreeItem::text(int column) const {
45 return buf->displayName();
47 return buf->networkName();
53 QColor BufferTreeItem::foreground(int /*column*/) const {
54 // for the time beeing we ignore the column :)
55 if(activity & Buffer::Highlight) {
56 return QColor(Qt::red);
57 } else if(activity & Buffer::NewMessage) {
58 return QColor(Qt::darkYellow);
59 } else if(activity & Buffer::OtherActivity) {
60 return QColor(Qt::darkGreen);
63 return QColor(Qt::black);
65 return QColor(Qt::gray);
70 QVariant BufferTreeItem::data(int column, int role) const {
74 case Qt::ForegroundRole:
75 return foreground(column);
76 case BufferTreeModel::BufferNameRole:
77 return buf->bufferName();
78 case BufferTreeModel::BufferTypeRole:
79 return buf->bufferType();
80 case BufferTreeModel::BufferActiveRole:
81 return buf->isActive();
82 case BufferTreeModel::BufferIdRole:
83 return buf->bufferId().uid();
89 /*****************************************
91 *****************************************/
92 NetworkTreeItem::NetworkTreeItem(const QString &network, TreeItem *parent) : TreeItem(parent) {
94 itemData << net << "";
97 uint NetworkTreeItem::id() const {
103 /*****************************************
105 *****************************************/
106 BufferTreeModel::BufferTreeModel(QObject *parent) : TreeModel(BufferTreeModel::defaultHeader(), parent) {
107 connect(this, SIGNAL(fakeUserInput(BufferId, QString)),
108 ClientProxy::instance(), SLOT(gsUserInput(BufferId, QString)));
111 QList<QVariant >BufferTreeModel::defaultHeader() {
112 QList<QVariant> data;
113 data << tr("Buffer") << tr("Network");
118 Qt::ItemFlags BufferTreeModel::flags(const QModelIndex &index) const {
120 return Qt::ItemIsDropEnabled;
123 // I think this is pretty ugly..
124 if(isBufferIndex(index)) {
125 Buffer *buffer = getBufferByIndex(index);
126 if(buffer->bufferType() == Buffer::QueryBuffer)
127 return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled;
129 return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
131 return Qt::ItemIsEnabled | Qt::ItemIsDropEnabled;
135 bool BufferTreeModel::isBufferIndex(const QModelIndex &index) const {
137 return parent(index) != QModelIndex();
140 Buffer *BufferTreeModel::getBufferByIndex(const QModelIndex &index) const {
141 BufferTreeItem *item = static_cast<BufferTreeItem *>(index.internalPointer());
142 return item->buffer();
145 QModelIndex BufferTreeModel::getOrCreateNetworkItemIndex(Buffer *buffer) {
146 QString net = buffer->networkName();
147 TreeItem *networkItem;
149 if(not(networkItem = rootItem->childById(qHash(net)))) {
150 int nextRow = rootItem->childCount();
151 networkItem = new NetworkTreeItem(net, rootItem);
153 beginInsertRows(QModelIndex(), nextRow, nextRow);
154 rootItem->appendChild(networkItem);
158 Q_ASSERT(networkItem);
159 return index(networkItem->row(), 0);
162 QModelIndex BufferTreeModel::getOrCreateBufferItemIndex(Buffer *buffer) {
163 QModelIndex networkItemIndex = getOrCreateNetworkItemIndex(buffer);
164 NetworkTreeItem *networkItem = static_cast<NetworkTreeItem*>(networkItemIndex.internalPointer());
165 TreeItem *bufferItem;
167 if(not(bufferItem = networkItem->childById(buffer->bufferId().uid()))) {
168 int nextRow = networkItem->childCount();
169 bufferItem = new BufferTreeItem(buffer, networkItem);
171 beginInsertRows(networkItemIndex, nextRow, nextRow);
172 networkItem->appendChild(bufferItem);
176 Q_ASSERT(bufferItem);
177 return index(bufferItem->row(), 0, networkItemIndex);
180 QStringList BufferTreeModel::mimeTypes() const {
182 types << "application/Quassel/BufferItem/row"
183 << "application/Quassel/BufferItem/network"
184 << "application/Quassel/BufferItem/bufferId";
188 QMimeData *BufferTreeModel::mimeData(const QModelIndexList &indexes) const {
189 QMimeData *mimeData = new QMimeData();
191 QModelIndex index = indexes.first();
193 mimeData->setData("application/Quassel/BufferItem/row", QByteArray::number(index.row()));
194 mimeData->setData("application/Quassel/BufferItem/network", getBufferByIndex(index)->networkName().toUtf8());
195 mimeData->setData("application/Quassel/BufferItem/bufferId", QByteArray::number(getBufferByIndex(index)->bufferId().uid()));
199 bool BufferTreeModel::dropMimeData(const QMimeData *data, Qt::DropAction /*action*/, int /*row*/, int /*column*/, const QModelIndex &parent) {
200 if(!(data->hasFormat("application/Quassel/BufferItem/row")
201 && data->hasFormat("application/Quassel/BufferItem/network")
202 && data->hasFormat("application/Quassel/BufferItem/bufferId")))
203 return false; // whatever the drop is... it's not a buffer...
205 int sourcerow = data->data("application/Quassel/BufferItem/row").toInt();
206 QString network = QString::fromUtf8(data->data("application/Quassel/BufferItem/network"));
208 Q_ASSERT(rootItem->childById(qHash(network)));
210 if(parent == QModelIndex()) // can't be a query...
213 Buffer *sourceBuffer = static_cast<BufferTreeItem *>(rootItem->childById(qHash(network))->child(sourcerow))->buffer();
214 Buffer *targetBuffer = getBufferByIndex(parent);
216 if(!(sourceBuffer->bufferType() & targetBuffer->bufferType() & Buffer::QueryBuffer)) // only queries can be merged
219 if(sourceBuffer == targetBuffer) // we won't merge with ourself :)
223 if(QMessageBox::warning(static_cast<QWidget *>(QObject::parent()),
224 tr("Merge Buffers?"),
225 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()),
226 QMessageBox::Yes|QMessageBox::No) == QMessageBox::No)
230 qDebug() << "merging" << sourceBuffer->bufferName() << "with" << targetBuffer->bufferName();
231 removeRow(parent.row(), BufferTreeModel::parent(parent));
236 void BufferTreeModel::bufferUpdated(Buffer *buffer) {
237 QModelIndex itemindex = getOrCreateBufferItemIndex(buffer);
238 emit invalidateFilter();
239 emit dataChanged(itemindex, itemindex);
242 // This Slot indicates that the user has selected a different buffer in the gui
243 void BufferTreeModel::changeCurrent(const QModelIndex ¤t, const QModelIndex &/*previous*/) {
244 if(isBufferIndex(current)) {
245 currentBuffer = getBufferByIndex(current);
246 bufferActivity(Buffer::NoActivity, currentBuffer);
247 emit bufferSelected(currentBuffer);
248 emit selectionChanged(current);
252 // we received a double click on a buffer, so we're going to join it
253 void BufferTreeModel::doubleClickReceived(const QModelIndex &clicked) {
254 if(isBufferIndex(clicked)) {
255 Buffer *buffer = getBufferByIndex(clicked);
256 if(!buffer->isStatusBuffer())
257 emit fakeUserInput(buffer->bufferId(), QString("/join " + buffer->bufferName()));
261 void BufferTreeModel::bufferActivity(Buffer::ActivityLevel level, Buffer *buffer) {
262 BufferTreeItem *bufferItem = static_cast<BufferTreeItem*>(getOrCreateBufferItemIndex(buffer).internalPointer());
263 if(buffer != currentBuffer)
264 bufferItem->setActivity(level);
266 bufferItem->setActivity(Buffer::NoActivity);
267 bufferUpdated(buffer);
270 void BufferTreeModel::selectBuffer(Buffer *buffer) {
271 QModelIndex index = getOrCreateBufferItemIndex(buffer);
272 emit selectionChanged(index);