refactor: 整合MCP命令到mcp目录 - 优化项目结构
- 移动MCPServerCommand.js和MCPStreamableHttpCommand.js到lib/mcp/目录 - 删除空的lib/commands/目录，避免代码分散 - 更新bin/promptx.js中的import路径指向新位置 - 统一MCP相关代码到同一目录，提升代码组织性 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
sean
252
src/lib/mcp/MCPServerCommand.js
Normal file
252
src/lib/mcp/MCPServerCommand.js
Normal file
@ -0,0 +1,252 @@
|
||||
const { Server } = require('@modelcontextprotocol/sdk/server/index.js');
|
||||
const { StdioServerTransport } = require('@modelcontextprotocol/sdk/server/stdio.js');
|
||||
const { cli } = require('../core/pouch');
|
||||
const { MCPOutputAdapter } = require('../mcp/MCPOutputAdapter');
|
||||
const { getExecutionContext, getDebugInfo } = require('../utils/executionContext');
|
||||
const { getToolDefinitions } = require('../mcp/toolDefinitions');
|
||||
const treeKill = require('tree-kill');
|
||||
|
||||
/**
|
||||
* MCP Server 适配器 - 函数调用架构
|
||||
* 将MCP协议请求转换为PromptX函数调用，实现零开销适配
|
||||
* 支持智能工作目录检测，确保MCP和CLI模式下的一致性
|
||||
*/
|
||||
class MCPServerCommand {
|
||||
constructor() {
|
||||
this.name = 'promptx-mcp-server';
|
||||
this.version = '1.0.0';
|
||||
this.debug = process.env.MCP_DEBUG === 'true';
|
||||
|
||||
// 智能检测执行上下文
|
||||
this.executionContext = getExecutionContext();
|
||||
|
||||
// 调试信息输出
|
||||
this.log(`🎯 检测到执行模式: ${this.executionContext.mode}`);
|
||||
this.log(`📍 原始工作目录: ${this.executionContext.originalCwd}`);
|
||||
this.log(`📁 目标工作目录: ${this.executionContext.workingDirectory}`);
|
||||
|
||||
// 如果需要切换工作目录
|
||||
if (this.executionContext.workingDirectory !== this.executionContext.originalCwd) {
|
||||
this.log(`🔄 切换工作目录: ${this.executionContext.originalCwd} -> ${this.executionContext.workingDirectory}`);
|
||||
try {
|
||||
process.chdir(this.executionContext.workingDirectory);
|
||||
this.log(`✅ 工作目录切换成功`);
|
||||
} catch (error) {
|
||||
this.log(`❌ 工作目录切换失败: ${error.message}`);
|
||||
this.log(`🔄 继续使用原始目录: ${this.executionContext.originalCwd}`);
|
||||
}
|
||||
}
|
||||
|
||||
// 基本调试信息
|
||||
this.log(`📂 最终工作目录: ${process.cwd()}`);
|
||||
this.log(`📋 预期记忆文件路径: ${require('path').join(process.cwd(), '.promptx/memory/declarative.md')}`);
|
||||
|
||||
// DirectoryService路径信息将在需要时异步获取
|
||||
|
||||
// 输出完整调试信息
|
||||
if (this.debug) {
|
||||
this.log(`🔍 完整调试信息: ${JSON.stringify(getDebugInfo(), null, 2)}`);
|
||||
}
|
||||
|
||||
// 创建输出适配器
|
||||
this.outputAdapter = new MCPOutputAdapter();
|
||||
|
||||
// 创建MCP服务器实例 - 使用正确的API
|
||||
this.server = new Server(
|
||||
{
|
||||
name: this.name,
|
||||
version: this.version
|
||||
},
|
||||
{
|
||||
capabilities: {
|
||||
tools: {}
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
this.setupHandlers();
|
||||
}
|
||||
|
||||
/**
|
||||
* 调试日志 - 输出到stderr，不影响MCP协议
|
||||
*/
|
||||
log(message) {
|
||||
if (this.debug) {
|
||||
console.error(`[MCP DEBUG] ${message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动MCP Server
|
||||
*/
|
||||
async execute(options = {}) {
|
||||
try {
|
||||
// 设置进程清理处理器
|
||||
this.setupProcessCleanup();
|
||||
|
||||
|
||||
this.log('🚀 启动MCP Server...');
|
||||
const transport = new StdioServerTransport();
|
||||
await this.server.connect(transport);
|
||||
this.log('✅ MCP Server 已启动，等待连接...');
|
||||
|
||||
// 保持进程运行
|
||||
return new Promise((resolve) => {
|
||||
// MCP服务器现在正在运行，监听stdin输入
|
||||
process.on('SIGINT', () => {
|
||||
this.log('🛑 收到SIGINT信号，正在关闭...');
|
||||
this.cleanup();
|
||||
resolve();
|
||||
});
|
||||
|
||||
process.on('SIGTERM', () => {
|
||||
this.log('🛑 收到SIGTERM信号，正在关闭...');
|
||||
this.cleanup();
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
} catch (error) {
|
||||
// 输出到stderr
|
||||
console.error(`❌ MCP Server 启动失败: ${error.message}`);
|
||||
this.cleanup();
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置进程清理处理器
|
||||
*/
|
||||
setupProcessCleanup() {
|
||||
// 处理各种退出情况
|
||||
const exitHandler = (signal) => {
|
||||
this.log(`收到信号: ${signal}`);
|
||||
this.cleanup();
|
||||
process.exit(0);
|
||||
};
|
||||
|
||||
// 捕获所有可能的退出信号
|
||||
process.on('exit', () => this.cleanup());
|
||||
process.on('SIGHUP', () => exitHandler('SIGHUP'));
|
||||
process.on('SIGQUIT', () => exitHandler('SIGQUIT'));
|
||||
process.on('uncaughtException', (err) => {
|
||||
console.error('未捕获的异常:', err);
|
||||
this.cleanup();
|
||||
process.exit(1);
|
||||
});
|
||||
process.on('unhandledRejection', (reason, promise) => {
|
||||
console.error('未处理的Promise拒绝:', reason);
|
||||
this.cleanup();
|
||||
process.exit(1);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 清理资源
|
||||
*/
|
||||
cleanup() {
|
||||
this.log('🔧 清理MCP Server资源');
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* 设置MCP工具处理程序 - 使用正确的MCP SDK API
|
||||
*/
|
||||
setupHandlers() {
|
||||
// 使用Schema常量进行注册
|
||||
const {
|
||||
ListToolsRequestSchema,
|
||||
CallToolRequestSchema
|
||||
} = require('@modelcontextprotocol/sdk/types.js');
|
||||
|
||||
// 注册工具列表处理程序
|
||||
this.server.setRequestHandler(ListToolsRequestSchema, async () => {
|
||||
this.log('📋 收到工具列表请求');
|
||||
return {
|
||||
tools: this.getToolDefinitions()
|
||||
};
|
||||
});
|
||||
|
||||
// 注册工具调用处理程序
|
||||
this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
||||
const { name, arguments: args } = request.params;
|
||||
this.log(`🔧 调用工具: ${name} 参数: ${JSON.stringify(args)}`);
|
||||
return await this.callTool(name, args || {});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取工具定义
|
||||
*/
|
||||
getToolDefinitions() {
|
||||
return getToolDefinitions();
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行工具调用
|
||||
*/
|
||||
async callTool(toolName, args) {
|
||||
try {
|
||||
// 将MCP参数转换为CLI函数调用参数
|
||||
const cliArgs = this.convertMCPToCliParams(toolName, args);
|
||||
this.log(`🎯 CLI调用: ${toolName} -> ${JSON.stringify(cliArgs)}`);
|
||||
this.log(`🗂️ 当前工作目录: ${process.cwd()}`);
|
||||
|
||||
// 直接调用PromptX CLI函数 - 启用静默模式避免console.log干扰MCP协议
|
||||
const result = await cli.execute(toolName.replace('promptx_', ''), cliArgs, true);
|
||||
this.log(`✅ CLI执行完成: ${toolName}`);
|
||||
|
||||
// 使用输出适配器转换为MCP响应格式
|
||||
return this.outputAdapter.convertToMCPFormat(result);
|
||||
|
||||
} catch (error) {
|
||||
this.log(`❌ 工具调用失败: ${toolName} - ${error.message}`);
|
||||
return this.outputAdapter.handleError(error);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 转换MCP参数为CLI函数调用参数
|
||||
*/
|
||||
convertMCPToCliParams(toolName, mcpArgs) {
|
||||
const paramMapping = {
|
||||
'promptx_init': (args) => args.workingDirectory ? [args] : [],
|
||||
|
||||
'promptx_welcome': () => [],
|
||||
|
||||
'promptx_action': (args) => [args.role],
|
||||
|
||||
'promptx_learn': (args) => args.resource ? [args.resource] : [],
|
||||
|
||||
'promptx_recall': (args) => {
|
||||
// 忽略random_string dummy参数，只处理query
|
||||
// 处理各种空值情况：undefined、null、空对象、空字符串
|
||||
if (!args || !args.query || typeof args.query !== 'string' || args.query.trim() === '') {
|
||||
return [];
|
||||
}
|
||||
return [args.query];
|
||||
},
|
||||
|
||||
'promptx_remember': (args) => {
|
||||
const result = [args.content];
|
||||
if (args.tags) {
|
||||
result.push('--tags', args.tags);
|
||||
}
|
||||
return result;
|
||||
},
|
||||
|
||||
|
||||
'promptx_tool': (args) => [args]
|
||||
};
|
||||
|
||||
const mapper = paramMapping[toolName];
|
||||
if (!mapper) {
|
||||
throw new Error(`未知工具: ${toolName}`);
|
||||
}
|
||||
|
||||
return mapper(mcpArgs);
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { MCPServerCommand };
|
||||
512
src/lib/mcp/MCPStreamableHttpCommand.js
Normal file
512
src/lib/mcp/MCPStreamableHttpCommand.js
Normal file
@ -0,0 +1,512 @@
|
||||
const express = require('express');
|
||||
const { randomUUID } = require('node:crypto');
|
||||
const { McpServer } = require('@modelcontextprotocol/sdk/server/mcp.js');
|
||||
const { StreamableHTTPServerTransport } = require('@modelcontextprotocol/sdk/server/streamableHttp.js');
|
||||
const { SSEServerTransport } = require('@modelcontextprotocol/sdk/server/sse.js');
|
||||
const { isInitializeRequest } = require('@modelcontextprotocol/sdk/types.js');
|
||||
const { cli } = require('../core/pouch');
|
||||
const { MCPOutputAdapter } = require('../mcp/MCPOutputAdapter');
|
||||
const { getToolDefinitions, getToolDefinition, getToolZodSchema } = require('../mcp/toolDefinitions');
|
||||
const logger = require('../utils/logger');
|
||||
|
||||
/**
|
||||
* MCP Streamable HTTP Server Command
|
||||
* 实现基于 Streamable HTTP 传输的 MCP 服务器
|
||||
* 同时提供 SSE 向后兼容支持
|
||||
*/
|
||||
class MCPStreamableHttpCommand {
|
||||
constructor() {
|
||||
this.name = 'promptx-mcp-streamable-http-server';
|
||||
this.version = '1.0.0';
|
||||
this.transport = 'http';
|
||||
this.port = 3000;
|
||||
this.host = 'localhost';
|
||||
this.transports = {}; // 存储会话传输
|
||||
this.outputAdapter = new MCPOutputAdapter();
|
||||
this.debug = process.env.MCP_DEBUG === 'true';
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行命令
|
||||
*/
|
||||
async execute(options = {}) {
|
||||
const {
|
||||
transport = 'http',
|
||||
port = 3000,
|
||||
host = 'localhost'
|
||||
} = options;
|
||||
|
||||
// 验证传输类型
|
||||
if (!['http', 'sse'].includes(transport)) {
|
||||
throw new Error(`Unsupported transport: ${transport}`);
|
||||
}
|
||||
|
||||
// 验证配置
|
||||
this.validatePort(port);
|
||||
this.validateHost(host);
|
||||
|
||||
if (transport === 'http') {
|
||||
return this.startStreamableHttpServer(port, host);
|
||||
} else if (transport === 'sse') {
|
||||
return this.startSSEServer(port, host);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动 Streamable HTTP 服务器
|
||||
*/
|
||||
async startStreamableHttpServer(port, host) {
|
||||
this.log(`🚀 启动 Streamable HTTP MCP Server...`);
|
||||
|
||||
const app = express();
|
||||
|
||||
// 中间件设置
|
||||
app.use(express.json());
|
||||
app.use(this.corsMiddleware.bind(this));
|
||||
|
||||
// 健康检查端点
|
||||
app.get('/health', (req, res) => {
|
||||
res.json({
|
||||
status: 'ok',
|
||||
name: this.name,
|
||||
version: this.version,
|
||||
transport: 'http'
|
||||
});
|
||||
});
|
||||
|
||||
// MCP 端点
|
||||
app.post('/mcp', this.handleMCPPostRequest.bind(this));
|
||||
app.get('/mcp', this.handleMCPGetRequest.bind(this));
|
||||
app.delete('/mcp', this.handleMCPDeleteRequest.bind(this));
|
||||
|
||||
// 错误处理中间件
|
||||
app.use(this.errorHandler.bind(this));
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const server = app.listen(port, host, () => {
|
||||
this.log(`✅ Streamable HTTP MCP Server 运行在 http://${host}:${port}`);
|
||||
this.server = server;
|
||||
resolve(server);
|
||||
});
|
||||
|
||||
server.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* CORS 中间件
|
||||
*/
|
||||
corsMiddleware(req, res, next) {
|
||||
res.setHeader('Access-Control-Allow-Origin', '*');
|
||||
res.setHeader('Access-Control-Allow-Methods', 'GET, POST, DELETE, OPTIONS');
|
||||
res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Accept, mcp-session-id');
|
||||
|
||||
if (req.method === 'OPTIONS') {
|
||||
res.sendStatus(200);
|
||||
return;
|
||||
}
|
||||
|
||||
next();
|
||||
}
|
||||
|
||||
/**
|
||||
* 错误处理中间件
|
||||
*/
|
||||
errorHandler(error, req, res, next) {
|
||||
this.log('Express 错误处理:', error);
|
||||
|
||||
if (!res.headersSent) {
|
||||
// 检查是否是JSON解析错误
|
||||
if (error.type === 'entity.parse.failed' || error.message?.includes('JSON')) {
|
||||
res.status(400).json({
|
||||
jsonrpc: '2.0',
|
||||
error: {
|
||||
code: -32700,
|
||||
message: 'Parse error: Invalid JSON'
|
||||
},
|
||||
id: null
|
||||
});
|
||||
} else {
|
||||
res.status(500).json({
|
||||
jsonrpc: '2.0',
|
||||
error: {
|
||||
code: -32603,
|
||||
message: 'Internal server error'
|
||||
},
|
||||
id: null
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动 SSE 服务器
|
||||
*/
|
||||
async startSSEServer(port, host) {
|
||||
const app = express();
|
||||
app.use(express.json());
|
||||
|
||||
this.log(`🚀 启动 SSE MCP Server...`);
|
||||
|
||||
// 健康检查端点
|
||||
app.get('/health', (req, res) => {
|
||||
res.json({ status: 'ok', name: this.name, version: this.version, transport: 'sse' });
|
||||
});
|
||||
|
||||
// SSE 端点 - 建立事件流
|
||||
app.get('/mcp', async (req, res) => {
|
||||
await this.handleSSEConnection(req, res);
|
||||
});
|
||||
|
||||
// 消息端点 - 接收客户端 JSON-RPC 消息
|
||||
app.post('/messages', async (req, res) => {
|
||||
await this.handleSSEMessage(req, res);
|
||||
});
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const server = app.listen(port, host, () => {
|
||||
this.log(`✅ SSE MCP Server 运行在 http://${host}:${port}`);
|
||||
resolve(server);
|
||||
});
|
||||
|
||||
server.on('error', reject);
|
||||
this.server = server;
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 SSE 连接建立
|
||||
*/
|
||||
async handleSSEConnection(req, res) {
|
||||
this.log('建立 SSE 连接');
|
||||
|
||||
try {
|
||||
// 创建 SSE 传输
|
||||
const transport = new SSEServerTransport('/messages', res);
|
||||
const sessionId = transport.sessionId;
|
||||
|
||||
// 存储传输
|
||||
this.transports[sessionId] = transport;
|
||||
|
||||
// 设置关闭处理程序
|
||||
transport.onclose = () => {
|
||||
this.log(`SSE 传输关闭: ${sessionId}`);
|
||||
delete this.transports[sessionId];
|
||||
};
|
||||
|
||||
// 连接到 MCP 服务器
|
||||
const server = this.setupMCPServer();
|
||||
await server.connect(transport);
|
||||
|
||||
this.log(`SSE 流已建立，会话ID: ${sessionId}`);
|
||||
} catch (error) {
|
||||
this.log('建立 SSE 连接错误:', error);
|
||||
if (!res.headersSent) {
|
||||
res.status(500).send('Error establishing SSE connection');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 SSE 消息
|
||||
*/
|
||||
async handleSSEMessage(req, res) {
|
||||
this.log('收到 SSE 消息:', req.body);
|
||||
|
||||
try {
|
||||
// 从查询参数获取会话ID
|
||||
const sessionId = req.query.sessionId;
|
||||
|
||||
if (!sessionId) {
|
||||
res.status(400).send('Missing sessionId parameter');
|
||||
return;
|
||||
}
|
||||
|
||||
const transport = this.transports[sessionId];
|
||||
if (!transport) {
|
||||
res.status(404).send('Session not found');
|
||||
return;
|
||||
}
|
||||
|
||||
// 处理消息
|
||||
await transport.handlePostMessage(req, res, req.body);
|
||||
} catch (error) {
|
||||
this.log('处理 SSE 消息错误:', error);
|
||||
if (!res.headersSent) {
|
||||
res.status(500).send('Error handling request');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置 MCP 服务器
|
||||
*/
|
||||
setupMCPServer() {
|
||||
const server = new McpServer({
|
||||
name: this.name,
|
||||
version: this.version
|
||||
}, {
|
||||
capabilities: {
|
||||
tools: {},
|
||||
logging: {}
|
||||
}
|
||||
});
|
||||
|
||||
// 注册所有 PromptX 工具
|
||||
this.setupMCPTools(server);
|
||||
|
||||
return server;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置 MCP 工具
|
||||
*/
|
||||
setupMCPTools(server) {
|
||||
// 使用共享的工具定义
|
||||
const toolDefinitions = getToolDefinitions();
|
||||
|
||||
toolDefinitions.forEach(toolDef => {
|
||||
const zodSchema = getToolZodSchema(toolDef.name);
|
||||
|
||||
server.tool(toolDef.name, toolDef.description, zodSchema, async (args, extra) => {
|
||||
this.log(`🔧 调用工具: ${toolDef.name} 参数: ${JSON.stringify(args)}`);
|
||||
return await this.callTool(toolDef.name, args || {});
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取工具定义
|
||||
*/
|
||||
getToolDefinitions() {
|
||||
return getToolDefinitions();
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 MCP POST 请求
|
||||
*/
|
||||
async handleMCPPostRequest(req, res) {
|
||||
this.log('收到 MCP 请求:', req.body);
|
||||
|
||||
try {
|
||||
// 检查现有会话 ID
|
||||
const sessionId = req.headers['mcp-session-id'];
|
||||
let transport;
|
||||
|
||||
if (sessionId && this.transports[sessionId]) {
|
||||
// 复用现有传输
|
||||
transport = this.transports[sessionId];
|
||||
} else if (!sessionId && isInitializeRequest(req.body)) {
|
||||
// 新的初始化请求
|
||||
transport = new StreamableHTTPServerTransport({
|
||||
sessionIdGenerator: () => randomUUID(),
|
||||
enableJsonResponse: true,
|
||||
onsessioninitialized: (sessionId) => {
|
||||
this.log(`会话初始化: ${sessionId}`);
|
||||
this.transports[sessionId] = transport;
|
||||
}
|
||||
});
|
||||
|
||||
// 设置关闭处理程序
|
||||
transport.onclose = () => {
|
||||
const sid = transport.sessionId;
|
||||
if (sid && this.transports[sid]) {
|
||||
this.log(`传输关闭: ${sid}`);
|
||||
delete this.transports[sid];
|
||||
}
|
||||
};
|
||||
|
||||
// 连接到 MCP 服务器
|
||||
const server = this.setupMCPServer();
|
||||
await server.connect(transport);
|
||||
await transport.handleRequest(req, res, req.body);
|
||||
return;
|
||||
} else if (!sessionId && this.isStatelessRequest(req.body)) {
|
||||
// 无状态请求（如 tools/list, prompts/list 等）
|
||||
transport = new StreamableHTTPServerTransport({
|
||||
sessionIdGenerator: undefined, // 无状态模式
|
||||
enableJsonResponse: true
|
||||
});
|
||||
|
||||
// 连接到 MCP 服务器
|
||||
const server = this.setupMCPServer();
|
||||
await server.connect(transport);
|
||||
await transport.handleRequest(req, res, req.body);
|
||||
return;
|
||||
} else {
|
||||
// 无效请求
|
||||
return res.status(400).json({
|
||||
jsonrpc: '2.0',
|
||||
error: {
|
||||
code: -32000,
|
||||
message: 'Bad Request: No valid session ID provided'
|
||||
},
|
||||
id: null
|
||||
});
|
||||
}
|
||||
|
||||
// 处理现有传输的请求
|
||||
await transport.handleRequest(req, res, req.body);
|
||||
} catch (error) {
|
||||
this.log('处理 MCP 请求错误:', error);
|
||||
if (!res.headersSent) {
|
||||
res.status(500).json({
|
||||
jsonrpc: '2.0',
|
||||
error: {
|
||||
code: -32603,
|
||||
message: 'Internal server error'
|
||||
},
|
||||
id: null
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 MCP GET 请求（SSE）
|
||||
*/
|
||||
async handleMCPGetRequest(req, res) {
|
||||
const sessionId = req.headers['mcp-session-id'];
|
||||
if (!sessionId || !this.transports[sessionId]) {
|
||||
return res.status(400).json({
|
||||
error: 'Invalid or missing session ID'
|
||||
});
|
||||
}
|
||||
|
||||
this.log(`建立 SSE 流: ${sessionId}`);
|
||||
const transport = this.transports[sessionId];
|
||||
await transport.handleRequest(req, res);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 MCP DELETE 请求（会话终止）
|
||||
*/
|
||||
async handleMCPDeleteRequest(req, res) {
|
||||
const sessionId = req.headers['mcp-session-id'];
|
||||
if (!sessionId || !this.transports[sessionId]) {
|
||||
return res.status(400).json({
|
||||
error: 'Invalid or missing session ID'
|
||||
});
|
||||
}
|
||||
|
||||
this.log(`终止会话: ${sessionId}`);
|
||||
try {
|
||||
const transport = this.transports[sessionId];
|
||||
await transport.handleRequest(req, res);
|
||||
} catch (error) {
|
||||
this.log('处理会话终止错误:', error);
|
||||
if (!res.headersSent) {
|
||||
res.status(500).json({
|
||||
error: 'Error processing session termination'
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 调用工具
|
||||
*/
|
||||
async callTool(toolName, args) {
|
||||
try {
|
||||
// 将 MCP 参数转换为 CLI 函数调用参数
|
||||
const cliArgs = this.convertMCPToCliParams(toolName, args);
|
||||
this.log(`🎯 CLI调用: ${toolName} -> ${JSON.stringify(cliArgs)}`);
|
||||
|
||||
// 直接调用 PromptX CLI 函数
|
||||
const result = await cli.execute(toolName.replace('promptx_', ''), cliArgs, true);
|
||||
this.log(`✅ CLI执行完成: ${toolName}`);
|
||||
|
||||
// 使用输出适配器转换为MCP响应格式（与stdio模式保持一致）
|
||||
return this.outputAdapter.convertToMCPFormat(result);
|
||||
|
||||
} catch (error) {
|
||||
this.log(`❌ 工具调用失败: ${toolName} - ${error.message}`);
|
||||
return this.outputAdapter.handleError(error);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 转换 MCP 参数为 CLI 函数调用参数
|
||||
*/
|
||||
convertMCPToCliParams(toolName, mcpArgs) {
|
||||
const paramMapping = {
|
||||
'promptx_init': () => [],
|
||||
'promptx_welcome': () => [],
|
||||
'promptx_action': (args) => args && args.role ? [args.role] : [],
|
||||
'promptx_learn': (args) => args && args.resource ? [args.resource] : [],
|
||||
'promptx_recall': (args) => {
|
||||
if (!args || !args.query || typeof args.query !== 'string' || args.query.trim() === '') {
|
||||
return [];
|
||||
}
|
||||
return [args.query];
|
||||
},
|
||||
'promptx_remember': (args) => {
|
||||
if (!args || !args.content) {
|
||||
throw new Error('content 参数是必需的');
|
||||
}
|
||||
const result = [args.content];
|
||||
if (args.tags) {
|
||||
result.push('--tags', args.tags);
|
||||
}
|
||||
return result;
|
||||
},
|
||||
};
|
||||
|
||||
const mapper = paramMapping[toolName];
|
||||
if (!mapper) {
|
||||
throw new Error(`未知工具: ${toolName}`);
|
||||
}
|
||||
|
||||
return mapper(mcpArgs || {});
|
||||
}
|
||||
|
||||
/**
|
||||
* 调试日志
|
||||
*/
|
||||
log(message, ...args) {
|
||||
if (this.debug) {
|
||||
logger.debug(`[MCP DEBUG] ${message}`, ...args);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证端口号
|
||||
*/
|
||||
validatePort(port) {
|
||||
if (typeof port !== 'number') {
|
||||
throw new Error('Port must be a number');
|
||||
}
|
||||
if (port < 1 || port > 65535) {
|
||||
throw new Error('Port must be between 1 and 65535');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证主机地址
|
||||
*/
|
||||
validateHost(host) {
|
||||
if (!host || typeof host !== 'string' || host.trim() === '') {
|
||||
throw new Error('Host cannot be empty');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断是否为无状态请求（不需要会话ID）
|
||||
*/
|
||||
isStatelessRequest(requestBody) {
|
||||
if (!requestBody || !requestBody.method) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// 这些方法可以无状态处理
|
||||
const statelessMethods = [
|
||||
'tools/list',
|
||||
'prompts/list',
|
||||
'resources/list'
|
||||
];
|
||||
|
||||
return statelessMethods.includes(requestBody.method);
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { MCPStreamableHttpCommand };
|
||||
Reference in New Issue
Block a user