src/client/buffertreemodel.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by The Quassel Team                             *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   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.                                   *
   9  *                                                                         *
  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.                          *
  14  *                                                                         *
  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  ***************************************************************************/
  20
  21 #include <QColor>  // FIXME Dependency on QtGui!
  22
  23 #include "buffertreemodel.h"
  24
  25 #include "mappedselectionmodel.h"
  26 #include <QAbstractItemView>
  27
  28 #include "bufferinfo.h"
  29 #include "client.h"
  30 #include "signalproxy.h"
  31
  32 /*****************************************
  33 *  Fancy Buffer Items
  34 *****************************************/
  35 BufferTreeItem::BufferTreeItem(Buffer *buffer, TreeItem *parent) : TreeItem(parent) {
  36   buf = buffer;
  37   activity = Buffer::NoActivity;
  38 }
  39
  40 uint BufferTreeItem::id() const {
  41   return buf->bufferInfo().uid();
  42 }
  43
  44 void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
  45   activity = level;
  46 }
  47
  48 QString BufferTreeItem::text(int column) const {
  49   switch(column) {
  50     case 0:
  51       return buf->displayName();
  52     case 1:
  53       return buf->networkName();
  54     default:
  55       return QString();
  56   }
  57 }
  58
  59 QColor BufferTreeItem::foreground(int /*column*/) const {
  60   // for the time beeing we ignore the column :)
  61   if(activity & Buffer::Highlight) {
  62     return QColor(Qt::red);
  63   } else if(activity & Buffer::NewMessage) {
  64     return QColor(Qt::darkYellow);
  65   } else if(activity & Buffer::OtherActivity) {
  66     return QColor(Qt::darkGreen);
  67   } else {
  68     if(buf->isActive())
  69       return QColor(Qt::black);
  70     else
  71       return QColor(Qt::gray);
  72   }
  73 }
  74
  75
  76 QVariant BufferTreeItem::data(int column, int role) const {
  77   switch(role) {
  78     case Qt::DisplayRole:
  79       return text(column);
  80     case Qt::ForegroundRole:
  81       return foreground(column);
  82     case BufferTreeModel::BufferNameRole:
  83       return buf->bufferName();
  84     case BufferTreeModel::BufferTypeRole:
  85       return int(buf->bufferType());
  86     case BufferTreeModel::BufferActiveRole:
  87       return buf->isActive();
  88     case BufferTreeModel::BufferUidRole:
  89       return buf->bufferInfo().uid();
  90     default:
  91       return QVariant();
  92   }
  93 }
  94
  95 Qt::ItemFlags BufferTreeItem::flags() const {
  96   Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
  97   if(buf->bufferType() == Buffer::QueryType)
  98     flags |= Qt::ItemIsDropEnabled;
  99
 100   return flags;
 101 }
 102
 103 /*****************************************
 104 *  Network Items
 105 *****************************************/
 106 NetworkTreeItem::NetworkTreeItem(const QString &network, TreeItem *parent) : TreeItem(parent) {
 107   net = network;
 108   itemData << net << "";
 109 }
 110
 111 uint NetworkTreeItem::id() const {
 112   return qHash(net);
 113 }
 114
 115 Qt::ItemFlags NetworkTreeItem::flags() const {
 116   return Qt::ItemIsEnabled | Qt::ItemIsDropEnabled;
 117 }
 118
 119 /*****************************************
 120  * BufferTreeModel
 121  *****************************************/
 122 BufferTreeModel::BufferTreeModel(QObject *parent)
 123   : TreeModel(BufferTreeModel::defaultHeader(), parent),
 124     _selectionModelSynchronizer(new SelectionModelSynchronizer(this)),
 125     _propertyMapper(new ModelPropertyMapper(this))
 126 {
 127   _propertyMapper->setModel(this);
 128   delete _propertyMapper->selectionModel();
 129   MappedSelectionModel *mappedSelectionModel = new MappedSelectionModel(this);
 130   _propertyMapper->setSelectionModel(mappedSelectionModel);
 131   synchronizeSelectionModel(mappedSelectionModel);
 132
 133   connect(_selectionModelSynchronizer, SIGNAL(setCurrentIndex(QModelIndex, QItemSelectionModel::SelectionFlags)),
 134           this, SLOT(setCurrentIndex(QModelIndex, QItemSelectionModel::SelectionFlags)));
 135 }
 136
 137 QList<QVariant >BufferTreeModel::defaultHeader() {
 138   QList<QVariant> data;
 139   data << tr("Buffer") << tr("Network");
 140   return data;
 141 }
 142
 143 void BufferTreeModel::synchronizeSelectionModel(MappedSelectionModel *selectionModel) {
 144   selectionModelSynchronizer()->addSelectionModel(selectionModel);
 145 }
 146
 147 void BufferTreeModel::synchronizeView(QAbstractItemView *view) {
 148   MappedSelectionModel *mappedSelectionModel = new MappedSelectionModel(view->model());
 149   selectionModelSynchronizer()->addSelectionModel(mappedSelectionModel);
 150   Q_ASSERT(mappedSelectionModel);
 151   delete view->selectionModel();
 152   view->setSelectionModel(mappedSelectionModel);
 153 }
 154
 155 void BufferTreeModel::mapProperty(int column, int role, QObject *target, const QByteArray &property) {
 156   propertyMapper()->addMapping(column, role, target, property);
 157 }
 158
 159 bool BufferTreeModel::isBufferIndex(const QModelIndex &index) const {
 160   // not so purdy...
 161   return parent(index) != QModelIndex();
 162 }
 163
 164 Buffer *BufferTreeModel::getBufferByIndex(const QModelIndex &index) const {
 165   BufferTreeItem *item = static_cast<BufferTreeItem *>(index.internalPointer());
 166   return item->buffer();
 167 }
 168
 169 QModelIndex BufferTreeModel::getOrCreateNetworkItemIndex(Buffer *buffer) {
 170   QString net = buffer->networkName();
 171   TreeItem *networkItem;
 172
 173   if(!(networkItem = rootItem->childById(qHash(net)))) {
 174     int nextRow = rootItem->childCount();
 175     networkItem = new NetworkTreeItem(net, rootItem);
 176
 177     beginInsertRows(QModelIndex(), nextRow, nextRow);
 178     rootItem->appendChild(networkItem);
 179     endInsertRows();
 180   }
 181
 182   Q_ASSERT(networkItem);
 183   return index(networkItem->row(), 0);
 184 }
 185
 186 QModelIndex BufferTreeModel::getOrCreateBufferItemIndex(Buffer *buffer) {
 187   QModelIndex networkItemIndex = getOrCreateNetworkItemIndex(buffer);
 188   NetworkTreeItem *networkItem = static_cast<NetworkTreeItem*>(networkItemIndex.internalPointer());
 189   TreeItem *bufferItem;
 190
 191   if(!(bufferItem = networkItem->childById(buffer->bufferInfo().uid()))) {
 192     int nextRow = networkItem->childCount();
 193     bufferItem = new BufferTreeItem(buffer, networkItem);
 194
 195     beginInsertRows(networkItemIndex, nextRow, nextRow);
 196     networkItem->appendChild(bufferItem);
 197     endInsertRows();
 198   }
 199
 200   Q_ASSERT(bufferItem);
 201   return index(bufferItem->row(), 0, networkItemIndex);
 202 }
 203
 204 QStringList BufferTreeModel::mimeTypes() const {
 205   QStringList types;
 206   types << "application/Quassel/BufferItem/row"
 207     << "application/Quassel/BufferItem/network"
 208     << "application/Quassel/BufferItem/bufferInfo";
 209   return types;
 210 }
 211
 212 QMimeData *BufferTreeModel::mimeData(const QModelIndexList &indexes) const {
 213   QMimeData *mimeData = new QMimeData();
 214
 215   QModelIndex index = indexes.first();
 216
 217   mimeData->setData("application/Quassel/BufferItem/row", QByteArray::number(index.row()));
 218   mimeData->setData("application/Quassel/BufferItem/network", getBufferByIndex(index)->networkName().toUtf8());
 219   mimeData->setData("application/Quassel/BufferItem/bufferInfo", QByteArray::number(getBufferByIndex(index)->bufferInfo().uid()));
 220   return mimeData;
 221 }
 222
 223 bool BufferTreeModel::dropMimeData(const QMimeData *data, Qt::DropAction /*action*/, int /*row*/, int /*column*/, const QModelIndex &parent) {
 224   foreach(QString mimeType, mimeTypes()) {
 225     if(!(data->hasFormat(mimeType)))
 226       return false; // whatever the drop is... it's not a buffer...
 227   }
 228
 229   int sourcerow = data->data("application/Quassel/BufferItem/row").toInt();
 230   QString network = QString::fromUtf8(data->data("application/Quassel/BufferItem/network"));
 231
 232   Q_ASSERT(rootItem->childById(qHash(network)));
 233
 234   if(parent == QModelIndex()) // can't be a query...
 235     return false;
 236
 237   Buffer *sourceBuffer = static_cast<BufferTreeItem *>(rootItem->childById(qHash(network))->child(sourcerow))->buffer();
 238   Buffer *targetBuffer = getBufferByIndex(parent);
 239
 240   if(!(sourceBuffer->bufferType() & targetBuffer->bufferType() & Buffer::QueryType)) // only queries can be merged
 241     return false;
 242
 243   if(sourceBuffer == targetBuffer) // we won't merge with ourself :)
 244     return false;
 245
 246   // TODO: warn user about buffermerge!
 247   qDebug() << "merging" << sourceBuffer->bufferName() << "with" << targetBuffer->bufferName();
 248   removeRow(parent.row(), BufferTreeModel::parent(parent));
 249
 250   return true;
 251 }
 252
 253 void BufferTreeModel::bufferUpdated(Buffer *buffer) {
 254   QModelIndex itemindex = getOrCreateBufferItemIndex(buffer);
 255   emit invalidateFilter();
 256   emit dataChanged(itemindex, itemindex);
 257 }
 258
 259 // This Slot indicates that the user has selected a different buffer in the gui
 260 void BufferTreeModel::setCurrentIndex(const QModelIndex &index, QItemSelectionModel::SelectionFlags command) {
 261   Q_UNUSED(command)
 262   if(isBufferIndex(index)) {
 263     currentBuffer = getBufferByIndex(index);
 264     bufferActivity(Buffer::NoActivity, currentBuffer);
 265     emit bufferSelected(currentBuffer);
 266     emit selectionChanged(index);
 267   }
 268 }
 269
 270 void BufferTreeModel::bufferActivity(Buffer::ActivityLevel level, Buffer *buffer) {
 271   BufferTreeItem *bufferItem = static_cast<BufferTreeItem*>(getOrCreateBufferItemIndex(buffer).internalPointer());
 272   if(buffer != currentBuffer)
 273     bufferItem->setActivity(level);
 274   else
 275     bufferItem->setActivity(Buffer::NoActivity);
 276   bufferUpdated(buffer);
 277 }
 278
 279 void BufferTreeModel::selectBuffer(Buffer *buffer) {
 280   QModelIndex index = getOrCreateBufferItemIndex(buffer);
 281   // SUPER UGLY!
 282   setCurrentIndex(index, 0);
 283 }