1 files changed, 49 insertions, 0 deletions
diff --git a/libs/installer_fomod_plus/patchwizard/lib/PatchFinder.h b/libs/installer_fomod_plus/patchwizard/lib/PatchFinder.h
new file mode 100644
index 0000000..02327e3
--- /dev/null
+++ b/libs/installer_fomod_plus/patchwizard/lib/PatchFinder.h
@@ -0,0 +1,49 @@
+#pragma once
+
+#include "../../installer/lib/Logger.h"
+
+#include <FomodDB.h>
+#include <imodinterface.h>
+#include <imoinfo.h>
+#include <ifiletree.h>
+
+struct AvailablePatch {
+    FomodOption fomod_option;
+    std::string installer_name;
+    std::string patch_for_mod;
+    bool installed = false;
+    bool hidden    = false;
+    bool userDeselected = false;  // True if user explicitly chose not to install
+};
+
+class PatchFinder {
+friend class FomodPlusPatchWizard;
+
+public:
+    explicit PatchFinder(MOBase::IOrganizer* m_organizer) : m_organizer(m_organizer)
+    {
+        mFomodDb = std::make_unique<FomodDB>(m_organizer->basePath().toStdString());
+        logMessage(DEBUG, "mFomodDb loaded.");
+    }
+
+    std::vector<AvailablePatch> getAvailablePatchesForMod(const MOBase::IModInterface* mod);
+    std::vector<AvailablePatch> getAvailablePatchesForModList();
+
+protected:
+    void populateInstalledPlugins();
+
+private:
+    Logger& log = Logger::getInstance();
+    MOBase::IOrganizer* m_organizer;
+    std::unique_ptr<FomodDB> mFomodDb;
+
+    // Map of { pluginPtr: [1.esp, 2.esp, 3.esp] }
+    std::unordered_map<const MOBase::IModInterface*, std::vector<std::string> > m_installedPlugins;
+    std::unordered_set<std::string> m_installedPluginsCacheSet;
+
+    void logMessage(const LogLevel level, const std::string& message) const
+    {
+        log.logMessage(level, "[PATCHFINDER] " + message);
+    }
+
+};
+\ No newline at end of file