140 lines
3.4 KiB
Python
140 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
||
"""session_log — agentStop 时记录本次对话的完整日志。
|
||
|
||
收集来源：
|
||
- 环境变量 AGENT_OUTPUT（Kiro 注入的 agent 输出）
|
||
- 环境变量 USER_PROMPT（最近一次用户输入）
|
||
- .kiro/state/.last_prompt_id.json（Prompt ID 溯源）
|
||
- .kiro/state/.audit_state.json（变更文件列表）
|
||
- git diff --stat（变更统计）
|
||
|
||
输出：docs/audit/session_logs/session_<timestamp>.md
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
from datetime import datetime, timezone, timedelta
|
||
|
||
# cwd 校验
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
from _ensure_root import ensure_repo_root
|
||
|
||
TZ_TAIPEI = timezone(timedelta(hours=8))
|
||
LOG_DIR = os.path.join("docs", "audit", "session_logs")
|
||
STATE_PATH = os.path.join(".kiro", "state", ".audit_state.json")
|
||
PROMPT_ID_PATH = os.path.join(".kiro", "state", ".last_prompt_id.json")
|
||
|
||
|
||
def now_taipei():
|
||
return datetime.now(TZ_TAIPEI)
|
||
|
||
|
||
def safe_read_json(path):
|
||
if not os.path.isfile(path):
|
||
return {}
|
||
try:
|
||
with open(path, "r", encoding="utf-8") as f:
|
||
return json.load(f)
|
||
except Exception:
|
||
return {}
|
||
|
||
|
||
def git_diff_stat():
|
||
try:
|
||
r = subprocess.run(
|
||
["git", "diff", "--stat", "HEAD"],
|
||
capture_output=True, text=True, timeout=10
|
||
)
|
||
return r.stdout.strip() if r.returncode == 0 else "(git diff failed)"
|
||
except Exception:
|
||
return "(git not available)"
|
||
|
||
|
||
def git_status_short():
|
||
try:
|
||
r = subprocess.run(
|
||
["git", "status", "--short"],
|
||
capture_output=True, text=True, timeout=10
|
||
)
|
||
return r.stdout.strip() if r.returncode == 0 else ""
|
||
except Exception:
|
||
return ""
|
||
|
||
|
||
def main():
|
||
ensure_repo_root()
|
||
now = now_taipei()
|
||
ts = now.strftime("%Y%m%d_%H%M%S")
|
||
timestamp_display = now.strftime("%Y-%m-%d %H:%M:%S %z")
|
||
|
||
# 收集数据
|
||
agent_output = os.environ.get("AGENT_OUTPUT", "")
|
||
user_prompt = os.environ.get("USER_PROMPT", "")
|
||
prompt_info = safe_read_json(PROMPT_ID_PATH)
|
||
audit_state = safe_read_json(STATE_PATH)
|
||
prompt_id = prompt_info.get("prompt_id", "unknown")
|
||
|
||
# 截断超长内容，避免日志文件过大
|
||
max_len = 50000
|
||
if len(agent_output) > max_len:
|
||
agent_output = agent_output[:max_len] + "\n\n[TRUNCATED: output exceeds 50KB]"
|
||
if len(user_prompt) > 10000:
|
||
user_prompt = user_prompt[:10000] + "\n\n[TRUNCATED: prompt exceeds 10KB]"
|
||
|
||
diff_stat = git_diff_stat()
|
||
status_short = git_status_short()
|
||
changed_files = audit_state.get("changed_files", [])
|
||
|
||
os.makedirs(LOG_DIR, exist_ok=True)
|
||
filename = f"session_{ts}.md"
|
||
filepath = os.path.join(LOG_DIR, filename)
|
||
|
||
content = f"""# Session Log — {timestamp_display}
|
||
|
||
- Prompt-ID: `{prompt_id}`
|
||
- Audit Required: `{audit_state.get('audit_required', 'N/A')}`
|
||
- Reasons: {', '.join(audit_state.get('reasons', [])) or 'none'}
|
||
|
||
## User Input
|
||
|
||
```text
|
||
{user_prompt or '(not captured)'}
|
||
```
|
||
|
||
## Agent Output
|
||
|
||
```text
|
||
{agent_output or '(not captured)'}
|
||
```
|
||
|
||
## Changed Files ({len(changed_files)})
|
||
|
||
```
|
||
{chr(10).join(changed_files[:80]) if changed_files else '(none)'}
|
||
```
|
||
|
||
## Git Diff Stat
|
||
|
||
```
|
||
{diff_stat}
|
||
```
|
||
|
||
## Git Status
|
||
|
||
```
|
||
{status_short or '(clean)'}
|
||
```
|
||
"""
|
||
|
||
with open(filepath, "w", encoding="utf-8") as f:
|
||
f.write(content)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
main()
|
||
except Exception:
|
||
pass
|