1 files changed, 128 insertions, 0 deletions
diff --git a/libs/installer_fomod_plus/scanner/FomodPlusScanner.cpp b/libs/installer_fomod_plus/scanner/FomodPlusScanner.cpp
new file mode 100644
index 0000000..4770178
--- /dev/null
+++ b/libs/installer_fomod_plus/scanner/FomodPlusScanner.cpp
@@ -0,0 +1,128 @@
+#include "FomodPlusScanner.h"
+
+#include "archiveparser.h"
+#include "FomodDBEntry.h"
+
+#include <QDialog>
+#include <QPushButton>
+#include <QProgressBar>
+#include <QVBoxLayout>
+#include <QMessageBox>
+#include <archive.h>
+
+#include <QLabel>
+#include <QMovie>
+#include <iostream>
+
+#include "stringutil.h"
+
+using ScanCallbackFn = std::function<bool(IModInterface*, ScanResult result)>;
+
+bool FomodPlusScanner::init(IOrganizer* organizer)
+{
+    mOrganizer = organizer;
+
+    mDialog = new QDialog();
+    mDialog->setWindowTitle(tr("FOMOD Scanner"));
+
+    const auto layout = new QVBoxLayout(mDialog);
+
+    const QString description = tr("Greetings, traveler.\n"
+        "This tool will scan your load order for mods installed via FOMOD.\n"
+        "It will also fix up any erroneous FOMOD flags from previous versions of FOMOD Plus :) \n\n"
+        "Safe travels, and may your load order be free of conflicts.");
+
+    const auto descriptionLabel = new QLabel(description, mDialog);
+    descriptionLabel->setWordWrap(true); // Enable word wrap for large text
+    descriptionLabel->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
+    descriptionLabel->setAlignment(Qt::AlignLeft | Qt::AlignTop); // Align text to the top-left corner
+
+    const auto gifLabel = new QLabel(mDialog);
+    const auto movie    = new QMovie(":/fomod/wizard");
+    gifLabel->setMovie(movie);
+    movie->start();
+
+    mProgressBar = new QProgressBar(mDialog);
+    mProgressBar->setRange(0, mOrganizer->modList()->allMods().size());
+    mProgressBar->setVisible(false);
+
+    const auto scanButton   = new QPushButton(tr("Scan"), mDialog);
+    const auto cancelButton = new QPushButton(tr("Cancel"), mDialog);
+
+    layout->addWidget(descriptionLabel, 1);
+    layout->addWidget(gifLabel, 1);
+    layout->addWidget(mProgressBar, 1);
+    layout->addWidget(scanButton, 1);
+    layout->addWidget(cancelButton, 1);
+
+    connect(cancelButton, &QPushButton::clicked, mDialog, &QDialog::reject);
+    connect(scanButton, &QPushButton::clicked, this, &FomodPlusScanner::onScanClicked);
+    connect(mDialog, &QDialog::finished, this, &FomodPlusScanner::cleanup);
+
+    mDialog->setLayout(layout);
+    mDialog->setMinimumSize(400, 300);
+    mDialog->adjustSize();
+    descriptionLabel->adjustSize();
+    return true;
+}
+
+void FomodPlusScanner::onScanClicked() const
+{
+    mProgressBar->setVisible(true);
+    const int added = scanLoadOrder(setFomodInfoForMod);
+    mDialog->accept();
+    QMessageBox::information(mDialog, tr("Scan Complete"),
+        tr("The load order scan is complete. Updated filter info for ") + QString::number(added) + tr(" mods."));
+    mOrganizer->refresh();
+}
+
+void FomodPlusScanner::cleanup() const
+{
+    mProgressBar->reset();
+    mProgressBar->setVisible(false);
+}
+
+void FomodPlusScanner::display() const
+{
+    mDialog->exec();
+}
+
+int FomodPlusScanner::scanLoadOrder(const ScanCallbackFn& callback) const
+{
+    int progress = 0;
+    int modified = 0;
+    for (const auto& modName : mOrganizer->modList()->allMods()) {
+        const auto mod          = mOrganizer->modList()->getMod(modName);
+        if (const ScanResult result = openInstallationArchive(mod); callback(mod, result)) {
+            modified++;
+        }
+        mProgressBar->setValue(++progress);
+    }
+    return modified;
+}
+
+ScanResult FomodPlusScanner::openInstallationArchive(const IModInterface* mod) const
+{
+    const auto downloadsDir         = mOrganizer->downloadsPath();
+    const auto installationFilePath = mod->installationFile();
+    return ArchiveParser::scanForFomodFiles(downloadsDir, installationFilePath, mod->name());
+}
+
+bool FomodPlusScanner::setFomodInfoForMod(IModInterface* mod, const ScanResult result)
+{
+    const auto pluginName = "FOMOD Plus";
+    const auto setting    = mod->pluginSetting(pluginName, "fomod", 0);
+    if (setting == 0 && ScanResult::HAS_FOMOD == result) {
+        return mod->setPluginSetting(pluginName, "fomod", "{}");
+    }
+    if (setting != 0 && ScanResult::NO_FOMOD == result) {
+        return mod->setPluginSetting(pluginName, "fomod", 0);
+    }
+    return false;
+}
+
+bool FomodPlusScanner::removeFomodInfoFromMod(IModInterface* mod, ScanResult)
+{
+    const auto pluginName = QString::fromStdString("FOMOD Plus");
+    return mod->setPluginSetting(pluginName, "fomod", 0);
+}