PrismLauncher/api/logic/modplatform/flame/FileResolvingTask.cpp

#include "FileResolvingTask.h"
#include "Json.h"

const char * metabase = "https://cursemeta.dries007.net";

Flame::FileResolvingTask::FileResolvingTask(Flame::Manifest& toProcess)
	: m_toProcess(toProcess)
{
}

void Flame::FileResolvingTask::executeTask()
{
	setStatus(tr("Resolving mod IDs..."));
	setProgress(0, m_toProcess.files.size());
	m_dljob.reset(new NetJob("Mod id resolver"));
	results.resize(m_toProcess.files.size());
	int index = 0;
	for(auto & file: m_toProcess.files)
	{
		auto projectIdStr = QString::number(file.projectId);
		auto fileIdStr = QString::number(file.fileId);
		QString metaurl = QString("%1/%2/%3.json").arg(metabase, projectIdStr, fileIdStr);
		auto dl = Net::Download::makeByteArray(QUrl(metaurl), &results[index]);
		m_dljob->addNetAction(dl);
		index ++;
	}
	connect(m_dljob.get(), &NetJob::finished, this, &Flame::FileResolvingTask::netJobFinished);
	m_dljob->start();
}

void Flame::FileResolvingTask::netJobFinished()
{
	bool failed = false;
	int index = 0;
	for(auto & bytes: results)
	{
		try
		{
			auto doc = Json::requireDocument(bytes);
			auto obj = Json::requireObject(doc);
			auto & out = m_toProcess.files[index];
			// result code signifies true failure.
			if(obj.contains("code"))
			{
				qCritical() << "Resolving of" << out.projectId << out.fileId << "failed because of a negative result:";
				qCritical() << bytes;
				failed = true;
				continue;
			}
			out.fileName = Json::requireString(obj, "FileNameOnDisk");
			QString rawUrl = Json::requireString(obj, "DownloadURL");
			out.url = QUrl(rawUrl, QUrl::TolerantMode);
			if(!out.url.isValid())
			{
				throw JSONValidationError(QString("Invalid URL: %1").arg(rawUrl));
			}
			// This is a piece of a Flame project JSON pulled out into the file metadata (here) for convenience
			// It is also optional
			QJsonObject projObj = Json::ensureObject(obj, "_Project", {});
			if(!projObj.isEmpty())
			{
				QString strType = Json::ensureString(projObj, "PackageType", "mod").toLower();
				if(strType == "singlefile")
				{
					out.type = File::Type::SingleFile;
				}
				else if(strType == "ctoc")
				{
					out.type = File::Type::Ctoc;
				}
				else if(strType == "cmod2")
				{
					out.type = File::Type::Cmod2;
				}
				else if(strType == "mod")
				{
					out.type = File::Type::Mod;
				}
				else if(strType == "folder")
				{
					out.type = File::Type::Folder;
				}
				else if(strType == "modpack")
				{
					out.type = File::Type::Modpack;
				}
				else
				{
					qCritical() << "Resolving of" << out.projectId << out.fileId << "failed because of unknown file type:" << strType;
					out.type = File::Type::Unknown;
					failed = true;
					continue;
				}
				out.targetFolder = Json::ensureString(projObj, "Path", "mods");
			}
			out.resolved = true;
		}
		catch (const JSONValidationError &e)
		{
			auto & out = m_toProcess.files[index];
			qCritical() << "Resolving of" << out.projectId << out.fileId << "failed because of a parsing error:";
			qCritical() << e.cause();
			qCritical() << "JSON:";
			qCritical() << bytes;
			failed = true;
		}
		index++;
	}
	if(!failed)
	{
		emitSucceeded();
	}
	else
	{
		emitFailed(tr("Some mod ID resolving tasks failed."));
	}
}