微信小程序页面迁移校验之前 P5任务处理之前

.kiro/scripts/prompt_on_submit.py

@@ -1,11 +1,13 @@
 #!/usr/bin/env python3
-"""prompt_on_submit — promptSubmit 合并 hook 脚本。
+"""prompt_on_submit — promptSubmit 合并 hook 脚本（v2：文件基线模式）。
 
 合并原 audit_flagger + prompt_audit_log 的功能：
-1. git status → 风险判定 → 写 .kiro/.audit_state.json
-2. 记录 prompt 日志 → docs/audit/prompt_logs/
-3. 记录当前 git fingerprint 快照 → .kiro/.git_snapshot.json（供 agentStop 对比）
+1. 扫描工作区文件 → 保存基线快照 → .kiro/state/.file_baseline.json
+2. 基于基线文件列表做风险判定 → .kiro/state/.audit_state.json
+3. 记录 prompt 日志 → docs/audit/prompt_logs/
 
+变更检测不再依赖 git status（解决不常 commit 导致的误判问题）。
+风险判定仍基于 git status（因为需要知道哪些文件相对于 commit 有变化）。
 所有功能块用 try/except 隔离，单个失败不影响其他。
 """
 
@@ -17,6 +19,11 @@ import subprocess
 import sys
 from datetime import datetime, timezone, timedelta
 
+# 同目录导入文件基线模块 + cwd 校验
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+from file_baseline import scan_workspace, save_baseline
+from _ensure_root import ensure_repo_root
+
 TZ_TAIPEI = timezone(timedelta(hours=8))
 
 # ── 风险规则（来自 audit_flagger） ──
@@ -43,9 +50,8 @@ DB_PATTERNS = [
     re.compile(r"\.prisma$"),
 ]
 
-STATE_PATH = os.path.join(".kiro", ".audit_state.json")
-SNAPSHOT_PATH = os.path.join(".kiro", ".git_snapshot.json")
-PROMPT_ID_PATH = os.path.join(".kiro", ".last_prompt_id.json")
+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():
@@ -56,7 +62,8 @@ def sha1hex(s: str) -> str:
     return hashlib.sha1(s.encode("utf-8")).hexdigest()
 
 
-def get_changed_files() -> list[str]:
+def get_git_changed_files() -> list[str]:
+    """通过 git status 获取变更文件（仅用于风险判定，不用于变更检测）"""
     try:
         r = subprocess.run(
             ["git", "status", "--porcelain"],
@@ -94,14 +101,14 @@ def safe_read_json(path):
 
 
 def write_json(path, data):
-    os.makedirs(os.path.dirname(path) or ".kiro", exist_ok=True)
+    os.makedirs(os.path.dirname(path) or os.path.join(".kiro", "state"), exist_ok=True)
     with open(path, "w", encoding="utf-8") as f:
         json.dump(data, f, indent=2, ensure_ascii=False)
 
 
-# ── 功能块 1：风险标记（audit_flagger） ──
-def do_audit_flag(all_files, now):
-    files = sorted(set(f for f in all_files if not is_noise(f)))
+# ── 功能块 1：风险标记（基于 git status，判定哪些文件需要审计） ──
+def do_audit_flag(git_files, now):
+    files = sorted(set(f for f in git_files if not is_noise(f)))
 
     if not files:
         write_json(STATE_PATH, {
@@ -184,47 +191,38 @@ def do_prompt_log(now):
     write_json(PROMPT_ID_PATH, {"prompt_id": prompt_id, "at": now.isoformat()})
 
 
-# ── 功能块 3：Git 快照（供 agentStop 对比检测非 Kiro 变更） ──
-def do_git_snapshot(all_files, now):
-    fp = sha1hex("\n".join(sorted(all_files))) if all_files else ""
-    write_json(SNAPSHOT_PATH, {
-        "files": sorted(all_files)[:100],
-        "fingerprint": fp,
-        "taken_at": now.isoformat(),
-    })
+# ── 功能块 3：文件基线快照（替代 git snapshot） ──
+def do_file_baseline():
+    """扫描工作区文件 mtime+size，保存为基线快照。
+    agentStop 时再扫一次对比，即可精确检测本次对话期间的变更。
+    """
+    baseline = scan_workspace(".")
+    save_baseline(baseline)
 
 
 def main():
-    # 非 git 仓库直接退出
-    try:
-        r = subprocess.run(
-            ["git", "rev-parse", "--is-inside-work-tree"],
-            capture_output=True, text=True, encoding="utf-8", errors="replace", timeout=5
-        )
-        if r.returncode != 0:
-            return
-    except Exception:
-        return
-
+    ensure_repo_root()
     now = now_taipei()
-    all_files = get_changed_files()
 
-    # 各功能块独立 try/except
+    # 功能块 3：文件基线快照（最先执行，记录对话开始时的文件状态）
     try:
-        do_audit_flag(all_files, now)
+        do_file_baseline()
     except Exception:
         pass
 
+    # 功能块 1：风险标记（仍用 git status，因为需要知道未提交的变更）
+    try:
+        git_files = get_git_changed_files()
+        do_audit_flag(git_files, now)
+    except Exception:
+        pass
+
+    # 功能块 2：Prompt 日志
     try:
         do_prompt_log(now)
     except Exception:
         pass
 
-    try:
-        do_git_snapshot(all_files, now)
-    except Exception:
-        pass
-
 
 if __name__ == "__main__":
     try: