Add Kingdom Come: Deliverance II Plugin

games/game_kingdomcomedeliverance2.py

@@ -0,0 +1,104 @@
+import os
+import xml.etree.ElementTree as ET
+from pathlib import Path
+
+from PyQt6.QtCore import QDir, qInfo, qWarning
+
+import mobase
+
+from ..basic_features import BasicModDataChecker, GlobPatterns
+from ..basic_game import BasicGame
+
+
+class KingdomComeDeliverance2Game(BasicGame):
+    Name = "Kingdom Come: Deliverance 2 Support Plugin"
+    Author = "TheForgotten69"
+    Version = "1.0.0"
+
+    GameName = "Kingdom Come: Deliverance II"
+    GameShortName = "kingdomcomedeliverance2"
+    GameNexusName = "kingdomcomedeliverance2"
+    GameNexusId = 7286
+    GameSteamId = [1771300]
+    GameBinary = "bin/Win64MasterMasterSteamPGO/KingdomCome.exe"
+    GameDataPath = "Mods"
+    GameSaveExtension = "whs"
+    GameDocumentsDirectory = "%GAME_PATH%"
+    GameSavesDirectory = "%USERPROFILE%/Saved Games/kingdomcome2/saves"
+
+    def init(self, organizer: mobase.IOrganizer) -> bool:
+        super().init(organizer)
+        self._register_feature(BasicModDataChecker(GlobPatterns(valid=["*"])))
+        organizer.onAboutToRun(self._write_mod_order)
+        return True
+
+    @staticmethod
+    def _get_mod_id(mod_path: Path) -> str | None:
+        """Return the mod ID for a KCD2 mod.
+
+        Checks mod.manifest for <modid> (preferred) or <name>, falling back
+        to the game mod folder name (the subfolder inside the MO2 mod directory).
+        """
+        for manifest in mod_path.rglob("mod.manifest"):
+            try:
+                root = ET.parse(manifest).getroot()
+                for tag in ("modid", "name"):
+                    elem = root.find(f".//{tag}")
+                    if elem is not None and elem.text and elem.text.strip():
+                        return elem.text.strip()
+            except ET.ParseError:
+                qWarning(f"KCD2: failed to parse {manifest}")
+            # Fall back to the folder that contains mod.manifest
+            return manifest.parent.name
+        # No manifest — use first subdirectory name (the game mod folder)
+        for subdir in mod_path.iterdir():
+            if subdir.is_dir():
+                return subdir.name
+        return None
+
+    def _write_mod_order(self, app_path: str, wd: QDir, args: str) -> bool:
+        if not self.isActive():
+            return True
+
+        modlist = self._organizer.modList()
+        mods_path = Path(self._organizer.modsPath())
+        mod_order_path = Path(self._organizer.overwritePath()) / "mod_order.txt"
+
+        mod_ids: list[str] = []
+        for mod_name in modlist.allModsByProfilePriority():
+            if not (modlist.state(mod_name) & mobase.ModState.ACTIVE):
+                continue
+            mod_id = self._get_mod_id(mods_path / mod_name)
+            if mod_id:
+                qInfo(f"KCD2: mod '{mod_name}' -> id '{mod_id}'")
+                mod_ids.append(mod_id)
+            else:
+                qWarning(f"KCD2: could not resolve id for mod '{mod_name}', skipping")
+
+        if not mod_ids:
+            mod_order_path.unlink(missing_ok=True)
+            return True
+
+        # MO2 priority 1 = top = highest priority. KCD2 is last-loaded-wins,
+        # so highest priority must be last in the file.
+        mod_ids.reverse()
+        mod_order_path.parent.mkdir(parents=True, exist_ok=True)
+        mod_order_path.write_text("\n".join(mod_ids))
+        qInfo(f"KCD2: wrote mod_order.txt with {len(mod_ids)} mods: {mod_ids}")
+        return True
+
+    def iniFiles(self):
+        return ["custom.cfg", "system.cfg", "user.cfg"]
+
+    def initializeProfile(self, directory: QDir, settings: mobase.ProfileSetting):
+        for iniFile in self.iniFiles():
+            iniPath = self.documentsDirectory().absoluteFilePath(iniFile)
+            if not os.path.exists(iniPath):
+                with open(iniPath, "w") as _:
+                    pass
+
+        modsPath = self.dataDirectory().absolutePath()
+        if not os.path.exists(modsPath):
+            os.mkdir(modsPath)
+
+        super().initializeProfile(directory, settings)