Lots of additions again. Working on implementing commands and prettifying the output.
[quassel.git] / network / server.cpp
index d55a9a7..e4d37a6 100644 (file)
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
-#include "quassel.h"
+#include "global.h"
 #include "server.h"
 #include "cmdcodes.h"
+#include "message.h"
 
-Server::Server() {
-  socket = new QTcpSocket();
+#include <QMetaObject>
+#include <QDateTime>
+
+Server::Server(QString net) : network(net) {
 
 }
 
 Server::~Server() {
-  delete socket;
-}
 
-void Server::init() {
-  Message::init(&dispatchServerMsg, &dispatchUserMsg);
 }
 
 void Server::run() {
-  connect(socket, SIGNAL(connected()), this, SLOT(socketConnected()));
-  connect(socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
-  connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
-  connect(socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
-  connect(socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
-
-  stream.setDevice(socket);
-  //connectToIrc("irc.quakenet.org", 6667);
-  exec();
-}
-
-/*
-QAbstractSocket::SocketState TcpConnection::state( ) const {
-  return socket.state();
-}
-*/
+  connect(&socket, SIGNAL(connected()), this, SLOT(socketConnected()));
+  connect(&socket, SIGNAL(disconnected()), this, SLOT(socketDisconnected()));
+  connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
+  connect(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(socketStateChanged(QAbstractSocket::SocketState)));
+  connect(&socket, SIGNAL(readyRead()), this, SLOT(socketHasData()));
 
-void Server::connectToIrc( const QString & host, quint16 port ) {
-  qDebug() << "Connecting...";
-  socket->connectToHost(host, port);
+  exec();
 }
 
-void Server::disconnectFromIrc( ) {
-  socket->disconnectFromHost();
+void Server::connectToIrc(QString net) {
+  if(net != network) return; // not me!
+  networkSettings = global->getData("Networks").toMap()[net].toMap();
+  identity = global->getData("Identities").toMap()[networkSettings["Identity"].toString()].toMap();
+  QList<QVariant> servers = networkSettings["Servers"].toList();
+  QString host = servers[0].toMap()["Address"].toString();
+  quint16 port = servers[0].toMap()["Port"].toUInt();
+  displayStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
+  socket.connectToHost(host, port);
 }
 
-void Server::putRawLine( const QString &s ) {
-  qDebug() << "Sent: " << s;
-  stream << s << "\r\n" << flush;
-  //Message::createFromServerString(this, s);
+void Server::disconnectFromIrc(QString net) {
+  if(net != network) return; // not me!
+  socket.disconnectFromHost();
 }
 
-void Server::socketHasData( ) {
-  while(socket->canReadLine()) {
-    QString s = stream.readLine();
+void Server::socketHasData() {
+  while(socket.canReadLine()) {
+    QString s = socket.readLine().trimmed();
     qDebug() << "Read: " << s;
     emit recvRawServerMsg(s);
-    Message *msg = Message::createFromServerString(this, s);
-    if(msg) {
-      try { handleServerMsg(msg); } catch(Exception e) {
-        emit recvLine(e.msg() + "\n");
-      }
-    }
-    delete msg;
+    //Message *msg = Message::createFromServerString(this, s);
+    handleServerMsg(s);
   }
 }
 
 void Server::socketError( QAbstractSocket::SocketError err ) {
-  qDebug() << "Socket Error!";
+  //qDebug() << "Socket Error!";
   //emit error(err);
 }
 
 void Server::socketConnected( ) {
-  qDebug() << "Socket connected!";
-  //emit connected();
+  putRawLine(QString("NICK :%1").arg(identity["NickList"].toStringList()[0]));
+  putRawLine(QString("USER %1 8 * :%2").arg(identity["Ident"].toString()).arg(identity["RealName"].toString()));
 }
 
 void Server::socketDisconnected( ) {
-  qDebug() << "Socket disconnected!";
-  //emit disconnected();
+  //qDebug() << "Socket disconnected!";
+  emit disconnected();
 }
 
 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
-  qDebug() << "Socket state changed: " << state;
-}
-
-/** Handle a message sent by the IRC server that does not have a custom handler. */
-void Server::handleServerMsg(Message *msg) {
-  int cmdCode = msg->getCmdCode();
-  QString prefix = msg->getPrefix();
-  QStringList params = msg->getParams();
-  if(cmdCode < 0) {
-    switch(-cmdCode) {
-      case CMD_PING:
-        // PING <server1> [<server2>]
-        if(params.size() == 1) {
-          putRawLine(QString("PONG :") + params[0]);
-        } else if(params.size() == 2) {
-          putRawLine(QString("PONG ") + params[0] + " :" + params[1]);
-        } else throw ParseError(msg);
+  //qDebug() << "Socket state changed: " << state;
+}
+
+QString Server::nickFromMask(QString mask) {
+  return mask.section('!', 0, 0);
+}
+
+QString Server::userFromMask(QString mask) {
+  QString userhost = mask.section('!', 1);
+  if(userhost.isEmpty()) return QString();
+  return userhost.section('@', 0, 0);
+}
+
+QString Server::hostFromMask(QString mask) {
+  QString userhost = mask.section('!', 1);
+  if(userhost.isEmpty()) return QString();
+  return userhost.section('@', 1);
+}
+
+void Server::userInput(QString net, QString buf, QString msg) {
+  if(net != network) return; // not me!
+  msg = msg.trimmed(); // remove whitespace from start and end
+  if(msg.isEmpty()) return;
+  if(!msg.startsWith('/')) {
+    msg = QString("/SAY ") + msg;
+  }
+  handleUserMsg(buf, msg);
+}
+
+void Server::putRawLine(QString s) {
+  qDebug() << "SentRaw: " << s;
+  s += "\r\n";
+  socket.write(s.toAscii());
+}
+
+void Server::putCmd(QString cmd, QStringList params, QString prefix) {
+  QString m;
+  if(!prefix.isEmpty()) m += ":" + prefix + " ";
+  m += cmd.toUpper();
+  for(int i = 0; i < params.size() - 1; i++) {
+    m += " " + params[i];
+  }
+  if(!params.isEmpty()) m += " :" + params.last();
+  qDebug() << "SentCmd: " << m;
+  m += "\r\n";
+  socket.write(m.toAscii());
+}
+
+/** Handle a raw message string sent by the server. We try to find a suitable handler, otherwise we call a default handler. */
+void Server::handleServerMsg(QString msg) {
+  try {
+    if(msg.isEmpty()) {
+      qWarning() << "Received empty string from server!";
+      return;
+    }
+    // OK, first we split the raw message into its various parts...
+    QString prefix;
+    QString cmd;
+    QStringList params;
+    if(msg[0] == ':') {
+      msg.remove(0,1);
+      prefix = msg.section(' ', 0, 0);
+      msg = msg.section(' ', 1);
+    }
+    cmd = msg.section(' ', 0, 0).toUpper();
+    msg = msg.section(' ', 1);
+    QString left = msg.section(':', 0, 0);
+    QString trailing = msg.section(':', 1);
+    if(!left.isEmpty()) {
+      params << left.split(' ', QString::SkipEmptyParts);
+    }
+    if(!trailing.isEmpty()) {
+      params << trailing;
+    }
+    // numeric replies usually have our own nick as first param. Remove this!
+    // BTW, this behavior is not in the RFC.
+    uint num = cmd.toUInt();
+    if(num > 1 && params.count() > 0) {  // 001 sets our nick, so we shouldn't remove anything
+      if(params[0] == currentNick) params.removeFirst();
+      else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
+    }
+    // Now we try to find a handler for this message. BTW, I do love the Trolltech guys ;-)
+    QString hname = cmd.toLower();
+    hname[0] = hname[0].toUpper();
+    hname = "handleServer" + hname;
+    if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, prefix), Q_ARG(QStringList, params))) {
+      // Ok. Default handler it is.
+      defaultServerHandler(cmd, prefix, params);
+    }
+  } catch(Exception e) {
+    emit displayMsg("", Message(e.msg(), "", Message::Error));
+  }
+}
+
+void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
+  uint num = cmd.toUInt();
+  if(num) {
+    // A lot of server messages don't really need their own handler because they don't do much.
+    // Catch and handle these here.
+    switch(num) {
+      // Welcome, status, info messages. Just display these.
+      case 2: case 3: case 4: case 5: case 251: case 252: case 253: case 254: case 255: case 372: case 375:
+        emit displayMsg("", Message(params.join(" "), prefix, Message::Server));
+        break;
+      // Ignore these commands.
+      case 366: case 376:
         break;
 
+      // Everything else will be marked in red, so we can add them somewhere.
       default:
-        throw Exception(QString("No handler installed for command: ") + msg->getCmd() + " " + msg->getParams().join(" "));
+        emit displayMsg("", Message(cmd + " " + params.join(" "), prefix, Message::Error));
     }
-  } else if(msg->getCmdCode() > 0) {
-    switch(msg->getCmdCode()) {
+    //qDebug() << prefix <<":"<<cmd<<params;
+  } else {
+    emit displayMsg("", Message(QString("Unknown: ") + cmd + " " + params.join(" "), prefix, Message::Error));
+    //qDebug() << prefix <<":"<<cmd<<params;
+  }
+}
 
-      default:
-        //
-        throw Exception(msg->getCmd() + " " + msg->getParams().join(" "));
+void Server::handleUserMsg(QString bufname, QString usrMsg) {
+  try {
+    Buffer *buffer = 0;
+    if(!bufname.isEmpty()) {
+      Q_ASSERT(buffers.contains(bufname));
+      buffer = buffers[bufname];
     }
+    QString cmd = usrMsg.section(' ', 0, 0).remove(0, 1).toUpper();
+    QString msg = usrMsg.section(' ', 1).trimmed();
+    QString hname = cmd.toLower();
+    hname[0] = hname[0].toUpper();
+    hname = "handleUser" + hname;
+    if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(QString, msg), Q_ARG(Buffer*, buffer))) {
+        // Ok. Default handler it is.
+      defaultUserHandler(cmd, msg, buffer);
+    }
+  } catch(Exception e) {
+    emit displayMsg("", Message(e.msg(), "", Message::Error));
+  }
+}
+
+void Server::defaultUserHandler(QString cmd, QString msg, Buffer *buf) {
+  emit displayMsg("", Message(QString("Error: %1 %2").arg(cmd).arg(msg), "", Message::Error));
+
+}
+
+/**********************************************************************************/
+
+/*
+void Server::handleUser(QString msg, Buffer *buf) {
+
+
+}
+*/
+
+void Server::handleUserJoin(QString msg, Buffer *buf) {
+  putCmd("JOIN", QStringList(msg));
+
+}
+
+void Server::handleUserQuote(QString msg, Buffer *buf) {
+  putRawLine(msg);
+}
+
+void Server::handleUserSay(QString msg, Buffer *buf) {
+  if(!buf) return;  // server buffer
+  QStringList params;
+  params << buf->name() << msg;
+  putCmd("PRIVMSG", params);
+  emit displayMsg(params[0], Message(msg, currentNick, Message::Msg, Message::Self));
+}
 
+/**********************************************************************************/
+
+void Server::handleServerJoin(QString prefix, QStringList params) {
+  Q_ASSERT(params.count() == 1);
+  QString nick = nickFromMask(prefix);
+  if(nick == currentNick) {
+    Q_ASSERT(!buffers.contains(params[0]));  // cannot join a buffer twice!
+    Buffer *buf = new Buffer(params[0]);
+    buffers[params[0]] = buf;
   } else {
-    throw UnknownCmdError(msg);
+    VarMap n;
+    if(nicks.contains(nick)) {
+      n = nicks[nick].toMap();
+      VarMap chans = n["Channels"].toMap();
+      // Q_ASSERT(!chans.keys().contains(params[0])); TODO uncomment
+      chans[params[0]] = VarMap();
+      n["Channels"] = chans;
+      nicks[nick] = n;
+      emit nickUpdated(network, nick, n);
+    } else {
+      VarMap chans;
+      chans[params[0]] = VarMap();
+      n["Channels"] = chans;
+      n["Nick"] = nick;
+      n["User"] = userFromMask(prefix);
+      n["Host"] = hostFromMask(prefix);
+      nicks[nick] = n;
+      emit nickAdded(network, nick, n);
+    }
+    QString user = n["User"].toString(); QString host = n["Host"].toString();
+    if(user.isEmpty() || host.isEmpty()) emit displayMsg(params[0], Message(tr("%1 has joined %2").arg(nick).arg(params[0]), "", Message::Join));
+    else emit displayMsg(params[0], Message(tr("%1 (%2@%3) has joined %4").arg(nick).arg(user).arg(host).arg(params[0]), "", Message::Join));
   }
 }
 
-QString Server::handleUserMsg(Message *msg) {
 
-  return "";
+
+void Server::handleServerNotice(QString prefix, QStringList params) {
+  Message msg(params[1], prefix, Message::Notice);
+  if(prefix == currentServer) emit displayMsg("", Message(params[1], prefix, Message::Server));
+  else emit displayMsg("", Message(params[1], prefix, Message::Notice));
+}
+
+void Server::handleServerPing(QString prefix, QStringList params) {
+  putCmd("PONG", params);
+}
+
+void Server::handleServerPrivmsg(QString prefix, QStringList params) {
+  emit displayMsg(params[0], Message(params[1], nickFromMask(prefix), Message::Msg));
+
+}
+
+/* RPL_WELCOME */
+void Server::handleServer001(QString prefix, QStringList params) {
+  currentServer = prefix;
+  currentNick = params[0];
+  emit ownNickSet(network, currentNick);
+  emit displayMsg("", Message(params[1], prefix, Message::Server));
+}
+
+/* RPL_NOTOPIC */
+void Server::handleServer331(QString prefix, QStringList params) {
+  emit topicSet(network, params[0], "");
+}
+
+/* RPL_TOPIC */
+void Server::handleServer332(QString prefix, QStringList params) {
+  emit topicSet(network, params[0], params[1]);
+  emit displayMsg(params[0], Message(tr("Topic for %1 is \"%2\"").arg(params[0]).arg(params[1]), "", Message::Server));
+}
+
+/* Topic set by... */
+void Server::handleServer333(QString prefix, QStringList params) {
+  emit displayMsg(params[0], Message(tr("Topic set by %1 on %2").arg(params[1]).arg(QDateTime::fromTime_t(params[2].toUInt()).toString()), "", Message::Server));
+}
+
+/* RPL_NAMREPLY */
+void Server::handleServer353(QString prefix, QStringList params) {
+  params.removeFirst(); // = or *
+  QString buf = params.takeFirst();
+  foreach(QString nick, params[0].split(' ')) {
+    // TODO: parse more prefix characters! use 005?
+    QString mode = "";
+    if(nick.startsWith('@')) { mode = "o"; nick.remove(0,1); }
+    else if(nick.startsWith('+')) { mode = "v"; nick.remove(0,1); }
+    VarMap c; c["Mode"] = mode;
+    if(nicks.contains(nick)) {
+      VarMap n = nicks[nick].toMap();
+      VarMap chans = n["Channels"].toMap();
+      chans[buf] = c;
+      n["Channels"] = chans;
+      nicks[nick] = n;
+      emit nickUpdated(network, nick, n);
+    } else {
+      VarMap n; VarMap c; VarMap chans;
+      c["Mode"] = mode;
+      chans[buf] = c;
+      n["Channels"] = chans;
+      n["Nick"] = nick;
+      nicks[nick] = n;
+      emit nickAdded(network, nick, n);
+    }
+  }
 }
+/***********************************************************************************/
 
 /* Exception classes for message handling */
-Server::ParseError::ParseError(Message *msg) {
-  _msg = QString("Command Parse Error: ") + msg->getCmd() + msg->getParams().join(" ");
+Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
+  _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
 
 }
 
-Server::UnknownCmdError::UnknownCmdError(Message *msg) {
-  _msg = QString("Unknown Command: ") + msg->getCmd();
+Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
+  _msg = QString("Unknown Command: ") + cmd;
 
 }