Merge pull request #69 from Qudix/stalkeranomaly

STALKER Anomaly: Add ModDataContent

games/game_stalkeranomaly.py

@@ -2,6 +2,7 @@
 
 from pathlib import Path
 from typing import List
+from enum import IntEnum
 
 from PyQt5.QtCore import QDir, QFileInfo
 
@@ -13,24 +14,123 @@ from ..basic_game import BasicGame
 
 class StalkerAnomalyModDataChecker(mobase.ModDataChecker):
     _valid_folders: List[str] = [
-        "db",
         "appdata",
+        "bin",
+        "db",
         "gamedata",
     ]
 
     def __init__(self):
         super().__init__()
 
-    def dataLooksValid(
-        self, tree: mobase.IFileTree
-    ) -> mobase.ModDataChecker.CheckReturn:
+    def hasValidFolders(self, tree: mobase.IFileTree) -> bool:
         for e in tree:
             if e.isDir():
                 if e.name().lower() in self._valid_folders:
-                    return mobase.ModDataChecker.VALID
+                    return True
+
+        return False
+
+    def findLostData(self, tree: mobase.IFileTree) -> List[mobase.FileTreeEntry]:
+        lost_db: List[mobase.FileTreeEntry] = []
+
+        for e in tree:
+            if e.isFile():
+                if e.suffix().lower().startswith("db"):
+                    lost_db.append(e)
+
+        return lost_db
+
+    def dataLooksValid(
+        self, tree: mobase.IFileTree
+    ) -> mobase.ModDataChecker.CheckReturn:
+        if self.hasValidFolders(tree):
+            return mobase.ModDataChecker.VALID
+
+        if self.findLostData(tree):
+            return mobase.ModDataChecker.FIXABLE
 
         return mobase.ModDataChecker.INVALID
 
+    def fix(self, tree: mobase.IFileTree) -> mobase.IFileTree:
+        lost_db = self.findLostData(tree)
+        if lost_db:
+            rfolder = tree.addDirectory("db").addDirectory("mods")
+            for r in lost_db:
+                rfolder.insert(r, mobase.IFileTree.REPLACE)
+
+        return tree
+
+
+class Content(IntEnum):
+    INTERFACE = 0
+    TEXTURE = 1
+    MESH = 2
+    SCRIPT = 3
+    SOUND = 4
+    MCM = 5
+    CONFIG = 6
+
+
+class StalkerAnomalyModDataContent(mobase.ModDataContent):
+    content: List[int] = []
+
+    def __init__(self):
+        super().__init__()
+
+    def getAllContents(self) -> List[mobase.ModDataContent.Content]:
+        return [
+            mobase.ModDataContent.Content(
+                Content.INTERFACE, "Interface", ":/MO/gui/content/interface"
+            ),
+            mobase.ModDataContent.Content(
+                Content.TEXTURE, "Textures", ":/MO/gui/content/texture"
+            ),
+            mobase.ModDataContent.Content(
+                Content.MESH, "Meshes", ":/MO/gui/content/mesh"
+            ),
+            mobase.ModDataContent.Content(
+                Content.SCRIPT, "Scripts", ":/MO/gui/content/script"
+            ),
+            mobase.ModDataContent.Content(
+                Content.SOUND, "Sounds", ":/MO/gui/content/sound"
+            ),
+            mobase.ModDataContent.Content(Content.MCM, "MCM", ":/MO/gui/content/menu"),
+            mobase.ModDataContent.Content(
+                Content.CONFIG, "Configs", ":/MO/gui/content/inifile"
+            ),
+        ]
+
+    def walkContent(
+        self, path: str, entry: mobase.FileTreeEntry
+    ) -> mobase.IFileTree.WalkReturn:
+        name = entry.name().lower()
+        if entry.isFile():
+            ext = entry.suffix().lower()
+            if ext in ["dds", "thm"]:
+                self.content.append(Content.TEXTURE)
+                if path.startswith("gamedata/textures/ui"):
+                    self.content.append(Content.INTERFACE)
+            elif ext in ["omf", "ogf"]:
+                self.content.append(Content.MESH)
+            elif ext in ["script"]:
+                self.content.append(Content.SCRIPT)
+                if "_mcm" in name:
+                    self.content.append(Content.MCM)
+            elif ext in ["ogg"]:
+                self.content.append(Content.SOUND)
+            elif ext in ["ltx", "xml"]:
+                self.content.append(Content.CONFIG)
+                if path.startswith("gamedata/configs/ui"):
+                    self.content.append(Content.INTERFACE)
+
+        return mobase.IFileTree.WalkReturn.CONTINUE
+
+    def getContentsFor(self, tree: mobase.IFileTree) -> List[int]:
+        self.content = []
+        tree.walk(self.walkContent, "/")
+        return self.content
+
 
 class StalkerAnomalySaveGame(BasicGameSaveGame):
     def allFiles(self) -> List[str]:
@@ -45,7 +145,7 @@ class StalkerAnomalySaveGame(BasicGameSaveGame):
 class StalkerAnomalyGame(BasicGame, mobase.IPluginFileMapper):
     Name = "STALKER Anomaly"
     Author = "Qudix"
-    Version = "0.3.1"
+    Version = "0.4.0"
     Description = "Adds support for STALKER Anomaly"
 
     GameName = "STALKER Anomaly"
@@ -63,6 +163,7 @@ class StalkerAnomalyGame(BasicGame, mobase.IPluginFileMapper):
     def init(self, organizer: mobase.IOrganizer):
         BasicGame.init(self, organizer)
         self._featureMap[mobase.ModDataChecker] = StalkerAnomalyModDataChecker()
+        self._featureMap[mobase.ModDataContent] = StalkerAnomalyModDataContent()
         return True
 
     def executables(self):