From 9e69b8fe1bf8b325f6c386b7578408da4b775177 Mon Sep 17 00:00:00 2001
From: Jamie Mansfield <jmansfield@cadixdev.org>
Date: Sat, 21 May 2022 19:11:37 +0100
Subject: [PATCH] ATLauncher: Parse keeps and deletes from pack manifests

---
 .../atlauncher/ATLPackManifest.cpp            | 64 +++++++++++++++++++
 .../modplatform/atlauncher/ATLPackManifest.h  | 23 +++++++
 2 files changed, 87 insertions(+)

diff --git a/launcher/modplatform/atlauncher/ATLPackManifest.cpp b/launcher/modplatform/atlauncher/ATLPackManifest.cpp
index 3af02a09..5a458f4e 100644
--- a/launcher/modplatform/atlauncher/ATLPackManifest.cpp
+++ b/launcher/modplatform/atlauncher/ATLPackManifest.cpp
@@ -224,6 +224,64 @@ static void loadVersionExtraArguments(ATLauncher::PackVersionExtraArguments& a,
     a.depends = Json::ensureString(obj, "depends", "");
 }
 
+static void loadVersionKeep(ATLauncher::VersionKeep& k, QJsonObject& obj)
+{
+    k.base = Json::requireString(obj, "base");
+    k.target = Json::requireString(obj, "target");
+}
+
+static void loadVersionKeeps(ATLauncher::VersionKeeps& k, QJsonObject& obj)
+{
+    if (obj.contains("files")) {
+        auto files = Json::requireArray(obj, "files");
+        for (const auto keepRaw : files) {
+            auto keepObj = Json::requireObject(keepRaw);
+            ATLauncher::VersionKeep keep;
+            loadVersionKeep(keep, keepObj);
+            k.files.append(keep);
+        }
+    }
+
+    if (obj.contains("folders")) {
+        auto folders = Json::requireArray(obj, "folders");
+        for (const auto keepRaw : folders) {
+            auto keepObj = Json::requireObject(keepRaw);
+            ATLauncher::VersionKeep keep;
+            loadVersionKeep(keep, keepObj);
+            k.folders.append(keep);
+        }
+    }
+}
+
+static void loadVersionDelete(ATLauncher::VersionDelete& d, QJsonObject& obj)
+{
+    d.base = Json::requireString(obj, "base");
+    d.target = Json::requireString(obj, "target");
+}
+
+static void loadVersionDeletes(ATLauncher::VersionDeletes& d, QJsonObject& obj)
+{
+    if (obj.contains("files")) {
+        auto files = Json::requireArray(obj, "files");
+        for (const auto deleteRaw : files) {
+            auto deleteObj = Json::requireObject(deleteRaw);
+            ATLauncher::VersionDelete versionDelete;
+            loadVersionDelete(versionDelete, deleteObj);
+            d.files.append(versionDelete);
+        }
+    }
+
+    if (obj.contains("folders")) {
+        auto folders = Json::requireArray(obj, "folders");
+        for (const auto deleteRaw : folders) {
+            auto deleteObj = Json::requireObject(deleteRaw);
+            ATLauncher::VersionDelete versionDelete;
+            loadVersionDelete(versionDelete, deleteObj);
+            d.folders.append(versionDelete);
+        }
+    }
+}
+
 void ATLauncher::loadVersion(PackVersion & v, QJsonObject & obj)
 {
     v.version = Json::requireString(obj, "version");
@@ -284,4 +342,10 @@ void ATLauncher::loadVersion(PackVersion & v, QJsonObject & obj)
 
     auto messages = Json::ensureObject(obj, "messages");
     loadVersionMessages(v.messages, messages);
+
+    auto keeps = Json::ensureObject(obj, "keeps");
+    loadVersionKeeps(v.keeps, keeps);
+
+    auto deletes = Json::ensureObject(obj, "deletes");
+    loadVersionDeletes(v.deletes, deletes);
 }
diff --git a/launcher/modplatform/atlauncher/ATLPackManifest.h b/launcher/modplatform/atlauncher/ATLPackManifest.h
index 43510c50..571c976d 100644
--- a/launcher/modplatform/atlauncher/ATLPackManifest.h
+++ b/launcher/modplatform/atlauncher/ATLPackManifest.h
@@ -150,6 +150,26 @@ struct VersionMessages
     QString update;
 };
 
+struct VersionKeep {
+    QString base;
+    QString target;
+};
+
+struct VersionKeeps {
+    QVector<VersionKeep> files;
+    QVector<VersionKeep> folders;
+};
+
+struct VersionDelete {
+    QString base;
+    QString target;
+};
+
+struct VersionDeletes {
+    QVector<VersionDelete> files;
+    QVector<VersionDelete> folders;
+};
+
 struct PackVersionMainClass
 {
     QString mainClass;
@@ -178,6 +198,9 @@ struct PackVersion
     QMap<QString, QString> colours;
     QMap<QString, QString> warnings;
     VersionMessages messages;
+
+    VersionKeeps keeps;
+    VersionDeletes deletes;
 };
 
 void loadVersion(PackVersion & v, QJsonObject & obj);