From 25ca6cf2cb9a5f862a0f363f41c4763ba57244c1 Mon Sep 17 00:00:00 2001
From: tretrauit <tretrauit@gmail.com>
Date: Mon, 26 Jun 2023 17:22:11 +0700
Subject: [PATCH] feat: implement configfile

Basically still ConfigParser
---
 vollerei/common/__init__.py   |  4 ++
 vollerei/common/configfile.py | 15 +++++++
 vollerei/hsr/launcher/game.py | 83 ++++++++++++++++++++++-------------
 3 files changed, 72 insertions(+), 30 deletions(-)
 create mode 100644 vollerei/common/__init__.py
 create mode 100644 vollerei/common/configfile.py

diff --git a/vollerei/common/__init__.py b/vollerei/common/__init__.py
new file mode 100644
index 0000000..2b3b5e4
--- /dev/null
+++ b/vollerei/common/__init__.py
@@ -0,0 +1,4 @@
+from vollerei.common.configfile import ConfigFile
+from vollerei.common.telemetry import block_telemetry, check_telemetry
+
+__all__ = ["ConfigFile", "block_telemetry", "check_telemetry"]
diff --git a/vollerei/common/configfile.py b/vollerei/common/configfile.py
new file mode 100644
index 0000000..86ca8bb
--- /dev/null
+++ b/vollerei/common/configfile.py
@@ -0,0 +1,15 @@
+from configparser import ConfigParser
+from pathlib import Path
+
+
+class ConfigFile(ConfigParser):
+    path: Path
+
+    def __init__(self, path, **kwargs):
+        super().__init__(**kwargs)
+        self.path = Path(path)
+        self.read(self.path)
+
+    def save(self):
+        with self.path.open("w") as f:
+            self.write(f)
diff --git a/vollerei/hsr/launcher/game.py b/vollerei/hsr/launcher/game.py
index 9a8326d..d90b71b 100644
--- a/vollerei/hsr/launcher/game.py
+++ b/vollerei/hsr/launcher/game.py
@@ -2,6 +2,7 @@ from hashlib import md5
 from os import PathLike
 from pathlib import Path
 from enum import Enum
+from vollerei.common import ConfigFile
 from vollerei.abc.launcher.game import GameABC
 from vollerei.hsr.constants import MD5SUMS
 
@@ -89,6 +90,24 @@ class Game(GameABC):
             return False
         return True
 
+    def _get_version_config(self) -> tuple[int, int, int]:
+        cfg_file = self._path.joinpath("config.ini")
+        if not cfg_file.exists():
+            return (0, 0, 0)
+        cfg = ConfigFile(cfg_file)
+        if "General" not in cfg.sections():
+            return (0, 0, 0)
+        if "game_version" not in cfg["General"]:
+            return (0, 0, 0)
+        version_str = cfg["General"]["game_version"]
+        if version_str.count(".") != 2:
+            return (0, 0, 0)
+        try:
+            version = tuple(int(i) for i in version_str.split("."))
+        except Exception:
+            return (0, 0, 0)
+        return version
+
     def get_version(self) -> tuple[int, int, int]:
         """
         Get the current installed game version.
@@ -116,36 +135,40 @@ class Game(GameABC):
         version_bytes: list[list[bytes]] = [[], [], []]
         version_ptr = 0
         correct = True
-        with self.data_folder().joinpath("data.unity3d").open("rb") as f:
-            f.seek(0x7D0)  # 2000 in decimal
-            for byte in f.read(10000):
-                match byte:
-                    case 0:
-                        version_bytes = [[], [], []]
-                        version_ptr = 0
-                        correct = True
-                    case 46:
-                        version_ptr += 1
-                        if version_ptr > 2:
-                            correct = False
-                    case 38:
-                        if (
-                            correct
-                            and len(version_bytes[0]) > 0
-                            and len(version_bytes[1]) > 0
-                            and len(version_bytes[2]) > 0
-                        ):
-                            return (
-                                bytes_to_int(version_bytes[0]),
-                                bytes_to_int(version_bytes[1]),
-                                bytes_to_int(version_bytes[2]),
-                            )
-                    case _:
-                        if correct and byte in allowed:
-                            version_bytes[version_ptr].append(byte)
-                        else:
-                            correct = False
-        return (0, 0, 0)
+        try:
+            with self.data_folder().joinpath("data.unity3d").open("rb") as f:
+                f.seek(0x7D0)  # 2000 in decimal
+                for byte in f.read(10000):
+                    match byte:
+                        case 0:
+                            version_bytes = [[], [], []]
+                            version_ptr = 0
+                            correct = True
+                        case 46:
+                            version_ptr += 1
+                            if version_ptr > 2:
+                                correct = False
+                        case 38:
+                            if (
+                                correct
+                                and len(version_bytes[0]) > 0
+                                and len(version_bytes[1]) > 0
+                                and len(version_bytes[2]) > 0
+                            ):
+                                return (
+                                    bytes_to_int(version_bytes[0]),
+                                    bytes_to_int(version_bytes[1]),
+                                    bytes_to_int(version_bytes[2]),
+                                )
+                        case _:
+                            if correct and byte in allowed:
+                                version_bytes[version_ptr].append(byte)
+                            else:
+                                correct = False
+        except Exception:
+            pass
+        # Fallback to config.ini
+        return self._get_version_config()
 
     def get_version_str(self) -> str:
         """