1 files changed, 140 insertions, 4 deletions

diff --git a/launcher/modplatform/atlauncher/ATLPackInstallTask.cpp b/launcher/modplatform/atlauncher/ATLPackInstallTask.cpp
index 0ed0ad29..5ed13470 100644
--- a/launcher/modplatform/atlauncher/ATLPackInstallTask.cpp
+++ b/launcher/modplatform/atlauncher/ATLPackInstallTask.cpp
@@ -60,12 +60,13 @@ namespace ATLauncher {
 
 static Meta::VersionPtr getComponentVersion(const QString& uid, const QString& version);
 
-PackInstallTask::PackInstallTask(UserInteractionSupport *support, QString packName, QString version)
+PackInstallTask::PackInstallTask(UserInteractionSupport *support, QString packName, QString version, InstallMode installMode)
 {
     m_support = support;
     m_pack_name = packName;
     m_pack_safe_name = packName.replace(QRegularExpression("[^A-Za-z0-9]"), "");
     m_version_name = version;
+    m_install_mode = installMode;
 }
 
 bool PackInstallTask::abort()
@@ -117,9 +118,30 @@ void PackInstallTask::onDownloadSucceeded()
     }
     m_version = version;
 
-    // Display install message if one exists
-    if (!m_version.messages.install.isEmpty())
-        m_support->displayMessage(m_version.messages.install);
+    // Derived from the installation mode
+    QString message;
+    bool resetDirectory;
+
+    switch (m_install_mode) {
+    case InstallMode::Reinstall:
+    case InstallMode::Update:
+        message = m_version.messages.update;
+        resetDirectory = true;
+        break;
+
+    case InstallMode::Install:
+        message = m_version.messages.install;
+        resetDirectory = false;
+        break;
+
+    default:
+        emitFailed(tr("Unsupported installation mode"));
+        break;
+    }
+
+    // Display message if one exists
+    if (!message.isEmpty())
+        m_support->displayMessage(message);
 
     auto ver = getComponentVersion("net.minecraft", m_version.minecraft);
     if (!ver) {
@@ -128,6 +150,10 @@ void PackInstallTask::onDownloadSucceeded()
     }
     minecraftVersion = ver;
 
+    if (resetDirectory) {
+        deleteExistingFiles();
+    }
+
     if(m_version.noConfigs) {
         downloadMods();
     }
@@ -143,6 +169,116 @@ void PackInstallTask::onDownloadFailed(QString reason)
     emitFailed(reason);
 }
 
+void PackInstallTask::deleteExistingFiles()
+{
+    setStatus(tr("Deleting existing files..."));
+
+    // Setup defaults, as per https://wiki.atlauncher.com/pack-admin/xml/delete
+    VersionDeletes deletes;
+    deletes.folders.append(VersionDelete{ "root", "mods%s%" });
+    deletes.folders.append(VersionDelete{ "root", "configs%s%" });
+    deletes.folders.append(VersionDelete{ "root", "bin%s%" });
+
+    // Setup defaults, as per https://wiki.atlauncher.com/pack-admin/xml/keep
+    VersionKeeps keeps;
+    keeps.files.append(VersionKeep{ "root", "mods%s%PortalGunSounds.pak" });
+    keeps.folders.append(VersionKeep{ "root", "mods%s%rei_minimap%s%" });
+    keeps.folders.append(VersionKeep{ "root", "mods%s%VoxelMods%s%" });
+    keeps.files.append(VersionKeep{ "root", "config%s%NEI.cfg" });
+    keeps.files.append(VersionKeep{ "root", "options.txt" });
+    keeps.files.append(VersionKeep{ "root", "servers.dat" });
+
+    // Merge with version deletes and keeps
+    for (const auto& item : m_version.deletes.files)
+        deletes.files.append(item);
+    for (const auto& item : m_version.deletes.folders)
+        deletes.folders.append(item);
+    for (const auto& item : m_version.keeps.files)
+        keeps.files.append(item);
+    for (const auto& item : m_version.keeps.folders)
+        keeps.folders.append(item);
+
+    auto getPathForBase = [this](const QString& base) {
+        auto minecraftPath = FS::PathCombine(m_stagingPath, "minecraft");
+
+        if (base == "root") {
+            return minecraftPath;
+        }
+        else if (base == "config") {
+            return FS::PathCombine(minecraftPath, "config");
+        }
+        else {
+            qWarning() << "Unrecognised base path" << base;
+            return minecraftPath;
+        }
+    };
+
+    auto convertToSystemPath = [](const QString& path) {
+        auto t = path;
+        t.replace("%s%", QDir::separator());
+        return t;
+    };
+
+    auto shouldKeep = [keeps, getPathForBase, convertToSystemPath](const QString& fullPath) {
+        for (const auto& item : keeps.files) {
+            auto basePath = getPathForBase(item.base);
+            auto targetPath = convertToSystemPath(item.target);
+            auto path = FS::PathCombine(basePath, targetPath);
+
+            if (fullPath == path) {
+                return true;
+            }
+        }
+
+        for (const auto& item : keeps.folders) {
+            auto basePath = getPathForBase(item.base);
+            auto targetPath = convertToSystemPath(item.target);
+            auto path = FS::PathCombine(basePath, targetPath);
+
+            if (fullPath.startsWith(path)) {
+                return true;
+            }
+        }
+
+        return false;
+    };
+
+    // Keep track of files to delete
+    QSet<QString> filesToDelete;
+
+    for (const auto& item : deletes.files) {
+        auto basePath = getPathForBase(item.base);
+        auto targetPath = convertToSystemPath(item.target);
+        auto fullPath = FS::PathCombine(basePath, targetPath);
+
+        if (shouldKeep(fullPath))
+            continue;
+
+        filesToDelete.insert(fullPath);
+    }
+
+    for (const auto& item : deletes.folders) {
+        auto basePath = getPathForBase(item.base);
+        auto targetPath = convertToSystemPath(item.target);
+        auto fullPath = FS::PathCombine(basePath, targetPath);
+
+        QDirIterator it(fullPath, QDirIterator::Subdirectories);
+        while (it.hasNext()) {
+            auto path = it.next();
+
+            if (shouldKeep(path))
+                continue;
+
+            filesToDelete.insert(path);
+        }
+    }
+
+    // Delete the files
+    for (const auto& item : filesToDelete) {
+        QFile::remove(item);
+    }
+}
+
 QString PackInstallTask::getDirForModType(ModType type, QString raw)
 {
     switch (type) {