@hupan56/wlkj 2.7.12 → 3.1.2

package/templates/qoder/scripts/kg_mcp_server.py

@@ -1,801 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-"""
-Knowledge Graph MCP Server for QoderWork (pure Python, no SDK dependency).
-
-Exposes the QODER knowledge graph as MCP tools that QoderWork's AI agent
-can call directly. Uses the MCP stdio protocol (JSON-RPC 2.0 over stdin/stdout)
-implemented from scratch — no external dependencies beyond the standard library.
-
-Registered in ~/.qoderwork/mcp.json as "qoder-knowledge-graph".
-Tools surface as mcp__qoder-knowledge-graph__<tool> to the AI.
-
-Each tool captures the stdout of the existing search_index/context_pack
-functions, so no logic is duplicated — this is a thin stdio→MCP wrapper.
-"""
-import io
-import os
-import sys
-import json
-import asyncio
-
-# Ensure UTF-8 output
-try:
-    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
-    sys.stderr.reconfigure(encoding='utf-8', errors='replace')
-except Exception:
-    pass
-
-# Locate the scripts dir
-SCRIPTS_DIR = os.path.dirname(os.path.abspath(__file__))
-BASE_DIR = os.path.dirname(os.path.dirname(SCRIPTS_DIR))  # repo root (.qoder 的上级)
-sys.path.insert(0, SCRIPTS_DIR)
-
-# Suppress all non-JSON output on stdout (MCP protocol uses stdout exclusively)
-# Redirect any stray prints to stderr
-_real_stdout = sys.stdout
-sys.stdout = sys.stderr  # default: prints go to stderr
-
-
-def _capture(func, *args, **kwargs):
-    """Run a function, capture its stdout (which the scripts print to)."""
-    old = sys.stdout
-    buf = io.StringIO()
-    sys.stdout = buf
-    try:
-        func(*args, **kwargs)
-    except Exception as e:
-        buf.write('\n[error] %s' % str(e))
-    finally:
-        sys.stdout = old
-    return buf.getvalue().strip()
-
-
-# Lazy imports
-_search = None
-_context = None
-
-def _ensure_imports():
-    global _search, _context
-    if _search is None:
-        import search_index as _search
-    if _context is None:
-        import context_pack as _context
-
-
-# ── MCP Tool Definitions ─────────────────────────────────────────────
-
-TOOLS = [
-    {
-        "name": "search_code",
-        "description": "搜索代码: 按关键词查找代码文件在哪。例: search_code(keyword='考勤') 或 search_code(keyword='车辆', platform='web')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "keyword": {"type": "string", "description": "搜索关键词(中文或英文)"},
-                "platform": {"type": "string", "enum": ["web", "app"],
-                             "description": "平台: web=PC管理端, app=移动端。可选"},
-            },
-            "required": ["keyword"],
-        },
-    },
-    {
-        "name": "search_api",
-        "description": "搜索API端点: 按关键词查找后端接口。例: search_api(keyword='salary')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {"keyword": {"type": "string", "description": "API关键词"}},
-            "required": ["keyword"],
-        },
-    },
-    {
-        "name": "search_prd",
-        "description": "搜索PRD: 查已有需求文档,防止重复造轮子。例: search_prd(keyword='考勤')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {"keyword": {"type": "string", "description": "PRD关键词"}},
-            "required": ["keyword"],
-        },
-    },
-    {
-        "name": "get_impact",
-        "description": "影响分析: 改某接口/路径会影响哪些前端页面和按钮?评估改动风险。例: get_impact(endpoint='/asset') 或 get_impact(endpoint='export')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {"endpoint": {"type": "string",
-                             "description": "API端点路径或关键词, 如 /asset/list 或 export"}},
-            "required": ["endpoint"],
-        },
-    },
-    {
-        "name": "context_360",
-        "description": "360度视图: 查看符号(端点/函数/处理器)的全部关联信息。例: context_360(symbol='handleExport') 或 context_360(symbol='/system/role')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {"symbol": {"type": "string", "description": "符号名: 端点/函数名/处理器名"}},
-            "required": ["symbol"],
-        },
-    },
-    {
-        "name": "context_pack",
-        "description": "上下文打包: 一次返回关键词相关的全部上下文(代码+页面+字段+PRD+API+布局指纹)。避免来回查。例: context_pack(keyword='车辆', platform='web')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "keyword": {"type": "string", "description": "业务关键词"},
-                "platform": {"type": "string", "enum": ["web", "app"],
-                             "description": "平台: web或app。可选"},
-                "page_type": {"type": "string",
-                              "description": "页面类型: table-page/form-page/detail-page。可选"},
-                "role": {"type": "string", "enum": ["pm", "design", "dev", "test", "admin"],
-                         "description": "角色裁剪: pm=字段+PRD+API, design=页面+布局+风格, dev=代码+API, test=代码+PRD+API。可选"},
-            },
-            "required": ["keyword"],
-        },
-    },
-    {
-        "name": "table_impact",
-        "description": "改表影响分析: 改某数据库表会影响哪些 Entity 类和接口? 跨层遍历(数据层→代码层)。例: table_impact(table='quality_case') → 找到 QualityCase 类 + 影响的 Controller/接口",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "table": {"type": "string", "description": "数据库表名 (如 quality_case)"},
-            },
-            "required": ["table"],
-        },
-    },
-    {
-        "name": "coverage_matrix",
-        "description": "功能×测试覆盖矩阵: 哪些功能有测试覆盖, 哪些是空白? 标出回归测试盲区。例: coverage_matrix()",
-        "inputSchema": {
-            "type": "object",
-            "properties": {},
-        },
-    },
-    {
-        "name": "feature_overview",
-        "description": "功能画像: 一个功能的完整画像(端点+按钮+测试+页面)。例: feature_overview(feature='资产管理') 或 feature_overview(feature='asset')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "feature": {"type": "string", "description": "功能名(中/英文都行): 资产管理/asset/hr"},
-            },
-            "required": ["feature"],
-        },
-    },
-    {
-        "name": "get_workflow",
-        "description": "业务流程: 查看一个模块的完整操作链(query→create→audit→archive→export→delete)。数据源: DuckDB workflows 表(168条/13模块)。优先匹配英文模块名(assets/cost/hr/maintenance等), 中文输入会返回候选模块列表。例: get_workflow(module='assets') 或 get_workflow(module='cost')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "module": {"type": "string", "description": "模块名(英文优先): assets/cost/hr/maintenance/operation/qualityAndSafety/veh/settings/dashboard/flow/systems/monitor/quasafe"},
-            },
-            "required": ["module"],
-        },
-    },
-    {
-        "name": "search_wiki",
-        "description": "搜索Repo Wiki: 查Qoder IDE生成的模块级语义文档(架构/职责/依赖), 比读源码省token。写PRD前优先查。数据源: wiki-index.json(730关键词)。注意: 不会自动构建索引, 索引缺失会提示手动跑。例: search_wiki(keyword='考勤') 或 search_wiki(keyword='权限')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {"keyword": {"type": "string", "description": "Wiki关键词(中/英文)"}},
-            "required": ["keyword"],
-        },
-    },
-    {
-        "name": "multi_hop",
-        "description": "知识图谱多跳遍历: 从一个符号出发, 沿edges递归N跳, 找出全部可达实体(页面→接口→控制器→测试)。知识图谱核心能力。数据源: DuckDB edges表(8251边)。支持别名解析(传中文/端点/函数名自动转entity_id)。例: multi_hop(symbol='资产管理') 或 multi_hop(symbol='/asset/list', max_depth=3)",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "symbol": {"type": "string", "description": "起点符号: 可传entity_id(E:/asset/list)、别名(资产管理)、端点(/asset/list)或函数名"},
-                "max_depth": {"type": "integer", "default": 3, "description": "最大跳数(1-5), 默认3。越大越全但越慢"},
-            },
-            "required": ["symbol"],
-        },
-    },
-    {
-        "name": "search_field",
-        "description": "搜索字段: 查某个字段在哪些页面用过、中文标题是什么。例: search_field(field='plateNo') 或 search_field(field='车牌号')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "field": {"type": "string", "description": "字段名(英文field或中文标题)"},
-            },
-            "required": ["field"],
-        },
-    },
-    {
-        "name": "list_modules",
-        "description": "模块清单: 列出项目所有业务模块及其中文名。例: list_modules() 或 list_modules(platform='web')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "platform": {"type": "string", "enum": ["web", "app"], "description": "平台过滤。可选"},
-            },
-        },
-    },
-    {
-        "name": "fill_prototype",
-        "description": "原型预填: 用真实代码数据(表格列/表单字段/布局参数)自动填模板,输出80%完成的HTML草稿+数据清单。AI只需微调20%。出原型前必须调。例: fill_prototype(keyword='营业外合同', platform='web')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "keyword": {"type": "string", "description": "业务关键词"},
-                "platform": {"type": "string", "enum": ["web", "app"],
-                             "description": "平台: web或app。默认web"},
-            },
-            "required": ["keyword"],
-        },
-    },
-    {
-        "name": "get_design_system",
-        "description": "设计系统查询: 返回项目完整设计规范(布局指纹/颜色token/页面模式/组件表/图标规则/behaviors)。出任何原型前必须先调。例: get_design_system() 或 get_design_system(platform='web')",
-        "inputSchema": {
-            "type": "object",
-            "properties": {
-                "platform": {"type": "string", "enum": ["web", "app"],
-                             "description": "平台: 只返回指定平台的指纹。不传则返回全部"},
-            },
-        },
-    },
-]
-
-
-
-
-# Role-based section filtering for context_pack output
-ROLE_SECTIONS = {
-    'pm':     {'1', '2', '2b', '3', '4', '5', '7'},  # PM: no code files, no style refs
-    'design': {'2', '2b', '3', '6'},                  # Design: pages+layout+fields+style
-    'dev':    {'1', '3', '5', '7'},                   # Dev: code+fields+API+wiki
-    'test':   {'1', '4', '5'},                        # Test: code+PRD+API
-    'admin':  {'4', '5'},                             # Admin: PRD+API overview
-}
-
-def _filter_by_role(text, role):
-    """Filter context_pack output by role: keep only relevant sections."""
-    import re
-    visible = ROLE_SECTIONS.get(role.lower())
-    if not visible:
-        return text
-    sections = re.split(r'(\n## \d+[ab]?\.)', text)
-    result = sections[0]  # header before first section
-    i = 1
-    while i < len(sections):
-        header = sections[i]
-        num_m = re.match(r'\n## (\d+[ab]?)\.', header)
-        if num_m and num_m.group(1) in visible:
-            result += header
-            if i + 1 < len(sections):
-                result += sections[i + 1]
-            i += 2
-        else:
-            i += 2  # skip this section
-    return result.strip()
-
-
-# ── Tool implementations (added 2026-06-17) ──────────────────────────
-# 三个工具的底层能力均已实测: get_workflow/multi_hop 走 DuckDB(有数据),
-# search_wiki 走 wiki-index.json(730词)。均不依赖已失效的旧 JSON 索引。
-
-def _tool_get_workflow(module):
-    """业务操作链查询。直查 DuckDB workflows 表, 带脏数据清洗 + 模糊匹配。
-
-    审核: workflows 表 168 行/13 模块, 脏数据约 25 行(endpoint='None' 或
-    button 泄漏模板语法 '{{'), 清洗后链路完整。模块名均为英文, 中文输入会
-    模糊查 entities.cn 返回候选。
-    """
-    import kg_duckdb as K
-    q = (module or '').strip()
-    if not q:
-        return "请提供模块名。可选: assets/cost/hr/maintenance/operation/qualityAndSafety/veh/settings/dashboard/flow/systems/monitor/quasafe"
-    con = K.get_db()
-    try:
-        # 1) 精确 + 模糊匹配 workflows.module (英文)
-        hits = con.execute(
-            "SELECT DISTINCT module FROM workflows WHERE module = ? OR module LIKE '%' || ? || '%' ORDER BY module",
-            [q.lower(), q.lower()]
-        ).fetchall()
-        if not hits:
-            # 2) 中文兜底: 查 entities 里 cn 匹配的, 给候选
-            cn_hits = con.execute(
-                "SELECT canonical, cn FROM entities WHERE type IN ('FEATURE','MODULE') AND cn <> '' AND (cn LIKE '%' || ? OR cn LIKE '%' || ? || '%')",
-                [q, q]
-            ).fetchall()
-            if cn_hits:
-                cands = ', '.join('%s(%s)' % (c[0], c[1]) for c in cn_hits[:8])
-                return "未在 workflows 表找到模块 '%s'。但 entities 表有相近的中文名, 可换这些英文模块名重试:\n  %s" % (q, cands)
-            all_mods = [r[0] for r in con.execute("SELECT DISTINCT module FROM workflows ORDER BY module").fetchall()]
-            return "未找到模块 '%s'。现有模块: %s" % (q, ', '.join(all_mods))
-
-        out = []
-        for (mod,) in hits:
-            # 清洗: 过滤空端点和模板泄漏按钮
-            rows = con.execute("""
-                SELECT state, seq, button, endpoint FROM workflows
-                WHERE module = ?
-                  AND endpoint <> 'None' AND endpoint <> ''
-                  AND button NOT LIKE '{{%'
-                ORDER BY seq, state
-            """, [mod]).fetchall()
-            if not rows:
-                continue
-            # 按 seq 分组展示操作链
-            out.append('=== %s (%d 步, 已清洗脏数据) ===' % (mod, len(rows)))
-            cur_seq = -1
-            for st, sq, btn, ep in rows:
-                if sq != cur_seq:
-                    out.append('  [%d] %s' % (sq, st))
-                    cur_seq = sq
-                out.append('        %-16s -> %s' % (btn, ep))
-            out.append('')
-        return '\n'.join(out).strip() or "模块 '%s' 清洗后无有效操作链" % q
-    finally:
-        try: con.close()
-        except Exception: pass
-
-
-def _tool_search_wiki(keyword):
-    """Repo Wiki 搜索。只读 wiki-index.json, 不触发自动构建(慢操作, 不适合 MCP 同步调用)。
-
-    审核: repowiki.search_wiki 函数在, wiki-index.json 有 730 关键词。
-    search_index.search_wiki_cli 会在索引缺失时调 build_wiki_index (可能跑很久),
-    这里绕开它直接调底层 search_wiki, 索引缺失即报错让用户手动跑。
-    """
-    kw = (keyword or '').strip()
-    if not kw:
-        return "请提供搜索关键词"
-    from common.paths import DATA_INDEX_DIR as _idx, PROJECT_ROOT as _base
-    wiki_path = os.path.join(str(_idx), 'wiki-index.json')
-    if not os.path.isfile(wiki_path):
-        return ("wiki-index.json 不存在。请在终端手动构建(一次性):\n"
-                "  python .qoder/scripts/search_index.py wiki\n"
-                "(需要先在 Qoder IDE 对项目生成 Repo Wiki)")
-    try:
-        from common.repowiki import search_wiki
-        from common.terms import get_cn_map_with_auto
-    except ImportError as e:
-        return "Wiki 搜索模块缺失: %s" % str(e)
-    code_dir = os.path.join(_base, 'data', 'code')
-    try:
-        cn_map = get_cn_map_with_auto()
-        results = search_wiki(kw, str(_idx), code_dir, cn_map)
-    except Exception as e:
-        return "Wiki 搜索出错: %s" % str(e)
-    if not results:
-        return "No wiki pages match: %s" % kw
-    out = ['Wiki pages matching: %s (%d found)' % (kw, len(results)), '']
-    for i, r in enumerate(results, 1):
-        out.append('  %d. [%s] %s' % (i, r.get('project', '?'), r.get('title', '')))
-        deps = r.get('dependent_files') or []
-        if deps:
-            out.append('     依赖: %s%s' % (', '.join(deps[:2]),
-                      (' (共 %d 个文件)' % len(deps)) if len(deps) > 2 else ''))
-        if r.get('md_path'):
-            try:
-                rel = os.path.relpath(r['md_path'], _base)
-            except ValueError:
-                rel = r['md_path']
-            out.append('     文档: %s' % rel)
-    out.append('')
-    out.append('  提示: 读 wiki 文档 (带行号引用) 比读源码更省 token')
-    return '\n'.join(out)
-
-
-def _tool_multi_hop(symbol, max_depth=3):
-    """知识图谱多跳遍历。先别名解析, 再递归 CTE 遍历。
-
-    审核: kg_duckdb.multi_hop 实测从 E:/code 3 跳到 C:SysSmsCodeController。
-    edges 表 8251 边。用户可能传中文名/端点/函数名, 需先走 aliases 解析成
-    entity_id。
-    """
-    import kg_duckdb as K
-    sym = (symbol or '').strip()
-    if not sym:
-        return "请提供起点符号 (entity_id / 端点 / 函数名 / 中文别名)"
-    try:
-        depth = int(max_depth)
-    except (TypeError, ValueError):
-        depth = 3
-    depth = max(1, min(5, depth))  # 钳制 1-5
-    con = K.get_db()
-    try:
-        # 1) 尝试解析为 entity_id: 直接匹配 / 别名表 / 模糊匹配 entity_id
-        start_id = sym
-        # 1a) 别名精确匹配 (小写)
-        resolved = con.execute(
-            "SELECT entity_id FROM aliases WHERE alias = ?", [sym.lower()]
-        ).fetchone()
-        if resolved:
-            start_id = resolved[0]
-        else:
-            # 1b) entity_id 模糊匹配 (用户传端点 /asset/list 或函数名)
-            fuzzy = con.execute(
-                "SELECT id FROM entities WHERE id = ? OR id LIKE '%' || ? || '%' LIMIT 1",
-                [sym, sym]
-            ).fetchone()
-            if fuzzy:
-                start_id = fuzzy[0]
-            else:
-                # 1c) edges 里直接含该符号作为端点
-                edge_hit = con.execute(
-                    "SELECT from_id FROM edges WHERE from_id LIKE '%' || ? || '%' LIMIT 1",
-                    [sym]
-                ).fetchone()
-                if edge_hit:
-                    start_id = edge_hit[0]
-
-        # 2) 递归 CTE 遍历
-        try:
-            hops = K.multi_hop(con, start_id, max_depth=depth)
-        except Exception as e:
-            return "遍历出错 (起点=%s): %s" % (start_id, str(e))
-
-        if not hops or (len(hops) == 1 and hops[0][1] == 0):
-            return "从 '%s' (解析为 %s) 出发无可达节点。可能该符号不在图谱中, 或无出边。" % (sym, start_id)
-
-        # 3) 按 depth 分组, 取每层的实体详情 (最多展示)
-        from collections import defaultdict
-        by_depth = defaultdict(list)
-        for eid, d in hops:
-            by_depth[d].append(eid)
-
-        out = ['=== 多跳遍历: %s (解析为 %s) ===' % (sym, start_id)]
-        out.append('可达 %d 个实体, 最大 %d 跳:' % (len(hops), max(by_depth.keys())))
-        out.append('')
-        for d in sorted(by_depth.keys()):
-            eids = by_depth[d]
-            out.append('[第 %d 跳] %d 个实体:' % (d, len(eids)))
-            # 拉每个实体的 canonical/cn 信息
-            shown = 0
-            for eid in eids:
-                if shown >= 12:
-                    out.append('  ... 还有 %d 个' % (len(eids) - 12))
-                    break
-                info = con.execute(
-                    "SELECT type, canonical, cn FROM entities WHERE id = ?", [eid]
-                ).fetchone()
-                if info:
-                    t, c, cn = info
-                    label = '%s / %s' % (c, cn) if cn and cn != c else c
-                    out.append('  %-14s %-12s %s' % ('[%s]' % t, eid[:40], label))
-                else:
-                    out.append('  %-14s %s' % ('[edge]', eid[:60]))
-                shown += 1
-            out.append('')
-        return '\n'.join(out).strip()
-    finally:
-        try: con.close()
-        except Exception: pass
-
-
-def _tool_search_field(field):
-    from common.paths import DATA_INDEX_DIR
-    from kg_duckdb import get_db
-    con = get_db()
-    rows = con.execute(
-        "SELECT field, title, count, sample_files FROM fields WHERE LOWER(field) LIKE ? OR LOWER(title) LIKE ? LIMIT 20",
-        ['%' + field.lower() + '%', '%' + field.lower() + '%']
-    ).fetchall()
-    con.close()
-    if not rows:
-        return "未找到字段 '%s'" % field
-    lines = ["字段搜索 '%s' (%d个匹配)" % (field, len(rows))]
-    for f, t, c, s in rows:
-        lines.append("  %s -> %s (用了%d次)" % (f, t, c))
-    return '\n'.join(lines)
-
-
-def _tool_list_modules(platform=''):
-    from kg_duckdb import get_db
-    con = get_db()
-    if platform == 'web':
-        rows = con.execute("SELECT dir, cn, project FROM modules WHERE project LIKE '%fywl%' ORDER BY dir").fetchall()
-    elif platform == 'app':
-        rows = con.execute("SELECT dir, cn, project FROM modules WHERE project LIKE '%Carmg%' OR project LIKE '%H5%' ORDER BY dir").fetchall()
-    else:
-        rows = con.execute("SELECT dir, cn, project FROM modules ORDER BY project, dir").fetchall()
-    con.close()
-    if not rows:
-        return "无模块数据"
-    lines = ["模块清单 (%d个)" % len(rows)]
-    cur = ""
-    for d, cn, proj in rows:
-        if proj != cur:
-            cur = proj
-            lines.append("[%s]" % proj)
-        lines.append("  %s -> %s" % (d, cn))
-    return '\n'.join(lines)
-
-
-def _execute_tool(name, arguments):
-    """Execute a tool and return result text."""
-    _ensure_imports()
-
-    if name == "search_code":
-        kw = arguments.get("keyword", "")
-        plat = arguments.get("platform") or ""
-        return _capture(_search.search_keywords, kw, plat) or "无匹配结果"
-
-    elif name == "search_api":
-        return _capture(_search.search_api, arguments.get("keyword", "")) or "无匹配结果"
-
-    elif name == "search_prd":
-        return _capture(_search.search_prds, arguments.get("keyword", "")) or "无匹配PRD"
-
-    elif name == "get_impact":
-        return _capture(_search.search_impact, arguments.get("endpoint", "")) or "未找到匹配端点"
-
-    elif name == "context_360":
-        return _capture(_search.search_context360, arguments.get("symbol", "")) or "未找到匹配符号"
-
-    elif name == "context_pack":
-        kw = arguments.get("keyword", "")
-        plat = arguments.get("platform", "")
-        pt = arguments.get("page_type", "")
-        role = arguments.get("role", "")
-        full = _capture(_context._compute_and_print, kw, plat, pt)
-        if not full:
-            return "无匹配上下文"
-        if role:
-            return _filter_by_role(full, role)
-        return full
-
-    elif name == "fill_prototype":
-        # 调用 fill_prototype.py 的 main 逻辑 (捕获 stdout)
-        kw = arguments.get("keyword", "")
-        plat = arguments.get("platform", "web")
-        import subprocess as _sp
-        script = os.path.join(SCRIPTS_DIR, 'fill_prototype.py')
-        try:
-            r = _sp.run([sys.executable, script, kw, '--platform', plat],
-                        capture_output=True, text=True, encoding='utf-8',
-                        errors='replace', timeout=30, cwd=SCRIPTS_DIR)
-            return r.stdout or "fill_prototype 无输出 (可能索引缺失)"
-        except Exception as e:
-            return "[error] fill_prototype: %s" % str(e)
-
-    elif name == "get_design_system":
-        # 返回 DESIGN.md 内容 (若不存在则生成)
-        import subprocess as _sp
-        design_md = os.path.join(BASE_DIR, 'data', 'design', 'DESIGN.md')
-        if not os.path.isfile(design_md):
-            gen_script = os.path.join(SCRIPTS_DIR, 'gen_design_doc.py')
-            if os.path.isfile(gen_script):
-                _sp.run([sys.executable, gen_script], capture_output=True, timeout=30)
-        if os.path.isfile(design_md):
-            content = open(design_md, encoding='utf-8').read()
-            # 若指定了 platform, 只返回该平台的布局指纹段
-            plat = arguments.get("platform")
-            if plat:
-                lines = content.split('\n')
-                result = []
-                in_section = False
-                for line in lines:
-                    if line.startswith('### ') and plat.lower() in line.lower():
-                        in_section = True
-                    elif line.startswith('### ') and in_section:
-                        break
-                    if in_section:
-                        result.append(line)
-                return '\n'.join(result) if result else content
-            return content
-        return "DESIGN.md 不存在且生成失败。请手动跑: python gen_design_doc.py"
-
-    elif name == "table_impact":
-        # 改表影响分析: 表 → entity_table_map → Entity → 代码图谱 Controller/接口
-        from common.paths import DATA_INDEX_DIR as _idx
-        import duckdb as _ddb
-        db_path = str(_idx / 'kg_db.duckdb')
-        if not os.path.isfile(db_path):
-            return "数据层图谱未构建。请先跑: python .qoder/scripts/kg_build_db.py"
-        table = arguments.get("table", "").strip()
-        try:
-            con = _ddb.connect(db_path, read_only=True)
-        except Exception as e:
-            return "数据层图谱暂时不可用(可能正在重建): %s。稍后重试。" % str(e)[:60]
-        try:
-            # 1. 表 → Entity
-            ents = con.execute(
-                "SELECT DISTINCT entity_class FROM entity_table_map WHERE table_name = ?",
-                [table]
-            ).fetchall()
-            ent_list = [r[0] for r in ents]
-            if not ent_list:
-                return "表 '%s' 在 entity_table_map 中无映射 (可能该表无 @TableName 注解的 Entity)" % table
-            lines = ["改表影响分析: %s" % table]
-            lines.append("  对应 Entity: %s" % ', '.join(ent_list))
-            # 2. 表的真实列 (给个概览)
-            cols = con.execute(
-                "SELECT column_name, data_type, column_comment FROM db_columns "
-                "WHERE table_name = ? LIMIT 10", [table]
-            ).fetchall()
-            if cols:
-                lines.append("  表字段 (%d+):" % len(cols))
-                for cn, dt, cc in cols:
-                    lines.append("    %s (%s) %s" % (cn, dt, (cc or '')[:30]))
-            # 3. Entity → 代码图谱 (attach kg.duckdb 查接口/Controller)
-            try:
-                con.execute("ATTACH '%s' AS code (READ_ONLY)" % str(_idx / 'kg.duckdb'))
-                lines.append("  影响的接口/控制器:")
-                found = 0
-                for ec in ent_list:
-                    refs = con.execute(
-                        "SELECT type, canonical FROM code.entities "
-                        "WHERE contains(canonical, ?) AND type IN ('CONTROLLER','ENDPOINT') "
-                        "LIMIT 12", [ec]
-                    ).fetchall()
-                    for t, c in refs:
-                        lines.append("    [%s] %s" % (t, c))
-                        found += 1
-                if not found:
-                    lines.append("    (代码图谱中未找到引用 %s 的接口)" % ent_list[0])
-                con.execute("DETACH code")
-            except Exception as e:
-                lines.append("  (代码图谱查询失败: %s)" % str(e)[:60])
-            return '\n'.join(lines)
-        finally:
-            con.close()
-
-    elif name == "coverage_matrix":
-        from common.graph_traverse import CodeGraph
-        from common.paths import DATA_INDEX_DIR as _idx
-        g = CodeGraph(str(_idx))
-        result = g.coverage_matrix()
-        return result or "无法生成覆盖矩阵"
-
-    elif name == "feature_overview":
-        from common.graph_traverse import CodeGraph
-        from common.paths import DATA_INDEX_DIR as _idx
-        g = CodeGraph(str(_idx))
-        result = g.feature_overview(arguments.get("feature", ""))
-        if result and '不存在' not in result:
-            return result
-        # DuckDB 兜底: entity-registry.json 缺失时, 用 DuckDB 直接查
-        feat = arguments.get("feature", "")
-        fl = feat.lower()
-        try:
-            import duckdb as _ddb
-            con = _ddb.connect(str(_idx / 'kg.duckdb'), read_only=True)
-            # 端点
-            eps = con.execute(
-                "SELECT canonical FROM entities WHERE type='ENDPOINT' AND "
-                "(lower(canonical) LIKE ? OR lower(canonical) LIKE ?) LIMIT 15",
-                ['%'+fl+'%', '%'+feat+'%']).fetchall()
-            # 按钮
-            btns = con.execute(
-                "SELECT canonical FROM entities WHERE type='BUTTON' AND "
-                "lower(canonical) LIKE ? LIMIT 10", ['%'+fl+'%']).fetchall()
-            con.close()
-            lines = ['功能画像: %s (DuckDB兜底)' % feat]
-            if eps:
-                lines.append('端点(%d):' % len(eps))
-                for (e,) in eps[:10]:
-                    lines.append('  %s' % e)
-            if btns:
-                lines.append('按钮(%d):' % len(btns))
-                for (b,) in btns[:5]:
-                    lines.append('  %s' % b)
-            if not eps and not btns:
-                lines.append('  (无匹配。换关键词重试)')
-            return chr(10).join(lines)
-        except Exception as e:
-            return '未找到功能 "%s" (DuckDB兜底也失败: %s)' % (feat, str(e)[:60])
-
-    elif name == "get_workflow":
-        return _tool_get_workflow(arguments.get("module", ""))
-
-    elif name == "search_wiki":
-        return _tool_search_wiki(arguments.get("keyword", ""))
-
-    elif name == "multi_hop":
-        return _tool_multi_hop(arguments.get("symbol", ""),
-                               arguments.get("max_depth", 3))
-
-    elif name == "search_field":
-        return _tool_search_field(arguments.get("field", ""))
-
-    elif name == "list_modules":
-        return _tool_list_modules(arguments.get("platform", ""))
-
-    return "未知工具: %s" % name
-
-
-# ── MCP Stdio Protocol (JSON-RPC 2.0) ─────────────────────────────────
-
-def _send(msg):
-    """Send a JSON-RPC message on stdout (MCP transport)."""
-    _real_stdout.write(json.dumps(msg, ensure_ascii=False) + '\n')
-    _real_stdout.flush()
-
-
-async def _handle(req):
-    """Handle a single JSON-RPC request."""
-    method = req.get("method", "")
-    req_id = req.get("id")
-    params = req.get("params", {})
-
-    if method == "initialize":
-        _send({
-            "jsonrpc": "2.0", "id": req_id,
-            "result": {
-                "protocolVersion": "2024-11-05",
-                "capabilities": {"tools": {}},
-                "serverInfo": {"name": "qoder-knowledge-graph", "version": "1.0.0"},
-            }
-        })
-
-    elif method == "notifications/initialized":
-        pass  # no response needed for notifications
-
-    elif method == "tools/list":
-        _send({"jsonrpc": "2.0", "id": req_id, "result": {"tools": TOOLS}})
-
-    elif method == "tools/call":
-        tool_name = params.get("name", "")
-        arguments = params.get("arguments", {})
-        try:
-            result_text = await asyncio.get_event_loop().run_in_executor(
-                None, _execute_tool, tool_name, arguments)
-            _send({
-                "jsonrpc": "2.0", "id": req_id,
-                "result": {
-                    "content": [{"type": "text", "text": result_text}],
-                    "isError": False,
-                }
-            })
-        except Exception as e:
-            _send({
-                "jsonrpc": "2.0", "id": req_id,
-                "result": {
-                    "content": [{"type": "text", "text": "[error] %s" % str(e)}],
-                    "isError": True,
-                }
-            })
-
-    elif method == "ping":
-        _send({"jsonrpc": "2.0", "id": req_id, "result": {}})
-
-    else:
-        if req_id is not None:
-            _send({"jsonrpc": "2.0", "id": req_id,
-                   "error": {"code": -32601, "message": "Method not found: %s" % method}})
-
-
-def main():
-    """Read JSON-RPC requests from stdin (synchronous readline), dispatch.
-
-    Synchronous is more reliable than asyncio on Windows Proactor event loop.
-    Tool execution runs in a thread pool to avoid blocking on long queries.
-    """
-    import threading
-
-    while True:
-        try:
-            line = sys.stdin.buffer.readline()
-        except (EOFError, OSError):
-            break
-        if not line:
-            break
-        line = line.strip()
-        if not line:
-            continue
-        try:
-            req = json.loads(line.decode('utf-8'))
-            # Handle synchronously (tools are fast enough; context_pack ~1-2s)
-            loop = asyncio.new_event_loop()
-            loop.run_until_complete(_handle(req))
-            loop.close()
-        except json.JSONDecodeError:
-            pass
-        except Exception as e:
-            print('[kg-mcp] error: %s' % str(e), file=sys.stderr)
-
-
-if __name__ == "__main__":
-    try:
-        main()
-    except KeyboardInterrupt:
-        pass