// vim: set tabstop=4 shiftwidth=4 noexpandtab:
/*
Gwenview: an image viewer
Copyright 2009 Aurélien Gâteau <agateau@kde.org>

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) any later version.

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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Cambridge, MA 02110-1301, USA.

*/
// Self
#include "savejob.h"

// Qt
#include <QFuture>
#include <QFutureWatcher>
#include <QScopedPointer>
#include <QtConcurrentRun>

// KDE
#include <kapplication.h>
#include <kio/copyjob.h>
#include <kio/jobuidelegate.h>
#include <klocale.h>
#include <ksavefile.h>
#include <ktemporaryfile.h>
#include <kurl.h>

// Local
#include "documentloadedimpl.h"

namespace Gwenview {


struct SaveJobPrivate {
	DocumentLoadedImpl* mImpl;
	KUrl mOldUrl;
	KUrl mNewUrl;
	QByteArray mFormat;
	QScopedPointer<KTemporaryFile> mTemporaryFile;
	QScopedPointer<KSaveFile> mSaveFile;
	QScopedPointer<QFutureWatcher<void> > mInternalSaveWatcher;

	bool mKillReceived;
};


SaveJob::SaveJob(DocumentLoadedImpl* impl, const KUrl& url, const QByteArray& format)
: d(new SaveJobPrivate) {
	d->mImpl = impl;
	d->mOldUrl = impl->document()->url();
	d->mNewUrl = url;
	d->mFormat = format;
	d->mKillReceived = false;
	setCapabilities(Killable);
}


SaveJob::~SaveJob() {
	delete d;
}


void SaveJob::saveInternal() {
	if (!d->mImpl->saveInternal(d->mSaveFile.data(), d->mFormat)) {
		d->mSaveFile->abort();
		setError(UserDefinedError + 2);
		setErrorText(d->mImpl->document()->errorString());
	}
}


void SaveJob::doStart() {
	if (d->mKillReceived) {
		return;
	}
	QString fileName;

	if (d->mNewUrl.isLocalFile()) {
		fileName = d->mNewUrl.toLocalFile();
	} else {
		d->mTemporaryFile.reset(new KTemporaryFile);
		d->mTemporaryFile->setAutoRemove(true);
		d->mTemporaryFile->open();
		fileName = d->mTemporaryFile->fileName();
	}

	d->mSaveFile.reset(new KSaveFile(fileName));

	if (!d->mSaveFile->open()) {
		KUrl dirUrl = d->mNewUrl;
		dirUrl.setFileName(QString());
		setError(UserDefinedError + 1);
		setErrorText(i18nc("@info", "Could not open file for writing, check that you have the necessary rights in <filename>%1</filename>.", dirUrl.pathOrUrl()));
		emitResult();
		return;
	}

	QFuture<void> future = QtConcurrent::run(this, &SaveJob::saveInternal);
	d->mInternalSaveWatcher.reset(new QFutureWatcher<void>(this));
	d->mInternalSaveWatcher->setFuture(future);
	connect(d->mInternalSaveWatcher.data(), SIGNAL(finished()), SLOT(finishSave()));
}


void SaveJob::finishSave() {
	d->mInternalSaveWatcher.reset(0);
	if (d->mKillReceived) {
		return;
	}

	if (error()) {
		emitResult();
		return;
	}

	if (!d->mSaveFile->finalize()) {
		setErrorText(i18nc("@info", "Could not overwrite file, check that you have the necessary rights to write in <filename>%1</filename>.", d->mNewUrl.pathOrUrl()));
		setError(UserDefinedError + 3);
		return;
	}

	if (d->mNewUrl.isLocalFile()) {
		emitResult();
	} else {
		KIO::Job* job = KIO::copy(KUrl::fromPath(d->mTemporaryFile->fileName()), d->mNewUrl);
		job->ui()->setWindow(KApplication::kApplication()->activeWindow());
		addSubjob(job);
	}
}


void SaveJob::slotResult(KJob* job) {
	DocumentJob::slotResult(job);
	if (!error()) {
		emitResult();
	}
}


KUrl SaveJob::oldUrl() const {
	return d->mOldUrl;
}


KUrl SaveJob::newUrl() const {
	return d->mNewUrl;
}


bool SaveJob::doKill() {
	d->mKillReceived = true;
	if (d->mInternalSaveWatcher) {
		d->mInternalSaveWatcher->waitForFinished();
	}
	return true;
}


} // namespace
