src/dashserver/project/project.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by David Cuadrado                                  *
   3  *   krawek@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
  22 #include "project.h"
  23
  24 #include <QSet>
  25
  26 #include <yamf/model/command/processor.h>
  27
  28 #include "dashserver/connection.h"
  29 #include "dashserver/xdbms/manager.h"
  30
  31 #include "dash/projectsaver.h"
  32 #include "dash/network/package/transfer.h"
  33
  34 #include "member.h"
  35 #include "commandprocessor.h"
  36
  37 #include <dcore/debug.h>
  38
  39 namespace DashServer {
  40 namespace Project {
  41
  42 struct Project::Private
  43 {
  44         Private(Project *q) : q(q), initialized(false) {}
  45
  46         Project *q;
  47         bool initialized;
  48         CommandProcessor *processor;
  49
  50         QSet<DashServer::Connection *> connections;
  51
  52         QHash<QString, Member *> members;
  53
  54
  55         void loadMembers();
  56 };
  57
  58 void Project::Private::loadMembers()
  59 {
  60         members.clear();
  61
  62         QList<XDBMS::Object *> objects = q->objects();
  63
  64         foreach(XDBMS::Object *obj, objects)
  65         {
  66                 if( obj->klass() == "member" )
  67                 {
  68                         Member *member = static_cast<Member *>(obj);
  69                         members[member->id()] = member;
  70                 }
  71         }
  72
  73         initialized = true;
  74 }
  75
  76 Project::Project(QObject *parent) : Dash::Project(parent), XDBMS::Object(), d(new Private(this))
  77 {
  78         d->processor = new CommandProcessor(this);
  79
  80         setClass("project");
  81 }
  82
  83
  84 Project::~Project()
  85 {
  86         delete d->processor;
  87         delete d;
  88 }
  89
  90 void Project::execute(const QString &xml, DashServer::Connection *source)
  91 {
  92         if( d->connections.contains(source) )
  93         {
  94                 d->processor->setCurrentMember( d->members.value(source->user()->login()) );
  95                 if( d->processor->execute(xml) )
  96                 {
  97                         QString tosend = source->signXml(xml);
  98
  99                         foreach(DashServer::Connection *cnx, d->connections)
 100                         {
 101                                 cnx->sendToClient( tosend, false );
 102                         }
 103                 }
 104                 else
 105                 {
 106                         source->sendToClient( "<command />", true ); // FIXME: create a package
 107                 }
 108         }
 109         else
 110         {
 111                 qWarning("INVALID SOURCE");
 112         }
 113 }
 114
 115 void Project::addClient(DashServer::Connection *cnx)
 116 {
 117         d->connections << cnx;
 118 }
 119
 120 void Project::removeClient(DashServer::Connection *client)
 121 {
 122         d->connections.remove(client);
 123
 124         if( d->connections.isEmpty() )
 125         {
 126                 save();
 127         }
 128 }
 129
 130 void Project::sendToClient(DashServer::Connection *client)
 131 {
 132         // FIXME: Poner en un thread
 133         {
 134                 QFile f(this->fileName());
 135                 Dash::Network::Package::Transfer transfer(f.size());
 136                 client->sendToClient(transfer.toString(), true);
 137
 138                 if( f.open(QIODevice::ReadOnly) )
 139                 {
 140                         while(!f.atEnd())
 141                         {
 142                                 client->sendBinaryData(f.read(1024));
 143                         }
 144                 }
 145         }
 146 }
 147
 148 void Project::sendToAll()
 149 {
 150         save();
 151
 152         {
 153                 QFile f(this->fileName());
 154
 155                 foreach(DashServer::Connection *client, d->connections)
 156                 {
 157                         Dash::Network::Package::Transfer transfer(f.size());
 158                         client->sendToClient(transfer.toString(), true);
 159                 }
 160
 161                 if( f.open(QIODevice::ReadOnly) )
 162                 {
 163                         while(!f.atEnd())
 164                         {
 165                                 foreach(DashServer::Connection *client, d->connections)
 166                                 {
 167                                         client->sendBinaryData(f.read(1024));
 168                                 }
 169                         }
 170                 }
 171         }
 172 }
 173
 174 QHash<QString, Member *> Project::members() const
 175 {
 176         return d->members;
 177 }
 178
 179 QSet<DashServer::Connection *> Project::connections() const
 180 {
 181         return d->connections;
 182 }
 183
 184 int Project::clientCount() const
 185 {
 186         return d->connections.size();
 187 }
 188
 189 void Project::aboutToSave()
 190 {
 191         if( projectName().isEmpty() ) return;
 192
 193         setId(projectName());
 194
 195         add("description", description());
 196         add("author", author());
 197
 198         if(!d->initialized) d->loadMembers();
 199 }
 200
 201 void Project::afterLoad()
 202 {
 203         setProjectName(id());
 204         setDescription(value("description"));
 205         setAuthor(value("author"));
 206
 207         if(!d->initialized) d->loadMembers();
 208 }
 209
 210 bool Project::save()
 211 {
 212         if(!d->initialized) d->loadMembers();
 213
 214         QString pname = projectName();
 215
 216         if( pname.isEmpty() ) return false;
 217
 218         return Dash::ProjectSaver::save(this, fileName());
 219 }
 220
 221 bool Project::load()
 222 {
 223         if(!d->initialized) d->loadMembers();
 224
 225         QString pname = projectName();
 226
 227         if( pname.isEmpty() )
 228         {
 229                 qWarning("Warning: Project name is empty");
 230                 return false;
 231         }
 232
 233         clear();
 234
 235         return Dash::ProjectSaver::load(this, fileName());
 236 }
 237
 238 QString Project::fileName() const
 239 {
 240         QString pname = projectName();
 241
 242         if( pname.isEmpty() )
 243                 return QString();
 244
 245         return XDBMS::Manager::self()->databaseDir().absoluteFilePath(pname+".dsh");
 246 }
 247
 248 void Project::addMember(Member *member)
 249 {
 250         if( d->members.contains(member->id()) )
 251         {
 252                 d->members.remove(member->id());
 253                 remove(member);
 254         }
 255
 256         d->members[member->id()] = member;
 257         add(member);
 258 }
 259
 260 QString Project::partName(int part)
 261 {
 262         switch(part)
 263         {
 264                 case All:
 265                 {
 266                         return "all";
 267                 }
 268                 break;
 269                 case Scene:
 270                 {
 271                         return "scene";
 272                 }
 273                 break;
 274                 case Layer:
 275                 {
 276                         return "layer";
 277                 }
 278                 break;
 279                 case Frame:
 280                 {
 281                         return "frame";
 282                 }
 283                 break;
 284                 case Object:
 285                 {
 286                         return "object";
 287                 }
 288                 break;
 289         }
 290
 291         return QString();
 292 }
 293
 294 }
 295 }
 296
 297