#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);
    }

};