151 lines
4.1 KiB
JavaScript
151 lines
4.1 KiB
JavaScript
/**
|
|
* MCP 桥接代理脚本
|
|
* 负责在标准 MCP 客户端 (stdin/stdout) 与 Cocos Creator 插件 (HTTP) 之间转发请求。
|
|
*/
|
|
|
|
const http = require("http");
|
|
|
|
/**
|
|
* 当前 Cocos Creator 插件监听的端口
|
|
* 支持通过环境变量 MCP_BRIDGE_PORT 或命令行参数指定端口
|
|
* @type {number}
|
|
*/
|
|
const COCOS_PORT = parseInt(process.env.MCP_BRIDGE_PORT || process.argv[2] || "3456", 10);
|
|
|
|
/**
|
|
* 发送调试日志到标准的错误输出流水
|
|
* @param {string} msg 日志消息
|
|
*/
|
|
function debugLog(msg) {
|
|
process.stderr.write(`[代理调试] ${msg}\n`);
|
|
}
|
|
|
|
// 监听标准输入以获取 MCP 请求
|
|
process.stdin.on("data", (data) => {
|
|
const lines = data.toString().split("\n");
|
|
lines.forEach((line) => {
|
|
if (!line.trim()) return;
|
|
try {
|
|
const request = JSON.parse(line);
|
|
handleRequest(request);
|
|
} catch (e) {
|
|
// 忽略非 JSON 输入
|
|
}
|
|
});
|
|
});
|
|
|
|
/**
|
|
* 处理 JSON-RPC 请求
|
|
* @param {Object} req RPC 请求对象
|
|
*/
|
|
function handleRequest(req) {
|
|
const { method, id, params } = req;
|
|
|
|
// 处理握手初始化
|
|
if (method === "initialize") {
|
|
sendToAI({
|
|
jsonrpc: "2.0",
|
|
id: id,
|
|
result: {
|
|
protocolVersion: "2024-11-05",
|
|
capabilities: { tools: {} },
|
|
serverInfo: { name: "cocos-bridge", version: "1.0.0" },
|
|
},
|
|
});
|
|
return;
|
|
}
|
|
|
|
// 获取工具列表
|
|
if (method === "tools/list") {
|
|
forwardToCocos("/list-tools", null, id, "GET");
|
|
return;
|
|
}
|
|
|
|
// 执行具体工具
|
|
if (method === "tools/call") {
|
|
forwardToCocos(
|
|
"/call-tool",
|
|
{
|
|
name: params.name,
|
|
arguments: params.arguments,
|
|
},
|
|
id,
|
|
"POST",
|
|
);
|
|
return;
|
|
}
|
|
|
|
// 默认空响应
|
|
if (id !== undefined) sendToAI({ jsonrpc: "2.0", id: id, result: {} });
|
|
}
|
|
|
|
/**
|
|
* 将请求通过 HTTP 转发给 Cocos Creator 插件
|
|
* @param {string} path API 路径
|
|
* @param {Object|null} payload 发送的数据体
|
|
* @param {string|number} id RPC 请求标识符
|
|
* @param {string} method HTTP 方法 (默认 POST)
|
|
*/
|
|
function forwardToCocos(path, payload, id, method = "POST") {
|
|
const postData = payload ? JSON.stringify(payload) : "";
|
|
|
|
const options = {
|
|
hostname: "127.0.0.1",
|
|
port: COCOS_PORT,
|
|
path: path,
|
|
method: method,
|
|
headers: { "Content-Type": "application/json" },
|
|
};
|
|
|
|
if (postData) {
|
|
options.headers["Content-Length"] = Buffer.byteLength(postData);
|
|
}
|
|
|
|
const request = http.request(options, (res) => {
|
|
let resData = "";
|
|
res.on("data", (d) => (resData += d));
|
|
res.on("end", () => {
|
|
try {
|
|
const cocosRes = JSON.parse(resData);
|
|
|
|
// 检查关键字段，确保 Cocos 插件返回了期望的数据格式
|
|
if (path === "/list-tools" && !cocosRes.tools) {
|
|
debugLog(`致命错误: Cocos 未返回工具列表。接收内容: ${resData}`);
|
|
sendError(id, -32603, "Cocos 响应无效：缺少 tools 数组");
|
|
} else {
|
|
sendToAI({ jsonrpc: "2.0", id: id, result: cocosRes });
|
|
}
|
|
} catch (e) {
|
|
debugLog(`JSON 解析错误。Cocos 发送内容: ${resData}`);
|
|
sendError(id, -32603, "Cocos 返回了非 JSON 数据");
|
|
}
|
|
});
|
|
});
|
|
|
|
request.on("error", (e) => {
|
|
debugLog(`Cocos 插件已离线: ${e.message}`);
|
|
sendError(id, -32000, "Cocos 插件离线");
|
|
});
|
|
|
|
if (postData) request.write(postData);
|
|
request.end();
|
|
}
|
|
|
|
/**
|
|
* 将结果发送给 AI (通过标准输出)
|
|
* @param {Object} obj 结果对象
|
|
*/
|
|
function sendToAI(obj) {
|
|
process.stdout.write(JSON.stringify(obj) + "\n");
|
|
}
|
|
|
|
/**
|
|
* 发送 RPC 错误响应
|
|
* @param {string|number} id RPC 请求标识符
|
|
* @param {number} code 错误码
|
|
* @param {string} message 错误消息
|
|
*/
|
|
function sendError(id, code, message) {
|
|
sendToAI({ jsonrpc: "2.0", id: id, error: { code, message } });
|
|
}
|