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 ***************************************************************************/
26 #include <QMetaObject>
29 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::sendState() {
51 foreach(QString key, nicks.keys()) { n[key] = nicks[key]; }
52 foreach(QString key, topics.keys()) { t[key] = topics[key];}
55 s["OwnNick"] = ownNick;
56 s["ServerSupports"] = serverSupports;
57 emit serverState(network, s);
60 void Server::connectToIrc(QString net) {
61 if(net != network) return; // not me!
62 networkSettings = global->getData("Networks").toMap()[net].toMap();
63 identity = global->getData("Identities").toMap()[networkSettings["Identity"].toString()].toMap();
64 QList<QVariant> servers = networkSettings["Servers"].toList();
65 QString host = servers[0].toMap()["Address"].toString();
66 quint16 port = servers[0].toMap()["Port"].toUInt();
67 displayStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
68 socket.connectToHost(host, port);
71 void Server::disconnectFromIrc(QString net) {
72 if(net != network) return; // not me!
73 socket.disconnectFromHost();
76 void Server::socketHasData() {
77 while(socket.canReadLine()) {
78 QString s = socket.readLine().trimmed();
79 //qDebug() << "Read" << s;
80 emit recvRawServerMsg(s);
81 //Message *msg = Message::createFromServerString(this, s);
86 void Server::socketError( QAbstractSocket::SocketError err ) {
87 //qDebug() << "Socket Error!";
91 void Server::socketConnected( ) {
92 emit connected(network);
93 putRawLine(QString("NICK :%1").arg(identity["NickList"].toStringList()[0]));
94 putRawLine(QString("USER %1 8 * :%2").arg(identity["Ident"].toString()).arg(identity["RealName"].toString()));
97 void Server::socketDisconnected( ) {
98 //qDebug() << "Socket disconnected!";
99 emit disconnected(network);
104 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
105 //qDebug() << "Socket state changed: " << state;
108 QString Server::updateNickFromMask(QString mask) {
109 QString user = userFromMask(mask);
110 QString host = hostFromMask(mask);
111 QString nick = nickFromMask(mask);
112 if(nicks.contains(nick) && !user.isEmpty() && !host.isEmpty()) {
113 VarMap n = nicks[nick];
114 if(n["User"].toString() != user || n["Host"].toString() != host) {
115 if(!n["User"].toString().isEmpty() || !n["Host"].toString().isEmpty())
116 qWarning(QString("Strange: Hostmask for nick %1 has changed!").arg(nick).toAscii());
117 n["User"] = user; n["Host"] = host;
119 emit nickUpdated(network, nick, n);
125 void Server::userInput(QString net, QString buf, QString msg) {
126 if(net != network) return; // not me!
127 //msg = msg.trimmed(); // remove whitespace from start and end
128 if(msg.isEmpty()) return;
129 if(!msg.startsWith('/')) {
130 msg = QString("/SAY ") + msg;
132 handleUserInput(buf, msg);
135 void Server::putRawLine(QString s) {
136 // qDebug() << "SentRaw: " << s;
138 socket.write(s.toAscii());
141 void Server::putCmd(QString cmd, QStringList params, QString prefix) {
143 if(!prefix.isEmpty()) m += ":" + prefix + " ";
145 for(int i = 0; i < params.size() - 1; i++) {
146 m += " " + params[i];
148 if(!params.isEmpty()) m += " :" + params.last();
149 // qDebug() << "Sent: " << m;
151 socket.write(m.toAscii());
154 /** Handle a raw message string sent by the server. We try to find a suitable handler, otherwise we call a default handler. */
155 void Server::handleServerMsg(QString msg) {
158 qWarning() << "Received empty string from server!";
161 // OK, first we split the raw message into its various parts...
166 // check for prefix by checking for a colon as the first char
169 prefix = msg.section(' ', 0, 0);
170 msg = msg.section(' ', 1);
173 // next string without a whitespace is the command
174 cmd = msg.section(' ', 0, 0).toUpper();
175 msg = msg.section(' ', 1);
177 // get the parameters
178 QString left, trailing;
179 left = msg.section(" :", 0, 0);
180 trailing = msg.section(" :", -1, -1);
181 if(!left.isEmpty()) {
182 params << left.split(' ', QString::SkipEmptyParts);
184 if(!trailing.isEmpty()) {
188 // numeric replies have the target as first param (RFC 2812 - 2.4). this is usually our own nick. Remove this!
189 uint num = cmd.toUInt();
191 Q_ASSERT(params.count() > 0); // Violation to RFC
192 params.removeFirst();
195 // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
196 QString hname = cmd.toLower();
197 hname[0] = hname[0].toUpper();
198 hname = "handleServer" + hname;
199 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
200 // Ok. Default handler it is.
201 defaultServerHandler(cmd, prefix, params);
203 } catch(Exception e) {
204 emit displayMsg(Message::error("", e.msg()));
208 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
209 uint num = cmd.toUInt();
211 // A lot of server messages don't really need their own handler because they don't do much.
212 // Catch and handle these here.
214 // Welcome, status, info messages. Just display these.
215 case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
216 emit displayMsg(Message::server("", params.join(" "), prefix));
218 // Server error messages without param, just display them
219 case 409: case 411: case 412: case 422: case 424: case 431: case 445: case 446: case 451: case 462:
220 case 463: case 464: case 465: case 466: case 472: case 481: case 483: case 485: case 491: case 501: case 502:
221 emit displayMsg(Message::error("", params.join(" "), prefix));
223 // Server error messages, display them in red. First param will be appended.
224 case 401: case 402: case 403: case 404: case 406: case 408: case 415: case 421: case 432: case 442:
225 { QString p = params.takeFirst();
226 emit displayMsg(Message::error("", params.join(" ") + " " + p, prefix));
229 // Server error messages which will be displayed with a colon between the first param and the rest
230 case 413: case 414: case 423: case 436: case 441: case 444: case 461:
231 case 467: case 471: case 473: case 474: case 475: case 476: case 477: case 478: case 482:
232 { QString p = params.takeFirst();
233 emit displayMsg(Message::error("", p + ": " + params.join(" ")));
236 // Ignore these commands.
240 // Everything else will be marked in red, so we can add them somewhere.
242 emit displayMsg(Message::error("", cmd + " " + params.join(" "), prefix));
244 //qDebug() << prefix <<":"<<cmd<<params;
246 emit displayMsg(Message::error("", QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
247 //qDebug() << prefix <<":"<<cmd<<params;
251 void Server::handleUserInput(QString bufname, QString usrMsg) {
253 /* Looks like we don't need core-side buffers...
255 if(!bufname.isEmpty()) {
256 Q_ASSERT(buffers.contains(bufname));
257 buffer = buffers[bufname];
260 QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
261 QString msg = usrMsg.section(' ', 1);
262 QString hname = cmd.toLower();
263 hname[0] = hname[0].toUpper();
264 hname = "handleUser" + hname;
265 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, bufname), Q_ARG(QString, msg))) {
266 // Ok. Default handler it is.
267 defaultUserHandler(bufname, cmd, msg);
269 } catch(Exception e) {
270 emit displayMsg(Message::error("", e.msg()));
274 void Server::defaultUserHandler(QString bufname, QString cmd, QString msg) {
275 emit displayMsg(Message::error("", QString("Error: %1 %2").arg(cmd).arg(msg)));
279 QStringList Server::providesUserHandlers() {
280 QStringList userHandlers;
281 for(int i=0; i < metaObject()->methodCount(); i++) {
282 QString methodSignature(metaObject()->method(i).signature());
283 if (methodSignature.startsWith("handleUser")) {
284 methodSignature = methodSignature.section('(',0,0); // chop the attribute list
285 methodSignature = methodSignature.mid(10); // strip "handleUser"
286 userHandlers << methodSignature;
293 /**********************************************************************************/
296 void Server::handleUser(QString bufname, QString msg) {
302 void Server::handleUserAway(QString bufname, QString msg) {
303 putCmd("AWAY", QStringList(msg));
306 void Server::handleUserDeop(QString bufname, QString msg) {
307 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
308 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'o';
310 params << bufname << m << nicks;
311 putCmd("MODE", params);
314 void Server::handleUserDevoice(QString bufname, QString msg) {
315 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
316 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'v';
318 params << bufname << m << nicks;
319 putCmd("MODE", params);
322 void Server::handleUserInvite(QString bufname, QString msg) {
324 params << msg << bufname;
325 putCmd("INVITE", params);
328 void Server::handleUserJoin(QString bufname, QString msg) {
329 putCmd("JOIN", QStringList(msg));
332 void Server::handleUserKick(QString bufname, QString msg) {
334 params << bufname << msg.split(' ', QString::SkipEmptyParts);
335 putCmd("KICK", params);
338 void Server::handleUserList(QString bufname, QString msg) {
339 putCmd("LIST", msg.split(' ', QString::SkipEmptyParts));
342 void Server::handleUserMode(QString bufname, QString msg) {
343 putCmd("MODE", msg.split(' ', QString::SkipEmptyParts));
346 void Server::handleUserMsg(QString bufname, QString msg) {
347 QString nick = msg.section(" ", 0, 0);
348 msg = msg.section(" ", 1);
349 if(nick.isEmpty() || msg.isEmpty()) return;
351 params << nick << msg;
352 putCmd("PRIVMSG", params);
355 void Server::handleUserNick(QString bufname, QString msg) {
356 QString nick = msg.section(' ', 0, 0);
357 putCmd("NICK", QStringList(nick));
360 void Server::handleUserOp(QString bufname, QString msg) {
361 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
362 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'o';
364 params << bufname << m << nicks;
365 putCmd("MODE", params);
368 void Server::handleUserPart(QString bufname, QString msg) {
370 params << bufname << msg;
371 putCmd("PART", params);
374 void Server::handleUserQuery(QString bufname, QString msg) {
375 QString nick = msg.section(' ', 0, 0);
376 if(!nick.isEmpty()) emit queryRequested(network, nick);
379 void Server::handleUserQuit(QString bufname, QString msg) {
380 putCmd("QUIT", QStringList(msg));
383 void Server::handleUserQuote(QString bufname, QString msg) {
387 void Server::handleUserSay(QString bufname, QString msg) {
388 if(bufname.isEmpty()) return; // server buffer
390 params << bufname << msg;
391 putCmd("PRIVMSG", params);
392 if(isChannelName(bufname)) {
393 emit displayMsg(Message::plain(params[0], msg, ownNick, Message::Self));
395 emit displayMsg(Message::plain(params[0], msg, ownNick, Message::Self|Message::PrivMsg));
399 void Server::handleUserTopic(QString bufname, QString msg) {
400 if(bufname.isEmpty()) return;
402 params << bufname << msg;
403 putCmd("TOPIC", params);
406 void Server::handleUserVoice(QString bufname, QString msg) {
407 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
408 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'v';
410 params << bufname << m << nicks;
411 putCmd("MODE", params);
414 /**********************************************************************************/
416 void Server::handleServerJoin(QString prefix, QStringList params) {
417 Q_ASSERT(params.count() == 1);
418 QString nick = updateNickFromMask(prefix);
419 if(nick == ownNick) {
420 // Q_ASSERT(!buffers.contains(params[0])); // cannot join a buffer twice!
421 // Buffer *buf = new Buffer(params[0]);
422 // buffers[params[0]] = buf;
423 topics[params[0]] = "";
424 emit topicSet(network, params[0], "");
427 if(nicks.contains(nick)) {
429 VarMap chans = n["Channels"].toMap();
430 // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
431 chans[params[0]] = VarMap();
432 n["Channels"] = chans;
434 emit nickUpdated(network, nick, n);
437 chans[params[0]] = VarMap();
438 n["Channels"] = chans;
439 n["User"] = userFromMask(prefix);
440 n["Host"] = hostFromMask(prefix);
442 emit nickAdded(network, nick, n);
444 emit displayMsg(Message::join(params[0], params[0], prefix));
448 void Server::handleServerKick(QString prefix, QStringList params) {
449 QString kicker = updateNickFromMask(prefix);
450 QString nick = params[1];
451 Q_ASSERT(nicks.contains(nick));
452 VarMap n = nicks[nick];
453 VarMap chans = n["Channels"].toMap();
454 Q_ASSERT(chans.contains(params[0]));
455 chans.remove(params[0]);
457 if(params.count() > 2) msg = QString("%1 %2").arg(msg).arg(params[2]);
458 emit displayMsg(Message::kick(params[0], msg, prefix));
459 if(chans.count() > 0) {
460 n["Channels"] = chans;
462 emit nickUpdated(network, nick, n);
465 emit nickRemoved(network, nick);
467 if(nick == ownNick) {
468 topics.remove(params[0]);
472 void Server::handleServerMode(QString prefix, QStringList params) {
473 if(isChannelName(params[0])) {
474 // TODO only channel-user modes supported by now
475 QString prefixes = serverSupports["PrefixModes"].toString();
476 QString modes = params[1];
480 while(m < modes.length()) {
481 if(modes[m] == '+') { add = true; m++; continue; }
482 if(modes[m] == '-') { add = false; m++; continue; }
483 if(prefixes.contains(modes[m])) { // it's a user channel mode
484 Q_ASSERT(params.count() > m);
485 QString nick = params[p++];
486 if(nicks.contains(nick)) { // sometimes, a server might try to set a MODE on a nick that is no longer there
487 VarMap n = nicks[nick]; VarMap clist = n["Channels"].toMap(); VarMap chan = clist[params[0]].toMap();
488 QString mstr = chan["Mode"].toString();
489 add ? mstr += modes[m] : mstr.remove(modes[m]);
490 chan["Mode"] = mstr; clist[params[0]] = chan; n["Channels"] = clist; nicks[nick] = n;
491 emit nickUpdated(network, nick, n);
495 // TODO add more modes
499 emit displayMsg(Message::mode(params[0], params.join(" "), prefix));
501 //Q_ASSERT(nicks.contains(params[0]));
502 //VarMap n = nicks[params[0]].toMap();
503 //QString mode = n["Mode"].toString();
504 emit displayMsg(Message::mode("", params.join(" ")));
508 void Server::handleServerNick(QString prefix, QStringList params) {
509 QString oldnick = updateNickFromMask(prefix);
510 QString newnick = params[0];
511 VarMap v = nicks.take(oldnick);
513 VarMap chans = v["Channels"].toMap();
514 foreach(QString c, chans.keys()) {
515 if(oldnick != ownNick) { emit displayMsg(Message::nick(c, newnick, prefix)); }
516 else { emit displayMsg(Message::nick(c, newnick, newnick)); }
518 emit nickRenamed(network, oldnick, newnick);
519 if(oldnick == ownNick) {
521 emit ownNickSet(network, newnick);
525 void Server::handleServerNotice(QString prefix, QStringList params) {
526 //Message msg(Message::Notice, params[1], prefix);
527 if(currentServer.isEmpty() || prefix == currentServer) emit displayMsg(Message::server("", params[1], prefix));
528 else emit displayMsg(Message::notice("", params[1], prefix));
531 void Server::handleServerPart(QString prefix, QStringList params) {
532 QString nick = updateNickFromMask(prefix);
533 Q_ASSERT(nicks.contains(nick));
534 VarMap n = nicks[nick];
535 VarMap chans = n["Channels"].toMap();
536 Q_ASSERT(chans.contains(params[0]));
537 chans.remove(params[0]);
539 if(params.count() > 1) msg = params[1];
540 emit displayMsg(Message::part(params[0], msg, prefix));
541 if(chans.count() > 0) {
542 n["Channels"] = chans;
544 emit nickUpdated(network, nick, n);
547 emit nickRemoved(network, nick);
549 if(nick == ownNick) {
550 Q_ASSERT(topics.contains(params[0]));
551 topics.remove(params[0]);
555 void Server::handleServerPing(QString prefix, QStringList params) {
556 putCmd("PONG", params);
559 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
560 updateNickFromMask(prefix);
561 Q_ASSERT(params.count() >= 2);
562 if(params.count()<2) emit displayMsg(Message::plain(params[0], "", prefix));
564 if(params[0] == ownNick) {
565 emit displayMsg(Message::plain("", params[1], prefix, Message::PrivMsg));
567 Q_ASSERT(isChannelName(params[0])); // should be channel!
568 emit displayMsg(Message::plain(params[0], params[1], prefix));
573 void Server::handleServerQuit(QString prefix, QStringList params) {
574 QString nick = updateNickFromMask(prefix);
575 Q_ASSERT(nicks.contains(nick));
576 VarMap chans = nicks[nick]["Channels"].toMap();
577 foreach(QString c, chans.keys()) {
578 emit displayMsg(Message::quit(c, params[0], prefix));
581 emit nickRemoved(network, nick);
584 void Server::handleServerTopic(QString prefix, QStringList params) {
585 QString nick = updateNickFromMask(prefix);
586 Q_ASSERT(nicks.contains(nick));
587 topics[params[0]] = params[1];
588 emit topicSet(network, params[0], params[1]);
589 emit displayMsg(Message::server(params[0], tr("%1 has changed topic for %2 to: \"%3\"").arg(nick).arg(params[0]).arg(params[1])));
593 void Server::handleServer001(QString prefix, QStringList params) {
594 // there should be only one param: "Welcome to the Internet Relay Network <nick>!<user>@<host>"
595 currentServer = prefix;
596 ownNick = params[0].section(' ', -1, -1).section('!', 0, 0);
598 n["Channels"] = VarMap();
600 emit ownNickSet(network, ownNick);
601 emit nickAdded(network, ownNick, VarMap());
602 emit displayMsg(Message::server("", params[0], prefix));
604 QStringList performList = networkSettings["Perform"].toString().split( "\n" );
605 int count = performList.count();
606 for(int a = 0; a < count; a++) {
607 if(!performList[a].isEmpty() ) {
608 userInput(network, "", performList[a]);
614 // TODO Complete 005 handling, also use sensible defaults for non-sent stuff
615 void Server::handleServer005(QString prefix, QStringList params) {
617 foreach(QString p, params) {
618 QString key = p.section("=", 0, 0);
619 QString val = p.section("=", 1);
620 serverSupports[key] = val;
621 // handle some special cases
622 if(key == "PREFIX") {
623 VarMap foo; QString modes, prefixes;
624 Q_ASSERT(val.contains(')') && val.startsWith('('));
626 for(p = 2; p < val.length(); p++) if(val[p] == ')') break;
628 for(; val[m] != ')'; m++, p++) {
629 Q_ASSERT(p < val.length());
630 foo[QString(val[m])] = QString(val[p]);
631 modes += val[m]; prefixes += val[p];
633 serverSupports["PrefixModes"] = modes; serverSupports["Prefixes"] = prefixes;
634 serverSupports["ModePrefixMap"] = foo;
641 void Server::handleServer331(QString prefix, QStringList params) {
642 topics[params[0]] = "";
643 emit topicSet(network, params[0], "");
644 emit displayMsg(Message::server(params[0], tr("No topic is set for %1.").arg(params[0])));
648 void Server::handleServer332(QString prefix, QStringList params) {
649 topics[params[0]] = params[1];
650 emit topicSet(network, params[0], params[1]);
651 emit displayMsg(Message::server(params[0], tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1])));
654 /* Topic set by... */
655 void Server::handleServer333(QString prefix, QStringList params) {
656 emit displayMsg(Message::server(params[0],
657 tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString())));
661 void Server::handleServer353(QString prefix, QStringList params) {
662 params.removeFirst(); // = or *
663 QString buf = params.takeFirst();
664 QString prefixes = serverSupports["Prefixes"].toString();
665 foreach(QString nick, params[0].split(' ')) {
666 QString mode = "", pfx = "";
667 if(prefixes.contains(nick[0])) {
670 if(prefixes[i] == nick[0]) { mode = serverSupports["PrefixModes"].toString()[i]; break; }
673 VarMap c; c["Mode"] = mode; c["Prefix"] = pfx;
674 if(nicks.contains(nick)) {
675 VarMap n = nicks[nick];
676 VarMap chans = n["Channels"].toMap();
678 n["Channels"] = chans;
680 emit nickUpdated(network, nick, n);
682 VarMap n; VarMap c; VarMap chans;
685 n["Channels"] = chans;
687 emit nickAdded(network, nick, n);
692 /* RPL_NICKNAMEINUSER */
693 void Server::handleServer433(QString prefix, QStringList params) {
694 QString errnick = params[0];
695 emit displayMsg(Message::error("", tr("Nick %1 is already taken").arg(errnick)));
696 // if there is a problem while connecting to the server -> we handle it
697 // TODO rely on another source...
698 if(currentServer.isEmpty()) {
699 QStringList desiredNicks = identity["NickList"].toStringList();
700 int nextNick = desiredNicks.indexOf(errnick) + 1;
701 if (desiredNicks.size() > nextNick) {
702 putCmd("NICK", QStringList(desiredNicks[nextNick]));
704 emit displayMsg(Message::error("", "All nicks in nicklist taken... use: /nick <othernick> to continue"));
709 /***********************************************************************************/
711 /* Exception classes for message handling */
712 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
713 _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
717 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
718 _msg = QString("Unknown Command: ") + cmd;