src/dash/network/filereceiver.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 "filereceiver.h"
  23
  24 #include <QTemporaryFile>
  25 #include <QtDebug>
  26
  27 namespace Dash {
  28 namespace Network {
  29
  30 struct FileReceiver::Private
  31 {
  32         Private() : bytes(0), readed(0), file(0), readData(false) {}
  33         qint64 bytes, readed;
  34
  35         QTemporaryFile *file;
  36         bool readData;
  37 };
  38
  39 FileReceiver::FileReceiver(QObject *parent)
  40  : QObject(parent), d(new Private)
  41 {
  42 }
  43
  44
  45 FileReceiver::~FileReceiver()
  46 {
  47         delete d;
  48 }
  49
  50 void FileReceiver::start(qint64 size)
  51 {
  52         d->bytes = size;
  53         d->readed = 0;
  54
  55         if( d->file )
  56         {
  57                 d->file->close();
  58                 delete d->file;
  59         }
  60
  61         d->file = new QTemporaryFile;
  62
  63
  64         d->readData = d->file->open();
  65         d->file->setAutoRemove(false);
  66
  67         Q_ASSERT(d->readData);
  68 }
  69
  70 qint64 FileReceiver::bytesReaded() const
  71 {
  72         return d->readed;
  73 }
  74
  75 qint64 FileReceiver::totalSize() const
  76 {
  77         return d->bytes;
  78 }
  79
  80 bool FileReceiver::isReading() const
  81 {
  82         return d->readData;
  83 }
  84
  85 void FileReceiver::readed(const QString &)
  86 {
  87 }
  88
  89 void FileReceiver::dataReaded(const QByteArray &data)
  90 {
  91         if( d->readData )
  92         {
  93                 d->readed += data.size();
  94                 d->file->write(data);
  95
  96                 if( d->readed >= d->bytes )
  97                 {
  98                         d->readData = false;
  99
 100                         QString fileName = d->file->fileName();
 101                         d->file->flush();
 102                         d->file->close();
 103
 104                         emit finished(fileName);
 105                 }
 106         }
 107 }
 108
 109 }
 110 }
 111