src/dashserver/com/commanager.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Jorge Cuadrado                                  *
   3  *   kuadrosxx@gmail.com                                                   *
   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 "commanager.h"
  22
  23 //base
  24 #include "package.h"
  25 #include "logger.h"
  26
  27 //Qt
  28 #include <QVariant>
  29 #include <QXmlStreamReader>
  30 #include <QXmlStreamWriter>
  31
  32 //server/core
  33 #include "server.h"
  34 #include "connection.h"
  35
  36 //server/users
  37 #include "users/user.h"
  38
  39 #include "project/store.h"
  40 #include "project/projectmanager.h"
  41
  42 namespace DashServer {
  43 namespace Com {
  44
  45 struct Manager::Private
  46 {
  47         QString writeFromAttribute(const QString &pkg, const QString &from);
  48 };
  49
  50 QString Manager::Private::writeFromAttribute(const QString &pkg, const QString &from)
  51 {
  52         QString xml;
  53
  54         QXmlStreamReader reader(pkg);
  55         QXmlStreamWriter writer(&xml);
  56
  57         while(!reader.atEnd() )
  58         {
  59                 reader.readNext();
  60                 writer.writeCurrentToken(reader);
  61
  62                 if( reader.tokenType() == QXmlStreamReader::StartElement)
  63                 {
  64                         if( reader.name().toString() == "message" )
  65                         {
  66                                 writer.writeAttribute("from", from);
  67                         }
  68                 }
  69         }
  70
  71         return xml;
  72 }
  73
  74 Manager::Manager()
  75  : DashServer::Observer(), d(new Private)
  76 {
  77 }
  78
  79
  80 Manager::~Manager()
  81 {
  82         delete d;
  83 }
  84
  85 void Manager::handlePackage(DashServer::Package *const pkg)
  86 {
  87
  88         DashServer::Connection *cnn = pkg->source();
  89         DashServer::TcpServer *server = cnn->server();
  90
  91         if( pkg->root() == "chat" )
  92         {
  93                 QString projectName = cnn->data(Project::Store::ProjectName).toString();
  94
  95                 if( !projectName.isEmpty() )
  96                 {
  97                         server->projectManager()->sendToMembers(projectName, d->writeFromAttribute(pkg->xml(), cnn->user()->login()));
  98                         pkg->accept();
  99                 }
 100         }
 101         else if( pkg->root() == "notice" )
 102         {
 103                 QString projectName = cnn->data(Project::Store::ProjectName).toString();
 104
 105                 if( !projectName.isEmpty() )
 106                 {
 107                         server->projectManager()->sendToMembers(projectName, d->writeFromAttribute(pkg->xml(), cnn->user()->login()));
 108                         pkg->accept();
 109                 }
 110         }
 111         else if( pkg->root() == "wall" )
 112         {
 113                 cnn->sendToAll(d->writeFromAttribute(pkg->xml(), cnn->user()->login()), true);
 114                 pkg->accept();
 115         }
 116 }
 117
 118 }
 119 }
 120