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 "bufferinfo.h"
  26 #include "client.h"
  27 #include "signalproxy.h"
  28
  29 /*****************************************
  30 *  Fancy Buffer Items
  31 *****************************************/
  32 BufferTreeItem::BufferTreeItem(Buffer *buffer, TreeItem *parent) : TreeItem(parent) {
  33   buf = buffer;
  34   activity = Buffer::NoActivity;
  35 }
  36
  37 uint BufferTreeItem::id() const {
  38   return buf->bufferInfo().uid();
  39 }
  40
  41 void BufferTreeItem::setActivity(const Buffer::ActivityLevel &level) {
  42   activity = level;
  43 }
  44
  45 QString BufferTreeItem::text(int column) const {
  46   switch(column) {
  47     case 0:
  48       return buf->displayName();
  49     case 1:
  50       return buf->networkName();
  51     default:
  52       return QString();
  53   }
  54 }
  55
  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);
  64   } else {
  65     if(buf->isActive())
  66       return QColor(Qt::black);
  67     else
  68       return QColor(Qt::gray);
  69   }
  70 }
  71
  72
  73 QVariant BufferTreeItem::data(int column, int role) const {
  74   switch(role) {
  75     case Qt::DisplayRole:
  76       return text(column);
  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();
  87     default:
  88       return QVariant();
  89   }
  90 }
  91
  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;
  96
  97   return flags;
  98 }
  99
 100 /*****************************************
 101 *  Network Items
 102 *****************************************/
 103 NetworkTreeItem::NetworkTreeItem(const QString &network, TreeItem *parent) : TreeItem(parent) {
 104   net = network;
 105   itemData << net << "";
 106 }
 107
 108 uint NetworkTreeItem::id() const {
 109   return qHash(net);
 110 }
 111
 112 Qt::ItemFlags NetworkTreeItem::flags() const {
 113   return Qt::ItemIsEnabled | Qt::ItemIsDropEnabled;
 114 }
 115
 116 /*****************************************
 117  * BufferTreeModel
 118  *****************************************/
 119 BufferTreeModel::BufferTreeModel(QObject *parent)
 120   : TreeModel(BufferTreeModel::defaultHeader(), parent)
 121 {
 122   Client::signalProxy()->attachSignal(this, SIGNAL(fakeUserInput(BufferInfo, QString)), SIGNAL(sendInput(BufferInfo, QString)));
 123 }
 124
 125 QList<QVariant >BufferTreeModel::defaultHeader() {
 126   QList<QVariant> data;
 127   data << tr("Buffer") << tr("Network");
 128   return data;
 129 }
 130
 131 bool BufferTreeModel::isBufferIndex(const QModelIndex &index) const {
 132   // not so purdy...
 133   return parent(index) != QModelIndex();
 134 }
 135
 136 Buffer *BufferTreeModel::getBufferByIndex(const QModelIndex &index) const {
 137   BufferTreeItem *item = static_cast<BufferTreeItem *>(index.internalPointer());
 138   return item->buffer();
 139 }
 140
 141 QModelIndex BufferTreeModel::getOrCreateNetworkItemIndex(Buffer *buffer) {
 142   QString net = buffer->networkName();
 143   TreeItem *networkItem;
 144
 145   if(!(networkItem = rootItem->childById(qHash(net)))) {
 146     int nextRow = rootItem->childCount();
 147     networkItem = new NetworkTreeItem(net, rootItem);
 148
 149     beginInsertRows(QModelIndex(), nextRow, nextRow);
 150     rootItem->appendChild(networkItem);
 151     endInsertRows();
 152   }
 153
 154   Q_ASSERT(networkItem);
 155   return index(networkItem->row(), 0);
 156 }
 157
 158 QModelIndex BufferTreeModel::getOrCreateBufferItemIndex(Buffer *buffer) {
 159   QModelIndex networkItemIndex = getOrCreateNetworkItemIndex(buffer);
 160   NetworkTreeItem *networkItem = static_cast<NetworkTreeItem*>(networkItemIndex.internalPointer());
 161   TreeItem *bufferItem;
 162
 163   if(!(bufferItem = networkItem->childById(buffer->bufferInfo().uid()))) {
 164     int nextRow = networkItem->childCount();
 165     bufferItem = new BufferTreeItem(buffer, networkItem);
 166
 167     beginInsertRows(networkItemIndex, nextRow, nextRow);
 168     networkItem->appendChild(bufferItem);
 169     endInsertRows();
 170   }
 171
 172   Q_ASSERT(bufferItem);
 173   return index(bufferItem->row(), 0, networkItemIndex);
 174 }
 175
 176 QStringList BufferTreeModel::mimeTypes() const {
 177   QStringList types;
 178   types << "application/Quassel/BufferItem/row"
 179     << "application/Quassel/BufferItem/network"
 180     << "application/Quassel/BufferItem/bufferInfo";
 181   return types;
 182 }
 183
 184 QMimeData *BufferTreeModel::mimeData(const QModelIndexList &indexes) const {
 185   QMimeData *mimeData = new QMimeData();
 186
 187   QModelIndex index = indexes.first();
 188
 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()));
 192   return mimeData;
 193 }
 194
 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...
 199   }
 200
 201   int sourcerow = data->data("application/Quassel/BufferItem/row").toInt();
 202   QString network = QString::fromUtf8(data->data("application/Quassel/BufferItem/network"));
 203
 204   Q_ASSERT(rootItem->childById(qHash(network)));
 205
 206   if(parent == QModelIndex()) // can't be a query...
 207     return false;
 208
 209   Buffer *sourceBuffer = static_cast<BufferTreeItem *>(rootItem->childById(qHash(network))->child(sourcerow))->buffer();
 210   Buffer *targetBuffer = getBufferByIndex(parent);
 211
 212   if(!(sourceBuffer->bufferType() & targetBuffer->bufferType() & Buffer::QueryBuffer)) // only queries can be merged
 213     return false;
 214
 215   if(sourceBuffer == targetBuffer) // we won't merge with ourself :)
 216     return false;
 217
 218   // TODO: warn user about buffermerge!
 219   qDebug() << "merging" << sourceBuffer->bufferName() << "with" << targetBuffer->bufferName();
 220   removeRow(parent.row(), BufferTreeModel::parent(parent));
 221
 222   return true;
 223 }
 224
 225 void BufferTreeModel::bufferUpdated(Buffer *buffer) {
 226   QModelIndex itemindex = getOrCreateBufferItemIndex(buffer);
 227   emit invalidateFilter();
 228   emit dataChanged(itemindex, itemindex);
 229 }
 230
 231 // This Slot indicates that the user has selected a different buffer in the gui
 232 void BufferTreeModel::changeCurrent(const QModelIndex &current, const QModelIndex &/*previous*/) {
 233   if(isBufferIndex(current)) {
 234     currentBuffer = getBufferByIndex(current);
 235     bufferActivity(Buffer::NoActivity, currentBuffer);
 236     emit bufferSelected(currentBuffer);
 237     emit selectionChanged(current);
 238   }
 239 }
 240
 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()));
 247   }
 248 }
 249
 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);
 254   else
 255     bufferItem->setActivity(Buffer::NoActivity);
 256   bufferUpdated(buffer);
 257 }
 258
 259 void BufferTreeModel::selectBuffer(Buffer *buffer) {
 260   QModelIndex index = getOrCreateBufferItemIndex(buffer);
 261   //emit selectionChanged(index);
 262   changeCurrent(index, QModelIndex());
 263 }