游戏文本数据读取 & 存档读写类封装

.vscode/launch.json

@@ -0,0 +1,28 @@
+{
+    "configurations": [
+        {
+            "name": "Python Debugger: Python File",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${file}"
+        },
+        {
+            "name": "Run: struct",
+            "type": "debugpy",
+            "request": "launch",
+            "module": "app.structs.structs"
+        },
+        {
+            "name": "Run: unpack.text_unpacker",
+            "type": "debugpy",
+            "request": "launch",
+            "module": "app.unpack.text_unpacker"
+        },
+        {
+            "name": "Run: editor",
+            "type": "debugpy",
+            "request": "launch",
+            "module": "app.editor.save_editor"
+        },
+    ]
+}

app/deserializer/types.py

@@ -91,7 +91,7 @@ def copy_ctypes_to_struct(ctype_ins: ctypes.Structure, t: Type[T]) -> T:
     if is_primitive(t):
         return ctype_ins # type: ignore
     elif is_fixed_string(t):
-        return ctype_ins # type: ignore
+        return ctype_ins.decode('utf-8') # type: ignore
     elif is_fixed_array(t):
         elem_type = get_args(t)[0]
         return [copy_ctypes_to_struct(e, elem_type) for e in ctype_ins] # type: ignore

app/editor/save_editor.py

@@ -0,0 +1,146 @@
+import os
+from dataclasses import dataclass
+from enum import IntEnum
+
+from app.utils import find_game_path, find_save_path
+from app.structs.structs import SystemData
+from app.deserializer.types import Int32
+from app.unpack import TextUnpacker, TitleTextID, SaveTextID, Language
+
+class TitleId(IntEnum):
+    GS1 = 0
+    GS2 = 1
+    GS3 = 2
+
+@dataclass
+class SaveSlot:
+    time: str
+    progress: str
+    title: str
+    scenario: str
+
+class SaveEditor:
+    def __init__(self, language: Language = 'en') -> None:
+        game_path = find_game_path()
+        save_path = find_save_path()
+        if not game_path:
+            raise FileNotFoundError('Could not find game path')
+        if not save_path:
+            raise FileNotFoundError('Could not find save path')
+        self.game_path = game_path
+        self.save_path = save_path
+
+        self.__system_data: SystemData|None = None
+        self.__text_unpacker = TextUnpacker(game_path, language)
+
+    def load(self, save_file_path: str|None = None):
+        if not save_file_path:
+            save_file_path = os.path.join(self.save_path, 'systemdata')
+        with open(save_file_path, 'rb') as f:
+            self.__system_data = SystemData.from_bytes(f.read())
+
+    def save(self, save_file_path: str|None = None):
+        assert self.__system_data is not None, 'No data loaded'
+        if not save_file_path:
+            save_file_path = os.path.join(self.save_path, 'systemdata')
+        with open(save_file_path, 'wb') as f:
+            f.write(self.__system_data.to_bytes())
+
+    def set_account_id(self, account_id: int):
+        """
+        设置存档数据中保存的 Steam 账号 ID
+        """
+        assert self.__system_data is not None, 'No data loaded'
+        self.__system_data.reserve_work_.reserve[1] = Int32(account_id)
+
+    def get_account_id(self) -> int:
+        """
+        获取存档数据中保存的 Steam 账号 ID
+        """
+        assert self.__system_data is not None, 'No data loaded'
+        return self.__system_data.reserve_work_.reserve[1]
+
+    def get_slots(self) -> list[SaveSlot]:
+        """
+        获取所有存档槽位的信息
+        """
+        assert self.__system_data is not None, 'No data loaded'
+        slots = []
+        for i in range(50, 60):
+            slot = self.__system_data.slot_data_.save_data_[i]
+            time = str(slot.time)
+            title = TitleId(slot.title)
+            scenario = int(slot.scenario)
+
+            if time != '':
+                # title
+                title = self.__text_unpacker.get_text(TitleTextID.TITLE_NAME, slot.title)
+
+                # scenario
+                if title == TitleId.GS2 and scenario >= 4:
+                    scenario = ''
+                else:
+                    if title == TitleId.GS1:
+                        scenario = self.__text_unpacker.get_text(TitleTextID.GS1_SCENARIO_NAME, scenario)
+                    elif title == TitleId.GS2:
+                        scenario = self.__text_unpacker.get_text(TitleTextID.GS2_SCENARIO_NAME, scenario)
+                    elif title == TitleId.GS3:
+                        scenario = self.__text_unpacker.get_text(TitleTextID.GS3_SCENARIO_NAME, scenario)
+                    else:
+                        scenario = ''
+                    episode = self.__text_unpacker.get_text(TitleTextID.EPISODE_NUMBER, slot.scenario)
+                    scenario = f'{episode} {scenario}'
+
+                # progress
+                def get_progress_text(in_title: int, in_progress: int):
+                    in_text_id = 0
+                    if in_title != TitleId.GS1:
+                        if in_title == TitleId.GS2:
+                            in_text_id = 44 + in_progress
+                        elif in_title == TitleId.GS3:
+                            in_text_id = 66 + in_progress
+                    else:
+                        if in_progress in (17, 18):
+                            in_text_id = 34
+                        elif in_progress in (19, 20):
+                            in_text_id = 35
+                        elif in_progress == 21:
+                            in_text_id = 36
+                        elif in_progress in (22, 23, 24):
+                            in_text_id = 37
+                        elif in_progress == 25:
+                            in_text_id = 38
+                        elif in_progress in (26, 27):
+                            in_text_id = 39
+                        elif in_progress in (28, 29, 30):
+                            in_text_id = 40
+                        elif in_progress == 31:
+                            in_text_id = 41
+                        elif in_progress == 32:
+                            in_text_id = 42
+                        elif in_progress in (33, 34):
+                            in_text_id = 43
+                        else:
+                            in_text_id = 17 + in_progress
+
+                    return self.__text_unpacker.get_text(SaveTextID(in_text_id), 0)
+                progress = get_progress_text(slot.title, slot.progress)
+            else:
+                title = ''
+                progress = ''
+                scenario = ''
+
+            slots.append(SaveSlot(
+                time=time,
+                progress=progress,
+                title=title,
+                scenario=scenario,
+            ))
+        return slots
+
+if __name__ == '__main__':
+    from pprint import pprint
+    se = SaveEditor(language='hans')
+    se.load()
+    pprint(se.get_slots())
+

app/structs/structs.py

@@ -25,7 +25,11 @@ class SaveData(Struct):
     """存档时间。`YYYY/MM/DD\\nHH:MM:SS`"""
 
     title: ushort
+    """TextDataCtrl.TitleTextID.TITLE_NAME"""
+
     scenario: ushort
+    """"""
+
     progress: ushort
     in_data: ushort
 

app/unpack/__init__.py

@@ -0,0 +1,3 @@
+from .decrypt import decrypt_bytes, decrypt_file
+from .text_unpacker import TextUnpacker, Language
+from .decompiled import TitleTextID, SaveTextID