network/message.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 "message.h"
  22 #include <QtDebug>
  23
  24 extern BuiltinCmd builtins[];
  25
  26 recvHandlerType Message::defaultRecvHandler;
  27 sendHandlerType Message::defaultSendHandler;
  28 QHash<QString, CmdType> Message::cmdTypes;
  29
  30 Message::Message(Server *srv, Buffer *buf, QString _cmd, QString _prefix, QStringList _params)
  31   : server(srv), buffer(buf), cmd(_cmd), prefix(_prefix), params(_params) {
  32
  33   // Check if it's a registered cmd (or a numeric reply with a custom entry)
  34   if(cmdTypes.contains(cmd)) {
  35     CmdType c = cmdTypes[cmd];
  36     recvHandler = ( c.recvHandler ? c.recvHandler : defaultRecvHandler);
  37     sendHandler = ( c.sendHandler ? c.sendHandler : defaultSendHandler);
  38     type = - c.type;
  39   } else {
  40     int t = cmd.toInt();
  41     if(t) {
  42       type = t;
  43       recvHandler = defaultRecvHandler;
  44       sendHandler = defaultSendHandler;
  45     } else {
  46       // Unknown cmd!
  47       qWarning() << "Unknown command: " << cmd;
  48       type = 0;
  49     }
  50   }
  51 }
  52
  53 void Message::init(recvHandlerType _r, sendHandlerType _s) {
  54   defaultRecvHandler = _r;
  55   defaultSendHandler = _s;
  56
  57   // Register builtin commands
  58   for(int i = 0; ; i++) {
  59     if(builtins[i].cmd.isEmpty()) break;
  60     CmdType c;
  61     c.cmd = builtins[i].cmd.toUpper();
  62     c.cmdDescr = builtins[i].cmdDescr;
  63     c.args = builtins[i].args;
  64     c.argsDescr = builtins[i].argsDescr;
  65     c.recvHandler = ( builtins[i].recvHandler ? builtins[i].recvHandler : defaultRecvHandler);
  66     c.sendHandler = ( builtins[i].sendHandler ? builtins[i].sendHandler : defaultSendHandler);
  67     cmdTypes.insert(c.cmd, c);
  68   }
  69
  70 }
  71
  72 recvHandlerType Message::getRecvHandler() {
  73   CmdType c = cmdTypes[cmd];
  74   if(c.recvHandler) return c.recvHandler;
  75   qDebug() << "No recvHandler defined for " << cmd << "!";
  76   return 0;
  77 }
  78
  79 sendHandlerType Message::getSendHandler() {
  80   CmdType c = cmdTypes[cmd];
  81   if(c.sendHandler) return c.sendHandler;
  82   qDebug() << "No sendHandler defined for " << cmd << "!";
  83   return 0;
  84 }
  85
  86 /** This parses a raw string as sent by the server and constructs a message object which is then returned.
  87  */
  88 Message * Message::createFromServerString(Server *srv, QString msg) {
  89   if(msg.isEmpty()) {
  90     qWarning() << "Received empty message from server!";
  91     return 0;
  92   }
  93   QString prefix;
  94   QString cmd;
  95   QStringList params;
  96   if(msg[0] == ':') {
  97     msg.remove(0,1);
  98     prefix = msg.section(' ', 0, 0);
  99     msg = msg.section(' ', 1);
 100   }
 101   cmd = msg.section(' ', 0, 0).toUpper();
 102   msg = msg.section(' ', 1);
 103   QString left = msg.section(':', 0, 0);
 104   QString trailing = msg.section(':', 1);
 105   if(!left.isEmpty()) {
 106     params << left.split(' ', QString::SkipEmptyParts);
 107   }
 108   if(!trailing.isEmpty()) {
 109     params << trailing;
 110   }
 111   return new Message(srv, 0, cmd, prefix, params);
 112   //qDebug() << "prefix: " << prefix << " cmd: " << cmd << " params: " << params;
 113 }