Always put own nick last in nick completion
[quassel.git] / src / uisupport / tabcompleter.cpp
index 2b13097..695923f 100644 (file)
 /***************************************************************************
- *   Copyright (C) 2005/06 by The Quassel Team                             *
- *   devel@quassel-irc.org                                                 *
- *                                                                         *
- *   This program is free software; you can redistribute it and/or modify  *
- *   it under the terms of the GNU General Public License as published by  *
- *   the Free Software Foundation; either version 2 of the License, or     *
- *   (at your option) any later version.                                   *
- *                                                                         *
- *   This program is distributed in the hope that it will be useful,       *
- *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
- *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
- *   GNU General Public License for more details.                          *
- *                                                                         *
- *   You should have received a copy of the GNU General Public License     *
- *   along with this program; if not, write to the                         *
- *   Free Software Foundation, Inc.,                                       *
- *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
- ***************************************************************************/
+*   Copyright (C) 2005-09 by the Quassel Project                          *
+*   devel@quassel-irc.org                                                 *
+*                                                                         *
+*   This program is free software; you can redistribute it and/or modify  *
+*   it under the terms of the GNU General Public License as published by  *
+*   the Free Software Foundation; either version 2 of the License, or     *
+*   (at your option) version 3.                                           *
+*                                                                         *
+*   This program is distributed in the hope that it will be useful,       *
+*   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+*   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+*   GNU General Public License for more details.                          *
+*                                                                         *
+*   You should have received a copy of the GNU General Public License     *
+*   along with this program; if not, write to the                         *
+*   Free Software Foundation, Inc.,                                       *
+*   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+***************************************************************************/
 
 #include "tabcompleter.h"
 
-TabCompleter::TabCompleter(QLineEdit *l, QObject *parent) : QObject(parent) {
-  lineEdit = l;
-  enabled = false;
-  startOfLineSuffix = QString(": "); // TODO make start of line suffix configurable
-}
+#include "inputline.h"
+#include "client.h"
+#include "buffermodel.h"
+#include "networkmodel.h"
+#include "network.h"
+#include "ircchannel.h"
+#include "ircuser.h"
+#include "uisettings.h"
 
-void TabCompleter::updateNickList(QStringList l) {
-  nickList = l;
-}
+#include <QRegExp>
+
+const Network *TabCompleter::_currentNetwork;
+BufferId TabCompleter::_currentBufferId;
 
-void TabCompleter::updateChannelList(QStringList l) {
-  channelList = l;
+TabCompleter::TabCompleter(InputLine *inputLine_)
+  : QObject(inputLine_),
+    inputLine(inputLine_),
+    enabled(false),
+    nickSuffix(": ")
+{
+  inputLine->installEventFilter(this);
 }
 
 void TabCompleter::buildCompletionList() {
+  // ensure a safe state in case we return early.
+  completionMap.clear();
+  nextCompletion = completionMap.begin();
+
   // this is the first time tab is pressed -> build up the completion list and it's iterator
-  QString tabAbbrev = lineEdit->text().left(lineEdit->cursorPosition()).section(' ',-1,-1);
-  completionList.clear();
-  foreach(QString nick, nickList) {
-    if(nick.toLower().startsWith(tabAbbrev.toLower())) {
-      completionList << nick;
+  QModelIndex currentIndex = Client::bufferModel()->currentIndex();
+  _currentBufferId = currentIndex.data(NetworkModel::BufferIdRole).value<BufferId>();
+  if(!_currentBufferId.isValid())
+    return;
+
+  NetworkId networkId = currentIndex.data(NetworkModel::NetworkIdRole).value<NetworkId>();
+  QString bufferName = currentIndex.sibling(currentIndex.row(), 0).data().toString();
+
+  _currentNetwork = Client::network(networkId);
+  if(!_currentNetwork)
+    return;
+
+  QString tabAbbrev = inputLine->text().left(inputLine->cursorPosition()).section(' ',-1,-1);
+  QRegExp regex(QString("^[^a-zA-Z]*").append(QRegExp::escape(tabAbbrev)), Qt::CaseInsensitive);
+
+  switch(static_cast<BufferInfo::Type>(currentIndex.data(NetworkModel::BufferTypeRole).toInt())) {
+  case BufferInfo::ChannelBuffer:
+    { // scope is needed for local var declaration
+      IrcChannel *channel = _currentNetwork->ircChannel(bufferName);
+      if(!channel)
+        return;
+      foreach(IrcUser *ircUser, channel->ircUsers()) {
+        if(regex.indexIn(ircUser->nick()) > -1)
+          completionMap[ircUser->nick().toLower()] = ircUser->nick();
+      }
     }
+    break;
+  case BufferInfo::QueryBuffer:
+    if(regex.indexIn(bufferName) > -1)
+      completionMap[bufferName.toLower()] = bufferName;
+  case BufferInfo::StatusBuffer:
+    if(!_currentNetwork->myNick().isEmpty() && regex.indexIn(_currentNetwork->myNick()) > -1)
+      completionMap[_currentNetwork->myNick().toLower()] = _currentNetwork->myNick();
+    break;
+  default:
+    return;
   }
-  completionList.sort();
-  nextCompletion = completionList.begin();
+
+  nextCompletion = completionMap.begin();
   lastCompletionLength = tabAbbrev.length();
 }
 
 void TabCompleter::complete() {
-  if (! enabled) {
+  NickCompletionSettings s;
+  nickSuffix = s.completionSuffix();
+
+  if(!enabled) {
     buildCompletionList();
-    enabled = true;  
+    enabled = true;
   }
-  
-  if (nextCompletion != completionList.end()) {
+
+  if (nextCompletion != completionMap.end()) {
     // clear previous completion
     for (int i = 0; i < lastCompletionLength; i++) {
-      lineEdit->backspace();
+      inputLine->backspace();
     }
-    
+
     // insert completion
-    lineEdit->insert(*nextCompletion);
-    
+    inputLine->insert(*nextCompletion);
+
     // remember charcount to delete next time and advance to next completion
     lastCompletionLength = nextCompletion->length();
     nextCompletion++;
-    
+
     // we're completing the first word of the line
-    if(lineEdit->text().length() == lastCompletionLength) {
-      lineEdit->insert(startOfLineSuffix);
-      lastCompletionLength += 2;
+    if(inputLine->cursorPosition() == lastCompletionLength) {
+      inputLine->insert(nickSuffix);
+      lastCompletionLength += nickSuffix.length();
     }
 
   // we're at the end of the list -> start over again
   } else {
-    nextCompletion = completionList.begin();
+    if(!completionMap.isEmpty()) {
+      nextCompletion = completionMap.begin();
+      complete();
+    }
   }
-  
 }
 
-void TabCompleter::disable() {
+void TabCompleter::reset() {
   enabled = false;
 }
 
+bool TabCompleter::eventFilter(QObject *obj, QEvent *event) {
+  if(obj != inputLine || event->type() != QEvent::KeyPress)
+    return QObject::eventFilter(obj, event);
+
+  QKeyEvent *keyEvent = static_cast<QKeyEvent *>(event);
+
+  if(keyEvent->key() == Qt::Key_Tab) {
+    complete();
+    return true;
+  } else {
+    reset();
+    return false;
+  }
+}
+
+// this determines the sort order
+bool TabCompleter::CompletionKey::operator<(const CompletionKey &other) const {
+  IrcUser *thisUser = _currentNetwork->ircUser(this->nick);
+  if(thisUser && _currentNetwork->isMe(thisUser))
+    return false;
+
+  IrcUser *thatUser = _currentNetwork->ircUser(other.nick);
+  if(thatUser && _currentNetwork->isMe(thatUser))
+    return true;
+
+  if(!thisUser || !thatUser)
+    return QString::localeAwareCompare(this->nick, other.nick) < 0;
+
+  QDateTime thisSpokenTo = thisUser->lastSpokenTo(_currentBufferId);
+  QDateTime thatSpokenTo = thatUser->lastSpokenTo(_currentBufferId);
+
+  if(thisSpokenTo.isValid() || thatSpokenTo.isValid())
+    return thisSpokenTo > thatSpokenTo;
+
+  QDateTime thisTime = thisUser->lastChannelActivity(_currentBufferId);
+  QDateTime thatTime = thatUser->lastChannelActivity(_currentBufferId);
+
+  if(thisTime.isValid() || thatTime.isValid())
+    return thisTime > thatTime;
+
+  return QString::localeAwareCompare(this->nick, other.nick) < 0;
+}