1 files changed, 116 insertions, 0 deletions
diff --git a/libs/game_bethesda/src/gamebryo/gamebryomoddatacontent.cpp b/libs/game_bethesda/src/gamebryo/gamebryomoddatacontent.cpp
new file mode 100644
index 0000000..85b66b0
--- /dev/null
+++ b/libs/game_bethesda/src/gamebryo/gamebryomoddatacontent.cpp
@@ -0,0 +1,116 @@
+#include "gamebryomoddatacontent.h"
+
+#include <igamefeatures.h>
+#include <scriptextender.h>
+
+GamebryoModDataContent::GamebryoModDataContent(
+    MOBase::IGameFeatures const* gameFeatures)
+    : m_GameFeatures(gameFeatures), m_Enabled(CONTENT_MODGROUP + 1, true)
+{}
+
+std::vector<GamebryoModDataContent::Content>
+GamebryoModDataContent::getAllContents() const
+{
+  static std::vector<Content> GAMEBRYO_CONTENTS{
+      {CONTENT_PLUGIN, QT_TR_NOOP("Plugins (ESP/ESM/ESL)"), ":/MO/gui/content/plugin"},
+      {CONTENT_OPTIONAL, QT_TR_NOOP("Optional Plugins"), "", true},
+      {CONTENT_INTERFACE, QT_TR_NOOP("Interface"), ":/MO/gui/content/interface"},
+      {CONTENT_MESH, QT_TR_NOOP("Meshes"), ":/MO/gui/content/mesh"},
+      {CONTENT_BSA, QT_TR_NOOP("Bethesda Archive"), ":/MO/gui/content/bsa"},
+      {CONTENT_SCRIPT, QT_TR_NOOP("Scripts (Papyrus)"), ":/MO/gui/content/script"},
+      {CONTENT_SKSE, QT_TR_NOOP("Script Extender Plugin"), ":/MO/gui/content/skse"},
+      {CONTENT_SKSE_FILES, QT_TR_NOOP("Script Extender Files"), "", true},
+      {CONTENT_SKYPROC, QT_TR_NOOP("SkyProc Patcher"), ":/MO/gui/content/skyproc"},
+      {CONTENT_SOUND, QT_TR_NOOP("Sound or Music"), ":/MO/gui/content/sound"},
+      {CONTENT_TEXTURE, QT_TR_NOOP("Textures"), ":/MO/gui/content/texture"},
+      {CONTENT_MCM, QT_TR_NOOP("MCM Configuration"), ":/MO/gui/content/menu"},
+      {CONTENT_INI, QT_TR_NOOP("INI Files"), ":/MO/gui/content/inifile"},
+      {CONTENT_FACEGEN, QT_TR_NOOP("FaceGen Data"), ":/MO/gui/content/facegen"},
+      {CONTENT_MODGROUP, QT_TR_NOOP("ModGroup Files"), ":/MO/gui/content/modgroup"}};
+
+  // Copy the list of enabled contents:
+  std::vector<Content> contents;
+  std::copy_if(std::begin(GAMEBRYO_CONTENTS), std::end(GAMEBRYO_CONTENTS),
+               std::back_inserter(contents), [this](auto e) {
+                 return m_Enabled[e.id()];
+               });
+  return contents;
+}
+
+std::vector<int> GamebryoModDataContent::getContentsFor(
+    std::shared_ptr<const MOBase::IFileTree> fileTree) const
+{
+  std::vector<int> contents;
+
+  for (auto e : *fileTree) {
+    if (e->isFile()) {
+      auto suffix = e->suffix().toLower();
+      if (m_Enabled[CONTENT_PLUGIN] &&
+          (suffix == "esp" || suffix == "esm" || suffix == "esl")) {
+        contents.push_back(CONTENT_PLUGIN);
+      } else if (m_Enabled[CONTENT_BSA] && (suffix == "bsa" || suffix == "ba2")) {
+        contents.push_back(CONTENT_BSA);
+      } else if (m_Enabled[CONTENT_INI] && suffix == "ini" &&
+                 e->compare("meta.ini") != 0) {
+        contents.push_back(CONTENT_INI);
+      } else if (m_Enabled[CONTENT_MODGROUP] && suffix == "modgroups") {
+        contents.push_back(CONTENT_MODGROUP);
+      }
+    } else {
+      if (m_Enabled[CONTENT_TEXTURE] &&
+          (e->compare("textures") == 0 || e->compare("icons") == 0 ||
+           e->compare("bookart") == 0)) {
+        contents.push_back(CONTENT_TEXTURE);
+      } else if (m_Enabled[CONTENT_MESH] && e->compare("meshes") == 0) {
+        contents.push_back(CONTENT_MESH);
+      } else if (m_Enabled[CONTENT_INTERFACE] &&
+                 (e->compare("interface") == 0 || e->compare("menus") == 0)) {
+        contents.push_back(CONTENT_INTERFACE);
+      } else if (m_Enabled[CONTENT_SOUND] && e->compare("music") == 0 ||
+                 e->compare("sound") == 0) {
+        contents.push_back(CONTENT_SOUND);
+      } else if (m_Enabled[CONTENT_SCRIPT] && e->compare("scripts") == 0) {
+        contents.push_back(CONTENT_SCRIPT);
+      } else if (m_Enabled[CONTENT_SKYPROC] && e->compare("SkyProc Patchers") == 0) {
+        contents.push_back(CONTENT_SKYPROC);
+      } else if (m_Enabled[CONTENT_MCM] && e->compare("MCM") == 0) {
+        contents.push_back(CONTENT_MCM);
+      } else if (m_Enabled[CONTENT_OPTIONAL] && e->compare("Optional") == 0 &&
+                 e->astree()->size() > 0) {
+        contents.push_back(CONTENT_OPTIONAL);
+      }
+    }
+  }
+
+  if (m_Enabled[CONTENT_FACEGEN]) {
+    auto e1 = fileTree->findDirectory("meshes/actors/character/facegendata");
+    if (e1) {
+      contents.push_back(CONTENT_FACEGEN);
+    } else {
+      auto e2 = fileTree->findDirectory("textures/actors/character/facegendata");
+      if (e2) {
+        contents.push_back(CONTENT_FACEGEN);
+      }
+    }
+  }
+
+  auto extender = m_GameFeatures->gameFeature<MOBase::ScriptExtender>();
+  if (extender != nullptr) {
+    auto e = fileTree->findDirectory(extender->PluginPath());
+    if (e) {
+      if (m_Enabled[CONTENT_SKSE_FILES]) {
+        contents.push_back(CONTENT_SKSE_FILES);
+      }
+      if (m_Enabled[CONTENT_SKSE]) {
+        for (auto f : *e) {
+          if (f->hasSuffix("dll")) {
+            contents.push_back(CONTENT_SKSE);
+            break;
+          }
+        }
+      }
+    }
+  }
+
+  return contents;
+}