@hupan56/wlkj 2.5.0 → 2.7.0

package/templates/qoder/scripts/install_qoderwork.py

@@ -23,6 +23,7 @@ QoderWork 从 %USERPROFILE%\\.qoderwork\\skills\\ 加载技能（每个子目录
 """
 
 import argparse
+import json
 import os
 import shutil
 import subprocess
@@ -52,6 +53,50 @@ else:
     _HOME = Path.home()
 QODERWORK_SKILLS_DIR = _HOME / ".qoderwork" / "skills"
 QODERWORK_COMMANDS_DIR = _HOME / ".qoderwork" / "commands"
+QODERWORK_MCP_FILE = _HOME / ".qoderwork" / "mcp.json"
+
+# Playwright MCP 配置 (让 QoderWork 的 AI 能用 Playwright 工具驱动浏览器, 快且稳)
+# 可选增强: 不装 playwright 也不影响 QoderWork 自带连接器
+# ⚠️ 实测: npx @playwright/mcp@latest 冷启动 11.7s, QoderWork 会超时杀掉 → 工具注册失败
+#    → AI 瞎猜工具名(BrowserNavigate) → 全失败。
+#    解法: 优先用"全局已装的 cli.js + node 直启"(0.57s), 探测不到才回退 npx。
+def _find_playwright_mcp_cli():
+    """探测全局已装的 @playwright/mcp/cli.js 路径, 返回 (node, cli.js) 或 None。
+    Windows 下 npm 是 .cmd, 必须 shell=True 才能调到。"""
+    import shutil, subprocess
+    node = shutil.which("node")
+    if not node:
+        return None
+    # npm root -g 拿全局 node_modules 路径 (最准, 跨自定义prefix)
+    try:
+        out = subprocess.run("npm root -g", capture_output=True, text=True,
+                             shell=True, timeout=10)
+        root = out.stdout.strip()
+        if root:
+            cli = os.path.join(root, "@playwright", "mcp", "cli.js")
+            if os.path.isfile(cli):
+                return node, cli
+    except Exception:
+        pass
+    # 回退: 试常见全局路径
+    for cand in [os.path.join(os.path.expanduser("~"), "AppData", "Roaming", "npm",
+                              "node_modules", "@playwright", "mcp", "cli.js")]:
+        if os.path.isfile(cand):
+            return node, cand
+    return None
+
+
+def _playwright_mcp_entry():
+    """生成 playwright MCP 配置: 优先 node+cli.js(快), 回退 npx(慢但通用)。"""
+    found = _find_playwright_mcp_cli()
+    if found:
+        node, cli = found
+        return {"playwright": {"command": node, "args": [cli]}}
+    # 回退: npx (冷启动慢, 可能被 QoderWork 超时杀, 但通用)
+    return {"playwright": {"command": "npx", "args": ["@playwright/mcp@latest"]}}
+
+
+PLAYWRIGHT_MCP_ENTRY = _playwright_mcp_entry()
 
 # 旧的错误路径 (迁移清理用)
 LEGACY_WRONG_DIR = _HOME / ".qoderworkcn" / "skills"
@@ -197,6 +242,165 @@ def uninstall_one(name: str) -> str:
         return f"error: {e}"
 
 
+def setup_playwright_mcp(dry=False):
+    """把 Playwright MCP 写进 ~/.qoderwork/mcp.json (可选增强, 不影响自带连接器)。
+
+    让 QoderWork 的 AI 能用 Playwright 工具(browser_navigate/click/type/screenshot)
+    驱动浏览器 —— 快(10s级 vs 自带连接器 5-8分钟)、稳(不折腾 tab)、不依赖扩展。
+    已有 playwright 配置则跳过(幂等); 没有则追加。
+    """
+    import json
+    mcp_path = QODERWORK_MCP_FILE
+    # 读现有配置
+    existing = {}
+    if mcp_path.is_file():
+        try:
+            existing = json.loads(mcp_path.read_text(encoding="utf-8"))
+            if not isinstance(existing, dict):
+                existing = {}
+        except (json.JSONDecodeError, OSError):
+            existing = {}
+    servers = existing.get("mcpServers", {})
+    if not isinstance(servers, dict):
+        servers = {}
+    if "playwright" in servers:
+        # 已有配置: 若是旧的 npx@latest(慢, 易超时失败), 升级成 node+cli.js(快)
+        cur = servers["playwright"]
+        if cur.get("command") == "npx":
+            new_entry = PLAYWRIGHT_MCP_ENTRY["playwright"]
+            if new_entry.get("command") != "npx":  # 探测到了 cli.js
+                if dry:
+                    return "would-upgrade"
+                servers["playwright"] = new_entry
+                existing["mcpServers"] = servers
+                mcp_path.write_text(json.dumps(existing, indent=2, ensure_ascii=False), encoding="utf-8")
+                return "upgraded"
+        return "ok-existing"
+    if dry:
+        return "would-add"
+    # 追加 playwright
+    servers["playwright"] = PLAYWRIGHT_MCP_ENTRY["playwright"]
+    existing["mcpServers"] = servers
+    mcp_path.parent.mkdir(parents=True, exist_ok=True)
+    mcp_path.write_text(
+        json.dumps(existing, indent=2, ensure_ascii=False) + "\n",
+        encoding="utf-8",
+    )
+    return "added"
+
+
+def setup_lanhu_mcp(dry=False):
+    """注册蓝湖 MCP 到 ~/.qoderwork/mcp.json (可选增强, 设计稿直读)。
+
+    蓝湖 MCP (dsphper/lanhu-mcp) 是第三方 HTTP server, 让设计师发蓝湖链接,
+    AI 直读设计稿(颜色/尺寸/字体/切图)。注册成 url 模式 (Streamable HTTP),
+    跟 playwright/kg/mysql 这些 stdio server 并列。
+
+    注意: 蓝湖是 HTTP server, 需要用户手动启动
+      python .qoder/scripts/setup_lanhu.py start
+    (启动脚本自动读当前开发者的 cookie 注入, 不用手动配 .env)
+    本函数只注册 url (幂等), 不负责 clone/装依赖/启动 —— 那些由 setup_lanhu.py
+    在 setup.py Step 4.5 完成。这里只做 mcp.json 条目同步, 保证团队所有人
+    跑 install_qoderwork 后都能看到蓝湖 server。
+
+    返回: 'ok-existing' / 'added' / 'would-add'
+    """
+    # 复用 setup_lanhu.py 的注册逻辑 (单一事实源)
+    try:
+        sys.path.insert(0, str(SCRIPT_DIR))
+        from setup_lanhu import register_lanhu_mcp
+    except ImportError:
+        return 'would-add'
+    return register_lanhu_mcp(developer_name='Developer', dry=dry)
+
+
+def detect_python_command():
+    """探测本机可用的 python 命令名 (跨平台)。
+    Windows: py > python (无 python3); mac/linux: python3 > python。
+    返回命令名字符串 (在 PATH 里的), 失败返回 None。"""
+    for cand in (('py', '-3'), ('python3', None), ('python', None)):
+        name, extra = cand
+        if not shutil.which(name):
+            continue
+        cmd = [name] + ([extra] if extra else []) + ['--version']
+        try:
+            r = subprocess.run(cmd, capture_output=True, text=True,
+                               encoding='utf-8', errors='replace')
+            if r.returncode == 0:
+                # 返回完整调用前缀 (py -3 / python3 / python)
+                return ' '.join(cmd[:-1])
+        except (OSError, subprocess.SubprocessError):
+            continue
+    return None
+
+
+def rewrite_mcp_json_for_launcher():
+    """把 mcp.json 里所有 python MCP (kg/mysql/lanhu) 改成 launcher 模式。
+
+    旧 (写死路径): {"command": "E:\\...\\python.exe", "args": ["E:\\...\\server.py"]}
+    新 (launcher): {"command": "<py cmd>", "args": ["<home>/.qoderwork/mcp_launcher.py", "<name>"]}
+
+    幂等: 已是 launcher 模式 (args 含 mcp_launcher.py) 则跳过。
+    全新机器: mcp.json 不存在时创建空骨架 + 写入 kg/mysql/lanhu 条目。
+    mysql env 抽到 ~/.qoderwork/.mcp-mysql-env.json (launcher 读)。
+    """
+    # 全新机器: mcp.json 不存在 → 创建空骨架 (playwright 后续会追加)
+    if not QODERWORK_MCP_FILE.is_file():
+        QODERWORK_MCP_FILE.parent.mkdir(parents=True, exist_ok=True)
+        QODERWORK_MCP_FILE.write_text(
+            '{"mcpServers": {}}\n', encoding="utf-8")
+    try:
+        cfg = json.loads(QODERWORK_MCP_FILE.read_text(encoding="utf-8"))
+    except (json.JSONDecodeError, OSError):
+        return
+    servers = cfg.get("mcpServers", {})
+    if not isinstance(servers, dict):
+        return
+
+    py_cmd = detect_python_command() or "python"
+    launcher = str(_HOME / ".qoderwork" / "mcp_launcher.py")
+    # MCP 名 → launcher 子命令
+    name_map = {
+        "qoder-knowledge-graph": "kg",
+        "qoder-mysql": "mysql",
+        "lanhu": "lanhu",
+    }
+    changed = False
+    for srv, sub in name_map.items():
+        entry = servers.get(srv)
+        # 全新机器: kg/mysql/lanhu 条目可能不存在 → 创建它 (launcher 模式)
+        if not entry:
+            servers[srv] = {"command": py_cmd, "args": [launcher, sub]}
+            changed = True
+            continue
+        # 已是 launcher 模式?
+        args = entry.get("args", []) if isinstance(entry, dict) else []
+        if args and any("mcp_launcher.py" in str(a) for a in args):
+            continue
+        # mysql env 抽离到独立文件
+        if srv == "qoder-mysql" and isinstance(entry, dict) and entry.get("env"):
+            env_file = _HOME / ".qoderwork" / ".mcp-mysql-env.json"
+            try:
+                env_file.write_text(json.dumps(entry["env"], indent=2, ensure_ascii=False) + "\n",
+                                    encoding="utf-8")
+            except OSError:
+                pass
+        # 改成 launcher 模式
+        servers[srv] = {"command": py_cmd, "args": [launcher, sub]}
+        changed = True
+
+    if changed:
+        cfg["mcpServers"] = servers
+        try:
+            QODERWORK_MCP_FILE.write_text(
+                json.dumps(cfg, indent=2, ensure_ascii=False) + "\n", encoding="utf-8")
+            print(f"\n--- mcp.json 重写为 launcher 模式 ---")
+            print(f"  [UPD]  kg/mysql/lanhu → command={py_cmd}, args=[mcp_launcher.py, <名>]")
+            print(f"         不再写死仓库路径/venv路径, 跨机器通用")
+        except OSError as e:
+            print(f"  [ERR]  mcp.json 重写失败: {e}")
+
+
 def main():
     parser = argparse.ArgumentParser(
         description="把 .qoder/skills/ 安装到 QoderWork 桌面端目录"
@@ -209,8 +413,49 @@ def main():
                         help="用拷贝代替 junction（非 Windows 或不想软链时用）")
     parser.add_argument("--force-commands", action="store_true",
                         help="强制覆盖已存在的 command 文件（升级时用；默认已存在则跳过）")
+    parser.add_argument("--mcp-only", action="store_true",
+                        help="只刷新 launcher + mcp.json，不装 skill（update 时调）")
     args = parser.parse_args()
 
+    # --mcp-only: 轻量模式, 只刷 launcher + mcp.json, 不装 skill
+    if args.mcp_only:
+        print("=" * 56)
+        print("MCP 配置刷新 (--mcp-only)")
+        print("=" * 56)
+        # 1. 复制 launcher
+        launcher_src = SOURCE_COMMANDS_DIR.parent / "scripts" / "mcp_launcher.py"
+        launcher_dst = _HOME / ".qoderwork" / "mcp_launcher.py"
+        try:
+            launcher_dst.parent.mkdir(parents=True, exist_ok=True)
+            if launcher_src.is_file():
+                import shutil as _sh
+                _sh.copy2(str(launcher_src), str(launcher_dst))
+                print(f"  [OK]   launcher 已刷新: {launcher_dst}")
+            else:
+                print(f"  [ERR]  源不存在: {launcher_src}")
+        except OSError as e:
+            print(f"  [ERR]  复制失败: {e}")
+        # 2. 重写 mcp.json 为 launcher 模式 (创建 kg/mysql/lanhu)
+        rewrite_mcp_json_for_launcher()
+        # 3. 补 playwright (rewrite 不含它)
+        try:
+            setup_playwright_mcp(dry=False)
+        except Exception:
+            pass
+        # 4. 写 .repo-root 锚点 (launcher 定位仓库根用)
+        # 用 PROJECT_ROOT (脚本自身定位的仓库根) 而非 cwd —— 修"安装目录不一致"bug:
+        # 用户可能从任意目录跑 install_qoderwork (home/其他项目), cwd ≠ 仓库根,
+        # 写 cwd 会让 .repo-root 指向错误目录 → MCP 全找不到仓库。
+        # PROJECT_ROOT 由本文件位置推出 (scripts/ -> .qoder/ -> repo), 与 cwd 无关。
+        anchor = _HOME / ".qoderwork" / ".repo-root"
+        try:
+            anchor.parent.mkdir(parents=True, exist_ok=True)
+            anchor.write_text(str(PROJECT_ROOT), encoding="utf-8")
+            print(f"  [OK]   .repo-root 锚点: {anchor} -> {PROJECT_ROOT}")
+        except OSError:
+            pass
+        return
+
     print("=" * 56)
     print("QoderWork 技能安装器")
     print(f"  源:   {SOURCE_SKILLS_DIR}")
@@ -308,8 +553,8 @@ def main():
     if action in ("install", "check") and SOURCE_COMMANDS_DIR.is_dir():
         print("\n--- Commands (/wl-* 斜杠命令) ---")
         cmd_count = {"ok": 0, "new": 0, "upd": 0}
-        for cmd_file in sorted(SOURCE_COMMANDS_DIR.glob("wl-*.md")):
-            name = cmd_file.name  # e.g. wl-prd.md
+        for cmd_file in sorted(SOURCE_COMMANDS_DIR.rglob("*.md")):
+            name = cmd_file.name  # e.g. wl-prd.md (子目录归类的也平铺到目标)
             target = QODERWORK_COMMANDS_DIR / name
             if action == "check":
                 if target.exists():
@@ -318,19 +563,21 @@ def main():
                     print(f"  [MISS] {name}")
             else:  # install
                 need_copy = False
+                is_update = False
                 if not target.exists():
                     need_copy = True
-                elif args.force_commands:
-                    # 强制刷新（升级场景）：仅当内容不同才覆盖，避免无谓写
-                    import shutil
+                else:
+                    # 始终按内容同步(幂等): 不同就刷新。
+                    # command 是纯文本小文件, 内容不同说明源已升级, 必须覆盖,
+                    # 否则 QoderWork 用旧 command → 走错流程(如旧 wl-test 搜代码)。
                     if not file_equal(cmd_file, target):
                         need_copy = True
+                        is_update = True
                 if need_copy:
                     try:
                         QODERWORK_COMMANDS_DIR.mkdir(parents=True, exist_ok=True)
-                        import shutil
                         shutil.copy2(str(cmd_file), str(target))
-                        if target.exists() and args.force_commands:
+                        if is_update:
                             cmd_count["upd"] += 1
                         else:
                             cmd_count["new"] += 1
@@ -343,11 +590,121 @@ def main():
             msg += f" / {cmd_count['upd']} 刷新"
         print(msg)
 
+    # === 2.5. repo_root.py (QoderWork 桌面端仓库根定位器) ===
+    # QoderWork 桌面端工作目录是临时会话目录, 不是仓库根, 相对路径失效。
+    # repo_root.py 从 mcp.json 反推仓库根, 命令文件靠它定位脚本/模板。
+    if action in ("install", "check"):
+        repo_root_src = SOURCE_COMMANDS_DIR.parent / "scripts" / "repo_root.py"
+        repo_root_dst = _HOME / ".qoderwork" / "repo_root.py"
+        if repo_root_src.is_file():
+            if action == "check":
+                if repo_root_dst.exists():
+                    print(f"\n--- repo_root.py (仓库根定位器) ---")
+                    print(f"  [OK]   {repo_root_dst}")
+                else:
+                    print(f"\n--- repo_root.py (仓库根定位器) ---")
+                    print(f"  [MISS] 未安装到 {repo_root_dst}")
+            else:
+                try:
+                    repo_root_dst.parent.mkdir(parents=True, exist_ok=True)
+                    shutil.copy2(str(repo_root_src), str(repo_root_dst))
+                    print(f"\n--- repo_root.py (仓库根定位器) ---")
+                    print(f"  [OK]   已安装: {repo_root_dst}")
+                except OSError as e:
+                    print(f"\n--- repo_root.py (仓库根定位器) ---")
+                    print(f"  [ERR]  {e}")
+
+    # === 2.6. mcp_launcher.py + .repo-root (统一 MCP 启动器, 消除写死路径) ===
+    # mcp_launcher.py 让 mcp.json 不再写死仓库路径/venv路径, 跨 Win/Mac 通用。
+    # .repo-root 锚点文件让 launcher 能定位仓库根 (不依赖 mcp.json 里的路径)。
+    if action in ("install", "check"):
+        print(f"\n--- mcp_launcher.py (统一 MCP 启动器) ---")
+        launcher_src = SOURCE_COMMANDS_DIR.parent / "scripts" / "mcp_launcher.py"
+        launcher_dst = _HOME / ".qoderwork" / "mcp_launcher.py"
+        anchor_dst = _HOME / ".qoderwork" / ".repo-root"
+        if action == "check":
+            if launcher_dst.exists():
+                print(f"  [OK]   launcher 已装: {launcher_dst}")
+            else:
+                print(f"  [MISS] launcher 未装")
+            if anchor_dst.exists():
+                print(f"  [OK]   .repo-root 锚点在: {anchor_dst}")
+            else:
+                print(f"  [MISS] .repo-root 锚点缺失 (launcher 会回退向上找)")
+        else:
+            try:
+                launcher_dst.parent.mkdir(parents=True, exist_ok=True)
+                if launcher_src.is_file():
+                    shutil.copy2(str(launcher_src), str(launcher_dst))
+                    print(f"  [OK]   launcher 已装: {launcher_dst}")
+                else:
+                    print(f"  [ERR]  源不存在: {launcher_src}")
+                # 写 .repo-root 锚点 (记录当前仓库根, launcher 读它定位)
+                anchor_dst.write_text(str(PROJECT_ROOT) + "\n", encoding="utf-8")
+                print(f"  [OK]   .repo-root 锚点已写: {PROJECT_ROOT}")
+            except OSError as e:
+                print(f"  [ERR]  {e}")
+
+    # === 2.7. 重写 mcp.json: 所有 python MCP 改 launcher 模式 (消除写死路径) ===
+    if action == "install":
+        rewrite_mcp_json_for_launcher()
+
+    # === 3. Playwright MCP (可选增强: 让 QoderWork 能用 Playwright 驱动浏览器) ===
+    if action in ("install", "check"):
+        print("\n--- Playwright MCP (浏览器自动化增强, 可选) ---")
+        pw_status = setup_playwright_mcp(dry=(action == "check"))
+        if pw_status == "ok-existing":
+            print(f"  [OK]   playwright MCP 已配置: {QODERWORK_MCP_FILE}")
+        elif pw_status == "added":
+            print(f"  [NEW]  已写入 playwright MCP: {QODERWORK_MCP_FILE}")
+            print("         QoderWork 重启后, AI 可用 Playwright 工具驱动浏览器(快、稳)")
+        elif pw_status == "upgraded":
+            print(f"  [FIX]  playwright MCP 已升级: npx(慢,易超时) → node+cli.js(0.5s启动)")
+            print("         ⚠️ 必须 重启 QoderWork 生效。旧 npx 配置会被 QoderWork 超时杀掉导致工具注册失败")
+        elif pw_status == "would-upgrade":
+            print(f"  [FIX]  检测到旧 npx 配置(慢), 安装时会升级为 node+cli.js")
+        elif pw_status == "would-add":
+            print(f"  [MISS] 未配置 playwright MCP, 安装时会写入 {QODERWORK_MCP_FILE}")
+        # 检查 playwright mcp 是否已装(优先看 cli.js, 决定能否走快路径)
+        pw_cli = _find_playwright_mcp_cli()
+        if pw_cli:
+            print(f"  启动方式: node + cli.js (0.5s, 快) ← {pw_cli[1]}")
+        else:
+            npx_ok = shutil.which("npx") is not None
+            print(f"  启动方式: {'npx (11s冷启动, 可能被QoderWork超时杀)' if npx_ok else '✗ 未装 Node.js'}")
+            if npx_ok:
+                print("  提示: 装 @playwright/mcp 全局后重跑本脚本, 会自动切换到快的 node 启动: npm i -g @playwright/mcp")
+
+    # === 4. 蓝湖 MCP (可选增强: 设计师发蓝湖链接, AI 直读设计稿) ===
+    if action in ("install", "check"):
+        print("\n--- 蓝湖 MCP (设计稿直读增强, 可选) ---")
+        lh_status = setup_lanhu_mcp(dry=(action == "check"))
+        if lh_status == "ok-existing":
+            print(f"  [OK]   蓝湖 MCP 已注册: {QODERWORK_MCP_FILE}")
+        elif lh_status == "added":
+            print(f"  [NEW]  已写入蓝湖 MCP (STDIO): {QODERWORK_MCP_FILE}")
+            print("         QoderWork 开自动起、关自动停; 命令内授权调 mcp__lanhu__*")
+        elif lh_status == "upgraded":
+            print(f"  [UPD]  蓝湖 MCP 升级 HTTP→STDIO: {QODERWORK_MCP_FILE}")
+            print("         重启 QoderWork 后自动开关, 无需手动 start")
+        elif lh_status == "would-add":
+            print(f"  [MISS] 未注册蓝湖 MCP, 安装时会写入 {QODERWORK_MCP_FILE}")
+        # STDIO 模式由 QoderWork 自动管, 不查端口; 只确认 wrapper + 源码就绪
+        wrapper = PROJECT_ROOT / ".qoder" / "scripts" / "lanhu_stdio_wrapper.py"
+        lanhu_dir = PROJECT_ROOT / "external" / "lanhu-mcp"
+        if wrapper.is_file():
+            print(f"  wrapper 就绪: {wrapper.name} (按角色读 cookie)")
+        else:
+            print(f"  [MISS] wrapper 缺失: {wrapper}")
+        if not lanhu_dir.is_dir():
+            print("  蓝湖源码: 未 clone (跑 setup.py 或 setup_lanhu.py 安装)")
+
     if action == "install" and counts["err"] == 0:
         print("\n✓ 安装完成。重启 QoderWork（或新建对话）后技能生效。")
         print("  在 QoderWork 里直接用自然语言即可，例如：")
-        print('    "写个保险异常筛选的 PRD"  -> prd-generator + prototype-generator')
-        print('    "查一下考勤代码在哪"      -> wl-search')
+        print('    "写个保险异常筛选的 PRD"  -> /wl-prd-full（正经需求）')
+        print('    "列表加个导出按钮"       -> /wl-prd-quick（小改动）')
+        print('    "查一下考勤代码在哪"      -> /wl-search')
     elif action == "install" and counts["err"] > 0:
         print(f"\n⚠ 有 {counts['err']} 个安装失败，见上方 [ERR] 行。")
         print("  常见原因: 目标路径被占用为非软链。手动删除该目录后重跑本脚本。")