Fixed tabcompleter not completing some special characters correctly.
[quassel.git] / src / uisupport / tabcompleter.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005/06 by the Quassel Project                          *
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) version 3.                                           *
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 "tabcompleter.h"
22
23 #include "inputline.h"
24 #include "client.h"
25 #include "buffermodel.h"
26 #include "networkmodel.h"
27 #include "network.h"
28 #include "ircchannel.h"
29 #include "ircuser.h"
30 #include "uisettings.h"
31
32 #include <QRegExp>
33
34 TabCompleter::TabCompleter(InputLine *inputLine_)
35   : QObject(inputLine_),
36     inputLine(inputLine_),
37     enabled(false),
38     nickSuffix(": ")
39 {
40   inputLine->installEventFilter(this);
41 }
42
43 void TabCompleter::buildCompletionList() {
44   completionMap.clear();
45   // this is the first time tab is pressed -> build up the completion list and it's iterator
46   QModelIndex currentIndex = Client::bufferModel()->currentIndex();
47   if(!currentIndex.data(NetworkModel::BufferIdRole).isValid()) {
48     nextCompletion = completionMap.begin();
49     return;
50   }
51   
52   NetworkId networkId = currentIndex.data(NetworkModel::NetworkIdRole).value<NetworkId>();
53   QString channelName = currentIndex.sibling(currentIndex.row(), 0).data().toString();
54
55   const Network *network = Client::network(networkId);
56   if(!network)
57     return;
58
59   IrcChannel *channel = network->ircChannel(channelName);
60   if(!channel)
61     return;
62
63   // FIXME commented for debugging
64   /*
65   disconnect(this, SLOT(ircUserJoinedOrParted(IrcUser *)));
66   connect(channel, SIGNAL(ircUserJoined(IrcUser *)),
67           this, SLOT(ircUserJoinedOrParted(IrcUser *)));
68   connect(channel, SIGNAL(ircUserParted(IrcUser *)),
69           this, SLOT(ircUserJoinedOrParted(IrcUser *)));
70   */
71
72   QString tabAbbrev = inputLine->text().left(inputLine->cursorPosition()).section(' ',-1,-1);
73   QRegExp regex(QString("^[^a-zA-Z]*").append(QRegExp::escape(tabAbbrev)), Qt::CaseInsensitive);
74
75   foreach(IrcUser *ircUser, channel->ircUsers()) {
76     if(regex.indexIn(ircUser->nick()) > -1) {
77       completionMap[ircUser->nick().toLower()] = ircUser->nick();
78     }
79   }
80
81   nextCompletion = completionMap.begin();
82   lastCompletionLength = tabAbbrev.length();
83 }
84
85 void TabCompleter::ircUserJoinedOrParted(IrcUser *ircUser) {
86   Q_UNUSED(ircUser)
87   buildCompletionList();
88 }
89
90 void TabCompleter::complete() {
91   UiSettings uiSettings;
92   nickSuffix = uiSettings.value("CompletionSuffix", QString(": ")).toString();
93   
94   if(!enabled) {
95     buildCompletionList();
96     enabled = true;
97   }
98   
99   if (nextCompletion != completionMap.end()) {
100     // clear previous completion
101     for (int i = 0; i < lastCompletionLength; i++) {
102       inputLine->backspace();
103     }
104     
105     // insert completion
106     inputLine->insert(*nextCompletion);
107     
108     // remember charcount to delete next time and advance to next completion
109     lastCompletionLength = nextCompletion->length();
110     nextCompletion++;
111     
112     // we're completing the first word of the line
113     if(inputLine->text().length() == lastCompletionLength) {
114       inputLine->insert(nickSuffix);
115       lastCompletionLength += nickSuffix.length();
116     }
117
118   // we're at the end of the list -> start over again
119   } else {
120     if(!completionMap.isEmpty()) {
121       nextCompletion = completionMap.begin();
122       complete();
123     }
124   }
125   
126 }
127
128 void TabCompleter::reset() {
129   enabled = false;
130 }
131
132 bool TabCompleter::eventFilter(QObject *obj, QEvent *event) {
133   if(obj != inputLine || event->type() != QEvent::KeyPress)
134     return QObject::eventFilter(obj, event);
135
136   QKeyEvent *keyEvent = static_cast<QKeyEvent *>(event);
137   
138   if(keyEvent->key() == Qt::Key_Tab) {
139     complete();
140     return true;
141   } else {
142     reset();
143     return false;
144   }
145 }
146