1 files changed, 81 insertions, 0 deletions
diff --git a/libs/basic_games/games/game_sts2.py b/libs/basic_games/games/game_sts2.py
new file mode 100644
index 0000000..fe992e8
--- /dev/null
+++ b/libs/basic_games/games/game_sts2.py
@@ -0,0 +1,81 @@
+from pathlib import Path
+
+from PyQt6.QtCore import QDir, QFileInfo
+
+import mobase
+
+from ..basic_features import BasicModDataChecker, GlobPatterns
+from ..basic_features.basic_save_game_info import BasicGameSaveGame
+from ..basic_game import BasicGame
+
+
+class SlayTheSpire2ModDataChecker(BasicModDataChecker):
+    def __init__(self):
+        super().__init__(
+            GlobPatterns(
+                valid=[
+                    "*.pck",
+                    "*.dll",
+                    "*.json",
+                ],
+                move={
+                    "*/*.pck": "",
+                    "*/*.dll": "",
+                    "*/*.json": "",
+                },
+            )
+        )
+
+
+class SlayTheSpire2Game(BasicGame):
+    Name = "Slay the Spire 2 Support Plugin"
+    Author = "Azlle"
+    Version = "1.0.1"
+
+    GameName = "Slay the Spire 2"
+    GameShortName = "slaythespire2"
+    GameNexusName = "slaythespire2"
+    GameNexusId = 8916
+    GameSteamId = 2868840
+    GameBinary = "SlayTheSpire2.exe"
+    GameDataPath = "mods"
+    GameDocumentsDirectory = "%USERPROFILE%/AppData/Roaming/SlayTheSpire2"
+
+    def init(self, organizer: mobase.IOrganizer) -> bool:
+        super().init(organizer)
+        self._register_feature(SlayTheSpire2ModDataChecker())
+        return True
+
+    def initializeProfile(self, directory: QDir, settings: mobase.ProfileSetting):
+        mods_path = Path(self.dataDirectory().absolutePath())
+        mods_path.mkdir(exist_ok=True)
+        super().initializeProfile(directory, settings)
+
+    def savesDirectory(self) -> QDir:
+        docs = QDir(self.documentsDirectory())
+        steam_dir = Path(docs.absoluteFilePath("steam"))
+        if steam_dir.exists():
+            entries = [e for e in steam_dir.iterdir() if e.is_dir()]
+            if entries:
+                return QDir(str(entries[0]))
+        return QDir(str(steam_dir))
+
+    def listSaves(self, folder: QDir) -> list[mobase.ISaveGame]:
+        base = Path(folder.absolutePath())
+        return [
+            BasicGameSaveGame(save)
+            for save in base.rglob("*")
+            if save.is_file() and save.suffix in (".save", ".run")
+        ]
+
+    def executables(self):
+        return [
+            mobase.ExecutableInfo(
+                "Slay the Spire 2",
+                QFileInfo(self.gameDirectory().absoluteFilePath(self.binaryName())),
+            ),
+            mobase.ExecutableInfo(
+                "Slay the Spire 2 (OpenGL)",
+                QFileInfo(self.gameDirectory().absoluteFilePath(self.binaryName())),
+            ).withArgument("--rendering-driver opengl3"),
+        ]