From 1dde573df01a761f6252f506992b9ec6895e0208 Mon Sep 17 00:00:00 2001
From: Nigel McKernan <nmck160@gmail.com>
Date: Thu, 30 Nov 2023 09:26:11 -0800
Subject: [PATCH] Add The Witcher 2 support.

---
 games/game_witcher2.py | 67 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 games/game_witcher2.py

diff --git a/games/game_witcher2.py b/games/game_witcher2.py
new file mode 100644
index 0000000..3442fb6
--- /dev/null
+++ b/games/game_witcher2.py
@@ -0,0 +1,67 @@
+# -*- encoding: utf-8 -*-
+
+from pathlib import Path
+from typing import List
+
+import mobase
+from PyQt6.QtCore import QDir
+
+from ..basic_features import BasicGameSaveGameInfo
+from ..basic_features.basic_save_game_info import BasicGameSaveGame
+from ..basic_game import BasicGame
+
+
+class Witcher2SaveGame(BasicGameSaveGame):
+    def allFiles(self):
+        return [
+            self._filepath.name,
+            self._filepath.name.replace(".sav", "_640x360.bmp"),
+        ]
+
+
+class Witcher2Game(BasicGame):
+
+    Name = "Witcher 2 Support Plugin"
+    Author = "DefinitelyNotSade"
+    Version = "1.0.0"
+
+    GameName = "The Witcher 2: Assassins of Kings"
+    GameShortName = "witcher2"
+    GaneNexusHame = "witcher2"
+    # GameNexusId = 952
+    GameSteamId = 20920
+    GameGogId = 1207658930
+    GameLauncher = "Launcher.exe"
+    GameBinary = "bin/witcher2.exe"
+    GameDataPath = "CookedPC"
+    GameSaveExtension = "sav"
+    GameDocumentsDirectory = "%DOCUMENTS%/witcher 2/Config"
+    GameSavesDirectory = "%GAME_DOCUMENTS%/../gamesaves"
+
+    def init(self, organizer: mobase.IOrganizer):
+        super().init(organizer)
+        self._featureMap[mobase.SaveGameInfo] = BasicGameSaveGameInfo(
+            lambda s: Path(str(s.parent) + "\\" + s.stem + "_640x360").with_suffix(
+                ".bmp"
+            )
+        )
+        return True
+
+    def iniFiles(self):
+        return [
+            "User.ini",
+            "Rendering.ini",
+            "Community.ini",
+            "UserContent.ini",
+            "DIMapping.ini",
+            "Input_QWERTY.ini",
+            "Input_AZERTY.ini",
+            "Input_QWERTZ.ini",
+        ]
+
+    def listSaves(self, folder: QDir) -> List[mobase.ISaveGame]:
+        ext = self._mappings.savegameExtension.get()
+        return [
+            Witcher2SaveGame(path)
+            for path in Path(folder.absolutePath()).glob(f"*.{ext}")
+        ]