add bg3_data_content.py

Author: d.schlatter

games/baldursgate3/bg3_data_content.py

@@ -0,0 +1,53 @@
+from enum import IntEnum, auto
+
+import mobase
+
+
+class Content(IntEnum):
+    PAK = auto()
+    WORKSPACE = auto()
+    NATIVE = auto()
+    LOOSE_FILES = auto()
+    SE_FILES = auto()
+
+
+class BG3DataContent(mobase.ModDataContent):
+    BG3_CONTENTS: list[tuple[Content, str, str, bool] | tuple[Content, str, str]] = [
+        (Content.WORKSPACE, "Mod workspaces", ":/MO/gui/content/plugin"),
+        (Content.PAK, "Paks", ":/MO/gui/content/geometries"),
+        (Content.LOOSE_FILES, "Loose file override mods", ":/MO/gui/content/skse"),
+        (Content.SE_FILES, "Script Extender Files", "", True),
+        (Content.NATIVE, "Native DLL mods", ":/MO/gui/content/script"),
+    ]
+
+    def getAllContents(self) -> list[mobase.ModDataContent.Content]:
+        return [
+            mobase.ModDataContent.Content(id, name, icon, *filter_only)
+            for id, name, icon, *filter_only in self.BG3_CONTENTS
+        ]
+
+    def getContentsFor(self, filetree: mobase.IFileTree) -> list[int]:
+        contents: set[int] = set()
+        for entry in filetree:
+            if isinstance(entry, mobase.IFileTree):
+                match entry.name().casefold():
+                    case "Script Extender":
+                        contents.add(Content.SE_FILES)
+                    case "Data":
+                        contents.add(Content.LOOSE_FILES)
+                    case x if x.endswith(".pak"):
+                        contents.add(Content.PAK)
+                    case "Mods":
+                        for e in entry:
+                            if e.name().endswith(".pak"):
+                                contents.add(Content.PAK)
+                                break
+                    case "bin":
+                        contents.add(Content.NATIVE)
+                    case _:
+                        for e in entry:
+                            match e.name().casefold():
+                                case "Mods" | "Localization" | "ScriptExtender" | "Public" | "Generated":
+                                    contents.add(Content.WORKSPACE)
+                                    break
+        return list(contents)

games/game_baldursgate3.py

@@ -53,7 +53,7 @@ def __init__(self):
 
     def init(self, organizer: mobase.IOrganizer) -> bool:
         super().init(organizer)
-        from baldursgate3 import bg3_data_checker, bg3_utils
+        from baldursgate3 import bg3_data_checker, bg3_utils, bg3_data_content
 
         self._utils = bg3_utils.BG3Utils(organizer, self.name())
         self._register_feature(
@@ -77,6 +77,7 @@ def init(self, organizer: mobase.IOrganizer) -> bool:
                 )
             )
         )
+        self._register_feature(bg3_data_content.BG3DataContent())
         self._register_feature(BasicGameSaveGameInfo(lambda s: s.with_suffix(".webp")))
         self._register_feature(BasicLocalSavegames(self.savesDirectory()))
         organizer.onAboutToRun(self._utils.construct_modsettings_xml)