1 files changed, 76 insertions, 0 deletions
diff --git a/libs/game_gamebryo/src/gamebryo/gamebryomoddatachecker.cpp b/libs/game_gamebryo/src/gamebryo/gamebryomoddatachecker.cpp
new file mode 100644
index 0000000..b5bace6
--- /dev/null
+++ b/libs/game_gamebryo/src/gamebryo/gamebryomoddatachecker.cpp
@@ -0,0 +1,76 @@
+#include <ifiletree.h>
+
+#include "gamebryomoddatachecker.h"
+
+/**
+ * @return the list of possible folder names in data.
+ */
+auto GamebryoModDataChecker::possibleFolderNames() const -> const FileNameSet&
+{
+  static FileNameSet result{"fonts",
+                            "interface",
+                            "menus",
+                            "meshes",
+                            "music",
+                            "scripts",
+                            "shaders",
+                            "sound",
+                            "strings",
+                            "textures",
+                            "trees",
+                            "video",
+                            "facegen",
+                            "materials",
+                            "skse",
+                            "obse",
+                            "mwse",
+                            "nvse",
+                            "fose",
+                            "f4se",
+                            "distantlod",
+                            "asi",
+                            "SkyProc Patchers",
+                            "Tools",
+                            "MCM",
+                            "icons",
+                            "bookart",
+                            "distantland",
+                            "mits",
+                            "splash",
+                            "dllplugins",
+                            "CalienteTools",
+                            "NetScriptFramework",
+                            "shadersfx"};
+  return result;
+}
+
+/**
+ * @return the extensions of possible files in data.
+ */
+auto GamebryoModDataChecker::possibleFileExtensions() const -> const FileNameSet&
+{
+  static FileNameSet result{"esp", "esm", "esl", "bsa", "ba2", "modgroups", "ini"};
+  return result;
+}
+
+GamebryoModDataChecker::GamebryoModDataChecker(const GameGamebryo* game) : m_Game(game)
+{}
+
+GamebryoModDataChecker::CheckReturn GamebryoModDataChecker::dataLooksValid(
+    std::shared_ptr<const MOBase::IFileTree> fileTree) const
+{
+  auto& folders  = possibleFolderNames();
+  auto& suffixes = possibleFileExtensions();
+  for (auto entry : *fileTree) {
+    if (entry->isDir()) {
+      if (folders.count(entry->name()) > 0) {
+        return CheckReturn::VALID;
+      }
+    } else {
+      if (suffixes.count(entry->suffix()) > 0) {
+        return CheckReturn::VALID;
+      }
+    }
+  }
+  return CheckReturn::INVALID;
+}