129 lines
3.6 KiB
C++
129 lines
3.6 KiB
C++
/*
|
|
* Copyright 2013 Albert Vaca <albertvaka@gmail.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation; either version 2 of
|
|
* the License or (at your option) version 3 or any later version
|
|
* accepted by the membership of KDE e.V. (or its successor approved
|
|
* by the membership of KDE e.V.), which shall act as a proxy
|
|
* defined in Section 14 of version 3 of the license.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "filetransferjob.h"
|
|
|
|
#include <KLocalizedString>
|
|
|
|
#include <QDebug>
|
|
#include <qalgorithms.h>
|
|
|
|
FileTransferJob::FileTransferJob(QIODevice* origin, int size, const KUrl& destination): KJob()
|
|
{
|
|
mDestination = destination;
|
|
mOrigin = origin;
|
|
mSize = size;
|
|
mWritten = 0;
|
|
}
|
|
|
|
void FileTransferJob::start()
|
|
{
|
|
//Open destination file
|
|
|
|
QTemporaryFile tmp;
|
|
tmp.setAutoRemove(false);
|
|
tmp.open();
|
|
|
|
mTempDestination = KIO::open(tmp.fileName(), QIODevice::WriteOnly);
|
|
connect(mTempDestination, SIGNAL(open(KIO::Job*)), this, SLOT(open(KIO::Job*)));
|
|
mTempDestination->start();
|
|
|
|
}
|
|
|
|
void FileTransferJob::open(KIO::Job* job)
|
|
{
|
|
Q_UNUSED(job);
|
|
|
|
qDebug() << "open";
|
|
|
|
if (!mOrigin) {
|
|
qDebug() << "FileTransferJob: Origin is null";
|
|
return;
|
|
}
|
|
|
|
//Open source file
|
|
mOrigin->open(QIODevice::ReadOnly);
|
|
Q_ASSERT(mOrigin->isOpen());
|
|
|
|
connect(mOrigin, SIGNAL(readyRead()),this, SLOT(readyRead()));
|
|
connect(mOrigin, SIGNAL(disconnected()),this, SLOT(sourceFinished()));
|
|
if (mOrigin->bytesAvailable() > 0) readyRead();
|
|
|
|
}
|
|
|
|
void FileTransferJob::readyRead()
|
|
{
|
|
qDebug() << "readyRead";
|
|
|
|
//Copy a chunk of data
|
|
|
|
int bytes = qMin(qint64(4096), mOrigin->bytesAvailable());
|
|
QByteArray data = mOrigin->read(bytes);
|
|
mTempDestination->write(data);
|
|
mWritten += bytes;
|
|
|
|
if (mSize > -1) {
|
|
setPercent((mWritten*100)/mSize);
|
|
}
|
|
|
|
qDebug() << mSize << mWritten << bytes;
|
|
|
|
if (mSize > -1 && mWritten >= mSize) { //At the end or expected size reached
|
|
qDebug() << "No more data to read";
|
|
disconnect(mOrigin, SIGNAL(readyRead()),this, SLOT(readyRead()));
|
|
mOrigin->close();
|
|
} else if (mOrigin->bytesAvailable() > 0) {
|
|
QMetaObject::invokeMethod(this, "readyRead", Qt::QueuedConnection);
|
|
}
|
|
}
|
|
|
|
void FileTransferJob::sourceFinished()
|
|
{
|
|
qDebug() << "sourceFinished";
|
|
|
|
//Make sure we do not enter this function again
|
|
disconnect(mOrigin, SIGNAL(aboutToClose()),this, SLOT(sourceFinished()));
|
|
|
|
//TODO: MD5 check the file
|
|
if (mSize > -1 && mWritten != mSize) {
|
|
qDebug() << "Received incomplete file";
|
|
setError(1);
|
|
setErrorText(i18n("Received incomplete file"));
|
|
emitResult();
|
|
}
|
|
|
|
qDebug() << "Finished" << mTempDestination->url() << mDestination;
|
|
KIO::FileCopyJob* job = KIO::file_move(mTempDestination->url(), mDestination);
|
|
connect(job, SIGNAL(result(KJob*)), this, SLOT(moveResult(KJob*)));
|
|
job->start();
|
|
|
|
//delete mOrigin; //TODO: Use shared pointers
|
|
}
|
|
|
|
void FileTransferJob::moveResult(KJob* job)
|
|
{
|
|
//TODO: Error handling, cleanup
|
|
qDebug() << "Move finished";
|
|
if (job->error()) {
|
|
qDebug() << job->errorText();
|
|
}
|
|
emitResult();
|
|
}
|
|
|