codebee 0.1.95 → 0.1.96

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (138) hide show
  1. package/CHANGELOG.md +725 -720
  2. package/LICENSE +21 -21
  3. package/README.md +515 -509
  4. package/app/core/acceptance.py +100 -100
  5. package/app/core/agent_context.py +100 -0
  6. package/app/core/aiflavor.py +197 -197
  7. package/app/core/analytics.py +38 -38
  8. package/app/core/artifacts.py +148 -148
  9. package/app/core/assets.py +110 -110
  10. package/app/core/attachments.py +623 -623
  11. package/app/core/audit.py +30 -30
  12. package/app/core/backup.py +470 -470
  13. package/app/core/beekeeper.py +183 -183
  14. package/app/core/benchstore.py +102 -95
  15. package/app/core/board.py +209 -209
  16. package/app/core/bookmeta.py +1036 -1036
  17. package/app/core/bookmeta_catalog.py +180 -180
  18. package/app/core/branching.py +137 -137
  19. package/app/core/builtin_agent.py +1680 -1680
  20. package/app/core/capability.py +144 -144
  21. package/app/core/chaptersafe.py +32 -32
  22. package/app/core/checkpoints.py +143 -0
  23. package/app/core/cleanup.py +303 -303
  24. package/app/core/compaction.py +201 -201
  25. package/app/core/contracts.py +255 -255
  26. package/app/core/covergen.py +356 -356
  27. package/app/core/defectretro.py +43 -43
  28. package/app/core/diagnostics.py +114 -114
  29. package/app/core/dispatch.py +198 -191
  30. package/app/core/dispatch_log.py +128 -128
  31. package/app/core/env_scrub.py +85 -85
  32. package/app/core/errorlog.py +137 -137
  33. package/app/core/eval_matrix.py +82 -0
  34. package/app/core/evalbench.py +184 -91
  35. package/app/core/evaluation.py +174 -174
  36. package/app/core/failure_notify.py +290 -290
  37. package/app/core/flow_graph.py +33 -0
  38. package/app/core/goal_service.py +167 -167
  39. package/app/core/health.py +389 -389
  40. package/app/core/history.py +38 -38
  41. package/app/core/hooks.py +162 -162
  42. package/app/core/jobs.py +25 -4
  43. package/app/core/knowledge_pipeline.py +86 -0
  44. package/app/core/manager.py +2232 -2232
  45. package/app/core/mocks.py +64 -64
  46. package/app/core/notify.py +201 -201
  47. package/app/core/novel_quality.py +181 -189
  48. package/app/core/openapi.py +177 -154
  49. package/app/core/operations.py +228 -228
  50. package/app/core/paihang.py +145 -145
  51. package/app/core/pipeline.py +43 -19
  52. package/app/core/planner.py +17 -11
  53. package/app/core/policy.py +52 -0
  54. package/app/core/portguard.py +111 -111
  55. package/app/core/presence.py +77 -77
  56. package/app/core/preview.py +293 -293
  57. package/app/core/prompt_guard.py +78 -78
  58. package/app/core/publish/__init__.py +7 -7
  59. package/app/core/publish/auto.py +26 -3
  60. package/app/core/publish/browser.py +655 -655
  61. package/app/core/publish/chaprows.py +138 -0
  62. package/app/core/publish/fanqie.py +210 -165
  63. package/app/core/publish/flow.py +753 -640
  64. package/app/core/publish/flows-fanqie-calibrated.json +349 -336
  65. package/app/core/publish/ledger.py +13 -1
  66. package/app/core/publish/manager.py +1150 -1118
  67. package/app/core/publish/qimao.py +176 -176
  68. package/app/core/publish/volumes.py +108 -0
  69. package/app/core/publish/ws.py +139 -139
  70. package/app/core/quality_gate.py +194 -194
  71. package/app/core/redact.py +38 -38
  72. package/app/core/repeat_guard.py +129 -129
  73. package/app/core/retrieval.py +137 -137
  74. package/app/core/router.py +398 -398
  75. package/app/core/runner.py +2238 -2238
  76. package/app/core/selfupdate.py +384 -384
  77. package/app/core/session_log.py +161 -161
  78. package/app/core/sessions.py +312 -312
  79. package/app/core/settings.py +20 -0
  80. package/app/core/settings_schema.py +270 -270
  81. package/app/core/share_page.py +114 -114
  82. package/app/core/skill_scan.py +139 -139
  83. package/app/core/skillpacks/fanqie-novel.md +8 -6
  84. package/app/core/skillpacks/market/character-bible.md +66 -66
  85. package/app/core/skillpacks/market/code-risk-checklist.md +58 -58
  86. package/app/core/skillpacks/market/git-workflow.md +57 -57
  87. package/app/core/skillpacks/market/release-notes.md +72 -72
  88. package/app/core/skillpacks/market/weekly-report.md +71 -71
  89. package/app/core/skillpacks/market/worldview-consistency.md +70 -70
  90. package/app/core/skillpacks/qimao-signing.md +14 -11
  91. package/app/core/skills.py +32 -21
  92. package/app/core/step_runner.py +132 -132
  93. package/app/core/store.py +9 -0
  94. package/app/core/story_tracking.py +275 -275
  95. package/app/core/task_compile.py +199 -199
  96. package/app/core/telemetry.py +292 -292
  97. package/app/core/tlsctx.py +64 -64
  98. package/app/core/token_meter.py +147 -147
  99. package/app/core/tracing.py +207 -144
  100. package/app/core/upstream.py +41 -41
  101. package/app/core/usage.py +1038 -1038
  102. package/app/core/volumes.py +389 -389
  103. package/app/core/webhooks.py +89 -0
  104. package/app/core/wxdigest.py +858 -858
  105. package/app/core/zentao.py +2692 -2692
  106. package/app/main.py +104 -9
  107. package/app/mcp_server.py +244 -244
  108. package/app/pick_dialog.py +110 -110
  109. package/app/ui/app.js +286 -18
  110. package/app/ui/artdiff.js +74 -74
  111. package/app/ui/assets/community-qr.png +0 -0
  112. package/app/ui/assets/donate-alipay.jpg +0 -0
  113. package/app/ui/assets/donate-qr.jpg +0 -0
  114. package/app/ui/board.html +162 -162
  115. package/app/ui/board.js +305 -305
  116. package/app/ui/i18n.js +33 -0
  117. package/app/ui/icons/bee.svg +79 -79
  118. package/app/ui/index.html +32 -3
  119. package/app/ui/manifest.json +16 -16
  120. package/app/ui/qrcode.js +2297 -2297
  121. package/app/ui/style.css +196 -2
  122. package/bin/tutti.js +147 -147
  123. package/package.json +48 -48
  124. package/tools/wx_reader.py +261 -261
  125. package/app/.playwright-cli/console-2026-09-23T08-26-03-635Z.log +0 -88
  126. package/app/.playwright-cli/page-2026-09-23T08-26-03-845Z.yml +0 -126
  127. package/app/.playwright-cli/page-2026-09-23T08-29-44-203Z.png +0 -0
  128. package/app/.playwright-cli/page-2026-09-23T08-35-42-046Z.yml +0 -210
  129. package/app/.playwright-cli/page-2026-09-23T08-36-52-858Z.png +0 -0
  130. package/app/.playwright-cli/page-2026-09-23T08-37-33-938Z.png +0 -0
  131. package/app/.playwright-cli/run-detail-after.png +0 -0
  132. package/app/output/_createbook_body.json +0 -1
  133. package/app/output/_probe_fq_click.py +0 -50
  134. package/app/output/_probe_fq_form.py +0 -42
  135. package/app/output/_probe_fq_modal.py +0 -55
  136. package/app/output/_probe_fq_points.py +0 -71
  137. package/app/output/_probe_fq_submit.py +0 -34
  138. package/app/output/_probe_fq_watch.py +0 -62
@@ -1,128 +1,128 @@
1
- # -*- coding: utf-8 -*-
2
- """调度事件台账:记录脱敏的候选、决策和结果,支持按运行回放。"""
3
- from __future__ import annotations
4
-
5
- import json
6
- import hashlib
7
- import threading
8
- import time
9
-
10
- from . import paths
11
- from .redact import scrub_text
12
-
13
- LOCK = threading.RLock()
14
-
15
-
16
- def _dispatch_dir():
17
- # 读取动态 DATA_DIR,测试和多实例运行可在启动后重定向数据目录。
18
- return paths.DATA_DIR / "dispatch"
19
-
20
-
21
- def _month_file(day):
22
- return _dispatch_dir() / ("dispatch-%s.jsonl" % day[:7].replace("-", ""))
23
-
24
-
25
- def _text(value, limit=240):
26
- value = str(value or "").replace("\r", " ").replace("\n", " ")
27
- return scrub_text(value, limit=limit)
28
-
29
-
30
- def _candidate(row):
31
- if not isinstance(row, dict):
32
- return {}
33
- return {"agent_id": _text(row.get("agent_id"), 64),
34
- "label": _text(row.get("label"), 80),
35
- "kind": _text(row.get("kind"), 32),
36
- "score": row.get("score", 0),
37
- "reason": _text(row.get("reason"), 240),
38
- "order": row.get("order", 0)}
39
-
40
-
41
- def record_event(run_id="", task_id="", task_type="", difficulty="", role="",
42
- phase="selected", selected="", participants=(), candidates=(),
43
- fallback=(), selection_reason="", result="", verify_pass=None,
44
- review_pass=None, trace_id="", span_id="", parent_event_id=""):
45
- """追加一条事件;仅保存路由元数据,不保存正文、提示词、密钥或文件内容。"""
46
- try:
47
- day = time.strftime("%Y-%m-%d")
48
- event = {
49
- "ts": time.strftime("%Y-%m-%d %H:%M:%S"),
50
- "run_id": _text(run_id, 64), "task_id": _text(task_id, 64),
51
- "task_type": _text(task_type, 32), "difficulty": _text(difficulty, 16),
52
- "role": _text(role, 40), "phase": _text(phase, 16),
53
- "selected": _text(selected, 64),
54
- "participants": [_text(x.get("id") if isinstance(x, dict) else x, 64)
55
- for x in (participants or ())],
56
- "candidates": [_candidate(x) for x in (candidates or ()) if isinstance(x, dict)],
57
- "fallback": [_text(x, 64) for x in (fallback or ())],
58
- "selection_reason": _text(selection_reason, 400),
59
- "result": _text(result, 32),
60
- }
61
- # Trace identifiers are routing metadata only. They deliberately do
62
- # not carry prompt/output contents and remain optional for old callers.
63
- if run_id:
64
- from . import tracing
65
- event["trace_id"] = _text(trace_id or tracing.trace_id(run_id), 64)
66
- if span_id:
67
- event["span_id"] = _text(span_id, 32)
68
- if parent_event_id:
69
- event["parent_event_id"] = _text(parent_event_id, 64)
70
- event["event_id"] = hashlib.sha256(
71
- ("%s|%s|%s|%s|%s|%s" % (
72
- time.time_ns(), run_id, phase, role, selected, result
73
- )).encode("utf-8")
74
- ).hexdigest()[:24]
75
- if verify_pass is not None:
76
- event["verify_pass"] = bool(verify_pass)
77
- if review_pass is not None:
78
- event["review_pass"] = bool(review_pass)
79
- with LOCK:
80
- _dispatch_dir().mkdir(parents=True, exist_ok=True)
81
- with open(_month_file(day), "a", encoding="utf-8") as fh:
82
- fh.write(json.dumps(event, ensure_ascii=False) + "\n")
83
- return event
84
- except Exception:
85
- return None
86
-
87
-
88
- def _iter_events(run_id="", task_type="", limit=100):
89
- """从新到旧读取,满足 limit 即停,避免回放请求解析全部历史。"""
90
- try:
91
- directory = _dispatch_dir()
92
- files = sorted(directory.glob("dispatch-*.jsonl"), reverse=True) \
93
- if directory.is_dir() else []
94
- except Exception:
95
- return []
96
- events = []
97
- for path in files:
98
- try:
99
- lines = path.read_text(encoding="utf-8", errors="replace").splitlines()
100
- for line in reversed(lines):
101
- if not line.strip().startswith("{"):
102
- continue
103
- try:
104
- item = json.loads(line)
105
- except Exception:
106
- continue
107
- if not isinstance(item, dict):
108
- continue
109
- if run_id and item.get("run_id") != run_id:
110
- continue
111
- if task_type and item.get("task_type") != task_type:
112
- continue
113
- events.append(item)
114
- if len(events) >= limit:
115
- return events
116
- except Exception:
117
- continue
118
- return events
119
-
120
-
121
- def replay(run_id="", task_type="", limit=100):
122
- """返回最新的脱敏调度事件,支持按 run_id/task_type 筛选。"""
123
- try:
124
- limit = max(1, min(1000, int(limit)))
125
- except (TypeError, ValueError):
126
- limit = 100
127
- rid, ttype = _text(run_id, 64), _text(task_type, 32)
128
- return _iter_events(rid, ttype, limit)
1
+ # -*- coding: utf-8 -*-
2
+ """调度事件台账:记录脱敏的候选、决策和结果,支持按运行回放。"""
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ import hashlib
7
+ import threading
8
+ import time
9
+
10
+ from . import paths
11
+ from .redact import scrub_text
12
+
13
+ LOCK = threading.RLock()
14
+
15
+
16
+ def _dispatch_dir():
17
+ # 读取动态 DATA_DIR,测试和多实例运行可在启动后重定向数据目录。
18
+ return paths.DATA_DIR / "dispatch"
19
+
20
+
21
+ def _month_file(day):
22
+ return _dispatch_dir() / ("dispatch-%s.jsonl" % day[:7].replace("-", ""))
23
+
24
+
25
+ def _text(value, limit=240):
26
+ value = str(value or "").replace("\r", " ").replace("\n", " ")
27
+ return scrub_text(value, limit=limit)
28
+
29
+
30
+ def _candidate(row):
31
+ if not isinstance(row, dict):
32
+ return {}
33
+ return {"agent_id": _text(row.get("agent_id"), 64),
34
+ "label": _text(row.get("label"), 80),
35
+ "kind": _text(row.get("kind"), 32),
36
+ "score": row.get("score", 0),
37
+ "reason": _text(row.get("reason"), 240),
38
+ "order": row.get("order", 0)}
39
+
40
+
41
+ def record_event(run_id="", task_id="", task_type="", difficulty="", role="",
42
+ phase="selected", selected="", participants=(), candidates=(),
43
+ fallback=(), selection_reason="", result="", verify_pass=None,
44
+ review_pass=None, trace_id="", span_id="", parent_event_id=""):
45
+ """追加一条事件;仅保存路由元数据,不保存正文、提示词、密钥或文件内容。"""
46
+ try:
47
+ day = time.strftime("%Y-%m-%d")
48
+ event = {
49
+ "ts": time.strftime("%Y-%m-%d %H:%M:%S"),
50
+ "run_id": _text(run_id, 64), "task_id": _text(task_id, 64),
51
+ "task_type": _text(task_type, 32), "difficulty": _text(difficulty, 16),
52
+ "role": _text(role, 40), "phase": _text(phase, 16),
53
+ "selected": _text(selected, 64),
54
+ "participants": [_text(x.get("id") if isinstance(x, dict) else x, 64)
55
+ for x in (participants or ())],
56
+ "candidates": [_candidate(x) for x in (candidates or ()) if isinstance(x, dict)],
57
+ "fallback": [_text(x, 64) for x in (fallback or ())],
58
+ "selection_reason": _text(selection_reason, 400),
59
+ "result": _text(result, 32),
60
+ }
61
+ # Trace identifiers are routing metadata only. They deliberately do
62
+ # not carry prompt/output contents and remain optional for old callers.
63
+ if run_id:
64
+ from . import tracing
65
+ event["trace_id"] = _text(trace_id or tracing.trace_id(run_id), 64)
66
+ if span_id:
67
+ event["span_id"] = _text(span_id, 32)
68
+ if parent_event_id:
69
+ event["parent_event_id"] = _text(parent_event_id, 64)
70
+ event["event_id"] = hashlib.sha256(
71
+ ("%s|%s|%s|%s|%s|%s" % (
72
+ time.time_ns(), run_id, phase, role, selected, result
73
+ )).encode("utf-8")
74
+ ).hexdigest()[:24]
75
+ if verify_pass is not None:
76
+ event["verify_pass"] = bool(verify_pass)
77
+ if review_pass is not None:
78
+ event["review_pass"] = bool(review_pass)
79
+ with LOCK:
80
+ _dispatch_dir().mkdir(parents=True, exist_ok=True)
81
+ with open(_month_file(day), "a", encoding="utf-8") as fh:
82
+ fh.write(json.dumps(event, ensure_ascii=False) + "\n")
83
+ return event
84
+ except Exception:
85
+ return None
86
+
87
+
88
+ def _iter_events(run_id="", task_type="", limit=100):
89
+ """从新到旧读取,满足 limit 即停,避免回放请求解析全部历史。"""
90
+ try:
91
+ directory = _dispatch_dir()
92
+ files = sorted(directory.glob("dispatch-*.jsonl"), reverse=True) \
93
+ if directory.is_dir() else []
94
+ except Exception:
95
+ return []
96
+ events = []
97
+ for path in files:
98
+ try:
99
+ lines = path.read_text(encoding="utf-8", errors="replace").splitlines()
100
+ for line in reversed(lines):
101
+ if not line.strip().startswith("{"):
102
+ continue
103
+ try:
104
+ item = json.loads(line)
105
+ except Exception:
106
+ continue
107
+ if not isinstance(item, dict):
108
+ continue
109
+ if run_id and item.get("run_id") != run_id:
110
+ continue
111
+ if task_type and item.get("task_type") != task_type:
112
+ continue
113
+ events.append(item)
114
+ if len(events) >= limit:
115
+ return events
116
+ except Exception:
117
+ continue
118
+ return events
119
+
120
+
121
+ def replay(run_id="", task_type="", limit=100):
122
+ """返回最新的脱敏调度事件,支持按 run_id/task_type 筛选。"""
123
+ try:
124
+ limit = max(1, min(1000, int(limit)))
125
+ except (TypeError, ValueError):
126
+ limit = 100
127
+ rid, ttype = _text(run_id, 64), _text(task_type, 32)
128
+ return _iter_events(rid, ttype, limit)
@@ -1,86 +1,86 @@
1
- # -*- coding: utf-8 -*-
2
- """环境变量净化:剥离含 KEY/SECRET/TOKEN/PASSWORD 等密钥名的变量。
3
-
4
- 设计稿:docs/migration/01-defense-patterns.md §5A。
5
- 参考 dsh:docs/defensive-patterns.zh.md 第 31 行("绝不将环境变量或可预测路径暴露给不可信输出")。
6
- """
7
- from __future__ import annotations
8
-
9
- import logging
10
- import re
11
-
12
- log = logging.getLogger(__name__)
13
-
14
- # 黑名单:环境变量名含这些子串(不区分大小写)即剥离。
15
- _DENY_RE = re.compile(r"(KEY|SECRET|TOKEN|PASSWORD|PASSWD|CREDENTIAL)", re.IGNORECASE)
16
-
17
- # 白名单:精确匹配(不分大小写),用于覆盖黑名单。
18
- _ALLOW_EXACT = frozenset({
19
- # 系统/环境(TEMP/TMP/PATHEXT 等是 Windows 名,TMPDIR 是 macOS 每用户
20
- # 临时目录——三边各留各的,多出的名字在别的平台上不存在、无副作用)
21
- "PATH", "PATHEXT", "SYSTEMROOT", "TEMP", "TMP", "TMPDIR",
22
- "USERPROFILE", "USERDNSDOMAIN", "HOMEPATH", "HOMEDRIVE",
23
- "LANG", "LC_ALL", "LC_CTYPE",
24
- "HOME", "SHELL", "TERM", "PYTHONIOENCODING",
25
- "PYTHONPATH", "PYTHONUNBUFFERED",
26
- "COMSPEC", "WINDIR",
27
- # Tutti / DSH 内部
28
- "DSH_HOME", "TUTTI_HOME", "ALLINONE_HOME",
29
- # Claude CLI 自有(不是密钥本身,是配置)
30
- "CLAUDE_CODE_ENTRYPOINT", "CLAUDE_CODE_SSE_PORT",
31
- # PowerShell / Windows
32
- "PSMODULEPATH", "PROMPT",
33
- })
34
-
35
- # 白名单前缀(区分大小写以匹配 Windows 习惯);含前缀视为放行。
36
- _ALLOW_PREFIX = (
37
- "CODEX_", "CLAUDE_CODE_", "OPENAI_", "ANTHROPIC_",
38
- "QWEN_", "AIDER_", "OPENCODE_", "GEMINI_",
39
- "TUTTI_", "ALLINONE_", "DSH_",
40
- "LANG", "LC_", # locale 系列
41
- "PYTHON", # PYTHONPATH / PYTHONIOENCODING 等
42
- "XDG_", # XDG_CONFIG_HOME 等(macOS/Linux CLI 认它找配置/缓存落点)
43
- "CHROME", "EDGE", "PLAYWRIGHT", # 本机测试工具
44
- )
45
-
46
-
47
- def _should_keep(k: str) -> bool:
48
- if k in _ALLOW_EXACT:
49
- return True
50
- for p in _ALLOW_PREFIX:
51
- if k.startswith(p):
52
- return True
53
- return False
54
-
55
-
56
- def scrub_env(env: dict, *, mode: str = "drop") -> dict:
57
- """返回净化后的 env 副本(原 dict 不修改)。
58
-
59
- Args:
60
- env: 原始环境变量字典
61
- mode:
62
- "drop" - 直接删除命中黑名单的 key(默认)
63
- "stub" - 把值替换为 "__REDACTED__"(子进程若检查值类型可能崩)
64
- "off" - 完全不过滤(用于测试 / 排查)
65
- """
66
- if mode == "off":
67
- return dict(env)
68
- out: dict = {}
69
- dropped: list = []
70
- stubbed: list = []
71
- for k, v in env.items():
72
- if _should_keep(k):
73
- out[k] = v
74
- elif _DENY_RE.search(k):
75
- if mode == "stub":
76
- out[k] = "__REDACTED__"
77
- stubbed.append(k)
78
- else:
79
- dropped.append(k)
80
- else:
81
- out[k] = v
82
- if dropped:
83
- log.info("env-scrub dropped %d keys: %s", len(dropped), dropped[:20])
84
- if stubbed:
85
- log.info("env-scrub stubbed %d keys: %s", len(stubbed), stubbed[:20])
1
+ # -*- coding: utf-8 -*-
2
+ """环境变量净化:剥离含 KEY/SECRET/TOKEN/PASSWORD 等密钥名的变量。
3
+
4
+ 设计稿:docs/migration/01-defense-patterns.md §5A。
5
+ 参考 dsh:docs/defensive-patterns.zh.md 第 31 行("绝不将环境变量或可预测路径暴露给不可信输出")。
6
+ """
7
+ from __future__ import annotations
8
+
9
+ import logging
10
+ import re
11
+
12
+ log = logging.getLogger(__name__)
13
+
14
+ # 黑名单:环境变量名含这些子串(不区分大小写)即剥离。
15
+ _DENY_RE = re.compile(r"(KEY|SECRET|TOKEN|PASSWORD|PASSWD|CREDENTIAL)", re.IGNORECASE)
16
+
17
+ # 白名单:精确匹配(不分大小写),用于覆盖黑名单。
18
+ _ALLOW_EXACT = frozenset({
19
+ # 系统/环境(TEMP/TMP/PATHEXT 等是 Windows 名,TMPDIR 是 macOS 每用户
20
+ # 临时目录——三边各留各的,多出的名字在别的平台上不存在、无副作用)
21
+ "PATH", "PATHEXT", "SYSTEMROOT", "TEMP", "TMP", "TMPDIR",
22
+ "USERPROFILE", "USERDNSDOMAIN", "HOMEPATH", "HOMEDRIVE",
23
+ "LANG", "LC_ALL", "LC_CTYPE",
24
+ "HOME", "SHELL", "TERM", "PYTHONIOENCODING",
25
+ "PYTHONPATH", "PYTHONUNBUFFERED",
26
+ "COMSPEC", "WINDIR",
27
+ # Tutti / DSH 内部
28
+ "DSH_HOME", "TUTTI_HOME", "ALLINONE_HOME",
29
+ # Claude CLI 自有(不是密钥本身,是配置)
30
+ "CLAUDE_CODE_ENTRYPOINT", "CLAUDE_CODE_SSE_PORT",
31
+ # PowerShell / Windows
32
+ "PSMODULEPATH", "PROMPT",
33
+ })
34
+
35
+ # 白名单前缀(区分大小写以匹配 Windows 习惯);含前缀视为放行。
36
+ _ALLOW_PREFIX = (
37
+ "CODEX_", "CLAUDE_CODE_", "OPENAI_", "ANTHROPIC_",
38
+ "QWEN_", "AIDER_", "OPENCODE_", "GEMINI_",
39
+ "TUTTI_", "ALLINONE_", "DSH_",
40
+ "LANG", "LC_", # locale 系列
41
+ "PYTHON", # PYTHONPATH / PYTHONIOENCODING 等
42
+ "XDG_", # XDG_CONFIG_HOME 等(macOS/Linux CLI 认它找配置/缓存落点)
43
+ "CHROME", "EDGE", "PLAYWRIGHT", # 本机测试工具
44
+ )
45
+
46
+
47
+ def _should_keep(k: str) -> bool:
48
+ if k in _ALLOW_EXACT:
49
+ return True
50
+ for p in _ALLOW_PREFIX:
51
+ if k.startswith(p):
52
+ return True
53
+ return False
54
+
55
+
56
+ def scrub_env(env: dict, *, mode: str = "drop") -> dict:
57
+ """返回净化后的 env 副本(原 dict 不修改)。
58
+
59
+ Args:
60
+ env: 原始环境变量字典
61
+ mode:
62
+ "drop" - 直接删除命中黑名单的 key(默认)
63
+ "stub" - 把值替换为 "__REDACTED__"(子进程若检查值类型可能崩)
64
+ "off" - 完全不过滤(用于测试 / 排查)
65
+ """
66
+ if mode == "off":
67
+ return dict(env)
68
+ out: dict = {}
69
+ dropped: list = []
70
+ stubbed: list = []
71
+ for k, v in env.items():
72
+ if _should_keep(k):
73
+ out[k] = v
74
+ elif _DENY_RE.search(k):
75
+ if mode == "stub":
76
+ out[k] = "__REDACTED__"
77
+ stubbed.append(k)
78
+ else:
79
+ dropped.append(k)
80
+ else:
81
+ out[k] = v
82
+ if dropped:
83
+ log.info("env-scrub dropped %d keys: %s", len(dropped), dropped[:20])
84
+ if stubbed:
85
+ log.info("env-scrub stubbed %d keys: %s", len(stubbed), stubbed[:20])
86
86
  return out