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 "global.h"
  22 #include "server.h"
  23 #include "cmdcodes.h"
  24 #include "message.h"
  25
  26 #include <QMetaObject>
  27
  28 Server::Server(QString net) : network(net) {
  29
  30 }
  31
  32 Server::~Server() {
  33
  34 }
  35
  36 void Server::run() {
  37   connect(&socket, SIGNAL(connected()), this, SLOT(socketConnected()));
  38   connect(&socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
  39   connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
  40   connect(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
  41   connect(&socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
  42
  43   exec();
  44 }
  45
  46 void Server::connectToIrc(QString net) {
  47   if(net != network) return; // not me!
  48   QList<QVariant> servers = global->getData("Networks").toMap()[net].toMap()["Servers"].toList();
  49   qDebug() << "Connecting to"<< servers[0].toMap();
  50   QString host = servers[0].toMap()["Address"].toString();
  51   quint16 port = servers[0].toMap()["Port"].toUInt();
  52   sendStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
  53   socket.connectToHost(host, port);
  54 }
  55
  56 void Server::disconnectFromIrc(QString net) {
  57   if(net != network) return; // not me!
  58   socket.disconnectFromHost();
  59 }
  60
  61 void Server::socketHasData() {
  62   while(socket.canReadLine()) {
  63     QString s = socket.readLine().trimmed();
  64     qDebug() << "Read: " << s;
  65     emit recvRawServerMsg(s);
  66     //Message *msg = Message::createFromServerString(this, s);
  67     handleServerMsg(s);
  68   }
  69 }
  70
  71 void Server::socketError( QAbstractSocket::SocketError err ) {
  72   //qDebug() << "Socket Error!";
  73   //emit error(err);
  74 }
  75
  76 void Server::socketConnected( ) {
  77   qDebug() << "Socket connected!";
  78   putRawLine("NICK :QuasselDev");
  79   putRawLine("USER Sputnick 8 * :Using Quassel IRC (WiP Version)");
  80 }
  81
  82 void Server::socketDisconnected( ) {
  83   //qDebug() << "Socket disconnected!";
  84   emit disconnected();
  85 }
  86
  87 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
  88   //qDebug() << "Socket state changed: " << state;
  89 }
  90
  91 void Server::userInput(QString net, QString buf, QString msg) {
  92   if(net != network) return; // not me!
  93   msg = msg.trimmed(); // remove whitespace from start and end
  94   if(msg.isEmpty()) return;
  95   if(!msg.startsWith('/')) {
  96     msg = QString("/SAY ") + msg;
  97   }
  98   handleUserMsg(buf, msg);
  99 }
 100
 101 void Server::putRawLine(QString s) {
 102   qDebug() << "SentRaw: " << s;
 103   s += "\r\n";
 104   socket.write(s.toAscii());
 105 }
 106
 107 void Server::putCmd(QString cmd, QStringList params, QString prefix) {
 108   QString m;
 109   if(!prefix.isEmpty()) m += ":" + prefix + " ";
 110   m += cmd.toUpper();
 111   for(int i = 0; i < params.size() - 1; i++) {
 112     m += " " + params[i];
 113   }
 114   if(!params.isEmpty()) m += " :" + params.last();
 115   qDebug() << "SentCmd: " << m;
 116   m += "\r\n";
 117   socket.write(m.toAscii());
 118 }
 119
 120 /** Handle a raw message string sent by the server. We try to find a suitable handler, otherwise we call a default handler. */
 121 void Server::handleServerMsg(QString msg) {
 122   try {
 123     if(msg.isEmpty()) {
 124       qWarning() << "Received empty string from server!";
 125       return;
 126     }
 127     // OK, first we split the raw message into its various parts...
 128     QString prefix;
 129     QString cmd;
 130     QStringList params;
 131     if(msg[0] == ':') {
 132       msg.remove(0,1);
 133       prefix = msg.section(' ', 0, 0);
 134       msg = msg.section(' ', 1);
 135     }
 136     cmd = msg.section(' ', 0, 0).toUpper();
 137     msg = msg.section(' ', 1);
 138     QString left = msg.section(':', 0, 0);
 139     QString trailing = msg.section(':', 1);
 140     if(!left.isEmpty()) {
 141       params << left.split(' ', QString::SkipEmptyParts);
 142     }
 143     if(!trailing.isEmpty()) {
 144       params << trailing;
 145     }
 146     // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
 147     QString hname = cmd.toLower();
 148     hname[0] = hname[0].toUpper();
 149     hname = "handleServer" + hname;
 150     if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
 151       // Ok. Default handler it is.
 152       defaultServerHandler(cmd, prefix, params);
 153     }
 154   } catch(Exception e) {
 155     emit sendMessage("", Message(e.msg(), "", Message::Error));
 156   }
 157 }
 158
 159 void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
 160   uint num = cmd.toUInt();
 161   if(num) {
 162     if(params.count() > 0) {
 163       if(params[0] == currentNick) params.removeFirst();  // remove nick if it is first arg
 164       else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
 165     }
 166     // A lot of server messages don't really need their own handler because they don't do much.
 167     // Catch and handle these here.
 168     switch(num) {
 169       // Welcome, status, info messages. Just display these.
 170       case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
 171         emit sendMessage("", Message(params.join(" "), prefix, Message::Server));
 172         break;
 173       // Ignore these commands.
 174       case 376:
 175         break;
 176
 177       // Everything else will be marked in red, so we can add them somewhere.
 178       default:
 179         emit sendMessage("", Message(cmd + " " + params.join(" "), prefix, Message::Error));
 180     }
 181     //qDebug() << prefix <<":"<<cmd<<params;
 182   } else {
 183     emit sendMessage("", Message(QString("Unknown: ") + cmd + " " + params.join(" "), prefix, Message::Error));
 184     //qDebug() << prefix <<":"<<cmd<<params;
 185   }
 186 }
 187
 188 void Server::handleUserMsg(QString bufname, QString usrMsg) {
 189   try {
 190     Buffer *buffer = 0;
 191     if(!bufname.isEmpty()) {
 192       Q_ASSERT(buffers.contains(bufname));
 193       buffer = buffers[bufname];
 194     }
 195     QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
 196     QString msg = usrMsg.section(' ', 1).trimmed();
 197     QString hname = cmd.toLower();
 198     hname[0] = hname[0].toUpper();
 199     hname = "handleUser" + hname;
 200     if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, msg), Q_ARG(Buffer*, buffer))) {
 201         // Ok. Default handler it is.
 202       defaultUserHandler(cmd, msg, buffer);
 203     }
 204   } catch(Exception e) {
 205     emit sendMessage("", Message(e.msg(), "", Message::Error));
 206   }
 207 }
 208
 209 void Server::defaultUserHandler(QString cmd, QString msg, Buffer *buf) {
 210   emit sendMessage("", Message(QString("Error: %1 %2").arg(cmd).arg(msg), "", Message::Error));
 211
 212 }
 213
 214 /**********************************************************************************/
 215
 216 /*
 217 void Server::handleUser(QString msg, Buffer *buf) {
 218
 219
 220 }
 221 */
 222
 223 void Server::handleUserJoin(QString msg, Buffer *buf) {
 224   putCmd("JOIN", QStringList(msg));
 225
 226 }
 227
 228 void Server::handleUserQuote(QString msg, Buffer *buf) {
 229   putRawLine(msg);
 230 }
 231
 232 void Server::handleUserSay(QString msg, Buffer *buf) {
 233   if(!buf) return;  // server buffer
 234   QStringList params;
 235   params << buf->name() << msg;
 236   putCmd("PRIVMSG", params);
 237 }
 238
 239 /**********************************************************************************/
 240
 241 void Server::handleServerJoin(QString prefix, QStringList params) {
 242   Q_ASSERT(params.count() == 1);
 243   QString bufname = params[0];
 244   if(!buffers.contains(bufname)) {
 245     Buffer *buf = new Buffer(bufname);
 246     buffers[bufname] = buf;
 247   }
 248   // handle user joins!
 249 }
 250
 251 void Server::handleServerNotice(QString prefix, QStringList params) {
 252   Message msg(params[1], prefix, Message::Notice);
 253   if(prefix == currentServer) emit sendMessage("", Message(params[1], prefix, Message::Server));
 254   else emit sendMessage("", Message(params[1], prefix, Message::Notice));
 255 }
 256
 257 void Server::handleServerPing(QString prefix, QStringList params) {
 258   putCmd("PONG", params);
 259 }
 260
 261 void Server::handleServerPrivmsg(QString prefix, QStringList params) {
 262   emit sendMessage(params[0], Message(params[1], prefix, Message::Msg));
 263
 264 }
 265
 266 /* RPL_WELCOME */
 267 void Server::handleServer001(QString prefix, QStringList params) {
 268   currentServer = prefix;
 269   currentNick = params[0];
 270   emit sendMessage("", Message(params[1], prefix, Message::Server));
 271 }
 272
 273 /* RPL_NOTOPIC */
 274 void Server::handleServer331(QString prefix, QStringList params) {
 275   if(params[0] == currentNick) params.removeFirst();
 276   emit setTopic(network, params[0], "");
 277 }
 278
 279 /* RPL_TOPIC */
 280 void Server::handleServer332(QString prefix, QStringList params) {
 281   if(params[0] == currentNick) params.removeFirst();
 282   emit setTopic(network, params[0], params[1]);
 283 }
 284
 285 /***********************************************************************************/
 286
 287 /* Exception classes for message handling */
 288 Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
 289   _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
 290
 291 }
 292
 293 Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
 294   _msg = QString("Unknown Command: ") + cmd;
 295
 296 }