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