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