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) {
38 connect(&socket, SIGNAL(connected()), this, SLOT(socketConnected()));
39 connect(&socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
40 connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
41 connect(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
42 connect(&socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
47 void Server::sendState() {
50 foreach(QString key, nicks.keys()) { n[key] = nicks[key]; }
51 foreach(QString key, topics.keys()) { t[key] = topics[key];}
54 s["OwnNick"] = ownNick;
55 s["ServerSupports"] = serverSupports;
56 emit serverState(network, s);
59 void Server::connectToIrc(QString net) {
60 if(net != network) return; // not me!
61 networkSettings = global->getData("Networks").toMap()[net].toMap();
62 identity = global->getData("Identities").toMap()[networkSettings["Identity"].toString()].toMap();
63 QList<QVariant> servers = networkSettings["Servers"].toList();
64 QString host = servers[0].toMap()["Address"].toString();
65 quint16 port = servers[0].toMap()["Port"].toUInt();
66 displayStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
67 socket.connectToHost(host, port);
70 void Server::disconnectFromIrc(QString net) {
71 if(net != network) return; // not me!
72 socket.disconnectFromHost();
75 void Server::socketHasData() {
76 while(socket.canReadLine()) {
77 QString s = socket.readLine().trimmed();
78 //qDebug() << "Read" << s;
79 emit recvRawServerMsg(s);
80 //Message *msg = Message::createFromServerString(this, s);
85 void Server::socketError( QAbstractSocket::SocketError err ) {
86 //qDebug() << "Socket Error!";
90 void Server::socketConnected( ) {
91 emit connected(network);
92 putRawLine(QString("NICK :%1").arg(identity["NickList"].toStringList()[0]));
93 putRawLine(QString("USER %1 8 * :%2").arg(identity["Ident"].toString()).arg(identity["RealName"].toString()));
96 void Server::socketDisconnected( ) {
97 //qDebug() << "Socket disconnected!";
98 emit disconnected(network);
103 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
104 //qDebug() << "Socket state changed: " << state;
107 QString Server::updateNickFromMask(QString mask) {
108 QString user = userFromMask(mask);
109 QString host = hostFromMask(mask);
110 QString nick = nickFromMask(mask);
111 if(nicks.contains(nick) && !user.isEmpty() && !host.isEmpty()) {
112 VarMap n = nicks[nick];
113 if(n["User"].toString() != user || n["Host"].toString() != host) {
114 if(!n["User"].toString().isEmpty() || !n["Host"].toString().isEmpty())
115 qWarning(QString("Strange: Hostmask for nick %1 has changed!").arg(nick).toAscii());
116 n["User"] = user; n["Host"] = host;
118 emit nickUpdated(network, nick, n);
124 void Server::userInput(QString net, QString buf, QString msg) {
125 if(net != network) return; // not me!
126 //msg = msg.trimmed(); // remove whitespace from start and end
127 if(msg.isEmpty()) return;
128 if(!msg.startsWith('/')) {
129 msg = QString("/SAY ") + msg;
131 handleUserInput(buf, msg);
134 void Server::putRawLine(QString s) {
135 // qDebug() << "SentRaw: " << s;
137 socket.write(s.toAscii());
140 void Server::putCmd(QString cmd, QStringList params, QString prefix) {
142 if(!prefix.isEmpty()) m += ":" + prefix + " ";
144 for(int i = 0; i < params.size() - 1; i++) {
145 m += " " + params[i];
147 if(!params.isEmpty()) m += " :" + params.last();
148 // qDebug() << "Sent: " << m;
150 socket.write(m.toAscii());
153 /** Handle a raw message string sent by the server. We try to find a suitable handler, otherwise we call a default handler. */
154 void Server::handleServerMsg(QString msg) {
157 qWarning() << "Received empty string from server!";
160 // OK, first we split the raw message into its various parts...
166 prefix = msg.section(' ', 0, 0);
167 msg = msg.section(' ', 1);
169 cmd = msg.section(' ', 0, 0).toUpper();
170 msg = msg.section(' ', 1);
171 QString left, trailing;
172 // RPL_ISUPPORT (005) can contain colons, so don't treat it like the rest of the commands
173 if(cmd.toUInt() == 5) {
174 left = msg.remove(QString(":are supported by this server"));
176 left = msg.section(':', 0, 0);
177 trailing = msg.section(':', 1);
179 if(!left.isEmpty()) {
180 params << left.split(' ', QString::SkipEmptyParts);
182 if(!trailing.isEmpty()) {
185 // numeric replies usually have our own nick as first param. Remove this!
186 // BTW, this behavior is not in the RFC.
187 uint num = cmd.toUInt();
188 if(num > 1 && params.count() > 0) { // 001 sets our nick, so we shouldn't remove anything
189 if(params[0] == ownNick) params.removeFirst();
190 else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
192 // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
193 QString hname = cmd.toLower();
194 hname[0] = hname[0].toUpper();
195 hname = "handleServer" + hname;
196 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
197 // Ok. Default handler it is.
198 defaultServerHandler(cmd, prefix, params);
200 } catch(Exception e) {
201 emit displayMsg(Message::error("", e.msg()));
205 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
206 uint num = cmd.toUInt();
208 // A lot of server messages don't really need their own handler because they don't do much.
209 // Catch and handle these here.
211 // Welcome, status, info messages. Just display these.
212 case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
213 emit displayMsg(Message::server("", params.join(" "), prefix));
215 // Server error messages without param, just display them
216 case 409: case 411: case 412: case 422: case 424: case 431: case 445: case 446: case 451: case 462:
217 case 463: case 464: case 465: case 466: case 472: case 481: case 483: case 485: case 491: case 501: case 502:
218 emit displayMsg(Message::error("", params.join(" "), prefix));
220 // Server error messages, display them in red. First param will be appended.
221 case 401: case 402: case 403: case 404: case 406: case 408: case 415: case 421: case 432: case 442:
222 { QString p = params.takeFirst();
223 emit displayMsg(Message::error("", params.join(" ") + " " + p, prefix));
226 // Server error messages which will be displayed with a colon between the first param and the rest
227 case 413: case 414: case 423: case 433: case 436: case 441: case 444: case 461:
228 case 467: case 471: case 473: case 474: case 475: case 476: case 477: case 478: case 482:
229 { QString p = params.takeFirst();
230 emit displayMsg(Message::error("", p + ": " + params.join(" ")));
233 // Ignore these commands.
237 // Everything else will be marked in red, so we can add them somewhere.
239 emit displayMsg(Message::error("", cmd + " " + params.join(" "), prefix));
241 //qDebug() << prefix <<":"<<cmd<<params;
243 emit displayMsg(Message::error("", QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
244 //qDebug() << prefix <<":"<<cmd<<params;
248 void Server::handleUserInput(QString bufname, QString usrMsg) {
250 /* Looks like we don't need core-side buffers...
252 if(!bufname.isEmpty()) {
253 Q_ASSERT(buffers.contains(bufname));
254 buffer = buffers[bufname];
257 QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
258 QString msg = usrMsg.section(' ', 1);
259 QString hname = cmd.toLower();
260 hname[0] = hname[0].toUpper();
261 hname = "handleUser" + hname;
262 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, bufname), Q_ARG(QString, msg))) {
263 // Ok. Default handler it is.
264 defaultUserHandler(bufname, cmd, msg);
266 } catch(Exception e) {
267 emit displayMsg(Message::error("", e.msg()));
271 void Server::defaultUserHandler(QString bufname, QString cmd, QString msg) {
272 emit displayMsg(Message::error("", QString("Error: %1 %2").arg(cmd).arg(msg)));
276 /**********************************************************************************/
279 void Server::handleUser(QString bufname, QString msg) {
285 void Server::handleUserAway(QString bufname, QString msg) {
286 putCmd("AWAY", QStringList(msg));
289 void Server::handleUserDeop(QString bufname, QString msg) {
290 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
291 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'o';
293 params << bufname << m << nicks;
294 putCmd("MODE", params);
297 void Server::handleUserDevoice(QString bufname, QString msg) {
298 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
299 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'v';
301 params << bufname << m << nicks;
302 putCmd("MODE", params);
305 void Server::handleUserInvite(QString bufname, QString msg) {
307 params << msg << bufname;
308 putCmd("INVITE", params);
311 void Server::handleUserJoin(QString bufname, QString msg) {
312 putCmd("JOIN", QStringList(msg));
315 void Server::handleUserKick(QString bufname, QString msg) {
317 params << bufname << msg.split(' ', QString::SkipEmptyParts);
318 putCmd("KICK", params);
321 void Server::handleUserList(QString bufname, QString msg) {
322 putCmd("LIST", msg.split(' ', QString::SkipEmptyParts));
325 void Server::handleUserMode(QString bufname, QString msg) {
326 putCmd("MODE", msg.split(' ', QString::SkipEmptyParts));
329 void Server::handleUserMsg(QString bufname, QString msg) {
330 QString nick = msg.section(" ", 0, 0);
331 msg = msg.section(" ", 1);
332 if(nick.isEmpty() || msg.isEmpty()) return;
334 params << nick << msg;
335 putCmd("PRIVMSG", params);
338 void Server::handleUserNick(QString bufname, QString msg) {
339 QString nick = msg.section(' ', 0, 0);
340 putCmd("NICK", QStringList(nick));
343 void Server::handleUserOp(QString bufname, QString msg) {
344 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
345 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'o';
347 params << bufname << m << nicks;
348 putCmd("MODE", params);
351 void Server::handleUserPart(QString bufname, QString msg) {
353 params << bufname << msg;
354 putCmd("PART", params);
357 void Server::handleUserQuery(QString bufname, QString msg) {
358 QString nick = msg.section(' ', 0, 0);
359 if(!nick.isEmpty()) emit queryRequested(network, nick);
362 void Server::handleUserQuit(QString bufname, QString msg) {
363 putCmd("QUIT", QStringList(msg));
366 void Server::handleUserQuote(QString bufname, QString msg) {
370 void Server::handleUserSay(QString bufname, QString msg) {
371 if(bufname.isEmpty()) return; // server buffer
373 params << bufname << msg;
374 putCmd("PRIVMSG", params);
375 if(isChannelName(bufname)) {
376 emit displayMsg(Message::plain(params[0], msg, ownNick, Message::Self));
378 emit displayMsg(Message::plain(params[0], msg, ownNick, Message::Self|Message::PrivMsg));
382 void Server::handleUserTopic(QString bufname, QString msg) {
383 if(bufname.isEmpty()) return;
385 params << bufname << msg;
386 putCmd("TOPIC", params);
389 void Server::handleUserVoice(QString bufname, QString msg) {
390 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
391 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'v';
393 params << bufname << m << nicks;
394 putCmd("MODE", params);
397 /**********************************************************************************/
399 void Server::handleServerJoin(QString prefix, QStringList params) {
400 Q_ASSERT(params.count() == 1);
401 QString nick = updateNickFromMask(prefix);
402 if(nick == ownNick) {
403 // Q_ASSERT(!buffers.contains(params[0])); // cannot join a buffer twice!
404 // Buffer *buf = new Buffer(params[0]);
405 // buffers[params[0]] = buf;
406 topics[params[0]] = "";
407 emit topicSet(network, params[0], "");
410 if(nicks.contains(nick)) {
412 VarMap chans = n["Channels"].toMap();
413 // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
414 chans[params[0]] = VarMap();
415 n["Channels"] = chans;
417 emit nickUpdated(network, nick, n);
420 chans[params[0]] = VarMap();
421 n["Channels"] = chans;
422 n["User"] = userFromMask(prefix);
423 n["Host"] = hostFromMask(prefix);
425 emit nickAdded(network, nick, n);
427 emit displayMsg(Message::join(params[0], params[0], prefix));
431 void Server::handleServerKick(QString prefix, QStringList params) {
432 QString kicker = updateNickFromMask(prefix);
433 QString nick = params[1];
434 Q_ASSERT(nicks.contains(nick));
435 VarMap n = nicks[nick];
436 VarMap chans = n["Channels"].toMap();
437 Q_ASSERT(chans.contains(params[0]));
438 chans.remove(params[0]);
440 if(params.count() > 2) msg = QString("%1 %2").arg(msg).arg(params[2]);
441 emit displayMsg(Message::kick(params[0], msg, prefix));
442 if(chans.count() > 0) {
443 n["Channels"] = chans;
445 emit nickUpdated(network, nick, n);
448 emit nickRemoved(network, nick);
450 if(nick == ownNick) {
451 topics.remove(params[0]);
455 void Server::handleServerMode(QString prefix, QStringList params) {
456 if(isChannelName(params[0])) {
457 // TODO only channel-user modes supported by now
458 QString prefixes = serverSupports["PrefixModes"].toString();
459 QString modes = params[1];
463 while(m < modes.length()) {
464 if(modes[m] == '+') { add = true; m++; continue; }
465 if(modes[m] == '-') { add = false; m++; continue; }
466 if(prefixes.contains(modes[m])) { // it's a user channel mode
467 Q_ASSERT(params.count() > m);
468 QString nick = params[p++];
469 if(nicks.contains(nick)) { // sometimes, a server might try to set a MODE on a nick that is no longer there
470 VarMap n = nicks[nick]; VarMap clist = n["Channels"].toMap(); VarMap chan = clist[params[0]].toMap();
471 QString mstr = chan["Mode"].toString();
472 add ? mstr += modes[m] : mstr.remove(modes[m]);
473 chan["Mode"] = mstr; clist[params[0]] = chan; n["Channels"] = clist; nicks[nick] = n;
474 emit nickUpdated(network, nick, n);
478 // TODO add more modes
482 emit displayMsg(Message::mode(params[0], params.join(" "), prefix));
484 //Q_ASSERT(nicks.contains(params[0]));
485 //VarMap n = nicks[params[0]].toMap();
486 //QString mode = n["Mode"].toString();
487 emit displayMsg(Message::mode("", params.join(" ")));
491 void Server::handleServerNick(QString prefix, QStringList params) {
492 QString oldnick = updateNickFromMask(prefix);
493 QString newnick = params[0];
494 VarMap v = nicks.take(oldnick);
496 VarMap chans = v["Channels"].toMap();
497 foreach(QString c, chans.keys()) {
498 if(oldnick != ownNick) { emit displayMsg(Message::nick(c, newnick, prefix)); }
499 else { emit displayMsg(Message::nick(c, newnick, newnick)); }
501 emit nickRenamed(network, oldnick, newnick);
502 if(oldnick == ownNick) {
504 emit ownNickSet(network, newnick);
508 void Server::handleServerNotice(QString prefix, QStringList params) {
509 //Message msg(Message::Notice, params[1], prefix);
510 if(prefix == currentServer) emit displayMsg(Message::server("", params[1], prefix));
511 else emit displayMsg(Message::notice("", params[1], prefix));
514 void Server::handleServerPart(QString prefix, QStringList params) {
515 QString nick = updateNickFromMask(prefix);
516 Q_ASSERT(nicks.contains(nick));
517 VarMap n = nicks[nick];
518 VarMap chans = n["Channels"].toMap();
519 Q_ASSERT(chans.contains(params[0]));
520 chans.remove(params[0]);
522 if(params.count() > 1) msg = params[1];
523 emit displayMsg(Message::part(params[0], msg, prefix));
524 if(chans.count() > 0) {
525 n["Channels"] = chans;
527 emit nickUpdated(network, nick, n);
530 emit nickRemoved(network, nick);
532 if(nick == ownNick) {
533 Q_ASSERT(topics.contains(params[0]));
534 topics.remove(params[0]);
538 void Server::handleServerPing(QString prefix, QStringList params) {
539 putCmd("PONG", params);
542 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
543 updateNickFromMask(prefix);
544 Q_ASSERT(params.count() >= 2);
545 if(params.count()<2) emit displayMsg(Message::plain(params[0], "", prefix));
547 if(params[0] == ownNick) {
548 emit displayMsg(Message::plain("", params[1], prefix, Message::PrivMsg));
550 Q_ASSERT(isChannelName(params[0])); // should be channel!
551 emit displayMsg(Message::plain(params[0], params[1], prefix));
556 void Server::handleServerQuit(QString prefix, QStringList params) {
557 QString nick = updateNickFromMask(prefix);
558 Q_ASSERT(nicks.contains(nick));
559 VarMap chans = nicks[nick]["Channels"].toMap();
560 foreach(QString c, chans.keys()) {
561 emit displayMsg(Message::quit(c, params[0], prefix));
564 emit nickRemoved(network, nick);
567 void Server::handleServerTopic(QString prefix, QStringList params) {
568 QString nick = updateNickFromMask(prefix);
569 Q_ASSERT(nicks.contains(nick));
570 topics[params[0]] = params[1];
571 emit topicSet(network, params[0], params[1]);
572 emit displayMsg(Message::server(params[0], tr("%1 has changed topic for %2 to: \"%3\"").arg(nick).arg(params[0]).arg(params[1])));
576 void Server::handleServer001(QString prefix, QStringList params) {
577 currentServer = prefix;
580 n["Channels"] = VarMap();
582 emit ownNickSet(network, ownNick);
583 emit nickAdded(network, ownNick, VarMap());
584 emit displayMsg(Message::server("", params[1], prefix));
586 QStringList performList = networkSettings["Perform"].toString().split( "\n" );
587 int count = performList.count();
588 for(int a = 0; a < count; a++) {
589 if(!performList[a].isEmpty() ) {
590 userInput(network, "", performList[a]);
596 // TODO Complete 005 handling, also use sensible defaults for non-sent stuff
597 void Server::handleServer005(QString prefix, QStringList params) {
598 foreach(QString p, params) {
599 QString key = p.section("=", 0, 0);
600 QString val = p.section("=", 1);
601 serverSupports[key] = val;
602 // handle some special cases
603 if(key == "PREFIX") {
604 VarMap foo; QString modes, prefixes;
605 Q_ASSERT(val.contains(')') && val.startsWith('('));
607 for(p = 2; p < val.length(); p++) if(val[p] == ')') break;
609 for(; val[m] != ')'; m++, p++) {
610 Q_ASSERT(p < val.length());
611 foo[QString(val[m])] = QString(val[p]);
612 modes += val[m]; prefixes += val[p];
614 serverSupports["PrefixModes"] = modes; serverSupports["Prefixes"] = prefixes;
615 serverSupports["ModePrefixMap"] = foo;
622 void Server::handleServer331(QString prefix, QStringList params) {
623 topics[params[0]] = "";
624 emit topicSet(network, params[0], "");
625 emit displayMsg(Message::server(params[0], tr("No topic is set for %1.").arg(params[0])));
629 void Server::handleServer332(QString prefix, QStringList params) {
630 topics[params[0]] = params[1];
631 emit topicSet(network, params[0], params[1]);
632 emit displayMsg(Message::server(params[0], tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1])));
635 /* Topic set by... */
636 void Server::handleServer333(QString prefix, QStringList params) {
637 emit displayMsg(Message::server(params[0],
638 tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString())));
642 void Server::handleServer353(QString prefix, QStringList params) {
643 params.removeFirst(); // = or *
644 QString buf = params.takeFirst();
645 QString prefixes = serverSupports["Prefixes"].toString();
646 foreach(QString nick, params[0].split(' ')) {
647 QString mode = "", pfx = "";
648 if(prefixes.contains(nick[0])) {
651 if(prefixes[i] == nick[0]) { mode = serverSupports["PrefixModes"].toString()[i]; break; }
654 VarMap c; c["Mode"] = mode; c["Prefix"] = pfx;
655 if(nicks.contains(nick)) {
656 VarMap n = nicks[nick];
657 VarMap chans = n["Channels"].toMap();
659 n["Channels"] = chans;
661 emit nickUpdated(network, nick, n);
663 VarMap n; VarMap c; VarMap chans;
666 n["Channels"] = chans;
668 emit nickAdded(network, nick, n);
672 /***********************************************************************************/
674 /* Exception classes for message handling */
675 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
676 _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
680 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
681 _msg = QString("Unknown Command: ") + cmd;