More neato stuff. A preliminary TabWidget now allows us to display multiple buffers,
[quassel.git] / network / server.cpp
index ae8cd57..d9e0a5b 100644 (file)
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include "global.h"
 #include "server.h"
+#include "cmdcodes.h"
+#include "message.h"
 
-Server::Server() {
-  socket = new QTcpSocket();
+#include <QMetaObject>
+
+Server::Server(QString net) : network(net) {
 
 }
 
 Server::~Server() {
-  delete socket;
-}
 
-void Server::init() {
-  Message::init(&handleServerMsg, &handleUserMsg);
 }
 
 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!
+  QList<QVariant> servers = global->getData("Networks").toMap()[net].toMap()["Servers"].toList();
+  qDebug() << "Connecting to"<< servers[0].toMap();
+  QString host = servers[0].toMap()["Address"].toString();
+  quint16 port = servers[0].toMap()["Port"].toUInt();
+  sendStatusMsg(QString("Connecting to %1:%2...").arg(host).arg(port));
+  socket.connectToHost(host, port);
 }
 
-void Server::putRawLine( const QString &s ) {
-  qDebug() << "Raw line: " << s;
-  stream << s << "\r\n" << flush;
+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 recvLine(s + "\n");
+    emit recvRawServerMsg(s);
+    //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("NICK :QuasselDev");
+  putRawLine("USER Sputnick 8 * :Using Quassel IRC (WiP Version)");
 }
 
 void Server::socketDisconnected( ) {
-  qDebug() << "Socket disconnected!";
-  //emit disconnected();
+  //qDebug() << "Socket disconnected!";
+  emit disconnected();
 }
 
 void Server::socketStateChanged(QAbstractSocket::SocketState state) {
-  qDebug() << "Socket state changed: " << state;
+  //qDebug() << "Socket state changed: " << state;
+}
+
+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());
 }
 
-void Server::handleServerMsg(Message *msg) {
+/** 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;
+    }
+    // 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 sendMessage("", Message(e.msg(), "", Message::Error));
+  }
+}
 
+void Server::defaultServerHandler(QString cmd, QString prefix, QStringList params) {
+  uint num = cmd.toUInt();
+  if(num) {
+    if(params.count() > 0) {
+      if(params[0] == currentNick) params.removeFirst();  // remove nick if it is first arg
+      else qWarning((QString("First param NOT nick: %1:%2 %3").arg(prefix).arg(cmd).arg(params.join(" "))).toAscii());
+    }
+    // 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 sendMessage("", Message(params.join(" "), prefix, Message::Server));
+        break;
+      // Ignore these commands.
+      case 376:
+        break;
 
+      // Everything else will be marked in red, so we can add them somewhere.
+      default:
+        emit sendMessage("", Message(cmd + " " + params.join(" "), prefix, Message::Error));
+    }
+    //qDebug() << prefix <<":"<<cmd<<params;
+  } else {
+    emit sendMessage("", Message(QString("Unknown: ") + cmd + " " + params.join(" "), prefix, Message::Error));
+    //qDebug() << prefix <<":"<<cmd<<params;
+  }
 }
 
-void Server::handleUserMsg(Message *msg) {
+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 sendMessage("", Message(e.msg(), "", Message::Error));
+  }
+}
 
+void Server::defaultUserHandler(QString cmd, QString msg, Buffer *buf) {
+  emit sendMessage("", 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);
+}
+
+/**********************************************************************************/
+
+void Server::handleServerJoin(QString prefix, QStringList params) {
+  Q_ASSERT(params.count() == 1);
+  QString bufname = params[0];
+  if(!buffers.contains(bufname)) {
+    Buffer *buf = new Buffer(bufname);
+    buffers[bufname] = buf;
+  }
+  // handle user joins!
+}
+
+void Server::handleServerNotice(QString prefix, QStringList params) {
+  Message msg(params[1], prefix, Message::Notice);
+  if(prefix == currentServer) emit sendMessage("", Message(params[1], prefix, Message::Server));
+  else emit sendMessage("", Message(params[1], prefix, Message::Notice));
+}
+
+void Server::handleServerPing(QString prefix, QStringList params) {
+  putCmd("PONG", params);
+}
+
+void Server::handleServerPrivmsg(QString prefix, QStringList params) {
+  emit sendMessage(params[0], Message(params[1], prefix, Message::Msg));
+
+}
+
+/* RPL_WELCOME */
+void Server::handleServer001(QString prefix, QStringList params) {
+  currentServer = prefix;
+  currentNick = params[0];
+  emit sendMessage("", Message(params[1], prefix, Message::Server));
+}
+
+/* RPL_NOTOPIC */
+void Server::handleServer331(QString prefix, QStringList params) {
+  if(params[0] == currentNick) params.removeFirst();
+  emit setTopic(network, params[0], "");
+}
+
+/* RPL_TOPIC */
+void Server::handleServer332(QString prefix, QStringList params) {
+  if(params[0] == currentNick) params.removeFirst();
+  emit setTopic(network, params[0], params[1]);
+}
+
+/***********************************************************************************/
+
+/* Exception classes for message handling */
+Server::ParseError::ParseError(QString cmd, QString prefix, QStringList params) {
+  _msg = QString("Command Parse Error: ") + cmd + params.join(" ");
+
+}
+
+Server::UnknownCmdError::UnknownCmdError(QString cmd, QString prefix, QStringList params) {
+  _msg = QString("Unknown Command: ") + cmd;
+
+}