@hupan56/wlkj 3.3.0 → 3.3.2

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

@@ -116,7 +116,7 @@ def _auto_learn_from_prd(prd_path):
     try:
         title = os.path.basename(prd_path).replace('.md', '')
         # 读 mcp_config 拿 token
-        cfg_path = os.path.join(BASE, 'wlinkj-workflow', 'mcp_config.json')
+        cfg_path = os.path.join(BASE, '.qoder', 'mcp_config.json')
         if not os.path.isfile(cfg_path):
             return
         import json as _j
@@ -143,6 +143,32 @@ def _auto_learn_from_prd(prd_path):
         pass  # 铁律：静默
 
 
+def _emit_ask_before_return(kind, path):
+    """PRD/原型写完 → 注入"先问用户再回流"指令（用户硬要求：上传前必须先问）。
+
+    hook 无法弹窗等 Y/n（工具执行后跑、输出即退），但能**确定性注入指令**让 AI
+    每次必问、默认不上传。stdout 注入会话上下文，AI 看到后必须先问用户。
+    """
+    name = os.path.basename(path)
+    if kind == 'prd':
+        cmd = '$PY "$R/.qoder/scripts/orchestration/wlkj.py" return prd "%s" "<标题>"' % path
+        what = 'PRD'
+    else:
+        cmd = '$PY "$R/.qoder/scripts/orchestration/wlkj.py" return prototype "%s" "<功能>"' % path
+        what = '原型(HTML)'
+    msg = (
+        "\n⚠️ 【回流前必须先问用户】刚写了 %s：%s\n"
+        "  → 必须先问用户：「要把这份 %s 回流到平台吗？」\n"
+        "  → 用户【明确同意】（说「回流 / 上传 / 确认」）才执行：%s\n"
+        "  → 用户拒绝 / 未明确同意 → 只留本地，绝不默认上传。\n"
+        "  （宿主确定性要求：不问就传 = 违规；这是用户明确要的确认环节）\n"
+    ) % (what, name, what, cmd)
+    try:
+        sys.stdout.write(msg)
+    except Exception:
+        pass
+
+
 def _append_event(event, data):
     """best-effort append 一条事件到当前 dev 的 feedback.jsonl。避开 learn import 断链, 直接写文件。"""
     try:
@@ -198,6 +224,8 @@ def main():
                     _append_event('prd_written', {'prd': os.path.basename(p), 'tool': tool_name})
                     # ★ 宿主能力接入：PRD 写完 → 自动调 MCP learn 沉淀业务规则
                     _auto_learn_from_prd(p)
+                    # ★ 用户硬要求：PRD 回流平台前必须先问（不问就传=违规）
+                    _emit_ask_before_return('prd', p)
                 if plow.endswith('task.json'):
                     _append_event('task_touched', {'task_json': os.path.basename(os.path.dirname(p)), 'tool': tool_name})
                 # ★ 宿主能力接入：代码文件改完 → 自动记 code_changed（知识层下次检测过时用）
@@ -205,6 +233,8 @@ def main():
                     _append_event('code_changed', {'file': os.path.basename(p), 'path': plow[:200]})
                 # v3.1 软提示: prototype HTML 写完但缺来源锚定 → stderr 提醒(不阻断)
                 if 'prototype' in os.path.basename(plow) and plow.endswith('.html'):
+                    # ★ 用户硬要求：原型回流平台前必须先问（不问就传=违规）
+                    _emit_ask_before_return('prototype', p)
                     try:
                         if os.path.isfile(p):
                             with open(p, encoding='utf-8', errors='ignore') as _f:

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)  # 永不阻塞