From a5dd6b6cd7050b4861a50a417d939c6a49d5cc33 Mon Sep 17 00:00:00 2001
From: TheKodeToad <TheKodeToad@proton.me>
Date: Thu, 2 Mar 2023 16:40:49 +0000
Subject: [PATCH] Export without file urls

Signed-off-by: TheKodeToad <TheKodeToad@proton.me>
---
 launcher/ui/dialogs/ExportInstanceDialog.cpp |  2 +-
 launcher/ui/dialogs/ExportMrPackDialog.cpp   | 99 ++++++++++++++++++--
 launcher/ui/dialogs/ExportMrPackDialog.h     |  3 +
 3 files changed, 93 insertions(+), 11 deletions(-)

diff --git a/launcher/ui/dialogs/ExportInstanceDialog.cpp b/launcher/ui/dialogs/ExportInstanceDialog.cpp
index 2c706b8a..f310a689 100644
--- a/launcher/ui/dialogs/ExportInstanceDialog.cpp
+++ b/launcher/ui/dialogs/ExportInstanceDialog.cpp
@@ -123,7 +123,7 @@ bool ExportInstanceDialog::doExport()
 
     const QString output = QFileDialog::getSaveFileName(
         this, tr("Export %1").arg(m_instance->name()),
-        FS::PathCombine(QDir::homePath(), name + ".zip"), "Zip (*.zip)", nullptr, QFileDialog::DontConfirmOverwrite);
+        FS::PathCombine(QDir::homePath(), name + ".zip"), "Zip (*.zip)", nullptr);
     if (output.isEmpty())
     {
         return false;
diff --git a/launcher/ui/dialogs/ExportMrPackDialog.cpp b/launcher/ui/dialogs/ExportMrPackDialog.cpp
index 1a839061..9015407d 100644
--- a/launcher/ui/dialogs/ExportMrPackDialog.cpp
+++ b/launcher/ui/dialogs/ExportMrPackDialog.cpp
@@ -17,9 +17,17 @@
  */
 
 #include "ExportMrPackDialog.h"
-#include <QFileSystemModel>
+#include <QJsonDocument>
+#include "FileSystem.h"
+#include "minecraft/MinecraftInstance.h"
+#include "minecraft/PackProfile.h"
 #include "ui_ExportMrPackDialog.h"
 
+#include <QFileDialog>
+#include <QFileSystemModel>
+#include <QMessageBox>
+#include "MMCZip.h"
+
 ExportMrPackDialog::ExportMrPackDialog(InstancePtr instance, QWidget* parent)
     : QDialog(parent), instance(instance), ui(new Ui::ExportMrPackDialog)
 {
@@ -41,16 +49,87 @@ ExportMrPackDialog::ExportMrPackDialog(InstancePtr instance, QWidget* parent)
     headerView->setSectionResizeMode(0, QHeaderView::Stretch);
 }
 
-void ExportMrPackDialog::done(int result)
-{
-    if (result != Accepted) {
-        QDialog::done(result);
-        return;
-    }
-    QDialog::done(result);
-}
-
 ExportMrPackDialog::~ExportMrPackDialog()
 {
     delete ui;
 }
+
+void ExportMrPackDialog::done(int result)
+{
+    if (result == Accepted)
+        runExport();
+
+    QDialog::done(result);
+}
+
+void ExportMrPackDialog::runExport()
+{
+    const QString filename = FS::RemoveInvalidFilenameChars(ui->name->text());
+    const QString output =
+        QFileDialog::getSaveFileName(this, tr("Export %1").arg(ui->name->text()), FS::PathCombine(QDir::homePath(), filename + ".mrpack"),
+                                     "Modrinth modpack (*.mrpack *.zip)", nullptr);
+
+    if (output.isEmpty())
+        return;
+
+    QFileInfoList files;
+    if (!MMCZip::collectFileListRecursively(instance->gameRoot(), nullptr, &files,
+                                            [this](const QString& path) { return proxy->blockedPaths().covers(path); })) {
+        QMessageBox::warning(this, tr("Unable to export modpack"), tr("Could not collect list of files"));
+        return;
+    }
+
+    QuaZip zip(output);
+    if (!zip.open(QuaZip::mdCreate)) {
+        QFile::remove(output);
+        QMessageBox::warning(this, tr("Unable to export modpack"), tr("Could not create file"));
+        return;
+    }
+
+    QuaZipFile indexFile(&zip);
+    indexFile.setFileName("modrinth.index.json");
+    if (!indexFile.open(QuaZipFile::NewOnly)) {
+        QFile::remove(output);
+        QMessageBox::warning(this, tr("Unable to export modpack"), tr("Could not create index"));
+        return;
+    }
+    indexFile.write(generateIndex());
+    indexFile.close();
+
+    // should exist
+    QDir dotMinecraft(instance->gameRoot());
+
+    for (const QFileInfo& file : files)
+        if (!JlCompress::compressFile(&zip, file.absoluteFilePath(), "overrides/" + dotMinecraft.relativeFilePath(file.absoluteFilePath())))
+            QMessageBox::warning(this, tr("Unable to export modpack"), tr("Could not compress %1").arg(file.absoluteFilePath()));
+
+    zip.close();
+
+    if (zip.getZipError() != 0) {
+        QFile::remove(output);
+        QMessageBox::warning(this, tr("Unable to export modpack"), tr("A zip error occured"));
+        return;
+    }
+}
+
+QByteArray ExportMrPackDialog::generateIndex()
+{
+    QJsonObject obj;
+    obj["formatVersion"] = 1;
+    obj["game"] = "minecraft";
+    obj["name"] = ui->name->text();
+    obj["versionId"] = ui->version->text();
+    obj["summary"] = ui->summary->text();
+
+    MinecraftInstance* mc = dynamic_cast<MinecraftInstance*>(instance.get());
+    if (mc) {
+        auto profile = mc->getPackProfile();
+        auto minecraft = profile->getComponent("net.minecraft");
+
+        QJsonObject dependencies;
+        dependencies["minecraft"] = minecraft->m_version;
+        obj["dependencies"] = dependencies;
+    }
+
+    return QJsonDocument(obj).toJson();
+}
\ No newline at end of file
diff --git a/launcher/ui/dialogs/ExportMrPackDialog.h b/launcher/ui/dialogs/ExportMrPackDialog.h
index 454c25eb..0cf4eb7f 100644
--- a/launcher/ui/dialogs/ExportMrPackDialog.h
+++ b/launcher/ui/dialogs/ExportMrPackDialog.h
@@ -39,4 +39,7 @@ class ExportMrPackDialog : public QDialog {
     InstancePtr instance;
     Ui::ExportMrPackDialog* ui;
     PackIgnoreProxy* proxy;
+
+    void runExport();
+    QByteArray generateIndex();
 };