1 files changed, 75 insertions, 0 deletions
diff --git a/libs/game_gamebryo/src/gamebryo/gamebryomoddatacontent.h b/libs/game_gamebryo/src/gamebryo/gamebryomoddatacontent.h
new file mode 100644
index 0000000..be02331
--- /dev/null
+++ b/libs/game_gamebryo/src/gamebryo/gamebryomoddatacontent.h
@@ -0,0 +1,75 @@
+#ifndef GAMEBRYO_MODDATACONTENT_H
+#define GAMEBRYO_MODDATACONTENT_H
+
+#include <ifiletree.h>
+#include <moddatacontent.h>
+
+namespace MOBase
+{
+class IGameFeatures;
+}
+
+/**
+ * @brief ModDataContent for GameBryo games.
+ *
+ */
+class GamebryoModDataContent : public MOBase::ModDataContent
+{
+protected:
+  /**
+   * Note: These are used to index m_Enabled so should have standard
+   * enum values, not custom ones.
+   */
+  enum EContent
+  {
+    CONTENT_PLUGIN,
+    CONTENT_OPTIONAL,
+    CONTENT_TEXTURE,
+    CONTENT_MESH,
+    CONTENT_BSA,
+    CONTENT_INTERFACE,
+    CONTENT_SOUND,
+    CONTENT_SCRIPT,
+    CONTENT_SKSE,
+    CONTENT_SKSE_FILES,
+    CONTENT_SKYPROC,
+    CONTENT_MCM,
+    CONTENT_INI,
+    CONTENT_FACEGEN,
+    CONTENT_MODGROUP
+  };
+
+  /**
+   * This is the first value that can be used for game-specific contents.
+   */
+  constexpr static auto CONTENT_NEXT_VALUE = CONTENT_MODGROUP + 1;
+
+public:
+  /**
+   *
+   */
+  GamebryoModDataContent(const MOBase::IGameFeatures* gameFeatures);
+
+  /**
+   * @return the list of all possible contents for the corresponding game.
+   */
+  virtual std::vector<Content> getAllContents() const override;
+
+  /**
+   * @brief Retrieve the list of contents in the given tree.
+   *
+   * @param fileTree The tree corresponding to the mod to retrieve contents for.
+   *
+   * @return the IDs of the content in the given tree.
+   */
+  virtual std::vector<int>
+  getContentsFor(std::shared_ptr<const MOBase::IFileTree> fileTree) const override;
+
+protected:
+  MOBase::IGameFeatures const* const m_GameFeatures;
+
+  // List of enabled contents:
+  std::vector<bool> m_Enabled;
+};
+
+#endif  // GAMEBRYO_MODDATACONTENT_H