Make sure NickView::customEvent() has a valid model
[quassel.git] / src / uisupport / nickview.cpp
index 6b883d9..11b40c7 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-08 by the Quassel Project                          *
+ *   Copyright (C) 2005-09 by the Quassel Project                          *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include "nickview.h"
+
+#include <QApplication>
 #include <QHeaderView>
 #include <QScrollBar>
 #include <QDebug>
 #include <QMenu>
 
+#include "buffermodel.h"
+#include "client.h"
+#include "contextmenuactionprovider.h"
+#include "graphicalui.h"
 #include "nickview.h"
 #include "nickviewfilter.h"
 #include "networkmodel.h"
-#include "buffermodel.h"
 #include "types.h"
-#include "client.h"
 
 class ExpandAllEvent : public QEvent {
 public:
@@ -46,11 +51,16 @@ NickView::NickView(QWidget *parent)
   sortByColumn(0, Qt::AscendingOrder);
 
   setContextMenuPolicy(Qt::CustomContextMenu);
+  setSelectionMode(QAbstractItemView::ExtendedSelection);
+
+  connect(this, SIGNAL(customContextMenuRequested(const QPoint&)), SLOT(showContextMenu(const QPoint&)));
 
-  connect(this, SIGNAL(customContextMenuRequested(const QPoint&)),
-         this, SLOT(showContextMenu(const QPoint&)));
-  connect(this, SIGNAL(activated( const QModelIndex& )),
-          this, SLOT(startQuery( const QModelIndex& )));
+#if defined Q_WS_QWS || defined Q_WS_X11
+  connect(this, SIGNAL(doubleClicked(QModelIndex)), SLOT(startQuery(QModelIndex)));
+#else
+  // afaik this is better on Mac and Windows
+  connect(this, SIGNAL(activated(QModelIndex)), SLOT(startQuery(QModelIndex)));
+#endif
 }
 
 void NickView::init() {
@@ -59,10 +69,16 @@ void NickView::init() {
 
   for(int i = 1; i < model()->columnCount(); i++)
     setColumnHidden(i, true);
+
+  connect(selectionModel(), SIGNAL(currentChanged(QModelIndex, QModelIndex)), SIGNAL(selectionUpdated()));
+  connect(selectionModel(), SIGNAL(selectionChanged(QItemSelection, QItemSelection)), SIGNAL(selectionUpdated()));
 }
 
-void NickView::setModel(QAbstractItemModel *model) {
-  QTreeView::setModel(model);
+void NickView::setModel(QAbstractItemModel *model_) {
+  if(model())
+    disconnect(model(), 0, this, 0);
+
+  QTreeView::setModel(model_);
   init();
 }
 
@@ -79,96 +95,40 @@ void NickView::setRootIndex(const QModelIndex &index) {
     QCoreApplication::postEvent(this, new ExpandAllEvent);
 }
 
-QString NickView::nickFromModelIndex(const QModelIndex & index) {
-  QString nick = index.sibling(index.row(), 0).data().toString();
-  return nick;
-}
-
-BufferInfo NickView::bufferInfoFromModelIndex(const QModelIndex & index) {
-  BufferInfo bufferInfo = index.data(NetworkModel::BufferInfoRole).value<BufferInfo>();
-  return bufferInfo;
-}
-
-void NickView::showContextMenu(const QPoint & pos ) {
-  QModelIndex index = indexAt(pos);
-  if(index.data(NetworkModel::ItemTypeRole) != NetworkModel::IrcUserItemType) return;
-
-  QString nick = nickFromModelIndex(index);
-
-  QMenu nickContextMenu(this);
-
-  QAction *whoisAction = nickContextMenu.addAction(tr("WHOIS"));
-  QAction *versionAction = nickContextMenu.addAction(tr("VERSION"));
-  QAction *pingAction = nickContextMenu.addAction(tr("PING"));
-
-  nickContextMenu.addSeparator();
+QModelIndexList NickView::selectedIndexes() const {
+  QModelIndexList indexList = QTreeView::selectedIndexes();
 
-  QMenu *modeMenu = nickContextMenu.addMenu(tr("Modes"));
-  QAction *opAction = modeMenu->addAction(tr("Op %1").arg(nick));
-  QAction *deOpAction = modeMenu->addAction(tr("Deop %1").arg(nick));
-  QAction *voiceAction = modeMenu->addAction(tr("Voice %1").arg(nick));
-  QAction *deVoiceAction = modeMenu->addAction(tr("Devoice %1").arg(nick));
-
-  QMenu *kickBanMenu = nickContextMenu.addMenu(tr("Kick/Ban"));
-  QAction *kickAction = kickBanMenu->addAction(tr("Kick %1").arg(nick));
-  QAction *banAction = kickBanMenu->addAction(tr("Ban %1").arg(nick));
-  QAction *kickBanAction = kickBanMenu->addAction(tr("Kickban %1").arg(nick));
-  QAction *ignoreAction = nickContextMenu.addAction(tr("Ignore"));
-  ignoreAction->setEnabled(false);
-
-  nickContextMenu.addSeparator();
-
-  QAction *queryAction = nickContextMenu.addAction(tr("Query"));
-  QAction *dccChatAction = nickContextMenu.addAction(tr("DCC-Chat"));
-  dccChatAction->setEnabled(false);
-  QAction *sendFileAction = nickContextMenu.addAction(tr("Send file"));
-  sendFileAction->setEnabled(false);
-
-  QAction *action = nickContextMenu.exec(QCursor::pos());
-  BufferInfo bufferInfo = bufferInfoFromModelIndex(index);
-
-  if(action == whoisAction)         { executeCommand(bufferInfo, QString("/WHOIS %1 %1").arg(nick)); }
-  else if(action == versionAction)  { executeCommand(bufferInfo, QString("/CTCP %1 VERSION").arg(nick)); }
-  else if(action == pingAction)     { executeCommand(bufferInfo, QString("/CTCP %1 PING ").arg(nick)); }
+  // make sure the item we clicked on is first
+  if(indexList.contains(currentIndex())) {
+    indexList.removeAll(currentIndex());
+    indexList.prepend(currentIndex());
+  }
 
-  else if(action == opAction)       { executeCommand(bufferInfo, QString("/OP %1").arg(nick)); }
-  else if(action == deOpAction)     { executeCommand(bufferInfo, QString("/DEOP %1").arg(nick)); }
-  else if(action == voiceAction)    { executeCommand(bufferInfo, QString("/VOICE %1").arg(nick)); }
-  else if(action == deVoiceAction)  { executeCommand(bufferInfo, QString("/DEVOICE %1").arg(nick)); }
+  return indexList;
+}
 
-  else if(action == kickAction)     { executeCommand(bufferInfo, QString("/KICK %1").arg(nick)); }
-  else if(action == banAction)      { executeCommand(bufferInfo, QString("/BAN %1").arg(nick)); }
-  else if(action == kickBanAction)  { executeCommand(bufferInfo, QString("/KICK %1").arg(nick)); 
-                                      executeCommand(bufferInfo, QString("/BAN %1").arg(nick)); }
-  else if(action == queryAction)    { startQuery(index); }
+void NickView::showContextMenu(const QPoint &pos ) {
+  Q_UNUSED(pos);
 
+  QMenu contextMenu(this);
+  GraphicalUi::contextMenuActionProvider()->addActions(&contextMenu, selectedIndexes());
+  contextMenu.exec(QCursor::pos());
 }
 
-void NickView::startQuery(const QModelIndex & index) {
-  QString nick = nickFromModelIndex(index);
-  bool activated = false;
-
-  if(QSortFilterProxyModel *nickviewFilter = qobject_cast<QSortFilterProxyModel *>(model())) {
-    // rootIndex() is the channel, parent() is the corresponding network
-    QModelIndex networkIndex = rootIndex().parent(); 
-    QModelIndex source_networkIndex = nickviewFilter->mapToSource(networkIndex);
-    for(int i = 0; i < Client::networkModel()->rowCount(source_networkIndex); i++) {
-      QModelIndex childIndex = source_networkIndex.child( i, 0);
-      if(nick.toLower() == childIndex.data().toString().toLower()) {
-        QModelIndex queryIndex = Client::bufferModel()->mapFromSource(childIndex);
-        Client::bufferModel()->setCurrentIndex(queryIndex);
-        activated = true;
-      }
-    }
-  }
-  if(!activated) {
-    BufferInfo bufferInfo = bufferInfoFromModelIndex(index);
-    executeCommand(bufferInfo, QString("/QUERY %1").arg(nick));
-  }
-}
+void NickView::startQuery(const QModelIndex &index) {
+  if(index.data(NetworkModel::ItemTypeRole) != NetworkModel::IrcUserItemType)
+    return;
+
+  IrcUser *ircUser = qobject_cast<IrcUser *>(index.data(NetworkModel::IrcUserRole).value<QObject *>());
+  NetworkId networkId = index.data(NetworkModel::NetworkIdRole).value<NetworkId>();
+  if(!ircUser || !networkId.isValid())
+    return;
 
-void NickView::executeCommand(const BufferInfo & bufferInfo, const QString & command) {
-  Client::instance()->userInput(bufferInfo, command);
+  BufferId bufId = Client::networkModel()->bufferId(networkId, ircUser->nick());
+  if(bufId.isValid())
+    Client::bufferModel()->switchToBuffer(bufId);
+  else
+    Client::userInput(index.data(NetworkModel::BufferInfoRole).value<BufferInfo>(), QString("/QUERY %1").arg(ircUser->nick()));
 }
 
 void NickView::customEvent(QEvent *event) {
@@ -184,6 +144,9 @@ void NickView::customEvent(QEvent *event) {
   if(event->type() != QEvent::User)
     return;
 
+  if(!model())
+    return;
+
   QModelIndex topLevelIdx;
   for(int i = 0; i < model()->rowCount(rootIndex()); i++) {
     topLevelIdx = model()->index(i, 0, rootIndex());