aboutsummaryrefslogtreecommitdiff
path: root/libs/installer_fomod_plus/share/FOMODData/FomodRescan.h
diff options
context:
space:
mode:
Diffstat (limited to 'libs/installer_fomod_plus/share/FOMODData/FomodRescan.h')
-rw-r--r--libs/installer_fomod_plus/share/FOMODData/FomodRescan.h277
1 files changed, 0 insertions, 277 deletions
diff --git a/libs/installer_fomod_plus/share/FOMODData/FomodRescan.h b/libs/installer_fomod_plus/share/FOMODData/FomodRescan.h
deleted file mode 100644
index c7d53a2..0000000
--- a/libs/installer_fomod_plus/share/FOMODData/FomodRescan.h
+++ /dev/null
@@ -1,277 +0,0 @@
-#pragma once
-
-#include "ArchiveExtractor.h"
-#include "FomodDB.h"
-#include "stringutil.h"
-#include "xml/ModuleConfiguration.h"
-
-#include <QDir>
-#include <QString>
-#include <functional>
-#include <imodinterface.h>
-#include <imoinfo.h>
-#include <nlohmann/json.hpp>
-
-struct RescanResult {
- int totalModsProcessed = 0;
- int successfullyScanned = 0;
- int missingArchives = 0;
- int parseErrors = 0;
- std::vector<std::string> failedMods;
-};
-
-/**
- * Orchestrates rescanning of all mods with stored FOMOD choices to repopulate the database.
- * Used when fomod.db is missing or needs to be regenerated from existing installations.
- */
-class FomodRescan {
-public:
- using ProgressCallback = std::function<void(int current, int total, const QString& modName)>;
-
- FomodRescan(MOBase::IOrganizer* organizer, FomodDB* db)
- : mOrganizer(organizer), mFomodDb(db) {}
-
- /**
- * Scan all mods that have stored FOMOD Plus choices and repopulate the database.
- * @param progressCallback Optional callback for progress updates
- * @return RescanResult with statistics about the scan
- */
- RescanResult scanAllModsWithChoices(const ProgressCallback& progressCallback = nullptr)
- {
- RescanResult result;
-
- const auto modList = mOrganizer->modList();
- if (!modList) {
- return result;
- }
-
- // First pass: gather all mods with stored choices
- std::vector<MOBase::IModInterface*> modsWithChoices;
- for (const auto& modName : modList->allMods()) {
- auto* mod = modList->getMod(modName);
- if (mod && hasStoredChoices(mod)) {
- modsWithChoices.push_back(mod);
- }
- }
-
- result.totalModsProcessed = static_cast<int>(modsWithChoices.size());
-
- // Second pass: process each mod
- int current = 0;
- for (auto* mod : modsWithChoices) {
- current++;
- if (progressCallback) {
- progressCallback(current, result.totalModsProcessed, mod->name());
- }
-
- const auto scanResult = processMod(mod);
- switch (scanResult) {
- case ScanOutcome::Success:
- result.successfullyScanned++;
- break;
- case ScanOutcome::MissingArchive:
- result.missingArchives++;
- result.failedMods.push_back(mod->name().toStdString() + " (missing archive)");
- break;
- case ScanOutcome::ParseError:
- result.parseErrors++;
- result.failedMods.push_back(mod->name().toStdString() + " (parse error)");
- break;
- case ScanOutcome::NoFomod:
- result.failedMods.push_back(mod->name().toStdString() + " (no FOMOD)");
- break;
- }
- }
-
- // Save the database
- mFomodDb->saveToFile();
-
- return result;
- }
-
-private:
- MOBase::IOrganizer* mOrganizer;
- FomodDB* mFomodDb;
-
- enum class ScanOutcome {
- Success,
- MissingArchive,
- ParseError,
- NoFomod
- };
-
- /**
- * Check if a mod has stored FOMOD Plus choices (non-zero pluginSetting).
- */
- bool hasStoredChoices(MOBase::IModInterface* mod) const
- {
- const auto fomodData = mod->pluginSetting(
- StringConstants::Plugin::NAME.data(), "fomod", 0);
-
- if (!fomodData.isValid() || fomodData.isNull()) {
- return false;
- }
-
- // Check if it's actually valid JSON with steps
- try {
- const auto json = nlohmann::json::parse(fomodData.toString().toStdString());
- return json.contains("steps") && json["steps"].is_array() && !json["steps"].empty();
- } catch (...) {
- return false;
- }
- }
-
- /**
- * Get the stored choices JSON from a mod's pluginSetting.
- */
- nlohmann::json getStoredChoices(MOBase::IModInterface* mod) const
- {
- const auto fomodData = mod->pluginSetting(
- StringConstants::Plugin::NAME.data(), "fomod", 0);
-
- try {
- return nlohmann::json::parse(fomodData.toString().toStdString());
- } catch (...) {
- return nlohmann::json();
- }
- }
-
- /**
- * Process a single mod: extract archive, parse FOMOD, create DB entry with selection states.
- */
- ScanOutcome processMod(MOBase::IModInterface* mod)
- {
- // Get the archive path
- const auto installationFile = mod->installationFile();
- if (installationFile.isEmpty()) {
- return ScanOutcome::MissingArchive;
- }
-
- const auto downloadsPath = mOrganizer->downloadsPath();
- const auto archivePath = QDir(installationFile).isAbsolute()
- ? installationFile
- : downloadsPath + "/" + installationFile;
-
- if (!QFile::exists(archivePath)) {
- return ScanOutcome::MissingArchive;
- }
-
- // Extract FOMOD data from archive
- auto extractionResult = ArchiveExtractor::extractFomodData(archivePath);
- if (!extractionResult.success) {
- return ScanOutcome::ParseError;
- }
-
- // Parse ModuleConfiguration
- auto moduleConfig = std::make_unique<ModuleConfiguration>();
- try {
- if (!moduleConfig->deserialize(extractionResult.moduleConfigPath)) {
- return ScanOutcome::ParseError;
- }
- } catch (...) {
- return ScanOutcome::ParseError;
- }
-
- // Get the mod's Nexus ID
- const int modId = mod->nexusId();
-
- // Create FomodDbEntry using existing logic
- auto entry = FomodDB::getEntryFromFomod(
- moduleConfig.get(),
- extractionResult.pluginPaths,
- modId
- );
-
- if (!entry || entry->getOptions().empty()) {
- return ScanOutcome::NoFomod;
- }
-
- // Apply selection states from stored choices
- const auto choices = getStoredChoices(mod);
- applySelectionsToEntry(*entry, choices);
-
- // Add to database (upsert)
- mFomodDb->addEntry(entry, true);
-
- return ScanOutcome::Success;
- }
-
- /**
- * Apply user selection states to a FomodDbEntry based on stored choices JSON.
- *
- * Choices JSON format:
- * {
- * "steps": [{
- * "name": "Step Name",
- * "groups": [{
- * "name": "Group Name",
- * "plugins": ["Selected Plugin 1"],
- * "deselected": ["Manually Deselected Plugin"]
- * }]
- * }]
- * }
- */
- void applySelectionsToEntry(FomodDbEntry& entry, const nlohmann::json& choices)
- {
- if (!choices.contains("steps") || !choices["steps"].is_array()) {
- // No choices data - mark all as Available
- for (auto& option : entry.getOptionsMutable()) {
- option.selectionState = SelectionState::Available;
- }
- return;
- }
-
- // Build a lookup map for quick matching: stepName/groupName/pluginName -> state
- struct PluginState {
- bool selected = false;
- bool deselected = false;
- };
- std::map<std::string, PluginState> stateMap;
-
- for (const auto& step : choices["steps"]) {
- if (!step.contains("name") || !step.contains("groups")) continue;
- const std::string stepName = step["name"];
-
- for (const auto& group : step["groups"]) {
- if (!group.contains("name")) continue;
- const std::string groupName = group["name"];
-
- // Process selected plugins
- if (group.contains("plugins") && group["plugins"].is_array()) {
- for (const auto& plugin : group["plugins"]) {
- const std::string pluginName = plugin;
- const auto key = stepName + "/" + groupName + "/" + pluginName;
- stateMap[key].selected = true;
- }
- }
-
- // Process deselected plugins
- if (group.contains("deselected") && group["deselected"].is_array()) {
- for (const auto& plugin : group["deselected"]) {
- const std::string pluginName = plugin;
- const auto key = stepName + "/" + groupName + "/" + pluginName;
- stateMap[key].deselected = true;
- }
- }
- }
- }
-
- // Apply states to options
- for (auto& option : entry.getOptionsMutable()) {
- const auto key = option.step + "/" + option.group + "/" + option.name;
-
- if (auto it = stateMap.find(key); it != stateMap.end()) {
- if (it->second.selected) {
- option.selectionState = SelectionState::Selected;
- } else if (it->second.deselected) {
- option.selectionState = SelectionState::Deselected;
- } else {
- option.selectionState = SelectionState::Available;
- }
- } else {
- // Plugin not found in choices - mark as Available
- option.selectionState = SelectionState::Available;
- }
- }
- }
-};
generated by cgit v1.3.1 (git 2.54.0) at 2026-07-21 23:04:27 +0000