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.
- package/CHANGELOG.md +725 -720
- package/LICENSE +21 -21
- package/README.md +515 -509
- package/app/core/acceptance.py +100 -100
- package/app/core/agent_context.py +100 -0
- package/app/core/aiflavor.py +197 -197
- package/app/core/analytics.py +38 -38
- package/app/core/artifacts.py +148 -148
- package/app/core/assets.py +110 -110
- package/app/core/attachments.py +623 -623
- package/app/core/audit.py +30 -30
- package/app/core/backup.py +470 -470
- package/app/core/beekeeper.py +183 -183
- package/app/core/benchstore.py +102 -95
- package/app/core/board.py +209 -209
- package/app/core/bookmeta.py +1036 -1036
- package/app/core/bookmeta_catalog.py +180 -180
- package/app/core/branching.py +137 -137
- package/app/core/builtin_agent.py +1680 -1680
- package/app/core/capability.py +144 -144
- package/app/core/chaptersafe.py +32 -32
- package/app/core/checkpoints.py +143 -0
- package/app/core/cleanup.py +303 -303
- package/app/core/compaction.py +201 -201
- package/app/core/contracts.py +255 -255
- package/app/core/covergen.py +356 -356
- package/app/core/defectretro.py +43 -43
- package/app/core/diagnostics.py +114 -114
- package/app/core/dispatch.py +198 -191
- package/app/core/dispatch_log.py +128 -128
- package/app/core/env_scrub.py +85 -85
- package/app/core/errorlog.py +137 -137
- package/app/core/eval_matrix.py +82 -0
- package/app/core/evalbench.py +184 -91
- package/app/core/evaluation.py +174 -174
- package/app/core/failure_notify.py +290 -290
- package/app/core/flow_graph.py +33 -0
- package/app/core/goal_service.py +167 -167
- package/app/core/health.py +389 -389
- package/app/core/history.py +38 -38
- package/app/core/hooks.py +162 -162
- package/app/core/jobs.py +25 -4
- package/app/core/knowledge_pipeline.py +86 -0
- package/app/core/manager.py +2232 -2232
- package/app/core/mocks.py +64 -64
- package/app/core/notify.py +201 -201
- package/app/core/novel_quality.py +181 -189
- package/app/core/openapi.py +177 -154
- package/app/core/operations.py +228 -228
- package/app/core/paihang.py +145 -145
- package/app/core/pipeline.py +43 -19
- package/app/core/planner.py +17 -11
- package/app/core/policy.py +52 -0
- package/app/core/portguard.py +111 -111
- package/app/core/presence.py +77 -77
- package/app/core/preview.py +293 -293
- package/app/core/prompt_guard.py +78 -78
- package/app/core/publish/__init__.py +7 -7
- package/app/core/publish/auto.py +26 -3
- package/app/core/publish/browser.py +655 -655
- package/app/core/publish/chaprows.py +138 -0
- package/app/core/publish/fanqie.py +210 -165
- package/app/core/publish/flow.py +753 -640
- package/app/core/publish/flows-fanqie-calibrated.json +349 -336
- package/app/core/publish/ledger.py +13 -1
- package/app/core/publish/manager.py +1150 -1118
- package/app/core/publish/qimao.py +176 -176
- package/app/core/publish/volumes.py +108 -0
- package/app/core/publish/ws.py +139 -139
- package/app/core/quality_gate.py +194 -194
- package/app/core/redact.py +38 -38
- package/app/core/repeat_guard.py +129 -129
- package/app/core/retrieval.py +137 -137
- package/app/core/router.py +398 -398
- package/app/core/runner.py +2238 -2238
- package/app/core/selfupdate.py +384 -384
- package/app/core/session_log.py +161 -161
- package/app/core/sessions.py +312 -312
- package/app/core/settings.py +20 -0
- package/app/core/settings_schema.py +270 -270
- package/app/core/share_page.py +114 -114
- package/app/core/skill_scan.py +139 -139
- package/app/core/skillpacks/fanqie-novel.md +8 -6
- package/app/core/skillpacks/market/character-bible.md +66 -66
- package/app/core/skillpacks/market/code-risk-checklist.md +58 -58
- package/app/core/skillpacks/market/git-workflow.md +57 -57
- package/app/core/skillpacks/market/release-notes.md +72 -72
- package/app/core/skillpacks/market/weekly-report.md +71 -71
- package/app/core/skillpacks/market/worldview-consistency.md +70 -70
- package/app/core/skillpacks/qimao-signing.md +14 -11
- package/app/core/skills.py +32 -21
- package/app/core/step_runner.py +132 -132
- package/app/core/store.py +9 -0
- package/app/core/story_tracking.py +275 -275
- package/app/core/task_compile.py +199 -199
- package/app/core/telemetry.py +292 -292
- package/app/core/tlsctx.py +64 -64
- package/app/core/token_meter.py +147 -147
- package/app/core/tracing.py +207 -144
- package/app/core/upstream.py +41 -41
- package/app/core/usage.py +1038 -1038
- package/app/core/volumes.py +389 -389
- package/app/core/webhooks.py +89 -0
- package/app/core/wxdigest.py +858 -858
- package/app/core/zentao.py +2692 -2692
- package/app/main.py +104 -9
- package/app/mcp_server.py +244 -244
- package/app/pick_dialog.py +110 -110
- package/app/ui/app.js +286 -18
- package/app/ui/artdiff.js +74 -74
- package/app/ui/assets/community-qr.png +0 -0
- package/app/ui/assets/donate-alipay.jpg +0 -0
- package/app/ui/assets/donate-qr.jpg +0 -0
- package/app/ui/board.html +162 -162
- package/app/ui/board.js +305 -305
- package/app/ui/i18n.js +33 -0
- package/app/ui/icons/bee.svg +79 -79
- package/app/ui/index.html +32 -3
- package/app/ui/manifest.json +16 -16
- package/app/ui/qrcode.js +2297 -2297
- package/app/ui/style.css +196 -2
- package/bin/tutti.js +147 -147
- package/package.json +48 -48
- package/tools/wx_reader.py +261 -261
- package/app/.playwright-cli/console-2026-09-23T08-26-03-635Z.log +0 -88
- package/app/.playwright-cli/page-2026-09-23T08-26-03-845Z.yml +0 -126
- package/app/.playwright-cli/page-2026-09-23T08-29-44-203Z.png +0 -0
- package/app/.playwright-cli/page-2026-09-23T08-35-42-046Z.yml +0 -210
- package/app/.playwright-cli/page-2026-09-23T08-36-52-858Z.png +0 -0
- package/app/.playwright-cli/page-2026-09-23T08-37-33-938Z.png +0 -0
- package/app/.playwright-cli/run-detail-after.png +0 -0
- package/app/output/_createbook_body.json +0 -1
- package/app/output/_probe_fq_click.py +0 -50
- package/app/output/_probe_fq_form.py +0 -42
- package/app/output/_probe_fq_modal.py +0 -55
- package/app/output/_probe_fq_points.py +0 -71
- package/app/output/_probe_fq_submit.py +0 -34
- package/app/output/_probe_fq_watch.py +0 -62
package/app/core/dispatch_log.py
CHANGED
|
@@ -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)
|
package/app/core/env_scrub.py
CHANGED
|
@@ -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
|