Time for another update. Most significantly, I implemented most user commands.
[quassel.git] / network / server.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005/06 by The Quassel Team                             *
3  *   devel@quassel-irc.org                                                 *
4  *                                                                         *
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.                                   *
9  *                                                                         *
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.                          *
14  *                                                                         *
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  ***************************************************************************/
20
21 #include "util.h"
22 #include "global.h"
23 #include "server.h"
24 #include "cmdcodes.h"
25 #include "message.h"
26
27 #include <QMetaObject>
28 #include <QDateTime>
29
30 Server::Server(QString net) : network(net) {
31
32 }
33
34 Server::~Server() {
35
36 }
37
38 void Server::run() {
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()));
44
45   exec();
46 }
47
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);
57 }
58
59 void Server::disconnectFromIrc(QString net) {
60   if(net != network) return; // not me!
61   socket.disconnectFromHost();
62 }
63
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);
70     handleServerMsg(s);
71   }
72 }
73
74 void Server::socketError( QAbstractSocket::SocketError err ) {
75   //qDebug() << "Socket Error!";
76   //emit error(err);
77 }
78
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()));
82 }
83
84 void Server::socketDisconnected( ) {
85   //qDebug() << "Socket disconnected!";
86   emit disconnected();
87 }
88
89 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
90   //qDebug() << "Socket state changed: " << state;
91 }
92
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;
103       nicks[nick] = n;
104       emit nickUpdated(network, nick, n);
105     }
106   }
107   return nick;
108 }
109
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;
116   }
117   handleUserInput(buf, msg);
118 }
119
120 void Server::putRawLine(QString s) {
121   qDebug() << "SentRaw: " << s;
122   s += "\r\n";
123   socket.write(s.toAscii());
124 }
125
126 void Server::putCmd(QString cmd, QStringList params, QString prefix) {
127   QString m;
128   if(!prefix.isEmpty()) m += ":" + prefix + " ";
129   m += cmd.toUpper();
130   for(int i = 0; i < params.size() - 1; i++) {
131     m += " " + params[i];
132   }
133   if(!params.isEmpty()) m += " :" + params.last();
134   qDebug() << "SentCmd: " << m;
135   m += "\r\n";
136   socket.write(m.toAscii());
137 }
138
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) {
141   try {
142     if(msg.isEmpty()) {
143       qWarning() << "Received empty string from server!";
144       return;
145     }
146     // OK, first we split the raw message into its various parts...
147     QString prefix;
148     QString cmd;
149     QStringList params;
150     if(msg[0] == ':') {
151       msg.remove(0,1);
152       prefix = msg.section(' ', 0, 0);
153       msg = msg.section(' ', 1);
154     }
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"));
161     } else {
162       left = msg.section(':', 0, 0);
163       trailing = msg.section(':', 1);
164     }
165     if(!left.isEmpty()) {
166       params << left.split(' ', QString::SkipEmptyParts);
167     }
168     if(!trailing.isEmpty()) {
169       params << trailing;
170     }
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());
177     }
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);
185     }
186   } catch(Exception e) {
187     emit displayMsg("", Message(Message::Error, e.msg()));
188   }
189 }
190
191 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
192   uint num = cmd.toUInt();
193   if(num) {
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.
196     switch(num) {
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));
200         break;
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));
205         break;
206       }
207       // Ignore these commands.
208       case 366: case 376:
209         break;
210
211       // Everything else will be marked in red, so we can add them somewhere.
212       default:
213         emit displayMsg("", Message(Message::Error, cmd + " " + params.join(" "), prefix));
214     }
215     //qDebug() << prefix <<":"<<cmd<<params;
216   } else {
217     emit displayMsg("", Message(Message::Error, QString("Unknown: ") + cmd + " " + params.join(" "), prefix));
218     //qDebug() << prefix <<":"<<cmd<<params;
219   }
220 }
221
222 void Server::handleUserInput(QString bufname, QString usrMsg) {
223   try {
224     /* Looks like we don't need core-side buffers...
225     Buffer *buffer = 0;
226     if(!bufname.isEmpty()) {
227       Q_ASSERT(buffers.contains(bufname));
228       buffer = buffers[bufname];
229     }
230     */
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);
239     }
240   } catch(Exception e) {
241     emit displayMsg("", Message(Message::Error, e.msg()));
242   }
243 }
244
245 void Server::defaultUserHandler(QString bufname, QString cmd, QString msg) {
246   emit displayMsg("", Message(Message::Error, QString("Error: %1 %2").arg(cmd).arg(msg)));
247
248 }
249
250 /**********************************************************************************/
251
252 /*
253 void Server::handleUser(QString bufname, QString msg) {
254
255
256 }
257 */
258
259 void Server::handleUserAway(QString bufname, QString msg) {
260   putCmd("AWAY", QStringList(msg));
261 }
262
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';
266   QStringList params;
267   params << bufname << m << nicks;
268   putCmd("MODE", params);
269 }
270
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';
274   QStringList params;
275   params << bufname << m << nicks;
276   putCmd("MODE", params);
277 }
278
279 void Server::handleUserInvite(QString bufname, QString msg) {
280   QStringList params;
281   params << msg << bufname;
282   putCmd("INVITE", params);
283 }
284
285 void Server::handleUserJoin(QString bufname, QString msg) {
286   putCmd("JOIN", QStringList(msg));
287 }
288
289 void Server::handleUserKick(QString bufname, QString msg) {
290   QStringList params;
291   params << bufname << msg.split(' ', QString::SkipEmptyParts);
292   putCmd("KICK", params);
293 }
294
295 void Server::handleUserList(QString bufname, QString msg) {
296   putCmd("LIST", msg.split(' ', QString::SkipEmptyParts));
297 }
298
299 void Server::handleUserMode(QString bufname, QString msg) {
300   putCmd("MODE", msg.split(' ', QString::SkipEmptyParts));
301 }
302
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;
307   QStringList params;
308   params << nick << msg;
309   putCmd("PRIVMSG", params);
310 }
311
312 void Server::handleUserNick(QString bufname, QString msg) {
313   QString nick = msg.section(' ', 0, 0);
314   putCmd("NICK", QStringList(nick));
315 }
316
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';
320   QStringList params;
321   params << bufname << m << nicks;
322   putCmd("MODE", params);
323 }
324
325 void Server::handleUserPart(QString bufname, QString msg) {
326   QStringList params;
327   params << bufname << msg;
328   putCmd("PART", params);
329 }
330
331 void Server::handleUserQuit(QString bufname, QString msg) {
332   putCmd("QUIT", QStringList(msg));
333 }
334
335 void Server::handleUserQuote(QString bufname, QString msg) {
336   putRawLine(msg);
337 }
338
339 void Server::handleUserSay(QString bufname, QString msg) {
340   if(bufname.isEmpty()) return;  // server buffer
341   QStringList params;
342   params << bufname << msg;
343   putCmd("PRIVMSG", params);
344   emit displayMsg(params[0], Message(Message::Msg, msg, currentNick, Message::Self));
345 }
346
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';
350   QStringList params;
351   params << bufname << m << nicks;
352   putCmd("MODE", params);
353 }
354
355 /**********************************************************************************/
356
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;
364   } else {
365     VarMap n;
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;
372       nicks[nick] = n;
373       emit nickUpdated(network, nick, n);
374     } else {
375       VarMap chans;
376       chans[params[0]] = VarMap();
377       n["Channels"] = chans;
378       n["User"] = userFromMask(prefix);
379       n["Host"] = hostFromMask(prefix);
380       nicks[nick] = n;
381       emit nickAdded(network, nick, n);
382     }
383     emit displayMsg(params[0], Message(Message::Join, params[0], prefix));
384   }
385 }
386
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]);
395   QString msg = nick;
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;
400     nicks[nick] = n;
401     emit nickUpdated(network, nick, n);
402   } else {
403     nicks.remove(nick);
404     emit nickRemoved(network, nick);
405   }
406 }
407
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];
413     int p = 2;
414     int m = 0;
415     bool add = true;
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);
427         m++;
428       } else {
429         // TODO add more modes
430         m++;
431       }
432     }
433     emit displayMsg(params[0], Message(Message::Mode, params.join(" "), prefix));
434   } else {
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(" ")));
439   }
440 }
441
442 void Server::handleServerNick(QString prefix, QStringList params) {
443   QString oldnick = updateNickFromMask(prefix);
444   QString newnick = params[0];
445   QVariant v = nicks.take(oldnick);
446   nicks[newnick] = v;
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)); }
451   }
452   emit nickRenamed(network, oldnick, newnick);
453   if(oldnick == currentNick) {
454     currentNick = newnick;
455     emit ownNickSet(network, newnick);
456   }
457 }
458
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));
463 }
464
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]);
472   QString msg;
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;
477     nicks[nick] = n;
478     emit nickUpdated(network, nick, n);
479   } else {
480     nicks.remove(nick);
481     emit nickRemoved(network, nick);
482   }
483 }
484
485 void Server::handleServerPing(QString prefix, QStringList params) {
486   putCmd("PONG", params);
487 }
488
489 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
490   updateNickFromMask(prefix);
491   emit displayMsg(params[0], Message(Message::Msg, params[1], prefix));
492
493 }
494
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));
501   }
502   nicks.remove(nick);
503   emit nickRemoved(network, nick);
504 }
505
506 /* RPL_WELCOME */
507 void Server::handleServer001(QString prefix, QStringList params) {
508   currentServer = prefix;
509   currentNick = params[0];
510   VarMap n;
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));
516 }
517
518 /* RPL_ISUPPORT */
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('('));
528       int m = 1, p;
529       for(p = 2; p < val.length(); p++) if(val[p] == ')') break;
530       p++;
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];
535       }
536       serverSupports["PrefixModes"] = modes; serverSupports["Prefixes"] = prefixes;
537       serverSupports["ModePrefixMap"] = foo;
538     }
539   }
540 }
541
542
543 /* RPL_NOTOPIC */
544 void Server::handleServer331(QString prefix, QStringList params) {
545   emit topicSet(network, params[0], "");
546 }
547
548 /* RPL_TOPIC */
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])));
552 }
553
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())));
558 }
559
560 /* RPL_NAMREPLY */
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])) {
568       pfx = nick[0];
569       for(int i = 0;; i++)
570         if(prefixes[i] == nick[0]) { mode = serverSupports["PrefixModes"].toString()[i]; break; }
571       nick.remove(0,1);
572     }
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();
577       chans[buf] = c;
578       n["Channels"] = chans;
579       nicks[nick] = n;
580       emit nickUpdated(network, nick, n);
581     } else {
582       VarMap n; VarMap c; VarMap chans;
583       c["Mode"] = mode;
584       chans[buf] = c;
585       n["Channels"] = chans;
586       nicks[nick] = n;
587       emit nickAdded(network, nick, n);
588     }
589   }
590 }
591 /***********************************************************************************/
592
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(" ");
596
597 }
598
599 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
600   _msg = QString("Unknown Command: ") + cmd;
601
602 }