1 /***************************************************************************
2 * Copyright (C) 2005-09 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 #include "userinputhandler.h"
24 #include "ctcphandler.h"
25 #include "coreidentity.h"
31 UserInputHandler::UserInputHandler(CoreNetwork *parent)
32 : BasicHandler(parent)
36 void UserInputHandler::handleUserInput(const BufferInfo &bufferInfo, const QString &msg) {
40 AliasManager::CommandList list = coreSession()->aliasManager().processInput(bufferInfo, msg);
42 for(int i = 0; i < list.count(); i++) {
43 QString cmd = list.at(i).second.section(' ', 0, 0).remove(0, 1).toUpper();
44 QString payload = list.at(i).second.section(' ', 1);
45 handle(cmd, Q_ARG(BufferInfo, list.at(i).first), Q_ARG(QString, payload));
49 // ====================
51 // ====================
52 void UserInputHandler::handleAway(const BufferInfo &bufferInfo, const QString &msg) {
55 QString awayMsg = msg;
56 IrcUser *me = network()->me();
58 // if there is no message supplied we have to check if we are already away or not
60 if(me && !me->isAway()) {
61 awayMsg = network()->identityPtr()->awayReason();
62 if(awayMsg.isEmpty()) {
68 me->setAwayMessage(awayMsg);
70 putCmd("AWAY", serverEncode(awayMsg));
73 void UserInputHandler::handleBan(const BufferInfo &bufferInfo, const QString &msg) {
74 banOrUnban(bufferInfo, msg, true);
77 void UserInputHandler::handleUnban(const BufferInfo &bufferInfo, const QString &msg) {
78 banOrUnban(bufferInfo, msg, false);
81 void UserInputHandler::banOrUnban(const BufferInfo &bufferInfo, const QString &msg, bool ban) {
85 QStringList params = msg.split(" ");
87 if(!params.isEmpty() && isChannelName(params[0])) {
88 banChannel = params.takeFirst();
89 } else if(bufferInfo.type() == BufferInfo::ChannelBuffer) {
90 banChannel = bufferInfo.bufferName();
92 emit displayMsg(Message::Error, BufferInfo::StatusBuffer, "", QString("Error: channel unknown in command: /BAN %1").arg(msg));
96 if(!params.isEmpty() && !params.contains("!") && network()->ircUser(params[0])) {
97 IrcUser *ircuser = network()->ircUser(params[0]);
98 // generalizedHost changes <nick> to *!ident@*.sld.tld.
99 QString generalizedHost = ircuser->host();
100 if(generalizedHost.isEmpty()) {
101 emit displayMsg(Message::Error, BufferInfo::StatusBuffer, "", QString("Error: host unknown in command: /BAN %1").arg(msg));
105 static QRegExp ipAddress("\\d+\\.\\d+\\.\\d+\\.\\d+");
106 if(ipAddress.exactMatch(generalizedHost)) {
107 int lastDotPos = generalizedHost.lastIndexOf('.') + 1;
108 generalizedHost.replace(lastDotPos, generalizedHost.length() - lastDotPos, '*');
109 } else if(generalizedHost.lastIndexOf(".") != -1 && generalizedHost.lastIndexOf(".", generalizedHost.lastIndexOf(".")-1) != -1) {
110 int secondLastPeriodPosition = generalizedHost.lastIndexOf(".", generalizedHost.lastIndexOf(".")-1);
111 generalizedHost.replace(0, secondLastPeriodPosition, "*");
113 banUser = QString("*!%1@%2").arg(ircuser->user(), generalizedHost);
115 banUser = params.join(" ");
118 QString banMode = ban ? "+b" : "-b";
119 QString banMsg = QString("MODE %1 %2 %3").arg(banChannel, banMode, banUser);
120 emit putRawLine(serverEncode(banMsg));
123 void UserInputHandler::handleCtcp(const BufferInfo &bufferInfo, const QString &msg) {
126 QString nick = msg.section(' ', 0, 0);
127 QString ctcpTag = msg.section(' ', 1, 1).toUpper();
128 if(ctcpTag.isEmpty())
131 QString message = "";
132 QString verboseMessage = tr("sending CTCP-%1 request").arg(ctcpTag);
134 if(ctcpTag == "PING") {
135 uint now = QDateTime::currentDateTime().toTime_t();
136 message = QString::number(now);
139 network()->ctcpHandler()->query(nick, ctcpTag, message);
140 emit displayMsg(Message::Action, BufferInfo::StatusBuffer, "", verboseMessage, network()->myNick());
143 void UserInputHandler::handleDeop(const BufferInfo &bufferInfo, const QString &msg) {
144 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
145 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'o';
147 params << bufferInfo.bufferName() << m << nicks;
148 emit putCmd("MODE", serverEncode(params));
151 void UserInputHandler::handleDevoice(const BufferInfo &bufferInfo, const QString &msg) {
152 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
153 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'v';
155 params << bufferInfo.bufferName() << m << nicks;
156 emit putCmd("MODE", serverEncode(params));
159 void UserInputHandler::handleInvite(const BufferInfo &bufferInfo, const QString &msg) {
161 params << msg << bufferInfo.bufferName();
162 emit putCmd("INVITE", serverEncode(params));
165 void UserInputHandler::handleJoin(const BufferInfo &bufferInfo, const QString &msg) {
166 Q_UNUSED(bufferInfo);
168 // trim spaces before chans or keys
169 QString sane_msg = msg;
170 sane_msg.replace(QRegExp(", +"), ",");
171 QStringList params = sane_msg.trimmed().split(" ");
172 QStringList chans = params[0].split(",");
175 for(i = 0; i < chans.count(); i++) {
176 if(!network()->isChannelName(chans[i]))
177 chans[i].prepend('#');
179 params[0] = chans.join(",");
180 if(params.count() > 1) keys = params[1].split(",");
181 emit putCmd("JOIN", serverEncode(params)); // FIXME handle messages longer than 512 bytes!
183 for(; i < keys.count(); i++) {
184 if(i >= chans.count()) break;
185 network()->addChannelKey(chans[i], keys[i]);
187 for(; i < chans.count(); i++) {
188 network()->removeChannelKey(chans[i]);
192 void UserInputHandler::handleKick(const BufferInfo &bufferInfo, const QString &msg) {
193 QString nick = msg.section(' ', 0, 0, QString::SectionSkipEmpty);
194 QString reason = msg.section(' ', 1, -1, QString::SectionSkipEmpty).trimmed();
196 reason = network()->identityPtr()->kickReason();
198 QList<QByteArray> params;
199 params << serverEncode(bufferInfo.bufferName()) << serverEncode(nick) << channelEncode(bufferInfo.bufferName(), reason);
200 emit putCmd("KICK", params);
203 void UserInputHandler::handleKill(const BufferInfo &bufferInfo, const QString &msg) {
205 QString nick = msg.section(' ', 0, 0, QString::SectionSkipEmpty);
206 QString pass = msg.section(' ', 1, -1, QString::SectionSkipEmpty);
207 QList<QByteArray> params;
208 params << serverEncode(nick) << serverEncode(pass);
209 emit putCmd("KILL", params);
213 void UserInputHandler::handleList(const BufferInfo &bufferInfo, const QString &msg) {
215 emit putCmd("LIST", serverEncode(msg.split(' ', QString::SkipEmptyParts)));
218 void UserInputHandler::handleMe(const BufferInfo &bufferInfo, const QString &msg) {
219 if(bufferInfo.bufferName().isEmpty()) return; // server buffer
220 network()->ctcpHandler()->query(bufferInfo.bufferName(), "ACTION", msg);
221 emit displayMsg(Message::Action, bufferInfo.type(), bufferInfo.bufferName(), msg, network()->myNick(), Message::Self);
224 void UserInputHandler::handleMode(const BufferInfo &bufferInfo, const QString &msg) {
227 QStringList params = msg.split(' ', QString::SkipEmptyParts);
228 // if the first argument is neither a channel nor us (user modes are only to oneself) the current buffer is assumed to be the target
229 if(!params.isEmpty() && !network()->isChannelName(params[0]) && !network()->isMyNick(params[0]))
230 params.prepend(bufferInfo.bufferName());
232 // TODO handle correct encoding for buffer modes (channelEncode())
233 emit putCmd("MODE", serverEncode(params));
236 // TODO: show privmsgs
237 void UserInputHandler::handleMsg(const BufferInfo &bufferInfo, const QString &msg) {
238 Q_UNUSED(bufferInfo);
239 if(!msg.contains(' '))
242 QByteArray target = serverEncode(msg.section(' ', 0, 0));
243 putPrivmsg(target, userEncode(target, msg.section(' ', 1)));
246 void UserInputHandler::handleNick(const BufferInfo &bufferInfo, const QString &msg) {
248 QString nick = msg.section(' ', 0, 0);
249 emit putCmd("NICK", serverEncode(nick));
252 void UserInputHandler::handleNotice(const BufferInfo &bufferInfo, const QString &msg) {
253 QString bufferName = msg.section(' ', 0, 0);
254 QString payload = msg.section(' ', 1);
255 QList<QByteArray> params;
256 params << serverEncode(bufferName) << channelEncode(bufferInfo.bufferName(), payload);
257 emit putCmd("NOTICE", params);
258 emit displayMsg(Message::Notice, bufferName, payload, network()->myNick(), Message::Self);
261 void UserInputHandler::handleOp(const BufferInfo &bufferInfo, const QString &msg) {
262 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
263 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'o';
265 params << bufferInfo.bufferName() << m << nicks;
266 emit putCmd("MODE", serverEncode(params));
269 void UserInputHandler::handleOper(const BufferInfo &bufferInfo, const QString &msg) {
271 emit putRawLine(serverEncode(QString("OPER %1").arg(msg)));
274 void UserInputHandler::handlePart(const BufferInfo &bufferInfo, const QString &msg) {
275 QList<QByteArray> params;
278 // msg might contain either a channel name and/or a reaon, so we have to check if the first word is a known channel
279 QString channelName = msg.section(' ', 0, 0);
280 if(channelName.isEmpty() || !network()->ircChannel(channelName)) {
281 channelName = bufferInfo.bufferName();
284 partReason = msg.mid(channelName.length() + 1);
287 if(partReason.isEmpty())
288 partReason = network()->identityPtr()->partReason();
290 params << serverEncode(channelName) << channelEncode(bufferInfo.bufferName(), partReason);
291 emit putCmd("PART", params);
294 void UserInputHandler::handlePing(const BufferInfo &bufferInfo, const QString &msg) {
299 param = QTime::currentTime().toString("hh:mm:ss.zzz");
301 putCmd("PING", serverEncode(param));
304 // TODO: implement queries
305 void UserInputHandler::handleQuery(const BufferInfo &bufferInfo, const QString &msg) {
307 QString target = msg.section(' ', 0, 0);
308 QString message = msg.section(' ', 1);
309 if(message.isEmpty())
310 emit displayMsg(Message::Server, BufferInfo::QueryBuffer, target, "Starting query with " + target, network()->myNick(), Message::Self);
312 emit displayMsg(Message::Plain, BufferInfo::QueryBuffer, target, message, network()->myNick(), Message::Self);
313 handleMsg(bufferInfo, msg);
316 void UserInputHandler::handleQuit(const BufferInfo &bufferInfo, const QString &msg) {
318 network()->disconnectFromIrc(true, msg);
321 void UserInputHandler::issueQuit(const QString &reason) {
322 emit putCmd("QUIT", serverEncode(reason));
325 void UserInputHandler::handleQuote(const BufferInfo &bufferInfo, const QString &msg) {
327 emit putRawLine(serverEncode(msg));
330 void UserInputHandler::handleSay(const BufferInfo &bufferInfo, const QString &msg) {
331 if(bufferInfo.bufferName().isEmpty())
332 return; // server buffer
333 putPrivmsg(serverEncode(bufferInfo.bufferName()), channelEncode(bufferInfo.bufferName(), msg));
334 emit displayMsg(Message::Plain, bufferInfo.type(), bufferInfo.bufferName(), msg, network()->myNick(), Message::Self);
337 void UserInputHandler::handleTopic(const BufferInfo &bufferInfo, const QString &msg) {
338 if(bufferInfo.bufferName().isEmpty()) return;
339 QList<QByteArray> params;
340 params << serverEncode(bufferInfo.bufferName());
342 params << channelEncode(bufferInfo.bufferName(), msg);
343 emit putCmd("TOPIC", params);
346 void UserInputHandler::handleVoice(const BufferInfo &bufferInfo, const QString &msg) {
347 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
348 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'v';
350 params << bufferInfo.bufferName() << m << nicks;
351 emit putCmd("MODE", serverEncode(params));
354 void UserInputHandler::handleWait(const BufferInfo &bufferInfo, const QString &msg) {
355 int splitPos = msg.indexOf(';');
360 int delay = msg.left(splitPos).trimmed().toInt(&ok);
366 QString command = msg.mid(splitPos + 1).trimmed();
367 if(command.isEmpty())
370 _delayedCommands[startTimer(delay)] = Command(bufferInfo, command);
373 void UserInputHandler::handleWho(const BufferInfo &bufferInfo, const QString &msg) {
375 emit putCmd("WHO", serverEncode(msg.split(' ')));
378 void UserInputHandler::handleWhois(const BufferInfo &bufferInfo, const QString &msg) {
380 emit putCmd("WHOIS", serverEncode(msg.split(' ')));
383 void UserInputHandler::handleWhowas(const BufferInfo &bufferInfo, const QString &msg) {
385 emit putCmd("WHOWAS", serverEncode(msg.split(' ')));
388 void UserInputHandler::defaultHandler(QString cmd, const BufferInfo &bufferInfo, const QString &msg) {
389 Q_UNUSED(bufferInfo);
390 emit displayMsg(Message::Error, BufferInfo::StatusBuffer, "", QString("Error: %1 %2").arg(cmd, msg));
393 void UserInputHandler::putPrivmsg(const QByteArray &target, const QByteArray &message) {
394 static const char *cmd = "PRIVMSG";
395 int overrun = lastParamOverrun(cmd, QList<QByteArray>() << message);
397 static const char *splitter = " .,-";
398 int maxSplitPos = message.count() - overrun;
400 for(const char *splitChar = splitter; *splitChar != 0; splitChar++) {
401 splitPos = qMax(splitPos, message.lastIndexOf(*splitChar, maxSplitPos));
404 splitPos = maxSplitPos;
406 putCmd(cmd, QList<QByteArray>() << target << message.left(splitPos));
407 putPrivmsg(target, message.mid(splitPos));
410 putCmd(cmd, QList<QByteArray>() << target << message);
414 // returns 0 if the message will not be chopped by the irc server or number of chopped bytes if message is too long
415 int UserInputHandler::lastParamOverrun(const QString &cmd, const QList<QByteArray> ¶ms) {
416 // the server will pass our message trunkated to 512 bytes including CRLF with the following format:
417 // ":prefix COMMAND param0 param1 :lastparam"
418 // where prefix = "nickname!user@host"
419 // that means that the last message can be as long as:
420 // 512 - nicklen - userlen - hostlen - commandlen - sum(param[0]..param[n-1])) - 2 (for CRLF) - 4 (":!@" + 1space between prefix and command) - max(paramcount - 1, 0) (space for simple params) - 2 (space and colon for last param)
421 IrcUser *me = network()->me();
422 int maxLen = 480 - cmd.toAscii().count(); // educated guess in case we don't know us (yet?)
425 maxLen = 512 - serverEncode(me->nick()).count() - serverEncode(me->user()).count() - serverEncode(me->host()).count() - cmd.toAscii().count() - 6;
427 if(!params.isEmpty()) {
428 for(int i = 0; i < params.count() - 1; i++) {
429 maxLen -= (params[i].count() + 1);
431 maxLen -= 2; // " :" last param separator;
433 if(params.last().count() > maxLen) {
434 return params.last().count() - maxLen;
444 void UserInputHandler::timerEvent(QTimerEvent *event) {
445 if(!_delayedCommands.contains(event->timerId())) {
446 QObject::timerEvent(event);
449 BufferInfo bufferInfo = _delayedCommands[event->timerId()].bufferInfo;
450 QString rawCommand = _delayedCommands[event->timerId()].command;
451 _delayedCommands.remove(event->timerId());
454 // the stored command might be the result of an alias expansion, so we need to split it up again
455 QStringList commands = rawCommand.split(QRegExp("; ?"));
456 foreach(QString command, commands) {
457 handleUserInput(bufferInfo, command);