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