rm subscript

Author: LevelDownRefine

src/config/set_config.py

@@ -5,17 +5,110 @@
 每个脚本的 config 格式、路径、字段名都不同，
 各脚本子类单独适配，上层无需关心差异。
 
-config 路径推导由 subscript_utils 统一处理：
+config 路径推导：
   1. 从 config.yml 中取得各脚本的 script_path（exe 路径），取其父目录作为脚本根目录
   2. _CONFIG_REL_PATHS 标注了各 config 相对于脚本根目录的路径
   3. 拼接根目录 + 相对路径 = config 文件绝对路径
 """
 
 import os
 import json
+import yaml
+from typing import Any
 
-from src.subscript_utils import load_config, save_config
-from src.utils import get_our_bgi_user_dir
+from src.utils import get_config_yml_path_under_root, get_our_bgi_user_dir
+
+
+# ============================================================
+# 各脚本 config 相对路径映射
+# ============================================================
+
+_CONFIG_REL_PATHS: dict[str, str] = {
+    "鸣潮":   "data/apps/ok-ww/working/configs/DailyTask.json",
+    "原神":   "User/OneDragon/默认配置.json",
+    "终末地": "data/apps/ok-ef/working/configs/DailyTask.json",
+    "绝区零": "config/01/one_dragon/charge_plan.yml",
+    "崩铁":   "config.yaml",
+    "异环":   "data/apps/ok-nte/working/configs/DailyTask.json",
+    "粥":     "config/gui.new.json",
+}
+
+
+def _load_config_yml() -> dict:
+    """读取主配置 config.yml"""
+    with open(get_config_yml_path_under_root(), 'r', encoding='utf-8') as f:
+        return yaml.safe_load(f)
+
+
+# ============================================================
+# 脚本根目录 & config 路径解析
+# ============================================================
+
+def _get_script_root_dir(script_display_name: str) -> str:
+    """
+    从 config.yml 中找到指定脚本的 script_path，
+    取其父目录作为脚本项目根目录。
+
+    script_path 可能是 Windows 风格路径（C:\\...\\xxx.exe），
+    统一将反斜杠转为正斜杠后再取 dirname，确保跨平台可用。
+    """
+    config_data = _load_config_yml()
+    for script in config_data.get('script_list', []):
+        if script.get('display_name') == script_display_name:
+            script_path = script.get('script_path', '')
+            assert script_path, f"[set_config] config.yml 中 {script_display_name} 的 script_path 为空"
+            normalized = script_path.replace('\\', '/')
+            return os.path.dirname(normalized)
+    assert False, f"[set_config] config.yml 中找不到脚本: {script_display_name}"
+
+
+def get_config_path(script_display_name: str) -> str:
+    """
+    获取指定脚本的 config 文件绝对路径。
+    拼接脚本根目录 + config 相对路径。
+    """
+    assert script_display_name in _CONFIG_REL_PATHS, \
+        f"[set_config] 未适配脚本: {script_display_name}"
+    root = _get_script_root_dir(script_display_name)
+    rel = _CONFIG_REL_PATHS[script_display_name]
+    config_path = os.path.join(root, rel)
+    assert os.path.exists(config_path), f"[set_config] config 文件不存在: {config_path}"
+    return config_path
+
+
+# ============================================================
+# config 读写
+# ============================================================
+
+def load_config(script_display_name: str) -> Any:
+    """
+    读取指定脚本的 config 文件，返回解析后的 dict/list。
+    支持 .json 和 .yaml/.yml 格式。
+    """
+    path = get_config_path(script_display_name)
+    ext = os.path.splitext(path)[1].lower()
+    with open(path, 'r', encoding='utf-8') as f:
+        if ext == '.json':
+            return json.load(f)
+        elif ext in ('.yaml', '.yml'):
+            return yaml.safe_load(f)
+        raise ValueError(f"[set_config] 不支持的 config 格式: {ext}")
+
+
+def save_config(script_display_name: str, data: Any) -> None:
+    """
+    将数据写回指定脚本的 config 文件。
+    保持原始格式（json / yaml）。
+    """
+    path = get_config_path(script_display_name)
+    ext = os.path.splitext(path)[1].lower()
+    with open(path, 'w', encoding='utf-8') as f:
+        if ext == '.json':
+            json.dump(data, f, ensure_ascii=False, indent=4)
+        elif ext in ('.yaml', '.yml'):
+            yaml.dump(data, f, allow_unicode=True, sort_keys=False)
+        else:
+            raise ValueError(f"[set_config] 不支持的 config 格式: {ext}")
 
 
 # ============================================================