@hupan56/wlkj 2.6.0 → 2.7.0

package/templates/qoder/scripts/init_doctor.py

@@ -24,6 +24,8 @@ Usage:
   7. PRD 模板 (docx 源 + 蒸馏 md, docx 更新会提示重新蒸馏)
   8. 周五自动构建 (检测 + 给出当前系统的设置命令)
   9. QoderWork 桌面端技能 (可选, 检查 ~/.qoderwork/skills/ 的 junction)
+  10. 个人测试数据 (可选, 登录类 /wl-test 用)
+  11. 蓝湖 MCP (可选, 设计师直读蓝湖设计稿)
 
 Exit: 0 = 全部健康, 1 = 有问题未修复 (--fix 可修的会自动修)
 """
@@ -111,6 +113,81 @@ def check_basics():
         say(WARN, 'git 未安装 — 团队同步/源码克隆将禁用, 本地 PRD/搜索/任务/报告仍可用')
 
 
+# ============================================================
+# 1.5 Python 依赖 (pip 包) — 新增: 不装则图谱/MCP/测试全崩
+# ============================================================
+
+# (import名, pip包名, 用途, 是否核心)
+PIP_DEPS = [
+    ('duckdb', 'duckdb', '知识图谱存储', True),
+    ('pymysql', 'pymysql', 'MySQL 只读 MCP', True),
+    ('yaml', 'PyYAML', 'config/learn 配置解析', True),
+    ('requests', 'requests', 'API 调用/通知', True),
+    ('tree_sitter', 'tree-sitter', '代码图谱构建(管理员)', False),
+    ('tree_sitter_typescript', 'tree-sitter-typescript', 'Vue/TS 解析(管理员)', False),
+    ('playwright', 'playwright', '浏览器测试(QA)', False),
+    ('ddddocr', 'ddddocr', '验证码识别(测试)', False),
+    ('cryptography', 'cryptography', '飞书集成加密(可选)', False),
+]
+
+
+def check_dependencies(fix=False):
+    """检查 pip 依赖装没装。--fix 时用镜像兜底安装 (失败切国内源重试)。"""
+    print('\n--- 1.5 Python 依赖 ---')
+    missing_core = []
+    missing_optional = []
+    for mod, pkg, usage, is_core in PIP_DEPS:
+        try:
+            __import__(mod)
+            say(OK, '%s (%s)' % (pkg, usage))
+        except ImportError:
+            if is_core:
+                missing_core.append(pkg)
+                problem('%s 未装 (%s)' % (pkg, usage),
+                        'pip install %s' % pkg)
+            else:
+                missing_optional.append(pkg)
+                say(WARN, '%s 未装 (%s, 按需装)' % (pkg, usage))
+
+    if fix and (missing_core or missing_optional):
+        req = os.path.join(BASE, 'requirements.txt')
+        if os.path.isfile(req):
+            # 用镜像兜底安装: 默认源失败 → 自动切清华/阿里/腾讯
+            from common.pip_install import install_with_fallback
+            print('  → 安装: pip install -r requirements.txt (失败自动切国内镜像)')
+            result = install_with_fallback(['-r', req])
+            if result['ok']:
+                src = result['source']
+                say(FIX, '依赖安装成功' + ('' if src == 'default' else ' (镜像源: %s)' % src))
+                # 装完后重新验证 (旧版装完不复查, 会把装失败误报为已装)
+                missing_core = []
+                for mod, pkg, usage, is_core in PIP_DEPS:
+                    if is_core:
+                        try:
+                            __import__(mod)
+                        except ImportError:
+                            missing_core.append(pkg)
+            else:
+                print('  [MISS] ' + result['hint'])
+                # 重新验证, 看哪些真的还缺
+                missing_core = []
+                for mod, pkg, usage, is_core in PIP_DEPS:
+                    if is_core:
+                        try:
+                            __import__(mod)
+                        except ImportError:
+                            missing_core.append(pkg)
+        else:
+            print('  requirements.txt 不存在, 手动装: pip install ' +
+                  ' '.join(missing_core + missing_optional))
+
+    if missing_core:
+        print('\n  ⚠️ 核心依赖缺失会导致知识图谱/MCP/配置全部不可用!')
+        print('  一键修复: python .qoder/scripts/init_doctor.py --fix')
+        print('  (装包慢/失败: pip config set global.index-url '
+              'https://pypi.tuna.tsinghua.edu.cn/simple)')
+
+
 # ============================================================
 # 2. 开发者身份
 # ============================================================
@@ -271,18 +348,35 @@ def check_source_repos(fix, config):
         else:
             problem('{} 源码缺失'.format(name), '加 --fix 自动克隆 ' + url)
 
+    # 校验 platforms 映射的项目名是否真实存在于 data/code/ (最常见的静默失败源:
+    # 管理员改了 git_sync.projects 的 URL 却忘了改 platforms.project -> 所有
+    # --platform 查询静默返回 0, 用户到生成原型时才发现啥都没有)
+    platforms = config.get('platforms') or {}
+    cloned = [d for d in os.listdir(CODE_DIR)] if os.path.isdir(CODE_DIR) else []
+    for pkey, pinfo in platforms.items():
+        proj_name = (pinfo or {}).get('project')
+        if not proj_name:
+            continue
+        if proj_name not in cloned:
+            problem(
+                'platforms.{}.project = {!r} 但 data/code/ 下无此目录'.format(pkey, proj_name),
+                'config.yaml platforms.{}.project 改成实际项目目录名 (现有: {})'.format(
+                    pkey, ', '.join(cloned) or '空'))
+        else:
+            say(OK, 'platform {} -> {} 目录存在'.format(pkey, proj_name))
+
 
 # ============================================================
 # 5. 知识图谱新鲜度 (增量的核心逻辑)
 # ============================================================
 
 def index_age_days():
-    """索引年龄: 取 .index-meta.json 的 last_sync, 退化用文件 mtime"""
-    meta_path = os.path.join(INDEX_DIR, '.index-meta.json')
-    ki_path = os.path.join(INDEX_DIR, 'keyword-index.json')
-    if not os.path.isfile(ki_path):
-        return None  # 索引不存在
-    ts = None
+    """索引年龄: 用 kg.duckdb 的 mtime (DuckDB 统一存储后不再依赖 JSON)"""
+    kg_path = os.path.join(INDEX_DIR, 'kg.duckdb')
+    if not os.path.isfile(kg_path):
+        return None
+    import time as _time
+    return int((_time.time() - os.path.getmtime(kg_path)) / 86400)
     if os.path.isfile(meta_path):
         try:
             import json
@@ -296,22 +390,61 @@ def index_age_days():
     return (datetime.now() - ts).total_seconds() / 86400
 
 
+def _has_source_code():
+    """data/code/ 下是否有真实项目 (非空目录)。"""
+    if not os.path.isdir(CODE_DIR):
+        return False
+    return any(d for d in os.listdir(CODE_DIR) if not d.startswith('.'))
+
+
+def print_degraded_status():
+    """无源码/无索引时的降级声明 —— 让新同事第一时间知道工作流什么能用。
+    在 main() 体检汇总后调用, 只在 data/code/ 为空时打印。
+    """
+    if _has_source_code():
+        return  # 有源码, 不需要降级声明
+
+    print('\n' + '=' * 56)
+    print('  ⚠ 当前没有源码 (data/code/ 为空), 工作流处于降级状态')
+    print('=' * 56)
+    print('  以下功能需要源码, 当前不可用:')
+    print('    /wl-search      搜代码 (无源码 = 搜不到)')
+    print('    /wl-prd-full    完整 PRD 的字段/API 锚定 (会跳过, 原型可用)')
+    print('    知识图谱 MCP     search_code/api 等工具 (启动但返回空)')
+    print()
+    print('  以下功能不依赖源码, 当前可用:')
+    print('    /wl-prd-quick   零星需求 (纯文案/小改动)')
+    print('    /wl-task        任务管理')
+    print('    /wl-report      日报周报')
+    print('    /wl-commit      代码提交 (提交的是工作流本身的改动)')
+    print()
+    print('  如何补上源码 (任选其一):')
+    print('    A. 配团队源码仓库: 打开 .qoder/config.yaml 的 git_sync.projects,')
+    print('       填 url, 然后 wlkj init 会自动 clone')
+    print('    B. 手动放代码: 把项目代码放到 data/code/<项目名>/ 下')
+    print('    C. 拿团队的图谱: team_sync.py pull (管理员已建好的 kg.duckdb)')
+    print('=' * 56)
+
+
 def check_index(fix):
     print('\n--- 5. 知识图谱 (data/index/) ---')
     age = index_age_days()
     if age is None:
         # 完全缺失 -> 需要全量 (唯一允许"重头再来"的情形)
-        if fix:
-            if os.path.isdir(CODE_DIR) and os.listdir(CODE_DIR):
+        if _has_source_code():
+            # 有源码 → 可以建索引
+            if fix:
                 print('       索引缺失, 全量构建 (一次性, 之后都是增量) ...')
                 if run_script('git_sync.py', '--index-only'):
                     actions.append('全量构建知识图谱')
                 else:
                     problem('索引构建失败, 查看上方报错')
             else:
-                problem('索引缺失且无源码', '先解决第 4 项再重跑')
+                problem('知识图谱不存在', '加 --fix 全量构建 (仅首次)')
         else:
-            problem('知识图谱不存在', '加 --fix 全量构建 (仅首次)')
+            # 无源码 → 建不了索引, 醒目告警 (不能只说"先解决第4项", 太含糊)
+            problem('索引缺失且无源码 (data/code/ 为空)',
+                    '无法构建知识图谱。看下方"降级状态"说明, 配源码后重跑 init')
         return
     if age <= FRESH_DAYS:
         say(OK, '知识图谱 {:.1f} 天前更新过, 新鲜 (阈值 {} 天), 跳过'.format(age, FRESH_DAYS))
@@ -332,37 +465,57 @@ def check_index(fix):
 # ============================================================
 
 def check_style(fix):
-    print('\n--- 6. 风格约束 ---')
-    required = {
-        'vben-style-reference.json': 'Web 设计 Token (Vben HSL)',
-        'chart-style-reference.json': '看板/大屏风格先例',
-        'icon-reference.json': '真实图标库 (antd SVG + Vant, 原型禁 emoji)',
-        'style-index.json': 'UI 模式索引 (--style/--field 数据源)',
-        'style-meta.json': '风格摘要 (会话注入)',
+    print('\n--- 6. 知识图谱 + 设计参考 ---')
+
+    # 手工参考文件 (必须存在, 提交在 git 里)
+    ref_files = {
+        'ref-vben-style.json': 'Web 设计 Token (Vben HSL)',
+        'ref-chart-style.json': '看板/大屏风格先例',
+        'ref-icon.json': '真实图标库 (antd SVG + Vant, 原型禁 emoji)',
     }
-    missing_buildable = False
-    for fname, desc in required.items():
-        if os.path.isfile(os.path.join(INDEX_DIR, fname)):
+    for fname, desc in ref_files.items():
+        fpath = os.path.join(INDEX_DIR, fname)
+        if os.path.isfile(fpath):
             say(OK, '{} - {}'.format(fname, desc))
-        elif fname in ('style-index.json', 'style-meta.json'):
-            missing_buildable = True
-            if not fix:
-                problem(fname + ' 缺失', '加 --fix 自动构建')
-        else:
-            problem(fname + ' 缺失', '该文件提交在 git 中, 先解决第 3 项团队同步')
-    tpls = ['prototype-web.html', 'prototype-app.html']
-    for t in tpls:
-        if os.path.isfile(os.path.join(TPL_DIR, t)):
-            say(OK, '原型模板 ' + t)
         else:
-            problem('原型模板缺失: ' + t)
-    if missing_buildable and fix:
-        if os.path.isdir(CODE_DIR) and os.listdir(CODE_DIR):
-            print('       构建风格索引 ...')
-            if run_script('build_style_index.py'):
-                actions.append('构建风格索引')
-        else:
-            problem('风格索引缺源码无法构建', '先解决第 4 项')
+            problem(fname + ' 缺失', '该文件提交在 git 中, 先解决团队同步 (team_sync pull)')
+
+    # kg.duckdb 知识图谱 (统一存储)
+    kg_path = os.path.join(INDEX_DIR, 'kg.duckdb')
+    role = ''
+    try:
+        dev_file = os.path.join(BASE, '.qoder', '.developer')
+        if os.path.isfile(dev_file):
+            for _line in open(dev_file, encoding='utf-8'):
+                if _line.strip().startswith('role='):
+                    role = _line.split('=', 1)[1].strip()
+    except Exception:
+        pass
+
+    if os.path.isfile(kg_path):
+        say(OK, 'kg.duckdb 知识图谱 (%d KB)' % (os.path.getsize(kg_path) // 1024))
+        # 过期检测
+        import time as _t
+        age_h = (_t.time() - os.path.getmtime(kg_path)) / 3600
+        if age_h > 168:
+            if role in ('admin', 'dev'):
+                print('       [WARN] kg.duckdb 已 %.0f 小时未更新, 建议重建' % age_h)
+                if fix:
+                    print('       重建知识图谱 (kg_build.py) ...')
+                    if run_script('kg_build.py'):
+                        actions.append('知识图谱重建')
+            else:
+                print('       [WARN] kg.duckdb 可能过期, 建议 team_sync pull')
+    elif role in ('admin', 'dev') and os.path.isdir(CODE_DIR) and os.listdir(CODE_DIR):
+        if fix:
+            print('       构建知识图谱 (kg_build.py, 首次2-4分钟) ...')
+            if run_script('kg_build.py'):
+                actions.append('知识图谱构建 (kg.duckdb)')
+                print('       提示: 运行 kg_semantic.py --check 标注模块中文名')
+    elif role not in ('admin', 'dev'):
+        problem('kg.duckdb 缺失', '非管理员: team_sync.py pull 获取管理员构建的知识图谱')
+    else:
+        problem('kg.duckdb 缺源码无法构建', '先解决源码 clone')
 
 
 # ============================================================
@@ -493,6 +646,95 @@ def check_qoderwork():
         print('       安装: python .qoder/scripts/install_qoderwork.py')
 
 
+# ============================================================
+# 10. 浏览器测试 · 个人测试数据文档 (可选, 测试时才需要)
+# ============================================================
+
+def check_autotest_data():
+    """检查个人测试数据文档是否就绪 (账号/密码, 登录类测试要用)。
+
+    这是个人文档 (members/{dev}/autotest-data.yaml), 已 gitignored 不共享。
+    缺了不阻塞普通工作流, 只是 /wl-test 登录类用例时会卡在"需要测试数据"。
+    """
+    print('\n--- 10. 浏览器测试 · 个人测试数据 (可选) ---')
+    dev = get_developer()
+    if not dev:
+        return  # 没初始化开发者, 前面 check_developer 已报过
+    data_doc = os.path.join(BASE, 'workspace', 'members', dev, 'autotest-data.yaml')
+    if os.path.isfile(data_doc):
+        say(OK, '个人测试数据文档就绪: workspace/members/%s/autotest-data.yaml' % dev)
+        print('       /wl-test 登录类用例会自动从中取账号密码 (已 gitignored, 不共享)')
+        return
+    say(WARN, '尚未准备个人测试数据文档 (登录类 /wl-test 会问你要账号密码)')
+    print('       生成模板: python .qoder/scripts/autotest.py init-data')
+    print('       然后打开 workspace/members/%s/autotest-data.yaml 填测试账号' % dev)
+    print('       (个人文档, 已 gitignored, 不会 push 给团队)')
+    print('       不填也能用: 每次 /wl-test 时临时 --data 传或对话里给)')
+
+
+# ============================================================
+# 11. 蓝湖 MCP (可选增强, 不阻塞)
+# ============================================================
+
+def check_lanhu():
+    """检查蓝湖 MCP 是否就绪 (第三方 HTTP server, 设计师直读蓝湖设计稿)。
+
+    三态检查:
+      - mcp.json 有 lanhu 条目 + 服务在跑 → OK
+      - mcp.json 有 lanhu 但服务没起 → WARN (用时再起)
+      - 完全没配 → MISS (团队不用蓝湖可忽略; 用的话跑 setup.py)
+    蓝湖是 opt-in 增强, 任何状态都不算 issue (不进未解决列表, 不阻塞)。
+    """
+    print('\n--- 11. 蓝湖 MCP (可选增强) ---')
+    import json
+    import socket
+
+    home = os.path.expanduser('~')
+    mcp_file = os.path.join(home, '.qoderwork', 'mcp.json')
+    if not os.path.isfile(mcp_file):
+        say(WARN, '未配置蓝湖 MCP (团队不用蓝湖可忽略)')
+        print('       用的话跑: python .qoder/scripts/setup.py  或  setup_lanhu.py')
+        return
+
+    try:
+        cfg = json.loads(open(mcp_file, encoding='utf-8').read())
+        servers = cfg.get('mcpServers', {})
+    except (ValueError, OSError):
+        say(WARN, 'mcp.json 解析失败, 跳过蓝湖检查')
+        return
+
+    if 'lanhu' not in servers:
+        say(WARN, '未配置蓝湖 MCP (团队不用蓝湖可忽略)')
+        print('       用的话跑: python .qoder/scripts/setup.py  或  setup_lanhu.py')
+        return
+
+    entry = servers['lanhu']
+    args = entry.get('args', []) if isinstance(entry, dict) else []
+    is_launcher = (isinstance(entry, dict) and 'command' in entry
+                   and any('mcp_launcher.py' in str(a) for a in args))
+    is_stdio = isinstance(entry, dict) and 'command' in entry and 'url' not in entry
+
+    if is_launcher:
+        # launcher 模式 (最佳): QoderWork 自动开关, 不写死路径
+        launcher = os.path.join(os.path.expanduser('~'), '.qoderwork', 'mcp_launcher.py')
+        say(OK, '蓝湖 MCP 已注册 (STDIO launcher, QoderWork 自动开关)')
+        if os.path.isfile(launcher):
+            print('       开 QoderWork 自动起, 关自动停, 无需手动 start')
+            print('       launcher 按当前 .developer 角色读 cookie (角色隔离)')
+        else:
+            say(WARN, 'launcher 缺失: ' + launcher)
+            print('       重跑: python .qoder/scripts/install_qoderwork.py')
+    elif is_stdio:
+        # 旧 STDIO (写死 venv 路径), 建议升级 launcher
+        say(WARN, '蓝湖 MCP 是旧 STDIO 格式 (command 写死路径)')
+        print('       建议升级 launcher 模式 (跨机器通用): python .qoder/scripts/install_qoderwork.py')
+    else:
+        # 旧 HTTP 模式
+        url = entry.get('url', '?')
+        say(WARN, '蓝湖 MCP 是旧 HTTP 模式: ' + url[:50])
+        print('       建议升级 STDIO launcher (QoderWork 自动开关): python .qoder/scripts/setup_lanhu.py')
+
+
 # ============================================================
 # Main
 # ============================================================
@@ -511,6 +753,7 @@ def main():
 
     config = load_config()
     check_basics()
+    check_dependencies(fix)
     check_developer(fix, name, role)
     check_team_sync(fix)
     check_source_repos(fix, config)
@@ -519,19 +762,28 @@ def main():
     check_prd_templates()
     check_weekly(fix)
     check_qoderwork()
+    check_autotest_data()
+    check_lanhu()
 
     print('\n' + '=' * 56)
     if actions:
         print('本次修复: ')
         for a in actions:
             print('  + ' + a)
+
+    # 无源码时的降级声明 (不算 issue, 但必须让用户知道当前能用什么)
+    print_degraded_status()
+
     if issues:
-        print('未解决问题 ({}):'.format(len(issues)))
+        print('\n未解决问题 ({}):'.format(len(issues)))
         for i in issues:
             print('  - ' + i)
         print('=' * 56)
         return 1
-    print('环境健康, 可以开始工作: /wl-prd <需求描述>')
+    if _has_source_code():
+        print('环境健康, 可以开始工作: /wl-prd-full <需求描述>')
+    else:
+        print('工作流核心已就绪 (见上方降级说明补源码)。')
     print('=' * 56)
     return 0