gjm
f33d42afe9
标准 MCP 客户端（Claude Desktop / CodeBuddy / MCP Inspector）只实现 stdio 与 Streamable HTTP，无法直连 cad_mcp_frame 的自定义裸 TCP。新增 mcp_bridge 包在 客户端与基座之间做协议适配，CAD 侧零改动： - 合成基座缺失的生命周期：initialize（含协议版本协商、capabilities、serverInfo） 与 ping；notifications/* 按规范不回包。 - tools/list、tools/call 原样转发到 127.0.0.1:8080，result 与顶层 error 忠实透传， 仅把 id 改写回客户端原始 id。 - 并发：读线程即时应答 initialize/ping，单工作线程串行转发工具调用（保证顺序、 慢工具不阻塞 ping）。 - stdout 只走 sys.stdout.buffer 并加锁，规避 Windows 的 \n→\r\n 翻译； 诊断信息一律走 stderr。 - CAD 未启动时桥照常启动，工具调用回 -32603（可重试）。 - 入口：python -m mcp_bridge，支持 --host/--port/--timeout/--verbose/--print-config。 dwg_review/mcp_client.py 增加 raw_rpc()（_rpc 新增 normalize 开关）， 返回基座原始响应 dict 供桥透传，避免重复实现 TCP 封装。 验证： - verify_stdio_bridge.py 24/24 PASS（版本协商、通知不回包、ping、tools/list 与基座 等价、tools/call、错误透传、-32700、string id、stdout 纯净性、死端口降级）。 - 真实标准客户端 MCP Inspector 冒烟通过：tools/list 拿到 26 个工具， tools/list --strict 退出码 0（无 schema 可移植性问题）， tools/call get_drawing_info 返回标准 content + isError:false。 - 回归：verify_mcp_frame.py 4/4、verify_annotate.py 9/9、dwg_review --list-tools/--check 正常。
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""入口：python -m mcp_bridge [options]"""
|
||
|
||
import argparse
|
||
import json
|
||
import sys
|
||
|
||
from dwg_review import config
|
||
|
||
from . import __version__
|
||
from .server import StdioBridge
|
||
|
||
|
||
def build_parser():
|
||
parser = argparse.ArgumentParser(
|
||
prog="mcp_bridge",
|
||
description="把标准 MCP 客户端（stdio）桥接到 CAD 基座（裸 TCP）。",
|
||
)
|
||
parser.add_argument("--host", default=None, help=f"CAD 基座主机（默认 {config.HOST}）")
|
||
parser.add_argument("--port", type=int, default=None,
|
||
help=f"CAD 基座端口（默认 {config.PORT}）")
|
||
parser.add_argument("--timeout", type=float, default=None,
|
||
help=f"单次调用超时秒数（默认 {config.CALL_TIMEOUT}）")
|
||
parser.add_argument("--verbose", action="store_true", help="把诊断信息打到 stderr")
|
||
parser.add_argument("--print-config", action="store_true",
|
||
help="打印可直接粘进 MCP 客户端的配置片段后退出")
|
||
return parser
|
||
|
||
|
||
def print_config(args):
|
||
host = args.host or config.HOST
|
||
port = args.port if args.port is not None else config.PORT
|
||
snippet = {
|
||
"mcpServers": {
|
||
"cad-autocad": {
|
||
"command": sys.executable,
|
||
"args": ["-m", "mcp_bridge"],
|
||
"env": {
|
||
# 用绝对路径，避免依赖客户端的工作目录
|
||
"PYTHONPATH": str(config.PROJECT_ROOT),
|
||
"CAD_MCP_HOST": host,
|
||
"CAD_MCP_PORT": str(port),
|
||
},
|
||
}
|
||
}
|
||
}
|
||
print(json.dumps(snippet, ensure_ascii=False, indent=2))
|
||
return 0
|
||
|
||
|
||
def main(argv=None):
|
||
args = build_parser().parse_args(argv)
|
||
if args.print_config:
|
||
return print_config(args)
|
||
|
||
bridge = StdioBridge(
|
||
host=args.host, port=args.port, timeout=args.timeout, verbose=args.verbose
|
||
)
|
||
if args.verbose:
|
||
print(
|
||
f"[mcp_bridge] v{__version__} 转发到 {bridge.host}:{bridge.port}，"
|
||
f"单次超时 {bridge.timeout}s",
|
||
file=sys.stderr, flush=True,
|
||
)
|
||
bridge.run()
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|