119 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			119 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
#include "Env.h"
 | 
						|
#include "ForgeMirrors.h"
 | 
						|
#include <QDebug>
 | 
						|
#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()
 | 
						|
{
 | 
						|
	qDebug() << "Downloading " << m_url.toString();
 | 
						|
	QNetworkRequest request(m_url);
 | 
						|
	request.setHeader(QNetworkRequest::UserAgentHeader, "MultiMC/5.0 (Uncached)");
 | 
						|
	auto worker = ENV.qnam();
 | 
						|
	QNetworkReply *rep = worker->get(request);
 | 
						|
 | 
						|
	m_reply.reset(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.
 | 
						|
	qCritical() << "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 netActionProgress(m_index_within_job, bytesReceived, bytesTotal);
 | 
						|
}
 | 
						|
 | 
						|
void ForgeMirrors::downloadReadyRead()
 | 
						|
{
 | 
						|
}
 |