1 /***************************************************************************
2 * Copyright (C) 2005/06 by The Quassel Team *
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) any later version. *
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 ***************************************************************************/
27 #include <QMetaObject>
30 Server::Server(QString net) : network(net) {
39 connect(&socket, SIGNAL(connected()), this, SLOT(socketConnected()));
40 connect(&socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
41 connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
42 connect(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
43 connect(&socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
48 void Server::connectToIrc(QString net) {
49 if(net != network) return; // not me!
50 networkSettings = global->getData("Networks").toMap()[net].toMap();
51 identity = global->getData("Identities").toMap()[networkSettings["Identity"].toString()].toMap();
52 QList<QVariant> servers = networkSettings["Servers"].toList();
53 QString host = servers[0].toMap()["Address"].toString();
54 quint16 port = servers[0].toMap()["Port"].toUInt();
55 displayStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
56 socket.connectToHost(host, port);
59 void Server::disconnectFromIrc(QString net) {
60 if(net != network) return; // not me!
61 socket.disconnectFromHost();
64 void Server::socketHasData() {
65 while(socket.canReadLine()) {
66 QString s = socket.readLine().trimmed();
67 qDebug() << "Read: " << s;
68 emit recvRawServerMsg(s);
69 //Message *msg = Message::createFromServerString(this, s);
74 void Server::socketError( QAbstractSocket::SocketError err ) {
75 //qDebug() << "Socket Error!";
79 void Server::socketConnected( ) {
80 putRawLine(QString("NICK :%1").arg(identity["NickList"].toStringList()[0]));
81 putRawLine(QString("USER %1 8 * :%2").arg(identity["Ident"].toString()).arg(identity["RealName"].toString()));
84 void Server::socketDisconnected( ) {
85 //qDebug() << "Socket disconnected!";
89 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
90 //qDebug() << "Socket state changed: " << state;
93 QString Server::updateNickFromMask(QString mask) {
94 QString user = userFromMask(mask);
95 QString host = hostFromMask(mask);
96 QString nick = nickFromMask(mask);
97 if(nicks.contains(nick) && !user.isEmpty() && !host.isEmpty()) {
98 VarMap n = nicks[nick].toMap();
99 if(n["User"].toString() != user || n["Host"].toString() != host) {
100 if(!n["User"].toString().isEmpty() || !n["Host"].toString().isEmpty())
101 qWarning(QString("Strange: Hostmask for nick %1 has changed!").arg(nick).toAscii());
102 n["User"] = user; n["Host"] = host;
104 emit nickUpdated(network, nick, n);
110 void Server::userInput(QString net, QString buf, QString msg) {
111 if(net != network) return; // not me!
112 //msg = msg.trimmed(); // remove whitespace from start and end
113 if(msg.isEmpty()) return;
114 if(!msg.startsWith('/')) {
115 msg = QString("/SAY ") + msg;
117 handleUserInput(buf, msg);
120 void Server::putRawLine(QString s) {
121 qDebug() << "SentRaw: " << s;
123 socket.write(s.toAscii());
126 void Server::putCmd(QString cmd, QStringList params, QString prefix) {
128 if(!prefix.isEmpty()) m += ":" + prefix + " ";
130 for(int i = 0; i < params.size() - 1; i++) {
131 m += " " + params[i];
133 if(!params.isEmpty()) m += " :" + params.last();
134 qDebug() << "Sent: " << m;
136 socket.write(m.toAscii());
139 /** Handle a raw message string sent by the server. We try to find a suitable handler, otherwise we call a default handler. */
140 void Server::handleServerMsg(QString msg) {
143 qWarning() << "Received empty string from server!";
146 // OK, first we split the raw message into its various parts...
152 prefix = msg.section(' ', 0, 0);
153 msg = msg.section(' ', 1);
155 cmd = msg.section(' ', 0, 0).toUpper();
156 msg = msg.section(' ', 1);
157 QString left, trailing;
158 // RPL_ISUPPORT (005) can contain colons, so don't treat it like the rest of the commands
159 if(cmd.toUInt() == 5) {
160 left = msg.remove(QString(":are supported by this server"));
162 left = msg.section(':', 0, 0);
163 trailing = msg.section(':', 1);
165 if(!left.isEmpty()) {
166 params << left.split(' ', QString::SkipEmptyParts);
168 if(!trailing.isEmpty()) {
171 // numeric replies usually have our own nick as first param. Remove this!
172 // BTW, this behavior is not in the RFC.
173 uint num = cmd.toUInt();
174 if(num > 1 && params.count() > 0) { // 001 sets our nick, so we shouldn't remove anything
175 if(params[0] == currentNick) params.removeFirst();
176 else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
178 // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
179 QString hname = cmd.toLower();
180 hname[0] = hname[0].toUpper();
181 hname = "handleServer" + hname;
182 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
183 // Ok. Default handler it is.
184 defaultServerHandler(cmd, prefix, params);
186 } catch(Exception e) {
187 emit displayMsg("", Message(Message::Error, e.msg()));
191 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
192 uint num = cmd.toUInt();
194 // A lot of server messages don't really need their own handler because they don't do much.
195 // Catch and handle these here.
197 // Welcome, status, info messages. Just display these.
198 case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
199 emit displayMsg("", Message(Message::Server, params.join(" "), prefix));
201 // Server error messages without param, just display them
202 case 409: case 411: case 412: case 422: case 424: case 431: case 445: case 446: case 451: case 462:
203 case 463: case 464: case 465: case 466: case 472: case 481: case 483: case 485: case 491: case 501: case 502:
204 emit displayMsg("", Message(Message::Error, params.join(" "), prefix));
206 // Server error messages, display them in red. First param will be appended.
207 case 401: case 402: case 403: case 404: case 406: case 408: case 415: case 421: case 432: case 442:
208 { QString p = params.takeFirst();
209 emit displayMsg("", Message(Message::Error, params.join(" ") + " " + p, prefix));
212 // Server error messages which will be displayed with a colon between the first param and the rest
213 case 413: case 414: case 423: case 433: case 436: case 441: case 444: case 461:
214 case 467: case 471: case 473: case 474: case 475: case 476: case 477: case 478: case 482:
215 { QString p = params.takeFirst();
216 emit displayMsg("", Message(Message::Error, p + ": " + params.join(" ")));
219 // Ignore these commands.
223 // Everything else will be marked in red, so we can add them somewhere.
225 emit displayMsg("", Message(Message::Error, cmd + " " + params.join(" "), prefix));
227 //qDebug() << prefix <<":"<<cmd<<params;
229 emit displayMsg("", Message(Message::Error, QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
230 //qDebug() << prefix <<":"<<cmd<<params;
234 void Server::handleUserInput(QString bufname, QString usrMsg) {
236 /* Looks like we don't need core-side buffers...
238 if(!bufname.isEmpty()) {
239 Q_ASSERT(buffers.contains(bufname));
240 buffer = buffers[bufname];
243 QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
244 QString msg = usrMsg.section(' ', 1);
245 QString hname = cmd.toLower();
246 hname[0] = hname[0].toUpper();
247 hname = "handleUser" + hname;
248 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, bufname), Q_ARG(QString, msg))) {
249 // Ok. Default handler it is.
250 defaultUserHandler(bufname, cmd, msg);
252 } catch(Exception e) {
253 emit displayMsg("", Message(Message::Error, e.msg()));
257 void Server::defaultUserHandler(QString bufname, QString cmd, QString msg) {
258 emit displayMsg("", Message(Message::Error, QString("Error: %1 %2").arg(cmd).arg(msg)));
262 /**********************************************************************************/
265 void Server::handleUser(QString bufname, QString msg) {
271 void Server::handleUserAway(QString bufname, QString msg) {
272 putCmd("AWAY", QStringList(msg));
275 void Server::handleUserDeop(QString bufname, QString msg) {
276 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
277 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'o';
279 params << bufname << m << nicks;
280 putCmd("MODE", params);
283 void Server::handleUserDevoice(QString bufname, QString msg) {
284 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
285 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'v';
287 params << bufname << m << nicks;
288 putCmd("MODE", params);
291 void Server::handleUserInvite(QString bufname, QString msg) {
293 params << msg << bufname;
294 putCmd("INVITE", params);
297 void Server::handleUserJoin(QString bufname, QString msg) {
298 putCmd("JOIN", QStringList(msg));
301 void Server::handleUserKick(QString bufname, QString msg) {
303 params << bufname << msg.split(' ', QString::SkipEmptyParts);
304 putCmd("KICK", params);
307 void Server::handleUserList(QString bufname, QString msg) {
308 putCmd("LIST", msg.split(' ', QString::SkipEmptyParts));
311 void Server::handleUserMode(QString bufname, QString msg) {
312 putCmd("MODE", msg.split(' ', QString::SkipEmptyParts));
315 void Server::handleUserMsg(QString bufname, QString msg) {
316 QString nick = msg.section(" ", 0, 0);
317 msg = msg.section(" ", 1);
318 if(nick.isEmpty() || msg.isEmpty()) return;
320 params << nick << msg;
321 putCmd("PRIVMSG", params);
324 void Server::handleUserNick(QString bufname, QString msg) {
325 QString nick = msg.section(' ', 0, 0);
326 putCmd("NICK", QStringList(nick));
329 void Server::handleUserOp(QString bufname, QString msg) {
330 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
331 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'o';
333 params << bufname << m << nicks;
334 putCmd("MODE", params);
337 void Server::handleUserPart(QString bufname, QString msg) {
339 params << bufname << msg;
340 putCmd("PART", params);
343 void Server::handleUserQuit(QString bufname, QString msg) {
344 putCmd("QUIT", QStringList(msg));
347 void Server::handleUserQuote(QString bufname, QString msg) {
351 void Server::handleUserSay(QString bufname, QString msg) {
352 if(bufname.isEmpty()) return; // server buffer
354 params << bufname << msg;
355 putCmd("PRIVMSG", params);
356 emit displayMsg(params[0], Message(Message::Msg, msg, currentNick, Message::Self));
359 void Server::handleUserVoice(QString bufname, QString msg) {
360 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
361 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'v';
363 params << bufname << m << nicks;
364 putCmd("MODE", params);
367 /**********************************************************************************/
369 void Server::handleServerJoin(QString prefix, QStringList params) {
370 Q_ASSERT(params.count() == 1);
371 QString nick = updateNickFromMask(prefix);
372 if(nick == currentNick) {
373 // Q_ASSERT(!buffers.contains(params[0])); // cannot join a buffer twice!
374 // Buffer *buf = new Buffer(params[0]);
375 // buffers[params[0]] = buf;
378 if(nicks.contains(nick)) {
379 n = nicks[nick].toMap();
380 VarMap chans = n["Channels"].toMap();
381 // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
382 chans[params[0]] = VarMap();
383 n["Channels"] = chans;
385 emit nickUpdated(network, nick, n);
388 chans[params[0]] = VarMap();
389 n["Channels"] = chans;
390 n["User"] = userFromMask(prefix);
391 n["Host"] = hostFromMask(prefix);
393 emit nickAdded(network, nick, n);
395 emit displayMsg(params[0], Message(Message::Join, params[0], prefix));
399 void Server::handleServerKick(QString prefix, QStringList params) {
400 QString kicker = updateNickFromMask(prefix);
401 QString nick = params[1];
402 Q_ASSERT(nicks.contains(nick));
403 VarMap n = nicks[nick].toMap();
404 VarMap chans = n["Channels"].toMap();
405 Q_ASSERT(chans.contains(params[0]));
406 chans.remove(params[0]);
408 if(params.count() > 2) msg = QString("%1 %2").arg(msg).arg(params[2]);
409 emit displayMsg(params[0], Message(Message::Kick, msg, prefix));
410 if(chans.count() > 0) {
411 n["Channels"] = chans;
413 emit nickUpdated(network, nick, n);
416 emit nickRemoved(network, nick);
420 void Server::handleServerMode(QString prefix, QStringList params) {
421 if(isChannelName(params[0])) {
422 // TODO only channel-user modes supported by now
423 QString prefixes = serverSupports["PrefixModes"].toString();
424 QString modes = params[1];
428 while(m < modes.length()) {
429 if(modes[m] == '+') { add = true; m++; continue; }
430 if(modes[m] == '-') { add = false; m++; continue; }
431 if(prefixes.contains(modes[m])) { // it's a user channel mode
432 Q_ASSERT(params.count() > m && nicks.contains(params[p]));
433 QString nick = params[p++];
434 VarMap n = nicks[nick].toMap(); VarMap clist = n["Channels"].toMap(); VarMap chan = clist[params[0]].toMap();
435 QString mstr = chan["Mode"].toString();
436 add ? mstr += modes[m] : mstr.remove(modes[m]);
437 chan["Mode"] = mstr; clist[params[0]] = chan; n["Channels"] = clist; nicks[nick] = n;
438 emit nickUpdated(network, nick, n);
441 // TODO add more modes
445 emit displayMsg(params[0], Message(Message::Mode, params.join(" "), prefix));
447 //Q_ASSERT(nicks.contains(params[0]));
448 //VarMap n = nicks[params[0]].toMap();
449 //QString mode = n["Mode"].toString();
450 emit displayMsg("", Message(Message::Mode, params.join(" ")));
454 void Server::handleServerNick(QString prefix, QStringList params) {
455 QString oldnick = updateNickFromMask(prefix);
456 QString newnick = params[0];
457 QVariant v = nicks.take(oldnick);
459 VarMap chans = v.toMap()["Channels"].toMap();
460 foreach(QString c, chans.keys()) {
461 if(oldnick != currentNick) { emit displayMsg(c, Message(Message::Nick, newnick, prefix)); }
462 else { emit displayMsg(c, Message(Message::Nick, newnick, newnick)); }
464 emit nickRenamed(network, oldnick, newnick);
465 if(oldnick == currentNick) {
466 currentNick = newnick;
467 emit ownNickSet(network, newnick);
471 void Server::handleServerNotice(QString prefix, QStringList params) {
472 //Message msg(Message::Notice, params[1], prefix);
473 if(prefix == currentServer) emit displayMsg("", Message(Message::Server, params[1], prefix));
474 else emit displayMsg("", Message(Message::Notice, params[1], prefix));
477 void Server::handleServerPart(QString prefix, QStringList params) {
478 QString nick = updateNickFromMask(prefix);
479 Q_ASSERT(nicks.contains(nick));
480 VarMap n = nicks[nick].toMap();
481 VarMap chans = n["Channels"].toMap();
482 Q_ASSERT(chans.contains(params[0]));
483 chans.remove(params[0]);
485 if(params.count() > 1) msg = params[1];
486 emit displayMsg(params[0], Message(Message::Part, msg, prefix));
487 if(chans.count() > 0) {
488 n["Channels"] = chans;
490 emit nickUpdated(network, nick, n);
493 emit nickRemoved(network, nick);
497 void Server::handleServerPing(QString prefix, QStringList params) {
498 putCmd("PONG", params);
501 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
502 updateNickFromMask(prefix);
503 emit displayMsg(params[0], Message(Message::Msg, params[1], prefix));
507 void Server::handleServerQuit(QString prefix, QStringList params) {
508 QString nick = updateNickFromMask(prefix);
509 Q_ASSERT(nicks.contains(nick));
510 VarMap chans = nicks[nick].toMap()["Channels"].toMap();
511 foreach(QString c, chans.keys()) {
512 emit displayMsg(c, Message(Message::Quit, params[0], prefix));
515 emit nickRemoved(network, nick);
519 void Server::handleServer001(QString prefix, QStringList params) {
520 currentServer = prefix;
521 currentNick = params[0];
523 n["Channels"] = VarMap();
524 nicks[currentNick] = n;
525 emit ownNickSet(network, currentNick);
526 emit nickAdded(network, currentNick, VarMap());
527 emit displayMsg("", Message(Message::Server, params[1], prefix));
531 void Server::handleServer005(QString prefix, QStringList params) {
532 foreach(QString p, params) {
533 QString key = p.section("=", 0, 0);
534 QString val = p.section("=", 1);
535 serverSupports[key] = val;
536 // handle some special cases
537 if(key == "PREFIX") {
538 VarMap foo; QString modes, prefixes;
539 Q_ASSERT(val.contains(')') && val.startsWith('('));
541 for(p = 2; p < val.length(); p++) if(val[p] == ')') break;
543 for(; val[m] != ')'; m++, p++) {
544 Q_ASSERT(p < val.length());
545 foo[QString(val[m])] = QString(val[p]);
546 modes += val[m]; prefixes += val[p];
548 serverSupports["PrefixModes"] = modes; serverSupports["Prefixes"] = prefixes;
549 serverSupports["ModePrefixMap"] = foo;
556 void Server::handleServer331(QString prefix, QStringList params) {
557 emit topicSet(network, params[0], "");
558 emit displayMsg(params[0], Message(Message::Server, tr("No topic is set for %1.").arg(params[0])));
562 void Server::handleServer332(QString prefix, QStringList params) {
563 emit topicSet(network, params[0], params[1]);
564 emit displayMsg(params[0], Message(Message::Server, tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1])));
567 /* Topic set by... */
568 void Server::handleServer333(QString prefix, QStringList params) {
569 emit displayMsg(params[0], Message(Message::Server,
570 tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString())));
574 void Server::handleServer353(QString prefix, QStringList params) {
575 params.removeFirst(); // = or *
576 QString buf = params.takeFirst();
577 QString prefixes = serverSupports["Prefixes"].toString();
578 foreach(QString nick, params[0].split(' ')) {
579 QString mode = "", pfx = "";
580 if(prefixes.contains(nick[0])) {
583 if(prefixes[i] == nick[0]) { mode = serverSupports["PrefixModes"].toString()[i]; break; }
586 VarMap c; c["Mode"] = mode; c["Prefix"] = pfx;
587 if(nicks.contains(nick)) {
588 VarMap n = nicks[nick].toMap();
589 VarMap chans = n["Channels"].toMap();
591 n["Channels"] = chans;
593 emit nickUpdated(network, nick, n);
595 VarMap n; VarMap c; VarMap chans;
598 n["Channels"] = chans;
600 emit nickAdded(network, nick, n);
604 /***********************************************************************************/
606 /* Exception classes for message handling */
607 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
608 _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
612 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
613 _msg = QString("Unknown Command: ") + cmd;