1 files changed, 55 insertions, 0 deletions
diff --git a/libs/game_bethesda/src/games/starfield/starfieldmoddatacontent.h b/libs/game_bethesda/src/games/starfield/starfieldmoddatacontent.h
new file mode 100644
index 0000000..d4b8014
--- /dev/null
+++ b/libs/game_bethesda/src/games/starfield/starfieldmoddatacontent.h
@@ -0,0 +1,55 @@
+#ifndef STARFIELD_MODDATACONTENT_H
+#define STARFIELD_MODDATACONTENT_H
+
+#include <gamebryomoddatacontent.h>
+#include <ifiletree.h>
+
+class StarfieldModDataContent : public GamebryoModDataContent
+{
+protected:
+  enum StarfieldContent
+  {
+    CONTENT_MATERIAL   = CONTENT_NEXT_VALUE,
+    CONTENT_GEOMETRIES = CONTENT_NEXT_VALUE + 1,
+    CONTENT_VIDEO      = CONTENT_NEXT_VALUE + 2
+  };
+
+public:
+  StarfieldModDataContent(MOBase::IGameFeatures const* gameFeatures)
+      : GamebryoModDataContent(gameFeatures)
+  {
+    m_Enabled[CONTENT_SKYPROC] = false;
+  }
+
+  std::vector<Content> getAllContents() const override
+  {
+    static std::vector<Content> STARFIELD_CONTENTS{
+        {CONTENT_MATERIAL, QT_TR_NOOP("Materials"), ":/MO/gui/content/material"},
+        {CONTENT_GEOMETRIES, QT_TR_NOOP("Geometries"), ":/MO/gui/content/geometries"},
+        {CONTENT_VIDEO, QT_TR_NOOP("Video"), ":/MO/gui/content/media"}};
+    auto contents = GamebryoModDataContent::getAllContents();
+    std::copy(std::begin(STARFIELD_CONTENTS), std::end(STARFIELD_CONTENTS),
+              std::back_inserter(contents));
+    return contents;
+  }
+
+  std::vector<int>
+  getContentsFor(std::shared_ptr<const MOBase::IFileTree> fileTree) const override
+  {
+    auto contents = GamebryoModDataContent::getContentsFor(fileTree);
+    for (auto e : *fileTree) {
+      if (e->isDir()) {
+        if (e->compare("materials") == 0) {
+          contents.push_back(CONTENT_MATERIAL);
+        } else if (e->compare("geometries") == 0) {
+          contents.push_back(CONTENT_GEOMETRIES);
+        } else if (e->compare("video") == 0) {
+          contents.push_back(CONTENT_VIDEO);
+        }
+      }
+    }
+    return contents;
+  }
+};
+
+#endif  // STARFIELD_MODDATACONTENT_H