added some more ctcp handlers

[quassel.git] / core / server.cpp

diff --git a/core/server.cpp b/core/server.cpp
index 442fe0f..01cf281 100644 (file)
--- a/core/server.cpp
+++ b/core/server.cpp
@@ -27,8 +27,16 @@
 #include <QDateTime>
 
 Server::Server(QString net) : network(net) {
+  QString MQUOTE = QString('\020');
+  ctcpMDequoteHash[MQUOTE + '0'] = QString('\000');
+  ctcpMDequoteHash[MQUOTE + 'n'] = QString('\n');
+  ctcpMDequoteHash[MQUOTE + 'r'] = QString('\r');
+  ctcpMDequoteHash[MQUOTE + MQUOTE] = MQUOTE;
 
-
+  XDELIM = QString('\001');
+  QString XQUOTE = QString('\134');
+  ctcpXDelimDequoteHash[XQUOTE + XQUOTE] = XQUOTE;
+  ctcpXDelimDequoteHash[XQUOTE + QString('a')] = XDELIM;
 }
 
 Server::~Server() {
@@ -293,6 +301,96 @@ QStringList Server::providesUserHandlers() {
   return userHandlers;
 }
 
+QString Server::ctcpDequote(QString message) {
+  QString dequotedMessage;
+  QString messagepart;
+  QHash<QString, QString>::iterator ctcpquote;
+  
+  // copy dequote Message
+  for(int i = 0; i < message.size(); i++) {
+    messagepart = message[i];
+    if(i+1 < message.size()) {
+      for(ctcpquote = ctcpMDequoteHash.begin(); ctcpquote != ctcpMDequoteHash.end(); ++ctcpquote) {
+        if(message.mid(i,2) == ctcpquote.key()) {
+          dequotedMessage += ctcpquote.value();
+          i++;
+          break;
+        }
+      }
+    }
+    dequotedMessage += messagepart;
+  }
+  return dequotedMessage;
+}
+
+
+QString Server::ctcpXdelimDequote(QString message) {
+  QString dequotedMessage;
+  QString messagepart;
+  QHash<QString, QString>::iterator xdelimquote;
+
+  for(int i = 0; i < message.size(); i++) {
+    messagepart = message[i];
+    if(i+1 < message.size()) {
+      for(xdelimquote = ctcpXDelimDequoteHash.begin(); xdelimquote != ctcpXDelimDequoteHash.end(); ++xdelimquote) {
+        if(message.mid(i,2) == xdelimquote.key()) {
+          messagepart = xdelimquote.value();
+          i++;
+          break;
+        }
+      }
+    }
+    dequotedMessage += messagepart;
+  }
+  return dequotedMessage;
+}
+
+QStringList Server::parseCtcp(CtcpType ctcptype, QString prefix, QString target, QString message) {
+  QStringList messages;
+  QString ctcp;
+  
+  //lowlevel message dequote
+  QString dequotedMessage = ctcpDequote(message);
+  
+  // extract tagged / extended data
+  while(dequotedMessage.contains(XDELIM)) {
+    messages << dequotedMessage.section(XDELIM,0,0);
+    ctcp = ctcpXdelimDequote(dequotedMessage.section(XDELIM,1,1));
+    dequotedMessage = dequotedMessage.section(XDELIM,2,2);
+    
+    //dispatch the ctcp command
+    QString ctcpcmd = ctcp.section(' ', 0, 0);
+    QString ctcpparam = ctcp.section(' ', 1);
+    
+    QString hname = ctcpcmd.toLower();
+    hname[0] = hname[0].toUpper();
+    hname = "handleCtcp" + hname;
+    if(!QMetaObject::invokeMethod(this, hname.toAscii(), Q_ARG(CtcpType, ctcptype), Q_ARG(QString, prefix), Q_ARG(QString, target), Q_ARG(QString, ctcpparam))) {
+      // Ok. Default handler it is.
+      defaultCtcpHandler(ctcptype, prefix, ctcpcmd, target, ctcpparam);
+    }
+  }
+  if(!dequotedMessage.isEmpty()) {
+    messages << dequotedMessage;
+  }
+  return messages;
+}
+
+QString Server::ctcpPack(QString ctcpTag, QString message) {
+  return XDELIM + ctcpTag + ' ' + message + XDELIM;
+}
+
+void Server::ctcpQuery(QString bufname, QString ctcpTag, QString message) {
+  QStringList params;
+  params << bufname << ctcpPack(ctcpTag, message);
+  putCmd("PRIVMSG", params); 
+}
+
+void Server::ctcpReply(QString bufname, QString ctcpTag, QString message) {
+  QStringList params;
+  params << bufname << ctcpPack(ctcpTag, message);
+  putCmd("NOTICE", params);
+}
 
 /**********************************************************************************/
 
@@ -402,6 +500,12 @@ void Server::handleUserSay(QString bufname, QString msg) {
   }
 }
 
+void Server::handleUserMe(QString bufname, QString msg) {
+  if(bufname.isEmpty()) return; // server buffer
+  ctcpQuery(bufname, "ACTION", msg);
+  emit displayMsg(Message::Action, bufname, msg, ownNick);
+}
+
 void Server::handleUserTopic(QString bufname, QString msg) {
   if(bufname.isEmpty()) return;
   QStringList params;
@@ -568,11 +672,23 @@ void Server::handleServerPrivmsg(QString prefix, QStringList params) {
   Q_ASSERT(params.count() >= 2);
   if(params.count()<2) emit displayMsg(Message::Plain, params[0], "", prefix);
   else {
-    if(params[0] == ownNick) {
-      emit displayMsg(Message::Plain, "", params[1], prefix, Message::PrivMsg);
+    // it's possible to pack multiple privmsgs into one param using ctcp
+    QStringList messages = parseCtcp(Server::CtcpQuery, prefix, params[0], params[1]);
+    if(params[0].toLower() == ownNick.toLower()) {  // Freenode sends nickname in lower case!
+      foreach(QString message, messages) {
+        if(!message.isEmpty()) {
+          emit displayMsg(Message::Plain, "", message, prefix, Message::PrivMsg);
+        }
+      }
+      
     } else {
+      //qDebug() << prefix << params;
       Q_ASSERT(isChannelName(params[0]));  // should be channel!
-      emit displayMsg(Message::Plain, params[0], params[1], prefix);
+      foreach(QString message, messages) {
+        if(!message.isEmpty()) {
+          emit displayMsg(Message::Plain, params[0], message, prefix);
+        }
+      }
     }
   }
 }
@@ -726,7 +842,6 @@ void Server::handleServer432(QString prefix, QStringList params) {
 
 /* ERR_NICKNAMEINUSE */
 void Server::handleServer433(QString prefix, QStringList params) {
-  qDebug() << "433:" << params;
   QString errnick = params[0];
   emit displayMsg(Message::Error, "", tr("Nick %1 is already taken").arg(errnick));
   // if there is a problem while connecting to the server -> we handle it
@@ -742,6 +857,37 @@ void Server::handleServer433(QString prefix, QStringList params) {
   }
 }
 
+/***********************************************************************************/
+// CTCP HANDLER
+
+void Server::handleCtcpAction(CtcpType ctcptype, QString prefix, QString target, QString param) {
+  emit displayMsg(Message::Action, target, param, prefix);
+}
+
+void Server::handleCtcpPing(CtcpType ctcptype, QString prefix, QString target, QString param) {
+  if(ctcptype == CtcpQuery) {
+    ctcpReply(userFromMask(prefix), "PING", param);
+    emit displayMsg(Message::Plain, "", tr("Received CTCP PING request by %1").arg(prefix));
+  } else {
+    // display ping answer
+  }
+}
+
+void Server::handleCtcpVersion(CtcpType ctcptype, QString prefix, QString target, QString param) {
+  if(ctcptype == CtcpQuery) {
+    // FIXME use real Info about quasel :)
+    ctcpReply(userFromMask(prefix), "VERSION", QString("Quassel:pre Release:*nix"));
+    emit displayMsg(Message::Plain, "", tr("Received CTCP VERSION request by %1").arg(prefix));
+  } else {
+    // TODO display Version answer
+  }
+}
+
+void Server::defaultCtcpHandler(CtcpType ctcptype, QString prefix, QString cmd, QString target, QString param) {
+  emit displayMsg(Message::Error, "", tr("Received unknown CTCP %1 by %2").arg(cmd).arg(prefix));
+}
+
+
 /***********************************************************************************/
 
 /* Exception classes for message handling */