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() << "SentCmd: " << 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, display them in red. First param will be appended.
202 case 401: case 402: case 403: case 404:
203 { QString p = params.takeFirst();
204 emit displayMsg("", Message(Message::Error, params.join(" ") + " " + p, prefix));
207 // Ignore these commands.
211 // Everything else will be marked in red, so we can add them somewhere.
213 emit displayMsg("", Message(Message::Error, cmd + " " + params.join(" "), prefix));
215 //qDebug() << prefix <<":"<<cmd<<params;
217 emit displayMsg("", Message(Message::Error, QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
218 //qDebug() << prefix <<":"<<cmd<<params;
222 void Server::handleUserInput(QString bufname, QString usrMsg) {
224 /* Looks like we don't need core-side buffers...
226 if(!bufname.isEmpty()) {
227 Q_ASSERT(buffers.contains(bufname));
228 buffer = buffers[bufname];
231 QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
232 QString msg = usrMsg.section(' ', 1).trimmed();
233 QString hname = cmd.toLower();
234 hname[0] = hname[0].toUpper();
235 hname = "handleUser" + hname;
236 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, bufname), Q_ARG(QString, msg))) {
237 // Ok. Default handler it is.
238 defaultUserHandler(bufname, cmd, msg);
240 } catch(Exception e) {
241 emit displayMsg("", Message(Message::Error, e.msg()));
245 void Server::defaultUserHandler(QString bufname, QString cmd, QString msg) {
246 emit displayMsg("", Message(Message::Error, QString("Error: %1 %2").arg(cmd).arg(msg)));
250 /**********************************************************************************/
253 void Server::handleUser(QString bufname, QString msg) {
259 void Server::handleUserAway(QString bufname, QString msg) {
260 putCmd("AWAY", QStringList(msg));
263 void Server::handleUserDeop(QString bufname, QString msg) {
264 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
265 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'o';
267 params << bufname << m << nicks;
268 putCmd("MODE", params);
271 void Server::handleUserDevoice(QString bufname, QString msg) {
272 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
273 QString m = "-"; for(int i = 0; i < nicks.count(); i++) m += 'v';
275 params << bufname << m << nicks;
276 putCmd("MODE", params);
279 void Server::handleUserInvite(QString bufname, QString msg) {
281 params << msg << bufname;
282 putCmd("INVITE", params);
285 void Server::handleUserJoin(QString bufname, QString msg) {
286 putCmd("JOIN", QStringList(msg));
289 void Server::handleUserKick(QString bufname, QString msg) {
291 params << bufname << msg.split(' ', QString::SkipEmptyParts);
292 putCmd("KICK", params);
295 void Server::handleUserList(QString bufname, QString msg) {
296 putCmd("LIST", msg.split(' ', QString::SkipEmptyParts));
299 void Server::handleUserMode(QString bufname, QString msg) {
300 putCmd("MODE", msg.split(' ', QString::SkipEmptyParts));
303 void Server::handleUserMsg(QString bufname, QString msg) {
304 QString nick = msg.section(" ", 0, 0);
305 msg = msg.section(" ", 1).trimmed();
306 if(nick.isEmpty() || msg.isEmpty()) return;
308 params << nick << msg;
309 putCmd("PRIVMSG", params);
312 void Server::handleUserNick(QString bufname, QString msg) {
313 QString nick = msg.section(' ', 0, 0);
314 putCmd("NICK", QStringList(nick));
317 void Server::handleUserOp(QString bufname, QString msg) {
318 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
319 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'o';
321 params << bufname << m << nicks;
322 putCmd("MODE", params);
325 void Server::handleUserPart(QString bufname, QString msg) {
327 params << bufname << msg;
328 putCmd("PART", params);
331 void Server::handleUserQuit(QString bufname, QString msg) {
332 putCmd("QUIT", QStringList(msg));
335 void Server::handleUserQuote(QString bufname, QString msg) {
339 void Server::handleUserSay(QString bufname, QString msg) {
340 if(bufname.isEmpty()) return; // server buffer
342 params << bufname << msg;
343 putCmd("PRIVMSG", params);
344 emit displayMsg(params[0], Message(Message::Msg, msg, currentNick, Message::Self));
347 void Server::handleUserVoice(QString bufname, QString msg) {
348 QStringList nicks = msg.split(' ', QString::SkipEmptyParts);
349 QString m = "+"; for(int i = 0; i < nicks.count(); i++) m += 'v';
351 params << bufname << m << nicks;
352 putCmd("MODE", params);
355 /**********************************************************************************/
357 void Server::handleServerJoin(QString prefix, QStringList params) {
358 Q_ASSERT(params.count() == 1);
359 QString nick = updateNickFromMask(prefix);
360 if(nick == currentNick) {
361 // Q_ASSERT(!buffers.contains(params[0])); // cannot join a buffer twice!
362 // Buffer *buf = new Buffer(params[0]);
363 // buffers[params[0]] = buf;
366 if(nicks.contains(nick)) {
367 n = nicks[nick].toMap();
368 VarMap chans = n["Channels"].toMap();
369 // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
370 chans[params[0]] = VarMap();
371 n["Channels"] = chans;
373 emit nickUpdated(network, nick, n);
376 chans[params[0]] = VarMap();
377 n["Channels"] = chans;
378 n["User"] = userFromMask(prefix);
379 n["Host"] = hostFromMask(prefix);
381 emit nickAdded(network, nick, n);
383 emit displayMsg(params[0], Message(Message::Join, params[0], prefix));
387 void Server::handleServerKick(QString prefix, QStringList params) {
388 QString kicker = updateNickFromMask(prefix);
389 QString nick = params[1];
390 Q_ASSERT(nicks.contains(nick));
391 VarMap n = nicks[nick].toMap();
392 VarMap chans = n["Channels"].toMap();
393 Q_ASSERT(chans.contains(params[0]));
394 chans.remove(params[0]);
396 if(params.count() > 2) msg = QString("%1 %2").arg(msg).arg(params[2]);
397 emit displayMsg(params[0], Message(Message::Kick, msg, prefix));
398 if(chans.count() > 0) {
399 n["Channels"] = chans;
401 emit nickUpdated(network, nick, n);
404 emit nickRemoved(network, nick);
408 void Server::handleServerMode(QString prefix, QStringList params) {
409 if(isChannelName(params[0])) {
410 // TODO only channel-user modes supported by now
411 QString prefixes = serverSupports["PrefixModes"].toString();
412 QString modes = params[1];
416 while(m < modes.length()) {
417 if(modes[m] == '+') { add = true; m++; continue; }
418 if(modes[m] == '-') { add = false; m++; continue; }
419 if(prefixes.contains(modes[m])) { // it's a user channel mode
420 Q_ASSERT(params.count() > m && nicks.contains(params[p]));
421 QString nick = params[p++];
422 VarMap n = nicks[nick].toMap(); VarMap clist = n["Channels"].toMap(); VarMap chan = clist[params[0]].toMap();
423 QString mstr = chan["Mode"].toString();
424 add ? mstr += modes[m] : mstr.remove(modes[m]);
425 chan["Mode"] = mstr; clist[params[0]] = chan; n["Channels"] = clist; nicks[nick] = n;
426 emit nickUpdated(network, nick, n);
429 // TODO add more modes
433 emit displayMsg(params[0], Message(Message::Mode, params.join(" "), prefix));
435 //Q_ASSERT(nicks.contains(params[0]));
436 //VarMap n = nicks[params[0]].toMap();
437 //QString mode = n["Mode"].toString();
438 emit displayMsg("", Message(Message::Mode, params.join(" ")));
442 void Server::handleServerNick(QString prefix, QStringList params) {
443 QString oldnick = updateNickFromMask(prefix);
444 QString newnick = params[0];
445 QVariant v = nicks.take(oldnick);
447 VarMap chans = v.toMap()["Channels"].toMap();
448 foreach(QString c, chans.keys()) {
449 if(oldnick != currentNick) { emit displayMsg(c, Message(Message::Nick, newnick, prefix)); }
450 else { emit displayMsg(c, Message(Message::Nick, newnick, newnick)); }
452 emit nickRenamed(network, oldnick, newnick);
453 if(oldnick == currentNick) {
454 currentNick = newnick;
455 emit ownNickSet(network, newnick);
459 void Server::handleServerNotice(QString prefix, QStringList params) {
460 //Message msg(Message::Notice, params[1], prefix);
461 if(prefix == currentServer) emit displayMsg("", Message(Message::Server, params[1], prefix));
462 else emit displayMsg("", Message(Message::Notice, params[1], prefix));
465 void Server::handleServerPart(QString prefix, QStringList params) {
466 QString nick = updateNickFromMask(prefix);
467 Q_ASSERT(nicks.contains(nick));
468 VarMap n = nicks[nick].toMap();
469 VarMap chans = n["Channels"].toMap();
470 Q_ASSERT(chans.contains(params[0]));
471 chans.remove(params[0]);
473 if(params.count() > 1) msg = params[1];
474 emit displayMsg(params[0], Message(Message::Part, msg, prefix));
475 if(chans.count() > 0) {
476 n["Channels"] = chans;
478 emit nickUpdated(network, nick, n);
481 emit nickRemoved(network, nick);
485 void Server::handleServerPing(QString prefix, QStringList params) {
486 putCmd("PONG", params);
489 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
490 updateNickFromMask(prefix);
491 emit displayMsg(params[0], Message(Message::Msg, params[1], prefix));
495 void Server::handleServerQuit(QString prefix, QStringList params) {
496 QString nick = updateNickFromMask(prefix);
497 Q_ASSERT(nicks.contains(nick));
498 VarMap chans = nicks[nick].toMap()["Channels"].toMap();
499 foreach(QString c, chans.keys()) {
500 emit displayMsg(c, Message(Message::Quit, params[0], prefix));
503 emit nickRemoved(network, nick);
507 void Server::handleServer001(QString prefix, QStringList params) {
508 currentServer = prefix;
509 currentNick = params[0];
511 n["Channels"] = VarMap();
512 nicks[currentNick] = n;
513 emit ownNickSet(network, currentNick);
514 emit nickAdded(network, currentNick, VarMap());
515 emit displayMsg("", Message(Message::Server, params[1], prefix));
519 void Server::handleServer005(QString prefix, QStringList params) {
520 foreach(QString p, params) {
521 QString key = p.section("=", 0, 0);
522 QString val = p.section("=", 1);
523 serverSupports[key] = val;
524 // handle some special cases
525 if(key == "PREFIX") {
526 VarMap foo; QString modes, prefixes;
527 Q_ASSERT(val.contains(')') && val.startsWith('('));
529 for(p = 2; p < val.length(); p++) if(val[p] == ')') break;
531 for(; val[m] != ')'; m++, p++) {
532 Q_ASSERT(p < val.length());
533 foo[QString(val[m])] = QString(val[p]);
534 modes += val[m]; prefixes += val[p];
536 serverSupports["PrefixModes"] = modes; serverSupports["Prefixes"] = prefixes;
537 serverSupports["ModePrefixMap"] = foo;
544 void Server::handleServer331(QString prefix, QStringList params) {
545 emit topicSet(network, params[0], "");
549 void Server::handleServer332(QString prefix, QStringList params) {
550 emit topicSet(network, params[0], params[1]);
551 emit displayMsg(params[0], Message(Message::Server, tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1])));
554 /* Topic set by... */
555 void Server::handleServer333(QString prefix, QStringList params) {
556 emit displayMsg(params[0], Message(Message::Server,
557 tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString())));
561 void Server::handleServer353(QString prefix, QStringList params) {
562 params.removeFirst(); // = or *
563 QString buf = params.takeFirst();
564 QString prefixes = serverSupports["Prefixes"].toString();
565 foreach(QString nick, params[0].split(' ')) {
566 QString mode = "", pfx = "";
567 if(prefixes.contains(nick[0])) {
570 if(prefixes[i] == nick[0]) { mode = serverSupports["PrefixModes"].toString()[i]; break; }
573 VarMap c; c["Mode"] = mode; c["Prefix"] = pfx;
574 if(nicks.contains(nick)) {
575 VarMap n = nicks[nick].toMap();
576 VarMap chans = n["Channels"].toMap();
578 n["Channels"] = chans;
580 emit nickUpdated(network, nick, n);
582 VarMap n; VarMap c; VarMap chans;
585 n["Channels"] = chans;
587 emit nickAdded(network, nick, n);
591 /***********************************************************************************/
593 /* Exception classes for message handling */
594 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
595 _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
599 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
600 _msg = QString("Unknown Command: ") + cmd;