235 lines
9.0 KiB
TypeScript
235 lines
9.0 KiB
TypeScript
declare const Editor: any;
|
||
|
||
/**
|
||
* MCP 工具测试器 - 直接测试通过 WebSocket 的 MCP 工具
|
||
*/
|
||
export class MCPToolTester {
|
||
private ws: WebSocket | null = null;
|
||
private messageId = 0;
|
||
private responseHandlers = new Map<number, (response: any) => void>();
|
||
|
||
async connect(port: number): Promise<boolean> {
|
||
return new Promise((resolve) => {
|
||
try {
|
||
this.ws = new WebSocket(`ws://localhost:${port}`);
|
||
|
||
this.ws.onopen = () => {
|
||
console.log('WebSocket 连接成功');
|
||
resolve(true);
|
||
};
|
||
|
||
this.ws.onerror = (error) => {
|
||
console.error('WebSocket 连接错误:', error);
|
||
resolve(false);
|
||
};
|
||
|
||
this.ws.onmessage = (event) => {
|
||
try {
|
||
const response = JSON.parse(event.data);
|
||
if (response.id && this.responseHandlers.has(response.id)) {
|
||
const handler = this.responseHandlers.get(response.id);
|
||
this.responseHandlers.delete(response.id);
|
||
handler?.(response);
|
||
}
|
||
} catch (error) {
|
||
console.error('处理响应时出错:', error);
|
||
}
|
||
};
|
||
} catch (error) {
|
||
console.error('创建 WebSocket 时出错:', error);
|
||
resolve(false);
|
||
}
|
||
});
|
||
}
|
||
|
||
async callTool(tool: string, args: any = {}): Promise<any> {
|
||
if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
|
||
throw new Error('WebSocket 未连接');
|
||
}
|
||
|
||
return new Promise((resolve, reject) => {
|
||
const id = ++this.messageId;
|
||
const request = {
|
||
jsonrpc: '2.0',
|
||
id,
|
||
method: 'tools/call',
|
||
params: {
|
||
name: tool,
|
||
arguments: args
|
||
}
|
||
};
|
||
|
||
const timeout = setTimeout(() => {
|
||
this.responseHandlers.delete(id);
|
||
reject(new Error('请求超时'));
|
||
}, 10000);
|
||
|
||
this.responseHandlers.set(id, (response) => {
|
||
clearTimeout(timeout);
|
||
if (response.error) {
|
||
reject(new Error(response.error.message));
|
||
} else {
|
||
resolve(response.result);
|
||
}
|
||
});
|
||
|
||
this.ws!.send(JSON.stringify(request));
|
||
});
|
||
}
|
||
|
||
async listTools(): Promise<any> {
|
||
if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
|
||
throw new Error('WebSocket 未连接');
|
||
}
|
||
|
||
return new Promise((resolve, reject) => {
|
||
const id = ++this.messageId;
|
||
const request = {
|
||
jsonrpc: '2.0',
|
||
id,
|
||
method: 'tools/list'
|
||
};
|
||
|
||
const timeout = setTimeout(() => {
|
||
this.responseHandlers.delete(id);
|
||
reject(new Error('请求超时'));
|
||
}, 10000);
|
||
|
||
this.responseHandlers.set(id, (response) => {
|
||
clearTimeout(timeout);
|
||
if (response.error) {
|
||
reject(new Error(response.error.message));
|
||
} else {
|
||
resolve(response.result);
|
||
}
|
||
});
|
||
|
||
this.ws!.send(JSON.stringify(request));
|
||
});
|
||
}
|
||
|
||
async testMCPTools() {
|
||
console.log('\n=== 测试 MCP 工具（通过 WebSocket）===');
|
||
|
||
try {
|
||
// 0. 获取工具列表
|
||
console.log('\n0. 获取工具列表...');
|
||
const toolsList = await this.listTools();
|
||
console.log(`找到 ${toolsList.tools?.length || 0} 个工具:`);
|
||
if (toolsList.tools) {
|
||
for (const tool of toolsList.tools.slice(0, 10)) { // 只显示前10个
|
||
console.log(` - ${tool.name}: ${tool.description}`);
|
||
}
|
||
if (toolsList.tools.length > 10) {
|
||
console.log(` ... 还有 ${toolsList.tools.length - 10} 个工具`);
|
||
}
|
||
}
|
||
|
||
// 1. 测试场景工具
|
||
console.log('\n1. 测试当前场景信息...');
|
||
const sceneInfo = await this.callTool('scene_get_current_scene');
|
||
console.log('场景信息:', JSON.stringify(sceneInfo).substring(0, 100) + '...');
|
||
|
||
// 2. 测试场景列表
|
||
console.log('\n2. 测试场景列表...');
|
||
const sceneList = await this.callTool('scene_get_scene_list');
|
||
console.log('场景列表:', JSON.stringify(sceneList).substring(0, 100) + '...');
|
||
|
||
// 3. 测试节点创建
|
||
console.log('\n3. 测试创建节点...');
|
||
const createResult = await this.callTool('node_create_node', {
|
||
name: 'MCPTestNode_' + Date.now(),
|
||
nodeType: 'cc.Node',
|
||
position: { x: 0, y: 0, z: 0 }
|
||
});
|
||
console.log('创建节点结果:', createResult);
|
||
|
||
// 解析创建节点的结果
|
||
let nodeUuid: string | null = null;
|
||
if (createResult.content && createResult.content[0] && createResult.content[0].text) {
|
||
try {
|
||
const resultData = JSON.parse(createResult.content[0].text);
|
||
if (resultData.success && resultData.data && resultData.data.uuid) {
|
||
nodeUuid = resultData.data.uuid;
|
||
console.log('成功获取节点UUID:', nodeUuid);
|
||
}
|
||
} catch (e) {
|
||
}
|
||
}
|
||
|
||
if (nodeUuid) {
|
||
// 4. 测试查询节点
|
||
console.log('\n4. 测试查询节点...');
|
||
const queryResult = await this.callTool('node_get_node_info', {
|
||
uuid: nodeUuid
|
||
});
|
||
console.log('节点信息:', JSON.stringify(queryResult).substring(0, 100) + '...');
|
||
|
||
// 5. 测试删除节点
|
||
console.log('\n5. 测试删除节点...');
|
||
const removeResult = await this.callTool('node_delete_node', {
|
||
uuid: nodeUuid
|
||
});
|
||
console.log('删除结果:', removeResult);
|
||
} else {
|
||
console.log('无法从创建结果获取节点UUID，尝试通过名称查找...');
|
||
|
||
// 备用方案：通过名称查找刚创建的节点
|
||
const findResult = await this.callTool('node_find_node_by_name', {
|
||
name: 'MCPTestNode_' + Date.now()
|
||
});
|
||
|
||
if (findResult.content && findResult.content[0] && findResult.content[0].text) {
|
||
try {
|
||
const findData = JSON.parse(findResult.content[0].text);
|
||
if (findData.success && findData.data && findData.data.uuid) {
|
||
nodeUuid = findData.data.uuid;
|
||
console.log('通过名称查找成功获取UUID:', nodeUuid);
|
||
}
|
||
} catch (e) {
|
||
}
|
||
}
|
||
|
||
if (!nodeUuid) {
|
||
console.log('所有方式都无法获取节点UUID，跳过后续节点操作测试');
|
||
}
|
||
}
|
||
|
||
// 6. 测试项目工具
|
||
console.log('\n6. 测试项目信息...');
|
||
const projectInfo = await this.callTool('project_get_project_info');
|
||
console.log('项目信息:', JSON.stringify(projectInfo).substring(0, 100) + '...');
|
||
|
||
// 7. 测试预制体工具
|
||
console.log('\n7. 测试预制体列表...');
|
||
const prefabResult = await this.callTool('prefab_get_prefab_list', {
|
||
folder: 'db://assets'
|
||
});
|
||
console.log('找到预制体:', prefabResult.data?.length || 0);
|
||
|
||
// 8. 测试组件工具
|
||
console.log('\n8. 测试可用组件...');
|
||
const componentsResult = await this.callTool('component_get_available_components');
|
||
console.log('可用组件:', JSON.stringify(componentsResult).substring(0, 100) + '...');
|
||
|
||
// 9. 测试调试工具
|
||
console.log('\n9. 测试编辑器信息...');
|
||
const editorInfo = await this.callTool('debug_get_editor_info');
|
||
console.log('编辑器信息:', JSON.stringify(editorInfo).substring(0, 100) + '...');
|
||
|
||
} catch (error) {
|
||
console.error('MCP 工具测试失败:', error);
|
||
}
|
||
}
|
||
|
||
disconnect() {
|
||
if (this.ws) {
|
||
this.ws.close();
|
||
this.ws = null;
|
||
}
|
||
this.responseHandlers.clear();
|
||
}
|
||
}
|
||
|
||
// 导出到全局方便测试
|
||
(global as any).MCPToolTester = MCPToolTester; |