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