GPXSee/src/downloader.cpp

#include <QFile>
#include <QFileInfo>
#include "config.h"
#include "downloader.h"

#include <QDebug>


#ifdef Q_OS_LINUX
#define PLATFORM_STR "Linux"
#elif Q_OS_WIN32
#define PLATFORM_STR "Windows"
#elif Q_OS_MAC
#define PLATFORM_STR "OS X"
#else
#define PLATFORM_STR "Unknown"
#endif

#define USER_AGENT APP_NAME"/"APP_VERSION" ("PLATFORM_STR"; Qt "QT_VERSION_STR")"


Downloader::Downloader()
{
	connect(&manager, SIGNAL(finished(QNetworkReply*)),
			SLOT(downloadFinished(QNetworkReply*)));
}

void Downloader::doDownload(const Download &dl)
{
	QUrl url(dl.url());
	QNetworkRequest request(url);
	request.setAttribute(QNetworkRequest::User, QVariant(dl.file()));
	request.setRawHeader("User-Agent", USER_AGENT);
	QNetworkReply *reply = manager.get(request);

	currentDownloads.append(reply);
}

bool Downloader::saveToDisk(const QString &filename, QIODevice *data)
{
	QFile file(filename);

	if (!file.open(QIODevice::WriteOnly)) {
		fprintf(stderr, "Error writing map tile: %s: %s\n",
		  qPrintable(filename), qPrintable(file.errorString()));
		return false;
	}

	file.write(data->readAll());
	file.close();

	return true;
}

void Downloader::downloadFinished(QNetworkReply *reply)
{
	QUrl url = reply->url();
	if (reply->error()) {
		fprintf(stderr, "Error downloading map tile: %s: %s\n",
		  url.toEncoded().constData(), qPrintable(reply->errorString()));
	} else {
		QString filename = reply->request().attribute(QNetworkRequest::User)
		  .toString();
		saveToDisk(filename, reply);
	}

	currentDownloads.removeAll(reply);
	reply->deleteLater();

	if (currentDownloads.isEmpty())
		emit finished();
}

void Downloader::get(const QList<Download> &list)
{
	for (int i = 0; i < list.count(); i++)
		doDownload(list.at(i));
}