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 handleUserMsg(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 = msg.section(':', 0, 0);
158 QString trailing = msg.section(':', 1);
159 if(!left.isEmpty()) {
160 params << left.split(' ', QString::SkipEmptyParts);
162 if(!trailing.isEmpty()) {
165 // numeric replies usually have our own nick as first param. Remove this!
166 // BTW, this behavior is not in the RFC.
167 uint num = cmd.toUInt();
168 if(num > 1 && params.count() > 0) { // 001 sets our nick, so we shouldn't remove anything
169 if(params[0] == currentNick) params.removeFirst();
170 else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
172 // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
173 QString hname = cmd.toLower();
174 hname[0] = hname[0].toUpper();
175 hname = "handleServer" + hname;
176 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
177 // Ok. Default handler it is.
178 defaultServerHandler(cmd, prefix, params);
180 } catch(Exception e) {
181 emit displayMsg("", Message(Message::Error, e.msg()));
185 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
186 uint num = cmd.toUInt();
188 // A lot of server messages don't really need their own handler because they don't do much.
189 // Catch and handle these here.
191 // Welcome, status, info messages. Just display these.
192 case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
193 emit displayMsg("", Message(Message::Server, params.join(" "), prefix));
195 // Ignore these commands.
199 // Everything else will be marked in red, so we can add them somewhere.
201 emit displayMsg("", Message(Message::Error, cmd + " " + params.join(" "), prefix));
203 //qDebug() << prefix <<":"<<cmd<<params;
205 emit displayMsg("", Message(Message::Error, QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
206 //qDebug() << prefix <<":"<<cmd<<params;
210 void Server::handleUserMsg(QString bufname, QString usrMsg) {
213 if(!bufname.isEmpty()) {
214 Q_ASSERT(buffers.contains(bufname));
215 buffer = buffers[bufname];
217 QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
218 QString msg = usrMsg.section(' ', 1).trimmed();
219 QString hname = cmd.toLower();
220 hname[0] = hname[0].toUpper();
221 hname = "handleUser" + hname;
222 if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, msg), Q_ARG(Buffer*, buffer))) {
223 // Ok. Default handler it is.
224 defaultUserHandler(cmd, msg, buffer);
226 } catch(Exception e) {
227 emit displayMsg("", Message(Message::Error, e.msg()));
231 void Server::defaultUserHandler(QString cmd, QString msg, Buffer *buf) {
232 emit displayMsg("", Message(Message::Error, QString("Error: %1 %2").arg(cmd).arg(msg)));
236 /**********************************************************************************/
239 void Server::handleUser(QString msg, Buffer *buf) {
245 void Server::handleUserJoin(QString msg, Buffer *buf) {
246 putCmd("JOIN", QStringList(msg));
250 void Server::handleUserQuote(QString msg, Buffer *buf) {
254 void Server::handleUserSay(QString msg, Buffer *buf) {
255 if(!buf) return; // server buffer
257 params << buf->name() << msg;
258 putCmd("PRIVMSG", params);
259 emit displayMsg(params[0], Message(Message::Msg, msg, currentNick, Message::Self));
262 /**********************************************************************************/
264 void Server::handleServerJoin(QString prefix, QStringList params) {
265 Q_ASSERT(params.count() == 1);
266 QString nick = updateNickFromMask(prefix);
267 if(nick == currentNick) {
268 Q_ASSERT(!buffers.contains(params[0])); // cannot join a buffer twice!
269 Buffer *buf = new Buffer(params[0]);
270 buffers[params[0]] = buf;
273 if(nicks.contains(nick)) {
274 n = nicks[nick].toMap();
275 VarMap chans = n["Channels"].toMap();
276 // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
277 chans[params[0]] = VarMap();
278 n["Channels"] = chans;
280 emit nickUpdated(network, nick, n);
283 chans[params[0]] = VarMap();
284 n["Channels"] = chans;
285 n["User"] = userFromMask(prefix);
286 n["Host"] = hostFromMask(prefix);
288 emit nickAdded(network, nick, n);
290 emit displayMsg(params[0], Message(Message::Join, params[0], prefix));
294 void Server::handleServerKick(QString prefix, QStringList params) {
295 QString kicker = updateNickFromMask(prefix);
296 QString nick = params[1];
297 Q_ASSERT(nicks.contains(nick));
298 VarMap n = nicks[nick].toMap();
299 VarMap chans = n["Channels"].toMap();
300 Q_ASSERT(chans.contains(params[0]));
301 chans.remove(params[0]);
303 if(params.count() > 2) msg = QString("%1 %2").arg(msg).arg(params[2]);
304 emit displayMsg(params[0], Message(Message::Kick, msg, prefix));
305 if(chans.count() > 0) {
306 n["Channels"] = chans;
308 emit nickUpdated(network, nick, n);
311 emit nickRemoved(network, nick);
315 void Server::handleServerNick(QString prefix, QStringList params) {
316 QString oldnick = updateNickFromMask(prefix);
317 QString newnick = params[0];
318 QVariant v = nicks.take(oldnick);
320 VarMap chans = v.toMap()["Channels"].toMap();
321 foreach(QString c, chans.keys()) {
322 if(oldnick != currentNick) { emit displayMsg(c, Message(Message::Nick, newnick, prefix)); }
323 else { emit displayMsg(c, Message(Message::Nick, newnick, newnick)); }
325 emit nickRenamed(network, oldnick, newnick);
326 if(oldnick == currentNick) {
327 currentNick == newnick;
328 emit ownNickSet(network, newnick);
332 void Server::handleServerNotice(QString prefix, QStringList params) {
333 //Message msg(Message::Notice, params[1], prefix);
334 if(prefix == currentServer) emit displayMsg("", Message(Message::Server, params[1], prefix));
335 else emit displayMsg("", Message(Message::Notice, params[1], prefix));
338 void Server::handleServerPart(QString prefix, QStringList params) {
339 QString nick = updateNickFromMask(prefix);
340 Q_ASSERT(nicks.contains(nick));
341 VarMap n = nicks[nick].toMap();
342 VarMap chans = n["Channels"].toMap();
343 Q_ASSERT(chans.contains(params[0]));
344 chans.remove(params[0]);
346 if(params.count() > 1) msg = params[1];
347 emit displayMsg(params[0], Message(Message::Part, msg, prefix));
348 if(chans.count() > 0) {
349 n["Channels"] = chans;
351 emit nickUpdated(network, nick, n);
354 emit nickRemoved(network, nick);
358 void Server::handleServerPing(QString prefix, QStringList params) {
359 putCmd("PONG", params);
362 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
363 updateNickFromMask(prefix);
364 emit displayMsg(params[0], Message(Message::Msg, params[1], prefix));
368 void Server::handleServerQuit(QString prefix, QStringList params) {
369 QString nick = updateNickFromMask(prefix);
370 Q_ASSERT(nicks.contains(nick));
371 VarMap chans = nicks[nick].toMap()["Channels"].toMap();
372 foreach(QString c, chans.keys()) {
373 emit displayMsg(c, Message(Message::Quit, params[0], prefix));
376 emit nickRemoved(network, nick);
380 void Server::handleServer001(QString prefix, QStringList params) {
381 currentServer = prefix;
382 currentNick = params[0];
383 emit ownNickSet(network, currentNick);
384 emit displayMsg("", Message(Message::Server, params[1], prefix));
388 void Server::handleServer331(QString prefix, QStringList params) {
389 emit topicSet(network, params[0], "");
393 void Server::handleServer332(QString prefix, QStringList params) {
394 emit topicSet(network, params[0], params[1]);
395 emit displayMsg(params[0], Message(Message::Server, tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1])));
398 /* Topic set by... */
399 void Server::handleServer333(QString prefix, QStringList params) {
400 emit displayMsg(params[0], Message(Message::Server,
401 tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString())));
405 void Server::handleServer353(QString prefix, QStringList params) {
406 params.removeFirst(); // = or *
407 QString buf = params.takeFirst();
408 foreach(QString nick, params[0].split(' ')) {
409 // TODO: parse more prefix characters! use 005?
411 if(nick.startsWith('@')) { mode = "o"; nick.remove(0,1); }
412 else if(nick.startsWith('+')) { mode = "v"; nick.remove(0,1); }
413 VarMap c; c["Mode"] = mode;
414 if(nicks.contains(nick)) {
415 VarMap n = nicks[nick].toMap();
416 VarMap chans = n["Channels"].toMap();
418 n["Channels"] = chans;
420 emit nickUpdated(network, nick, n);
422 VarMap n; VarMap c; VarMap chans;
425 n["Channels"] = chans;
428 emit nickAdded(network, nick, n);
432 /***********************************************************************************/
434 /* Exception classes for message handling */
435 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
436 _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
440 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
441 _msg = QString("Unknown Command: ") + cmd;