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     cmdCode = - c.cmdCode; // named commands have a _negative_ code!
  39   } else {
  40     int t = cmd.toInt();
  41     if(t) {
  42       cmdCode = t;
  43       recvHandler = defaultRecvHandler;
  44       sendHandler = defaultSendHandler;
  45     } else {
  46       // Unknown cmd!
  47       qWarning() << "Unknown command: " << cmd;
  48       cmdCode = 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.cmdCode = builtins[i].cmdCode;
  62     c.cmd = builtins[i].cmd.toUpper();
  63     c.cmdDescr = builtins[i].cmdDescr;
  64     c.args = builtins[i].args;
  65     c.argsDescr = builtins[i].argsDescr;
  66     c.recvHandler = ( builtins[i].recvHandler ? builtins[i].recvHandler : defaultRecvHandler);
  67     c.sendHandler = ( builtins[i].sendHandler ? builtins[i].sendHandler : defaultSendHandler);
  68     cmdTypes.insert(c.cmd, c);
  69   }
  70
  71 }
  72
  73 recvHandlerType Message::getRecvHandler() {
  74   CmdType c = cmdTypes[cmd];
  75   if(c.recvHandler) return c.recvHandler;
  76   qDebug() << "No recvHandler defined for " << cmd << "!";
  77   return 0;
  78 }
  79
  80 sendHandlerType Message::getSendHandler() {
  81   CmdType c = cmdTypes[cmd];
  82   if(c.sendHandler) return c.sendHandler;
  83   qDebug() << "No sendHandler defined for " << cmd << "!";
  84   return 0;
  85 }
  86
  87 /** This parses a raw string as sent by the server and constructs a message object which is then returned.
  88  */
  89 Message * Message::createFromServerString(Server *srv, QString msg) {
  90   if(msg.isEmpty()) {
  91     qWarning() << "Received empty message from server!";
  92     return 0;
  93   }
  94   QString prefix;
  95   QString cmd;
  96   QStringList params;
  97   if(msg[0] == ':') {
  98     msg.remove(0,1);
  99     prefix = msg.section(' ', 0, 0);
 100     msg = msg.section(' ', 1);
 101   }
 102   cmd = msg.section(' ', 0, 0).toUpper();
 103   msg = msg.section(' ', 1);
 104   QString left = msg.section(':', 0, 0);
 105   QString trailing = msg.section(':', 1);
 106   if(!left.isEmpty()) {
 107     params << left.split(' ', QString::SkipEmptyParts);
 108   }
 109   if(!trailing.isEmpty()) {
 110     params << trailing;
 111   }
 112   return new Message(srv, 0, cmd, prefix, params);
 113   //qDebug() << "prefix: " << prefix << " cmd: " << cmd << " params: " << params;
 114 }