pollymc/launcher/modplatform/flame/FileResolvingTask.cpp

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

Flame::FileResolvingTask::FileResolvingTask(shared_qobject_ptr<QNetworkAccessManager> network, Flame::Manifest& toProcess)
    : m_network(network), m_toProcess(toProcess)
{}

void Flame::FileResolvingTask::executeTask()
{
    setStatus(tr("Resolving mod IDs..."));
    setProgress(0, m_toProcess.files.size());
    m_dljob = new NetJob("Mod id resolver", m_network);
    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("https://api.curseforge.com/v1/mods/%1/files/%2").arg(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) {
        auto& out = m_toProcess.files[index];
        try {
            bool fail = (!out.parseFromBytes(bytes));
            if(fail){
                //failed :( probably disabled mod, try to add to the list
                auto doc = Json::requireDocument(bytes);
                if (!doc.isObject()) {
                    throw JSONValidationError(QString("data is not an object? that's not supposed to happen"));
                }
                auto obj = Json::ensureObject(doc.object(), "data");
                //FIXME : HACK, MAY NOT WORK FOR LONG
                out.url = QUrl(QString("https://media.forgecdn.net/files/%1/%2/%3")
                        .arg(QString::number(QString::number(out.fileId).leftRef(4).toInt())
                             ,QString::number(QString::number(out.fileId).rightRef(3).toInt())
                             ,QUrl::toPercentEncoding(out.fileName)), QUrl::TolerantMode);
            }
            failed &= fail;
        } catch (const JSONValidationError& e) {
            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."));
    }
}