重构: 巨函数拆分 + 清理历史标记注释 + custom_prompt/停止按钮/tunnel 改进

src-tauri/src/commands/ai/audit/approval.rs

@@ -1,4 +1,4 @@
-//! F-#97 / AE-04 / 阶段4 审批门控逻辑。
+//! 审批门控逻辑。
 //!
 //! 审批策略(方案 B:Persona 维度 + 扩展点):
 //! - reviewer/analyst 人设的工具调用全 auto(只读/分析,审批无意义)
@@ -88,12 +88,12 @@ pub(super) fn should_auto_for_persona(
     classify_risk_and_auto(risk_level, auto_exec_mode, tool_name, args)
 }
 
-/// AE-2025-04 检查会话信任命中。
+/// 检查会话信任命中。
 ///
 /// 首批信任工具：write_file / run_command。同会话已批准过同工具+同目录 →
 /// `TrustKey` 命中，返回 `Some(TrustKey)`；否则返回 `None`（走原审批流程）。
 ///
-/// BUG-260624-03/P0 重构:签名改 `session_arc: &Arc<Mutex<AiSession>>`,内部短 lock 读
+/// 签名改 `session_arc: &Arc<Mutex<AiSession>>`,内部短 lock 读
 /// `session_trust` 后立即 drop,信任查询不持锁,与 process_tool_calls 持锁 await 反模式解耦。
 pub(super) async fn check_trust_hits(
     draft: &ToolCallDraft,
@@ -118,7 +118,7 @@ pub(super) async fn check_trust_hits(
 /// **注意**：调用方应在调用前先 +=1 `pending_count`（保持与原 `handle_approval_tool`
 /// 行为一致——重试 skip 分支也在 `pending_count += 1` 之后返回）。
 ///
-/// BUG-260624-03/P0 重构:签名改 `session_arc: &Arc<Mutex<AiSession>>`,所有慢操作
+/// 签名改 `session_arc: &Arc<Mutex<AiSession>>`,所有慢操作
 /// (build_write_file_diff/detect_retry_count/build_approval_reason/audit_tool_call)
 /// 在锁外 await,仅 `pending_approvals.insert` + `messages.push` 两处纯写改短 lock 段。
 pub(super) async fn insert_pending_approval(
@@ -132,7 +132,7 @@ pub(super) async fn insert_pending_approval(
     db: &Arc<Database>,
     current_message_id: Option<&str>,
 ) {
-    // AE-2025-03（路径 B）：write_file 挂起审批前预读旧文件生成 diff。
+    // （路径 B）：write_file 挂起审批前预读旧文件生成 diff。
     // 仅 write_file（覆盖整文件，有完整新旧内容可对比）；其他工具 diff=None。
     // 旧文件不存在（新建）→ diff=None，前端回退显新 content。
     // 读失败不阻断审批（容错：文件无读权限等极端情况降级为无 diff 预览）。
@@ -142,7 +142,7 @@ pub(super) async fn insert_pending_approval(
         None
     };
 
-    // 阶段4（容错/恢复，开关 df-ai-approval-retry）：同 tc_id 重试检测。
+    // (容错/恢复，开关 df-ai-approval-retry)：同 tc_id 重试检测。
     // 与 High risk 的 find_cached_high_risk_result 互补：去重按 (tool_name,args) 匹配
     // （High only），本 guard 按 tc_id 匹配（覆盖 Med + High 残留场景）。
     // 同 tc_id 已有审计落定记录 → retry_count≥1，跳过审批 + emit Completed，断死循环。
@@ -179,12 +179,12 @@ pub(super) async fn insert_pending_approval(
             arguments: args.clone(),
             conversation_id: Some(conv_id.to_string()),
             recovered: false,
-            // 阶段3a：普通 RiskLevel 审批标 kind=Risk{diff}（下沉原 diff 字段）。
+            // 普通风险审批标 kind=Risk{diff}（下沉原 diff 字段）。
             kind: ApprovalKind::Risk { diff: approval_diff.clone() },
             retry_count,
             created_at: Some(std::time::SystemTime::now()),
         });
-        // 阶段2：占位带 __PENDING__:tc_id 标记，供 sanitize 豁免保留 + 出口断言自愈（防 400 orphan）
+        // 占位带 __PENDING__:tc_id 标记，供 sanitize 豁免保留 + 出口断言自愈（防 400 orphan）
         session.conv(conv_id).messages.push(ChatMessage::tool_result(&draft.id, &pending_placeholder_for(&draft.id)));
     }
     // 慢操作锁外:拼 reason(DB 读)+ emit + 审计落 pending 纪录(DB 写)
@@ -203,7 +203,7 @@ pub(super) async fn insert_pending_approval(
     audit_tool_call(audit_repo, conv_id, &draft.id, &draft.name, &draft.args, "pending", risk_level, None, None, current_message_id).await;
 }
 
-/// 阶段4（容错/恢复，开关 `df-ai-approval-retry`）：查审计表推算同 tc_id 重试计数。
+/// (容错/恢复，开关 `df-ai-approval-retry`)：查审计表推算同 tc_id 重试计数。
 ///
 /// 返回语义：
 /// - 0：审计表无该 tc_id 落定记录（或仅 pending），属首次审批执行，正常挂起。
@@ -224,7 +224,7 @@ pub(super) async fn detect_retry_count(audit_repo: &AiToolExecutionRepo, tc_id:
             None => return 0, // 无记录 = 首次
         },
         Err(e) => {
-            tracing::warn!("[阶段4-retry] 查审计表 tc_id={} 失败（降级无重试防护）: {}", tc_id, e);
+            tracing::warn!("[approval-retry] 查审计表 tc_id={} 失败（降级无重试防护）: {}", tc_id, e);
             return 0;
         }
     };
@@ -240,10 +240,10 @@ pub(super) async fn detect_retry_count(audit_repo: &AiToolExecutionRepo, tc_id:
 ///
 /// 1. 按 `auto_exec_mode`（low/medium/all） + `risk_level` + `patch_file` 小改动特例
 ///    判定是否应自动执行（`classify_risk_and_auto`）。若是 → 推入 `low_risk` 返回。
-/// 2. 否则走审批分支：会话信任（`check_trust_hits`）→ F-05 高危去重缓存 → 阶段4 重试 guard →
+/// 2. 否则走审批分支：会话信任（`check_trust_hits`）→ 高危去重缓存 → 重试 guard →
 ///    `insert_pending_approval`（write_file diff + 挂起 + emit + 审计落 pending 记录）。
 ///
-/// BUG-260624-03/P0 重构:签名改 `session_arc: &Arc<Mutex<AiSession>>`,所有 session 访问
+/// 签名改 `session_arc: &Arc<Mutex<AiSession>>`,所有 session 访问
 /// 都改短 lock 段(check_trust_hits/find_cached_high_risk_result 内部短 lock,命中后 push
 /// tool_result 短 lock 段)。慢操作(audit_tool_call/find_cached_high_risk_result 的 DB 查)
 /// 全在锁外 await,根治 process_tool_calls 持 session lock 期间 await 慢操作死锁反模式。
@@ -270,9 +270,9 @@ pub(super) async fn handle_approval_tool(
     }
 
     // ── Step 2: 会话信任检查 ──
-    // AE-2025-04 会话级信任（Session Trust）：首批 write_file / run_command，
+    // 会话级信任（Session Trust）：首批 write_file / run_command，
     // 同会话已批准过同工具+同目录 → TrustKey 命中 → 自动放行（跳过 pending + 二次确认）。
-    // 命中后走与 F-05 去重命中相似的「直接执行 + Completed + 审计 decided_by=auto_trust」路径，
+    // 命中后走与去重命中相似的「直接执行 + Completed + 审计 decided_by=auto_trust」路径，
     // 但与 F-05 不同：F-05 复用缓存 tool_result 跳过执行；trust 放行**真实执行工具**
     // （用户信任同目录同类操作，但仍要看每次的真实结果）。
     if let Some(key) = check_trust_hits(&draft, &args, session_arc, conv_id).await {
@@ -283,7 +283,7 @@ pub(super) async fn handle_approval_tool(
             tool = %draft.name,
             dir = %dir_label,
             new_tool_call_id = %draft.id,
-            "[AE-2025-04] 会话信任命中: 同会话已批准同类操作，自动放行（跳过审批+二次确认）"
+            "[会话信任] 命中: 同会话已批准同类操作，自动放行（跳过审批+二次确认）"
         );
         // emit 轻量 toast 事件（前端 AiChat.vue 显示"🔓 自动放行: tool(dir)"）
         // L3 emit 双写：会话信任自动放行 toast 双路发布（tunnel 透传 miniapp 即时反馈）。