2023-03-02 19:48:41 +00:00
|
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
/*
|
|
|
|
* Prism Launcher - Minecraft Launcher
|
|
|
|
* Copyright (C) 2023 TheKodeToad <TheKodeToad@proton.me>
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, version 3.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "ModrinthPackExportTask.h"
|
2023-03-03 15:00:07 +00:00
|
|
|
|
2023-03-18 12:21:13 +00:00
|
|
|
#include <QCryptographicHash>
|
2023-03-03 15:00:07 +00:00
|
|
|
#include <QFileInfo>
|
2023-03-02 19:48:41 +00:00
|
|
|
#include <QMessageBox>
|
2023-04-06 19:58:09 +01:00
|
|
|
#include <QtConcurrentRun>
|
2023-03-03 15:00:07 +00:00
|
|
|
#include "Json.h"
|
2023-03-02 19:48:41 +00:00
|
|
|
#include "MMCZip.h"
|
|
|
|
#include "minecraft/PackProfile.h"
|
2023-04-06 19:18:36 +01:00
|
|
|
#include "minecraft/mod/ModFolderModel.h"
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-04-06 19:18:36 +01:00
|
|
|
const QStringList ModrinthPackExportTask::PREFIXES({ "mods", "coremods", "resourcepacks", "texturepacks", "shaderpacks" });
|
2023-04-17 13:18:25 +01:00
|
|
|
const QStringList ModrinthPackExportTask::FILE_EXTENSIONS({ "jar", "litemod", "zip" });
|
2023-03-04 10:24:25 +00:00
|
|
|
|
2023-03-02 19:48:41 +00:00
|
|
|
ModrinthPackExportTask::ModrinthPackExportTask(const QString& name,
|
|
|
|
const QString& version,
|
|
|
|
const QString& summary,
|
|
|
|
InstancePtr instance,
|
|
|
|
const QString& output,
|
|
|
|
MMCZip::FilterFunction filter)
|
2023-04-06 19:24:19 +01:00
|
|
|
: name(name)
|
|
|
|
, version(version)
|
|
|
|
, summary(summary)
|
|
|
|
, instance(instance)
|
|
|
|
, mcInstance(dynamic_cast<const MinecraftInstance*>(instance.get()))
|
2023-04-17 13:18:25 +01:00
|
|
|
, gameRoot(instance->gameRoot())
|
2023-04-06 19:24:19 +01:00
|
|
|
, output(output)
|
|
|
|
, filter(filter)
|
2023-03-02 19:48:41 +00:00
|
|
|
{}
|
|
|
|
|
|
|
|
void ModrinthPackExportTask::executeTask()
|
|
|
|
{
|
2023-03-03 15:00:07 +00:00
|
|
|
setStatus(tr("Searching for files..."));
|
2023-03-04 10:24:25 +00:00
|
|
|
setProgress(0, 0);
|
|
|
|
collectFiles();
|
2023-03-04 10:37:52 +00:00
|
|
|
}
|
|
|
|
|
2023-03-04 19:55:38 +00:00
|
|
|
bool ModrinthPackExportTask::abort()
|
|
|
|
{
|
2023-03-06 17:22:20 +00:00
|
|
|
if (task != nullptr) {
|
|
|
|
if (!task->abort())
|
|
|
|
return false;
|
|
|
|
|
2023-03-04 11:07:07 +00:00
|
|
|
task = nullptr;
|
2023-03-18 12:21:13 +00:00
|
|
|
emitAborted();
|
2023-03-04 11:07:07 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-03-06 17:22:20 +00:00
|
|
|
pendingAbort = true;
|
|
|
|
return true;
|
2023-03-04 10:24:25 +00:00
|
|
|
}
|
2023-03-03 15:00:07 +00:00
|
|
|
|
2023-03-04 10:24:25 +00:00
|
|
|
void ModrinthPackExportTask::collectFiles()
|
|
|
|
{
|
|
|
|
files.clear();
|
2023-03-03 15:00:07 +00:00
|
|
|
if (!MMCZip::collectFileListRecursively(instance->gameRoot(), nullptr, &files, filter)) {
|
2023-04-06 19:59:15 +01:00
|
|
|
emitFailed(tr("Could not search for files"));
|
2023-03-03 15:00:07 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2023-03-04 19:55:38 +00:00
|
|
|
pendingHashes.clear();
|
|
|
|
resolvedFiles.clear();
|
2023-03-03 15:00:07 +00:00
|
|
|
|
2023-04-06 19:24:19 +01:00
|
|
|
if (mcInstance) {
|
|
|
|
mcInstance->loaderModList()->update();
|
|
|
|
connect(mcInstance->loaderModList().get(), &ModFolderModel::updateFinished, this, &ModrinthPackExportTask::collectHashes);
|
|
|
|
} else
|
|
|
|
collectHashes();
|
2023-04-06 19:18:36 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void ModrinthPackExportTask::collectHashes()
|
|
|
|
{
|
2023-04-17 13:18:25 +01:00
|
|
|
for (const QFileInfo& file : files) {
|
|
|
|
const QString relative = gameRoot.relativeFilePath(file.absoluteFilePath());
|
2023-03-03 15:00:07 +00:00
|
|
|
// require sensible file types
|
2023-03-04 10:24:25 +00:00
|
|
|
if (!std::any_of(PREFIXES.begin(), PREFIXES.end(),
|
2023-03-03 15:00:07 +00:00
|
|
|
[&relative](const QString& prefix) { return relative.startsWith(prefix + QDir::separator()); }))
|
|
|
|
continue;
|
2023-04-17 13:18:25 +01:00
|
|
|
if (!std::any_of(FILE_EXTENSIONS.begin(), FILE_EXTENSIONS.end(), [&relative](const QString& extension) {
|
|
|
|
return relative.endsWith('.' + extension) || relative.endsWith('.' + extension + ".disabled");
|
|
|
|
})) {
|
|
|
|
continue;
|
|
|
|
}
|
2023-03-03 15:00:07 +00:00
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
QCryptographicHash sha512(QCryptographicHash::Algorithm::Sha512);
|
2023-03-03 15:00:07 +00:00
|
|
|
|
|
|
|
QFile openFile(file.absoluteFilePath());
|
|
|
|
if (!openFile.open(QFile::ReadOnly)) {
|
|
|
|
qWarning() << "Could not open" << file << "for hashing";
|
|
|
|
continue;
|
2023-03-03 11:14:57 +00:00
|
|
|
}
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
const QByteArray data = openFile.readAll();
|
2023-04-06 19:18:36 +01:00
|
|
|
if (openFile.error() != QFileDevice::NoError) {
|
|
|
|
qWarning() << "Could not read" << file;
|
2023-03-03 15:00:07 +00:00
|
|
|
continue;
|
|
|
|
}
|
2023-04-17 13:18:25 +01:00
|
|
|
sha512.addData(data);
|
2023-04-06 19:18:36 +01:00
|
|
|
|
2023-04-06 19:24:19 +01:00
|
|
|
auto allMods = mcInstance->loaderModList()->allMods();
|
2023-04-06 19:18:36 +01:00
|
|
|
if (auto modIter = std::find_if(allMods.begin(), allMods.end(), [&file](Mod* mod) { return mod->fileinfo() == file; });
|
|
|
|
modIter != allMods.end()) {
|
2023-04-17 13:18:25 +01:00
|
|
|
const Mod* mod = *modIter;
|
2023-04-06 19:18:36 +01:00
|
|
|
if (mod->metadata() != nullptr) {
|
|
|
|
QUrl& url = mod->metadata()->url;
|
|
|
|
// most likely some of these may be from curseforge
|
|
|
|
if (!url.isEmpty() && BuildConfig.MODRINTH_MRPACK_HOSTS.contains(url.host())) {
|
|
|
|
qDebug() << "Resolving" << relative << "from index";
|
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
QCryptographicHash sha1(QCryptographicHash::Algorithm::Sha1);
|
|
|
|
sha1.addData(data);
|
2023-04-06 19:18:36 +01:00
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
ResolvedFile file{ sha1.result().toHex(), sha512.result().toHex(), url.toString(), openFile.size() };
|
2023-04-06 19:18:36 +01:00
|
|
|
resolvedFiles[relative] = file;
|
|
|
|
|
|
|
|
// nice! we've managed to resolve based on local metadata!
|
|
|
|
// no need to enqueue it
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-03-03 15:00:07 +00:00
|
|
|
|
2023-04-06 19:18:36 +01:00
|
|
|
qDebug() << "Enqueueing" << relative << "for Modrinth query";
|
2023-04-17 13:18:25 +01:00
|
|
|
pendingHashes[relative] = sha512.result().toHex();
|
2023-03-03 15:00:07 +00:00
|
|
|
}
|
2023-04-06 19:18:36 +01:00
|
|
|
|
|
|
|
makeApiRequest();
|
|
|
|
}
|
|
|
|
|
|
|
|
void ModrinthPackExportTask::makeApiRequest()
|
|
|
|
{
|
|
|
|
if (pendingHashes.isEmpty())
|
|
|
|
buildZip();
|
|
|
|
else {
|
|
|
|
QByteArray* response = new QByteArray;
|
|
|
|
task = api.currentVersions(pendingHashes.values(), "sha512", response);
|
|
|
|
connect(task.get(), &NetJob::succeeded, [this, response]() { parseApiResponse(response); });
|
|
|
|
connect(task.get(), &NetJob::failed, this, &ModrinthPackExportTask::emitFailed);
|
|
|
|
task->start();
|
|
|
|
}
|
2023-03-04 10:24:25 +00:00
|
|
|
}
|
2023-03-03 15:00:07 +00:00
|
|
|
|
2023-03-04 10:24:25 +00:00
|
|
|
void ModrinthPackExportTask::parseApiResponse(QByteArray* response)
|
|
|
|
{
|
2023-03-06 17:22:20 +00:00
|
|
|
task = nullptr;
|
|
|
|
|
2023-03-04 10:24:25 +00:00
|
|
|
try {
|
2023-04-07 11:03:11 +01:00
|
|
|
const QJsonDocument doc = Json::requireDocument(*response);
|
2023-03-04 10:24:25 +00:00
|
|
|
|
2023-03-04 19:55:38 +00:00
|
|
|
QMapIterator<QString, QString> iterator(pendingHashes);
|
2023-03-04 10:24:25 +00:00
|
|
|
while (iterator.hasNext()) {
|
|
|
|
iterator.next();
|
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
const QJsonObject obj = doc[iterator.value()].toObject();
|
2023-03-04 10:24:25 +00:00
|
|
|
if (obj.isEmpty())
|
|
|
|
continue;
|
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
const QJsonArray files = obj["files"].toArray();
|
2023-03-04 10:24:25 +00:00
|
|
|
if (auto fileIter = std::find_if(files.begin(), files.end(),
|
|
|
|
[&iterator](const QJsonValue& file) { return file["hashes"]["sha512"] == iterator.value(); });
|
|
|
|
fileIter != files.end()) {
|
|
|
|
// map the file to the url
|
2023-03-04 19:55:38 +00:00
|
|
|
resolvedFiles[iterator.key()] =
|
|
|
|
ResolvedFile{ fileIter->toObject()["hashes"].toObject()["sha1"].toString(), iterator.value(),
|
|
|
|
fileIter->toObject()["url"].toString(), fileIter->toObject()["size"].toInt() };
|
2023-03-03 15:00:07 +00:00
|
|
|
}
|
|
|
|
}
|
2023-04-17 13:18:25 +01:00
|
|
|
} catch (const Json::JsonException& e) {
|
2023-03-04 10:24:25 +00:00
|
|
|
qWarning() << "Failed to parse versions response" << e.what();
|
|
|
|
}
|
2023-03-04 19:55:38 +00:00
|
|
|
pendingHashes.clear();
|
2023-04-07 11:03:11 +01:00
|
|
|
buildZip();
|
2023-03-04 10:24:25 +00:00
|
|
|
}
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-03-04 19:55:38 +00:00
|
|
|
void ModrinthPackExportTask::buildZip()
|
2023-03-04 10:24:25 +00:00
|
|
|
{
|
2023-04-06 19:58:09 +01:00
|
|
|
static_cast<void>(QtConcurrent::run(QThreadPool::globalInstance(), [this]() {
|
2023-03-06 17:22:20 +00:00
|
|
|
setStatus("Adding files...");
|
|
|
|
QuaZip zip(output);
|
|
|
|
if (!zip.open(QuaZip::mdCreate)) {
|
|
|
|
QFile::remove(output);
|
|
|
|
emitFailed(tr("Could not create file"));
|
|
|
|
return;
|
|
|
|
}
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-03-06 17:22:20 +00:00
|
|
|
if (pendingAbort) {
|
2023-03-19 21:26:25 +00:00
|
|
|
QMetaObject::invokeMethod(this, &ModrinthPackExportTask::emitAborted, Qt::QueuedConnection);
|
2023-03-06 17:22:20 +00:00
|
|
|
return;
|
|
|
|
}
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-03-06 17:22:20 +00:00
|
|
|
QuaZipFile indexFile(&zip);
|
|
|
|
if (!indexFile.open(QIODevice::WriteOnly, QuaZipNewInfo("modrinth.index.json"))) {
|
|
|
|
QFile::remove(output);
|
2023-03-18 12:21:13 +00:00
|
|
|
QMetaObject::invokeMethod(
|
|
|
|
this, [this]() { emitFailed(tr("Could not create index")); }, Qt::QueuedConnection);
|
2023-03-06 17:22:20 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
indexFile.write(generateIndex());
|
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
size_t progress = 0;
|
2023-03-06 17:22:20 +00:00
|
|
|
for (const QFileInfo& file : files) {
|
|
|
|
if (pendingAbort) {
|
|
|
|
QFile::remove(output);
|
2023-03-19 21:26:25 +00:00
|
|
|
QMetaObject::invokeMethod(this, &ModrinthPackExportTask::emitAborted, Qt::QueuedConnection);
|
2023-03-06 17:22:20 +00:00
|
|
|
return;
|
|
|
|
}
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
setProgress(progress, files.length());
|
|
|
|
const QString relative = gameRoot.relativeFilePath(file.absoluteFilePath());
|
2023-03-28 14:22:28 +01:00
|
|
|
if (!resolvedFiles.contains(relative) && !JlCompress::compressFile(&zip, file.absoluteFilePath(), "overrides/" + relative)) {
|
|
|
|
QFile::remove(output);
|
|
|
|
QMetaObject::invokeMethod(
|
2023-04-06 19:18:36 +01:00
|
|
|
this, [this, relative]() { emitFailed(tr("Could not read and compress %1").arg(relative)); }, Qt::QueuedConnection);
|
2023-03-28 14:22:28 +01:00
|
|
|
return;
|
|
|
|
}
|
2023-04-17 13:18:25 +01:00
|
|
|
progress++;
|
2023-03-06 17:22:20 +00:00
|
|
|
}
|
2023-03-03 11:14:57 +00:00
|
|
|
|
2023-03-06 17:22:20 +00:00
|
|
|
zip.close();
|
|
|
|
|
|
|
|
if (zip.getZipError() != 0) {
|
|
|
|
QFile::remove(output);
|
2023-03-18 12:21:13 +00:00
|
|
|
QMetaObject::invokeMethod(
|
|
|
|
this, [this]() { emitFailed(tr("A zip error occurred")); }, Qt::QueuedConnection);
|
2023-03-06 17:22:20 +00:00
|
|
|
return;
|
|
|
|
}
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-03-19 21:26:25 +00:00
|
|
|
QMetaObject::invokeMethod(this, &ModrinthPackExportTask::emitSucceeded, Qt::QueuedConnection);
|
2023-04-06 19:58:09 +01:00
|
|
|
}));
|
2023-03-02 19:48:41 +00:00
|
|
|
}
|
|
|
|
|
2023-03-04 19:55:38 +00:00
|
|
|
QByteArray ModrinthPackExportTask::generateIndex()
|
2023-03-02 19:48:41 +00:00
|
|
|
{
|
|
|
|
QJsonObject obj;
|
|
|
|
obj["formatVersion"] = 1;
|
|
|
|
obj["game"] = "minecraft";
|
|
|
|
obj["name"] = name;
|
|
|
|
obj["versionId"] = version;
|
2023-03-18 14:01:41 +00:00
|
|
|
if (!summary.isEmpty())
|
|
|
|
obj["summary"] = summary;
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-04-17 13:18:25 +01:00
|
|
|
if (mcInstance) {
|
|
|
|
auto profile = mcInstance->getPackProfile();
|
2023-03-03 11:24:10 +00:00
|
|
|
// collect all supported components
|
2023-04-17 13:18:25 +01:00
|
|
|
const ComponentPtr minecraft = profile->getComponent("net.minecraft");
|
|
|
|
const ComponentPtr quilt = profile->getComponent("org.quiltmc.quilt-loader");
|
|
|
|
const ComponentPtr fabric = profile->getComponent("net.fabricmc.fabric-loader");
|
|
|
|
const ComponentPtr forge = profile->getComponent("net.minecraftforge");
|
2023-03-02 19:48:41 +00:00
|
|
|
|
2023-03-03 11:24:10 +00:00
|
|
|
// convert all available components to mrpack dependencies
|
2023-03-02 19:48:41 +00:00
|
|
|
QJsonObject dependencies;
|
2023-03-03 11:24:10 +00:00
|
|
|
if (minecraft != nullptr)
|
|
|
|
dependencies["minecraft"] = minecraft->m_version;
|
|
|
|
if (quilt != nullptr)
|
|
|
|
dependencies["quilt-loader"] = quilt->m_version;
|
|
|
|
if (fabric != nullptr)
|
|
|
|
dependencies["fabric-loader"] = fabric->m_version;
|
|
|
|
if (forge != nullptr)
|
|
|
|
dependencies["forge"] = forge->m_version;
|
2023-03-19 21:26:25 +00:00
|
|
|
|
2023-03-02 19:48:41 +00:00
|
|
|
obj["dependencies"] = dependencies;
|
|
|
|
}
|
|
|
|
|
2023-03-03 15:00:07 +00:00
|
|
|
QJsonArray files;
|
2023-03-04 10:24:25 +00:00
|
|
|
QMapIterator<QString, ResolvedFile> iterator(resolvedFiles);
|
2023-03-03 15:00:07 +00:00
|
|
|
while (iterator.hasNext()) {
|
|
|
|
iterator.next();
|
|
|
|
|
|
|
|
const ResolvedFile& value = iterator.value();
|
|
|
|
|
|
|
|
QJsonObject file;
|
2023-03-19 21:25:12 +00:00
|
|
|
QString path = iterator.key();
|
|
|
|
path.replace(QDir::separator(), "/");
|
|
|
|
file["path"] = path;
|
2023-03-03 15:00:07 +00:00
|
|
|
file["downloads"] = QJsonArray({ iterator.value().url });
|
|
|
|
|
|
|
|
QJsonObject hashes;
|
|
|
|
hashes["sha1"] = value.sha1;
|
|
|
|
hashes["sha512"] = value.sha512;
|
2023-04-06 19:58:09 +01:00
|
|
|
|
2023-03-03 15:00:07 +00:00
|
|
|
file["hashes"] = hashes;
|
|
|
|
file["fileSize"] = value.size;
|
|
|
|
|
|
|
|
files << file;
|
|
|
|
}
|
|
|
|
obj["files"] = files;
|
|
|
|
|
2023-03-02 19:48:41 +00:00
|
|
|
return QJsonDocument(obj).toJson(QJsonDocument::Compact);
|
2023-04-09 21:28:40 +01:00
|
|
|
}
|