mirror of
https://github.com/vcmi/vcmi.git
synced 2024-12-20 20:23:03 +02:00
116 lines
2.6 KiB
C++
116 lines
2.6 KiB
C++
#include "StdInc.h"
|
|
#include "cdownloadmanager_moc.h"
|
|
|
|
#include "../launcherdirs.h"
|
|
|
|
CDownloadManager::CDownloadManager()
|
|
{
|
|
connect(&manager, SIGNAL(finished(QNetworkReply*)),
|
|
SLOT(downloadFinished(QNetworkReply*)));
|
|
}
|
|
|
|
void CDownloadManager::downloadFile(const QUrl &url, const QString &file)
|
|
{
|
|
QNetworkRequest request(url);
|
|
FileEntry entry;
|
|
entry.file.reset(new QFile(CLauncherDirs::get().downloadsPath() + '/' + file));
|
|
entry.bytesReceived = 0;
|
|
entry.totalSize = 0;
|
|
|
|
if (entry.file->open(QIODevice::WriteOnly | QIODevice::Truncate))
|
|
{
|
|
entry.status = FileEntry::IN_PROGRESS;
|
|
entry.reply = manager.get(request);
|
|
|
|
connect(entry.reply, SIGNAL(downloadProgress(qint64, qint64)),
|
|
SLOT(downloadProgressChanged(qint64, qint64)));
|
|
}
|
|
else
|
|
{
|
|
entry.status = FileEntry::FAILED;
|
|
entry.reply = nullptr;
|
|
encounteredErrors += entry.file->errorString();
|
|
}
|
|
|
|
// even if failed - add it into list to report it in finished() call
|
|
currentDownloads.push_back(entry);
|
|
}
|
|
|
|
CDownloadManager::FileEntry & CDownloadManager::getEntry(QNetworkReply * reply)
|
|
{
|
|
assert(reply);
|
|
for (auto & entry : currentDownloads)
|
|
{
|
|
if (entry.reply == reply)
|
|
return entry;
|
|
}
|
|
assert(0);
|
|
static FileEntry errorValue;
|
|
return errorValue;
|
|
}
|
|
|
|
void CDownloadManager::downloadFinished(QNetworkReply *reply)
|
|
{
|
|
FileEntry & file = getEntry(reply);
|
|
|
|
if (file.reply->error())
|
|
{
|
|
encounteredErrors += file.reply->errorString();
|
|
file.file->remove();
|
|
file.status = FileEntry::FAILED;
|
|
}
|
|
else
|
|
{
|
|
file.file->write(file.reply->readAll());
|
|
file.file->close();
|
|
file.status = FileEntry::FINISHED;
|
|
}
|
|
|
|
bool downloadComplete = true;
|
|
for (auto & entry : currentDownloads)
|
|
{
|
|
if (entry.status == FileEntry::IN_PROGRESS)
|
|
{
|
|
downloadComplete = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
QStringList successful;
|
|
QStringList failed;
|
|
|
|
for (auto & entry : currentDownloads)
|
|
{
|
|
if (entry.status == FileEntry::FINISHED)
|
|
successful += entry.file->fileName();
|
|
else
|
|
failed += entry.file->fileName();
|
|
}
|
|
|
|
if (downloadComplete)
|
|
emit finished(successful, failed, encounteredErrors);
|
|
|
|
file.reply->deleteLater();
|
|
file.reply = nullptr;
|
|
}
|
|
|
|
void CDownloadManager::downloadProgressChanged(qint64 bytesReceived, qint64 bytesTotal)
|
|
{
|
|
auto reply = dynamic_cast<QNetworkReply *>(sender());
|
|
FileEntry & entry = getEntry(reply);
|
|
|
|
entry.file->write(entry.reply->readAll());
|
|
entry.bytesReceived = bytesReceived;
|
|
entry.totalSize = bytesTotal;
|
|
|
|
quint64 total = 0;
|
|
for (auto & entry : currentDownloads)
|
|
total += entry.totalSize > 0 ? entry.totalSize : 0;
|
|
|
|
quint64 received = 0;
|
|
for (auto & entry : currentDownloads)
|
|
received += entry.bytesReceived > 0 ? entry.bytesReceived : 0;
|
|
|
|
emit downloadProgress(received, total);
|
|
}
|