1 files changed, 177 insertions, 0 deletions
diff --git a/libs/installer_bsplugins/src/TESData/FileInfo.cpp b/libs/installer_bsplugins/src/TESData/FileInfo.cpp
new file mode 100644
index 0000000..8429a11
--- /dev/null
+++ b/libs/installer_bsplugins/src/TESData/FileInfo.cpp
@@ -0,0 +1,177 @@
+#include "FileInfo.h"
+#include "FileEntry.h"
+#include "MOPlugin/Settings.h"
+#include "PluginList.h"
+
+#include <algorithm>
+
+using namespace Qt::Literals::StringLiterals;
+
+namespace TESData
+{
+
+FileInfo::FileInfo(PluginList* pluginList, const QString& name, bool forceLoaded,
+                   bool forceEnabled, bool forceDisabled, bool lightSupported)
+    : m_PluginList{pluginList},
+      m_FileSystemData{
+          .name               = name,
+          .hasMasterExtension = name.endsWith(u".esm"_s, Qt::CaseInsensitive),
+          .hasLightExtension =
+              lightSupported && name.endsWith(u".esl"_s, Qt::CaseInsensitive),
+          .forceLoaded   = forceLoaded,
+          .forceEnabled  = forceEnabled,
+          .forceDisabled = forceDisabled,
+      },
+      m_Conflicts{[this]() {
+        return doConflictCheck();
+      }}
+{}
+
+bool FileInfo::isMasterFile() const
+{
+  return m_Metadata.isMasterFlagged || m_FileSystemData.hasMasterExtension ||
+         m_FileSystemData.hasLightExtension;
+}
+
+bool FileInfo::isMediumFile() const
+{
+  return m_Metadata.isMediumFlagged;
+}
+
+bool FileInfo::isSmallFile() const
+{
+  return m_Metadata.isLightFlagged || m_FileSystemData.hasLightExtension;
+}
+
+bool FileInfo::isBlueprintFile() const
+{
+  return isMasterFile() && m_Metadata.isBlueprintFlagged;
+}
+
+bool FileInfo::isAlwaysEnabled() const
+{
+  return m_FileSystemData.forceLoaded || m_FileSystemData.forceEnabled;
+}
+
+bool FileInfo::canBeToggled() const
+{
+  return !m_FileSystemData.forceLoaded && !m_FileSystemData.forceEnabled &&
+         !m_FileSystemData.forceDisabled;
+}
+
+bool FileInfo::mustLoadAfter(const FileInfo& other) const
+{
+  if (this->isBlueprintFile() && !other.isBlueprintFile()) {
+    return true;
+  } else if (other.isBlueprintFile() && !this->isBlueprintFile()) {
+    return false;
+  }
+
+  const bool hasMaster = this->masters().contains(other.name(), Qt::CaseInsensitive);
+  const bool isMaster  = other.masters().contains(this->name(), Qt::CaseInsensitive);
+
+  if (hasMaster && !isMaster) {
+    return true;
+  } else if (isMaster) {
+    return false;
+  }
+
+  if (other.forceLoaded() && !this->forceLoaded()) {
+    return true;
+  }
+
+  if (other.isMasterFile() && !this->isMasterFile()) {
+    return true;
+  }
+
+  return false;
+}
+
+static void checkConflict(QSet<int>& winning, QSet<int>& losing, const FileInfo& file,
+                          const TESData::PluginList* pluginList,
+                          TESFileHandle alternative, bool ignoreMasters)
+{
+  const auto entry      = pluginList->findEntryByName(file.name().toStdString());
+  const auto otherEntry = pluginList->findEntryByHandle(alternative);
+  if (otherEntry == nullptr || otherEntry == entry) {
+    return;
+  }
+
+  const auto otherFile =
+      pluginList->getPluginByName(QString::fromStdString(otherEntry->name()));
+  if (otherFile == nullptr) {
+    return;
+  }
+
+  const QString otherName = QString::fromStdString(otherEntry->name());
+  const int otherIndex    = pluginList->getIndex(otherName);
+
+  if (file.priority() > otherFile->priority()) {
+    if (!ignoreMasters ||
+        !file.masters().contains(otherFile->name(), Qt::CaseInsensitive)) {
+      winning.insert(otherIndex);
+    }
+  } else {
+    if (!ignoreMasters ||
+        !otherFile->masters().contains(file.name(), Qt::CaseInsensitive)) {
+      losing.insert(otherIndex);
+    }
+  }
+}
+
+FileInfo::Conflicts FileInfo::doConflictCheck() const
+{
+  Conflicts conflicts;
+
+  const auto entry = m_PluginList->findEntryByName(name().toStdString());
+  if (entry == nullptr) {
+    return conflicts;
+  }
+
+  const bool ignoreMasters =
+      Settings::instance()->get<bool>("ignore_master_conflicts", false);
+
+  entry->forEachRecord([&](auto&& record) {
+    if (record->ignored())
+      return;
+
+    for (const auto alternative : record->alternatives()) {
+      checkConflict(conflicts.m_OverridingList, conflicts.m_OverriddenList, *this,
+                    m_PluginList, alternative, ignoreMasters);
+    }
+  });
+
+  for (const auto& archive : m_FileSystemData.archives) {
+    const auto archiveEntry = m_PluginList->findArchive(archive);
+    if (!archiveEntry) {
+      continue;
+    }
+
+    archiveEntry->forEachMember([&](auto&& item) {
+      for (const auto alternative : item->alternatives) {
+        checkConflict(conflicts.m_OverwritingArchiveList,
+                      conflicts.m_OverwrittenArchiveList, *this, m_PluginList,
+                      alternative, ignoreMasters);
+      }
+    });
+  }
+
+  uint conflictState = CONFLICT_NONE;
+  if (!conflicts.m_OverridingList.empty()) {
+    conflictState |= CONFLICT_OVERRIDE;
+  }
+  if (!conflicts.m_OverriddenList.empty()) {
+    conflictState |= CONFLICT_OVERRIDDEN;
+  }
+  if (!conflicts.m_OverwritingArchiveList.empty()) {
+    conflictState |= CONFLICT_ARCHIVE_OVERWRITE;
+  }
+  if (!conflicts.m_OverwrittenArchiveList.empty()) {
+    conflictState |= CONFLICT_ARCHIVE_OVERWRITTEN;
+  }
+  conflicts.m_CurrentConflictState = static_cast<EConflictFlag>(conflictState);
+
+  return conflicts;
+}
+
+}  // namespace TESData