03d49ecd3ac0fb6e537375c9304da69d7b277f3d
[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
31 #include <QRegExp>
32
33 TabCompleter::TabCompleter(InputLine *inputLine_)
34   : QObject(inputLine_),
35     inputLine(inputLine_),
36     enabled(false),
37     nickSuffix(": ")
38 {
39   inputLine->installEventFilter(this);
40 }
41
42 void TabCompleter::buildCompletionList() {
43   // ensure a safe state in case we return early.
44   completionMap.clear();
45   nextCompletion = completionMap.begin();
46
47   // this is the first time tab is pressed -> build up the completion list and it's iterator
48   QModelIndex currentIndex = Client::bufferModel()->currentIndex();
49   if(!currentIndex.data(NetworkModel::BufferIdRole).isValid())
50     return;
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   if(!enabled) {
92     buildCompletionList();
93     enabled = true;
94   }
95   
96   if (nextCompletion != completionMap.end()) {
97     // clear previous completion
98     for (int i = 0; i < lastCompletionLength; i++) {
99       inputLine->backspace();
100     }
101     
102     // insert completion
103     inputLine->insert(*nextCompletion);
104     
105     // remember charcount to delete next time and advance to next completion
106     lastCompletionLength = nextCompletion->length();
107     nextCompletion++;
108     
109     // we're completing the first word of the line
110     if(inputLine->text().length() == lastCompletionLength) {
111       inputLine->insert(nickSuffix);
112       lastCompletionLength += nickSuffix.length();
113     }
114
115   // we're at the end of the list -> start over again
116   } else {
117     if(!completionMap.isEmpty()) {
118       nextCompletion = completionMap.begin();
119       complete();
120     }
121   }
122   
123 }
124
125 void TabCompleter::reset() {
126   enabled = false;
127 }
128
129 bool TabCompleter::eventFilter(QObject *obj, QEvent *event) {
130   if(obj != inputLine || event->type() != QEvent::KeyPress)
131     return QObject::eventFilter(obj, event);
132
133   QKeyEvent *keyEvent = static_cast<QKeyEvent *>(event);
134   
135   if(keyEvent->key() == Qt::Key_Tab) {
136     complete();
137     return true;
138   } else {
139     reset();
140     return false;
141   }
142 }
143