@hupan56/wlkj 3.3.0 → 3.3.1

package/templates/qoder/hooks/pre-tool-use.py

@@ -0,0 +1,136 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""pre-tool-use.py - PreToolUse hook: 改码前 get_impact 影响面 priming（#16 / CLAUDE.md 铁律#1 宿主级强制）
+
+吸收宿主确定性 hook 能力：把"改代码前必调 get_impact"从"靠 AI 记得（老忘→裸改崩一片）"
+升级为"宿主在 Write/Edit/MultiEdit 落到代码文件前，确定性提取符号 + 强提醒"。
+
+做法（确定性、零外部依赖、不依赖后端在线）：
+  1. 解析 stdin {tool_name, tool_input:{file_path}}
+  2. 只对代码文件(*.py/java/vue/js/ts/go/kt 等，非 .md/.json/.txt)触发
+  3. 正则提取文件里的 class/def/function 名（要改的符号）
+  4. 输出影响面 advisory 到会话：列符号 + 强制提醒改前调 get_impact(entity=<主符号>)
+     实际 get_impact 由工作流 cap.mcp.call 跑（robust，处理 token/SSE）。
+
+铁律：
+  - 永远 exit 0（advisory 不阻断；阻断会卡死工作流）
+  - 失败 try/except 静默（hook 崩绝不阻塞会话）
+  - 非 .py/.java 等代码文件 / 非编辑类工具 → 静默放行（不滥提醒）
+
+stdin 字段名官方未公开，兼容 tool_name/tool_input 等多种写法（同 post-tool-use.py）。
+"""
+import json
+import os
+import re
+import sys
+
+if sys.platform == 'win32':
+    try:
+        sys.stdout.reconfigure(encoding='utf-8')
+        sys.stderr.reconfigure(encoding='utf-8')
+    except Exception:
+        pass
+
+NL = chr(10)
+BASE = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+HOOK_ERRORS_LOG = os.path.join(BASE, '.qoder', '.runtime', 'hook-errors.log')
+
+# 触发的工具 + 代码文件后缀
+_EDIT_TOOLS = {"Write", "Edit", "MultiEdit", "write", "edit", "multi_edit"}
+_CODE_EXT = {".py", ".java", ".vue", ".js", ".ts", ".tsx", ".jsx", ".go", ".kt", ".scala", ".cs", ".c", ".cpp", ".h", ".rs"}
+
+# 符号提取正则（class/def/function/interface）
+_SYM_RE = re.compile(
+    r"^\s*(?:export\s+)?(?:public|private|protected|static|final|abstract|async|\s)*"
+    r"(?:class|def|function|interface|struct|enum|object)\s+([A-Za-z_][A-Za-z0-9_]*)",
+    re.MULTILINE,
+)
+
+
+def _log_err(line):
+    from datetime import datetime
+    try:
+        os.makedirs(os.path.dirname(HOOK_ERRORS_LOG), exist_ok=True)
+        with open(HOOK_ERRORS_LOG, 'a', encoding='utf-8') as f:
+            f.write('[{}] pre-tool-use: {}{}'.format(
+                datetime.now().strftime('%Y-%m-%d %H:%M:%S'), str(line)[:200], NL))
+    except Exception:
+        pass
+
+
+def _read_stdin():
+    try:
+        raw = sys.stdin.read()
+        if raw and raw.strip():
+            return json.loads(raw)
+    except Exception as e:
+        _log_err("stdin parse: %s" % str(e)[:120])
+    return {}
+
+
+def _is_code(path):
+    _, ext = os.path.splitext(path.lower())
+    return ext in _CODE_EXT
+
+
+def _extract_symbols(path, limit=8):
+    """从文件提取 class/def/function 名（要改的符号）。best-effort，读不到返空。"""
+    try:
+        if not os.path.isfile(path):
+            # 相对路径 → 试 BASE 下定位
+            cand = path
+            if not os.path.isabs(path):
+                for root in (BASE, os.getcwd()):
+                    c = os.path.join(root, path)
+                    if os.path.isfile(c):
+                        cand = c
+                        break
+            if not os.path.isfile(cand):
+                return []
+            path = cand
+        with open(path, 'r', encoding='utf-8', errors='replace') as f:
+            text = f.read()[:200000]  # 限 200K 防超大文件慢
+        return _SYM_RE.findall(text)[:limit]
+    except Exception as e:
+        _log_err("extract %s: %s" % (path, str(e)[:80]))
+        return []
+
+
+def main():
+    data = _read_stdin()
+    tool = data.get("tool_name") or data.get("tool") or ""
+    tinput = data.get("tool_input") or data.get("input") or {}
+    path = tinput.get("file_path") or tinput.get("path") or ""
+
+    # 非编辑类工具 / 非代码文件 → 静默放行
+    if tool not in _EDIT_TOOLS or not path or not _is_code(path):
+        sys.exit(0)
+
+    syms = _extract_symbols(path)
+    rel = path
+    try:
+        rel = os.path.relpath(path, BASE)
+    except Exception:
+        pass
+
+    lines = []
+    lines.append("⚠️ **改码前影响面 priming**（CLAUDE.md 铁律#1：改代码前必调 get_impact）")
+    lines.append("即将编辑代码文件：`%s`" % rel)
+    if syms:
+        lines.append("含符号 %d 个：%s" % (len(syms), ", ".join("`%s`" % s for s in syms)))
+        lines.append("→ **改前必调** `cap.mcp.call('get_impact', {entity: '%s', depth: 2})` 看 upstream(谁依赖我)/downstream(我依赖谁)。" % syms[0])
+        lines.append("  影响面大 / risk level≥high → 先提示风险再动手；高风险裸改须拦停问用户。")
+    else:
+        lines.append("（未提取到已知符号，可能是新文件）→ 仍建议改前 rag_search 找相关实现 + get_impact 看上下游。")
+
+    # advisory 输出到 stdout（注入会话上下文），exit 0 不阻断
+    sys.stdout.write(NL.join(lines) + NL)
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except Exception as e:
+        _log_err("fatal: %s" % str(e)[:150])
+        sys.exit(0)  # 永不阻塞