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