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!
24 #include "buffertreemodel.h"
26 /*****************************************
28 *****************************************/
29 BufferTreeItem::BufferTreeItem(Buffer *buffer, TreeItem *parent) : TreeItem(parent) {
31 activity = Buffer::NoActivity;
34 void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
38 QString BufferTreeItem::text(int column) const {
41 return buf->displayName();
43 return buf->networkName();
49 QColor BufferTreeItem::foreground(int column) const {
50 // for the time beeing we ignore the column :)
51 if(activity & Buffer::Highlight) {
52 return QColor(Qt::red);
53 } else if(activity & Buffer::NewMessage) {
54 return QColor(Qt::darkYellow);
55 } else if(activity & Buffer::OtherActivity) {
56 return QColor(Qt::darkGreen);
59 return QColor(Qt::black);
61 return QColor(Qt::gray);
66 QVariant BufferTreeItem::data(int column, int role) const {
70 case Qt::ForegroundRole:
71 return foreground(column);
72 case BufferTreeModel::BufferNameRole:
73 return buf->bufferName();
74 case BufferTreeModel::BufferTypeRole:
75 return buf->bufferType();
76 case BufferTreeModel::BufferActiveRole:
77 return buf->isActive();
78 case BufferTreeModel::BufferIdRole:
79 return buf->bufferId().uid();
85 /*****************************************
87 *****************************************/
88 BufferTreeModel::BufferTreeModel(QObject *parent) : TreeModel(BufferTreeModel::defaultHeader(), parent) {
89 connect(this, SIGNAL(fakeUserInput(BufferId, QString)), ClientProxy::instance(), SLOT(gsUserInput(BufferId, QString)));
92 QList<QVariant >BufferTreeModel::defaultHeader() {
94 data << tr("Buffer") << tr("Network");
99 Qt::ItemFlags BufferTreeModel::flags(const QModelIndex &index) const {
101 return Qt::ItemIsDropEnabled;
104 // I think this is pretty ugly..
105 if(isBufferIndex(index)) {
106 Buffer *buffer = getBufferByIndex(index);
107 if(buffer->bufferType() == Buffer::QueryBuffer)
108 return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled;
110 return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
112 return Qt::ItemIsEnabled | Qt::ItemIsDropEnabled;
116 bool BufferTreeModel::isBufferIndex(const QModelIndex &index) const {
117 return parent(index) != QModelIndex();
120 Buffer *BufferTreeModel::getBufferByIndex(const QModelIndex &index) const {
121 BufferTreeItem *item = static_cast<BufferTreeItem *>(index.internalPointer());
122 return item->buffer();
125 QModelIndex BufferTreeModel::getOrCreateNetworkItemIndex(Buffer *buffer) {
126 QString net = buffer->networkName();
128 if(networkItem.contains(net)) {
129 return index(networkItem[net]->row(), 0);
131 QList<QVariant> data;
134 int nextRow = rootItem->childCount();
136 beginInsertRows(QModelIndex(), nextRow, nextRow);
137 rootItem->appendChild(new TreeItem(data, rootItem));
140 networkItem[net] = rootItem->child(nextRow);
141 return index(nextRow, 0);
145 QModelIndex BufferTreeModel::getOrCreateBufferItemIndex(Buffer *buffer) {
146 QModelIndex networkItemIndex = getOrCreateNetworkItemIndex(buffer);
148 if(bufferItem.contains(buffer)) {
149 return index(bufferItem[buffer]->row(), 0, networkItemIndex);
151 // first we determine the parent of the new Item
152 TreeItem *networkItem = static_cast<TreeItem*>(networkItemIndex.internalPointer());
154 int nextRow = networkItem->childCount();
156 beginInsertRows(networkItemIndex, nextRow, nextRow);
157 networkItem->appendChild(new BufferTreeItem(buffer, networkItem));
160 bufferItem[buffer] = static_cast<BufferTreeItem *>(networkItem->child(nextRow));
161 return index(nextRow, 0, networkItemIndex);
165 QStringList BufferTreeModel::mimeTypes() const {
167 types << "application/Quassel/BufferItem/row"
168 << "application/Quassel/BufferItem/network"
169 << "application/Quassel/BufferItem/bufferId";
173 QMimeData *BufferTreeModel::mimeData(const QModelIndexList &indexes) const {
174 QMimeData *mimeData = new QMimeData();
176 QModelIndex index = indexes.first();
178 mimeData->setData("application/Quassel/BufferItem/row", QByteArray::number(index.row()));
179 mimeData->setData("application/Quassel/BufferItem/network", getBufferByIndex(index)->networkName().toUtf8());
180 mimeData->setData("application/Quassel/BufferItem/bufferId", QByteArray::number(getBufferByIndex(index)->bufferId().uid()));
184 bool BufferTreeModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) {
185 if(!(data->hasFormat("application/Quassel/BufferItem/row")
186 && data->hasFormat("application/Quassel/BufferItem/network")
187 && data->hasFormat("application/Quassel/BufferItem/bufferId")))
188 return false; // whatever the drop is... it's not a buffer...
190 int sourcerow = data->data("application/Quassel/BufferItem/row").toInt();
191 QString network = QString::fromUtf8(data->data("application/Quassel/BufferItem/network"));
193 if(!networkItem.contains(network))
197 Buffer *sourceBuffer = static_cast<BufferTreeItem *>(networkItem[network]->child(sourcerow))->buffer();
199 if(parent == QModelIndex()) { // droping into empty space
200 emit addBuffer(sourceBuffer->bufferId().uid(), network);
204 if(!isBufferIndex(parent)) { // dropping at a network
205 emit addBuffer(sourceBuffer->bufferId().uid(), network);
210 Buffer *targetBuffer = getBufferByIndex(parent);
212 if(sourceBuffer == targetBuffer) // we won't merge with ourself :)
216 if(QMessageBox::warning(static_cast<QWidget *>(QObject::parent()),
217 tr("Merge Buffers?"),
218 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()),
219 QMessageBox::Yes|QMessageBox::No) == QMessageBox::No)
223 qDebug() << "merging" << sourceBuffer->bufferName() << "with" << targetBuffer->bufferName();
224 bufferItem.remove(getBufferByIndex(parent));
225 removeRow(parent.row(), BufferTreeModel::parent(parent));
230 void BufferTreeModel::bufferUpdated(Buffer *buffer) {
231 QModelIndex itemindex = getOrCreateBufferItemIndex(buffer);
232 emit invalidateFilter();
233 emit dataChanged(itemindex, itemindex);
236 // This Slot indicates that the user has selected a different buffer in the gui
237 void BufferTreeModel::changeCurrent(const QModelIndex ¤t, const QModelIndex &previous) {
238 if(isBufferIndex(current)) {
239 currentBuffer = getBufferByIndex(current);
240 bufferActivity(Buffer::NoActivity, currentBuffer);
241 emit bufferSelected(currentBuffer);
242 emit updateSelection(current, QItemSelectionModel::ClearAndSelect);
246 // we received a double click on a buffer, so we're going to join it
247 void BufferTreeModel::doubleClickReceived(const QModelIndex &clicked) {
248 if(isBufferIndex(clicked)) {
249 Buffer *buffer = getBufferByIndex(clicked);
250 if(!buffer->isStatusBuffer())
251 emit fakeUserInput(buffer->bufferId(), QString("/join " + buffer->bufferName()));
256 void BufferTreeModel::bufferActivity(Buffer::ActivityLevel level, Buffer *buffer) {
257 if(bufferItem.contains(buffer) and buffer != currentBuffer)
258 bufferItem[buffer]->setActivity(level);
260 bufferItem[buffer]->setActivity(Buffer::NoActivity);
261 bufferUpdated(buffer);
264 void BufferTreeModel::selectBuffer(Buffer *buffer) {
265 QModelIndex index = getOrCreateBufferItemIndex(buffer);
266 emit updateSelection(index, QItemSelectionModel::ClearAndSelect);