summaryrefslogtreecommitdiffstats
path: root/src/DownloadManager.cpp
blob: b16bfe4c5b94010a1a27e1ffe61e85a45b8ca797 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#include "DownloadManager.h"

// ----------------------------------------------------------------------------------------
void DownloadManager::get(QString& filename)
{
	// Forge URL from the given filename and the base URL.
	QUrl u = this->baseUrl.resolved(filename);
	// If download in progress, enqueue file and return.
	if (dip)
	{
		qDebug() << "Download in progress! Enqueueing:" << u.toString();
		dlQ.enqueue(u);
		return;
	}
	// No running downloads: enqueue and start next download.
	dlQ.enqueue(u);
	qDebug() << "Enqueueing:" << u.toString() << endl;
	startNextDownload();
}
// ----------------------------------------------------------------------------------------
void DownloadManager::startNextDownload()
{
	qDebug() << "Starting next download: " << dlQ.head().toString()
			 << "(" << dlQ.size() << "in queue.)";
	// Set flag for download in progress.
	dip = true;
	// TODO: needed ?
	if (dlQ.isEmpty())
	{
		qDebug() << "Download queue empty! Exiting...";
		return;
	}
	// Dequeue next URL to download.
	QUrl url = dlQ.dequeue();
	//qDebug() << "Dequeueing..." << url.toString();
	// Extract the filename from the URL
	QString path = url.path();
	QString basename = QFileInfo(path).fileName();
	if (basename.isEmpty())
		this->filename = "download";
	else
		this->filename = basename;
	//qDebug() << "Path is: " << path << endl;
	//qDebug() << "Basename is: " << basename << endl;
	//qDebug() << "Filename is: " << this->filename;
	outfile.setFileName(this->filename);
	// If error upon opening, skip this file.
	if (!outfile.open(QIODevice::WriteOnly))
	{
		qDebug() << "Couldn't open file! Exiting...";
		startNextDownload();
		return;
	}
	// Start the request for this URL.
	QNetworkRequest request(url);
	currentDownload = (*qnam).get(request);
	QObject::connect(currentDownload, SIGNAL(readyRead()), this, SLOT(downloadReady()));
	QObject::connect(currentDownload, SIGNAL(downloadProgress(qint64, qint64)),
					 this, SLOT(downloadProgress(qint64, qint64)));
	QObject::connect(currentDownload, SIGNAL(finished()), this, SLOT(downloadFinished()));
}
// ----------------------------------------------------------------------------------------
//                  Slots for the signals emmited by currentDownload.
// ----------------------------------------------------------------------------------------
// This slot listens to readyRead() emmited when data is available for reading.
void DownloadManager::downloadReady()
{
	// readyRead() fired, so save the readable data.
	outfile.write(this->currentDownload->readAll());
}
// ----------------------------------------------------------------------------------------
// This slot listens to the downloadProgress(..)
// which provides information on the download progress of the file.
void DownloadManager::downloadProgress(qint64 bytesIn, qint64 bytesTotal)
{
	qDebug() << "Download progress of " << currentDownload->url().toString()
			 << ": " << bytesIn << "/" << bytesTotal;
}
// ----------------------------------------------------------------------------------------
// This slot listens to the finished() which is emmited
// when all the data from the reply has been read.
void DownloadManager::downloadFinished()
{
	// Second check if the download actually is finished.
	if (currentDownload->isFinished())
		qDebug() << "Downloaded " << currentDownload->url().toString() << endl;
	// Close output file.
	outfile.close();
	// If queue is empty, we are done.
	// TODO: not sure if this is actually needed...
	if (dlQ.isEmpty())
		return;
	// Queue not empty.
	// Delete current reply object.
	currentDownload->deleteLater();
	// Initialise next download.
	dip = false;
	startNextDownload();
}
// ----------------------------------------------------------------------------------------
// Constructor.
DownloadManager::DownloadManager(const QUrl& baseUrl)
{
	qDebug() << endl << "Initialising Download Manager...";
	this->qnam = new QNetworkAccessManager();
	this->baseUrl = baseUrl;
	dip = false;
}
// Destructor.
DownloadManager::~DownloadManager()
{
	delete[] this->qnam;
}