lcode-agent 0.1.0__py3-none-any.whl

lcode/settings.py

@@ -0,0 +1,229 @@
+"""应用配置:所有配置统一在 ~/.lcode/config.json,不放进仓库。
+
+用户只改这一个文件:url / model / models / apiKey / maxTokens / pricing /
+permissionMode 全在里面,dev 和打包后行为一致。缺的键启动时自动按出厂
+默认补全,打开文件就能看到所有可配置项。
+出厂默认值随包分发(src/lcode/default_config.json),不含密钥。
+旧的 secrets.json / settings.json 迁移完成后删除。
+"""
+
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+
+HOME_DIR = Path.home() / ".lcode"
+SECRETS_PATH = HOME_DIR / "secrets.json"
+SETTINGS_PATH = HOME_DIR / "settings.json"
+# 出厂默认值:随包分发,dev 和安装后都读这一份
+PROJECT_CONFIG = Path(__file__).with_name("default_config.json")
+# 唯一的用户配置点:模型、密钥都改这里,dev/打包一致
+USER_CONFIG = HOME_DIR / "config.json"
+
+_runtime: dict | None = None
+
+
+def _read_json(path: Path) -> dict:
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+    except (FileNotFoundError, json.JSONDecodeError, OSError):
+        return {}
+    return data if isinstance(data, dict) else {}
+
+
+def _write_json(path: Path, data: dict) -> None:
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(json.dumps(data, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
+
+
+def _migrate_key_from_project(project: dict, secrets: dict) -> tuple[dict, dict, bool]:
+    leaked = str(project.get("apiKey") or "").strip()
+    if not leaked:
+        if "apiKey" in project:
+            project = dict(project)
+            project.pop("apiKey", None)
+            return project, secrets, True
+        return project, secrets, False
+    secrets = dict(secrets)
+    if not str(secrets.get("apiKey") or "").strip():
+        secrets["apiKey"] = leaked
+        _write_json(SECRETS_PATH, secrets)
+    project = dict(project)
+    project.pop("apiKey", None)
+    return project, secrets, True
+
+
+def _migrate_key_from_git(secrets: dict) -> dict:
+    """旧提交里若有密钥,搬到 ~/.lcode/secrets.json,不写回项目文件。
+
+    兼容注意:git show 出来的历史文件是 UTF-8,Windows 默认按 GBK 解码会炸
+    (中文注释一炸,check_output 在读线程里吞异常返回 None),所以拿 bytes
+    手动解码,全程兜异常——这是尽力而为的遗留迁移,绝不能挂掉启动。
+    """
+    import subprocess
+
+    root = Path.cwd()
+    try:
+        log = subprocess.check_output(
+            ["git", "log", "--format=%H", "--", "config.json"],
+            cwd=str(root),
+            timeout=15,
+            stderr=subprocess.DEVNULL,
+        ).decode("utf-8", errors="replace")
+    except Exception:
+        return secrets
+    for commit in log.split():
+        try:
+            raw = subprocess.check_output(
+                ["git", "show", f"{commit}:config.json"],
+                cwd=str(root),
+                timeout=15,
+                stderr=subprocess.DEVNULL,
+            )
+        except Exception:
+            continue
+        if not raw:
+            continue
+        try:
+            data = json.loads(raw.decode("utf-8", errors="replace"))
+        except (ValueError, UnicodeDecodeError):
+            continue
+        if not isinstance(data, dict):
+            continue
+        key = str(data.get("apiKey") or "").strip()
+        if not key:
+            continue
+        secrets = dict(secrets)
+        secrets["apiKey"] = key
+        _write_json(SECRETS_PATH, secrets)
+        return secrets
+    return secrets
+
+
+def _backfill_user_config(project: dict, user: dict) -> dict:
+    """用户 config.json 缺的键按出厂默认补全,让文件本身就是完整模板。
+
+    用户已设置过的键不动;写回成功返回补全后的 dict,失败返回原样。
+    """
+    filled = dict(user)
+    changed = False
+    for key, value in project.items():
+        if key == "apiKey":
+            continue
+        if key not in filled:
+            filled[key] = value
+            changed = True
+    if changed:
+        try:
+            _write_json(USER_CONFIG, filled)
+        except OSError:
+            return user
+    return filled
+
+
+def load_runtime() -> dict:
+    """合并出厂默认 + 用户 config.json + 密钥 + 权限设置。apiKey 只留在内存里。
+
+    用户只改 ~/.lcode/config.json:url / model / models / apiKey / maxTokens /
+    permissionMode 全在这一个文件,缺的键自动补成完整模板,dev 和打包一致。
+    旧的 secrets.json / settings.json 迁移完成后删除。
+    """
+    global _runtime
+    project = _read_json(PROJECT_CONFIG)
+    secrets = _read_json(SECRETS_PATH)
+    project, secrets, stripped = _migrate_key_from_project(project, secrets)
+    user = _backfill_user_config(project, _read_json(USER_CONFIG))
+    # 环境变量 / 用户 config / 旧 secrets 都没有 key,才去翻 git 历史;
+    # 用户已经配好 key 的话,这个遗留迁移一次都不该跑
+    if not str(
+        os.environ.get("LCODE_API_KEY")
+        or user.get("apiKey")
+        or secrets.get("apiKey")
+        or ""
+    ).strip():
+        try:
+            secrets = _migrate_key_from_git(secrets)
+        except Exception:
+            secrets = dict(secrets)
+    if stripped:
+        _write_json(PROJECT_CONFIG, project)
+    merged = {**project, **user}
+
+    # 密钥:环境变量 > 用户 config.json > 旧 secrets.json;
+    # 用户 config 里确立了 key 之后,旧 secrets.json 使命完成,删掉
+    key = str(
+        os.environ.get("LCODE_API_KEY")
+        or user.get("apiKey")
+        or secrets.get("apiKey")
+        or ""
+    ).strip()
+    if (
+        key
+        and not str(user.get("apiKey") or "").strip()
+        and not os.environ.get("LCODE_API_KEY")
+    ):
+        user["apiKey"] = key
+        _write_json(USER_CONFIG, user)
+    if str(user.get("apiKey") or "").strip() and SECRETS_PATH.exists() and SECRETS_PATH != USER_CONFIG:
+        try:
+            SECRETS_PATH.unlink(missing_ok=True)
+        except OSError:
+            pass
+
+    # 权限模式:config.json 优先;但 settings.json 是旧位置,迁移期间旧值优先,
+    # 迁进 config 并删掉旧文件后就只剩一个来源
+    legacy_mode = str(_read_json(SETTINGS_PATH).get("permissionMode") or "").strip().lower()
+    mode = str(legacy_mode or user.get("permissionMode") or "ask").strip().lower()
+    if mode not in ("ask", "pass"):
+        mode = "ask"
+    if user.get("permissionMode") != mode:
+        user["permissionMode"] = mode
+        _write_json(USER_CONFIG, user)
+    if SETTINGS_PATH.exists() and SETTINGS_PATH != USER_CONFIG:
+        try:
+            SETTINGS_PATH.unlink(missing_ok=True)
+        except OSError:
+            pass
+
+    out = dict(merged)
+    out["apiKey"] = key
+    out["permissionMode"] = mode
+    _runtime = out
+    return out
+
+
+def get_runtime() -> dict:
+    if _runtime is None:
+        return load_runtime()
+    return _runtime
+
+
+def api_key() -> str:
+    return str(get_runtime().get("apiKey") or "")
+
+
+def get_permission_mode() -> str:
+    mode = str(get_runtime().get("permissionMode") or "ask").strip().lower()
+    return mode if mode in ("ask", "pass") else "ask"
+
+
+def save_permission_mode(mode: str) -> str:
+    mode = "pass" if str(mode).strip().lower() == "pass" else "ask"
+    save_project_config({"permissionMode": mode})
+    runtime = get_runtime()
+    runtime["permissionMode"] = mode
+    return mode
+
+
+def save_project_config(fields: dict) -> dict:
+    """把模型等用户配置写进 ~/.lcode/config.json,并同步进运行时。
+
+    dev 和打包后都在这里改,用户始终只看这一个文件。
+    """
+    base = _read_json(USER_CONFIG)
+    base.update(fields)
+    _write_json(USER_CONFIG, base)
+    runtime = get_runtime()
+    runtime.update(fields)
+    return runtime