119 lines
2.9 KiB
C++
119 lines
2.9 KiB
C++
#include "MultiMC.h"
|
|
#include "ForgeMirrors.h"
|
|
#include "logger/QsLog.h"
|
|
#include <algorithm>
|
|
#include <random>
|
|
|
|
ForgeMirrors::ForgeMirrors(QList<ForgeXzDownloadPtr> &libs, NetJobPtr parent_job,
|
|
QString mirrorlist)
|
|
{
|
|
m_libs = libs;
|
|
m_parent_job = parent_job;
|
|
m_url = QUrl(mirrorlist);
|
|
m_status = Job_NotStarted;
|
|
}
|
|
|
|
void ForgeMirrors::start()
|
|
{
|
|
QLOG_INFO() << "Downloading " << m_url.toString();
|
|
QNetworkRequest request(m_url);
|
|
request.setHeader(QNetworkRequest::UserAgentHeader, "MultiMC/5.0 (Uncached)");
|
|
auto worker = MMC->qnam();
|
|
QNetworkReply *rep = worker->get(request);
|
|
|
|
m_reply = std::shared_ptr<QNetworkReply>(rep);
|
|
connect(rep, SIGNAL(downloadProgress(qint64, qint64)),
|
|
SLOT(downloadProgress(qint64, qint64)));
|
|
connect(rep, SIGNAL(finished()), SLOT(downloadFinished()));
|
|
connect(rep, SIGNAL(error(QNetworkReply::NetworkError)),
|
|
SLOT(downloadError(QNetworkReply::NetworkError)));
|
|
connect(rep, SIGNAL(readyRead()), SLOT(downloadReadyRead()));
|
|
}
|
|
|
|
void ForgeMirrors::downloadError(QNetworkReply::NetworkError error)
|
|
{
|
|
// error happened during download.
|
|
QLOG_ERROR() << "Error getting URL:" << m_url.toString().toLocal8Bit()
|
|
<< "Network error: " << error;
|
|
m_status = Job_Failed;
|
|
}
|
|
|
|
void ForgeMirrors::downloadFinished()
|
|
{
|
|
// if the download succeeded
|
|
if (m_status != Job_Failed)
|
|
{
|
|
// nothing went wrong... ?
|
|
parseMirrorList();
|
|
return;
|
|
}
|
|
// else the download failed, we use a fixed list
|
|
else
|
|
{
|
|
m_status = Job_Finished;
|
|
m_reply.reset();
|
|
deferToFixedList();
|
|
return;
|
|
}
|
|
}
|
|
|
|
void ForgeMirrors::deferToFixedList()
|
|
{
|
|
m_mirrors.clear();
|
|
m_mirrors.append(
|
|
{"Minecraft Forge", "http://files.minecraftforge.net/forge_logo.png",
|
|
"http://files.minecraftforge.net/", "http://files.minecraftforge.net/maven/"});
|
|
m_mirrors.append({"Creeper Host",
|
|
"http://files.minecraftforge.net/forge_logo.png",
|
|
"https://www.creeperhost.net/link.php?id=1",
|
|
"http://new.creeperrepo.net/forge/maven/"});
|
|
injectDownloads();
|
|
emit succeeded(m_index_within_job);
|
|
}
|
|
|
|
void ForgeMirrors::parseMirrorList()
|
|
{
|
|
m_status = Job_Finished;
|
|
auto data = m_reply->readAll();
|
|
m_reply.reset();
|
|
auto dataLines = data.split('\n');
|
|
for(auto line: dataLines)
|
|
{
|
|
auto elements = line.split('!');
|
|
if (elements.size() == 4)
|
|
{
|
|
m_mirrors.append({elements[0],elements[1],elements[2],elements[3]});
|
|
}
|
|
}
|
|
if(!m_mirrors.size())
|
|
deferToFixedList();
|
|
injectDownloads();
|
|
emit succeeded(m_index_within_job);
|
|
}
|
|
|
|
void ForgeMirrors::injectDownloads()
|
|
{
|
|
// shuffle the mirrors randomly
|
|
std::random_device rd;
|
|
std::mt19937 rng(rd());
|
|
std::shuffle(m_mirrors.begin(), m_mirrors.end(), rng);
|
|
|
|
// tell parent to download the libs
|
|
for(auto lib: m_libs)
|
|
{
|
|
lib->setMirrors(m_mirrors);
|
|
m_parent_job->addNetAction(lib);
|
|
}
|
|
}
|
|
|
|
void ForgeMirrors::downloadProgress(qint64 bytesReceived, qint64 bytesTotal)
|
|
{
|
|
m_total_progress = bytesTotal;
|
|
m_progress = bytesReceived;
|
|
emit progress(m_index_within_job, bytesReceived, bytesTotal);
|
|
}
|
|
|
|
void ForgeMirrors::downloadReadyRead()
|
|
{
|
|
}
|