"use strict"; const http = require("http"); const path = require("path"); let logBuffer = []; // 存储所有日志 let mcpServer = null; let isSceneBusy = false; let serverConfig = { port: 3456, active: false, }; // 封装日志函数，同时发送给面板和编辑器控制台 function addLog(type, message) { const logEntry = { time: new Date().toLocaleTimeString(), type: type, content: message, }; logBuffer.push(logEntry); Editor.Ipc.sendToPanel("mcp-bridge", "mcp-bridge:on-log", logEntry); // 【修改】移除 Editor.log，保持编辑器控制台干净 // 仅在非常严重的系统错误时才输出到编辑器 if (type === "error") { Editor.error(`[MCP] ${message}`); // 如果你完全不想在编辑器看，可以注释掉 } } const getNewSceneTemplate = () => { // 尝试获取 UUID 生成函数 let newId = ""; if (Editor.Utils && Editor.Utils.uuid) { newId = Editor.Utils.uuid(); } else if (Editor.Utils && Editor.Utils.UuidUtils && Editor.Utils.UuidUtils.uuid) { newId = Editor.Utils.UuidUtils.uuid(); } else { // 兜底方案：如果找不到编辑器 API，生成一个随机字符串 newId = Math.random().toString(36).substring(2, 15); } const sceneData = [ { __type__: "cc.SceneAsset", _name: "", _objFlags: 0, _native: "", scene: { __id__: 1 }, }, { __id__: 1, __type__: "cc.Scene", _name: "", _objFlags: 0, _parent: null, _children: [], _active: true, _level: 0, _components: [], autoReleaseAssets: false, _id: newId, }, ]; return JSON.stringify(sceneData); }; const getToolsList = () => { return [ { name: "get_selected_node", description: "获取当前编辑器中选中的节点 ID", inputSchema: { type: "object", properties: {} }, }, { name: "set_node_name", description: "修改指定节点的名称", inputSchema: { type: "object", properties: { id: { type: "string", description: "节点的 UUID" }, newName: { type: "string", description: "新的节点名称" }, }, required: ["id", "newName"], }, }, { name: "save_scene", description: "保存当前场景的修改", inputSchema: { type: "object", properties: {} }, }, { name: "get_scene_hierarchy", description: "获取当前场景的完整节点树结构（包括 UUID、名称和层级关系）", inputSchema: { type: "object", properties: {} }, }, { name: "update_node_transform", description: "修改节点的坐标、缩放或颜色", inputSchema: { type: "object", properties: { id: { type: "string", description: "节点 UUID" }, x: { type: "number" }, y: { type: "number" }, scaleX: { type: "number" }, scaleY: { type: "number" }, color: { type: "string", description: "HEX 颜色代码如 #FF0000" }, }, required: ["id"], }, }, { name: "create_scene", description: "在 assets 目录下创建一个新的场景文件", inputSchema: { type: "object", properties: { sceneName: { type: "string", description: "场景名称" }, }, required: ["sceneName"], }, }, { name: "create_prefab", description: "将场景中的某个节点保存为预制体资源", inputSchema: { type: "object", properties: { nodeId: { type: "string", description: "节点 UUID" }, prefabName: { type: "string", description: "预制体名称" }, }, required: ["nodeId", "prefabName"], }, }, { name: "open_scene", description: "打开场景文件。注意：这是一个异步且耗时的操作，打开后请等待几秒再进行节点创建或保存操作。", inputSchema: { type: "object", properties: { url: { type: "string", description: "场景资源路径，如 db://assets/NewScene.fire", }, }, required: ["url"], }, }, { name: "create_node", description: "在当前场景中创建一个新节点", inputSchema: { type: "object", properties: { name: { type: "string", description: "节点名称" }, parentId: { type: "string", description: "父节点 UUID (可选，不传则挂在场景根部)", }, type: { type: "string", enum: ["empty", "sprite", "label"], description: "节点预设类型", }, }, required: ["name"], }, }, { name: "manage_components", description: "管理节点组件", inputSchema: { type: "object", properties: { nodeId: { type: "string", description: "节点 UUID" }, action: { type: "string", enum: ["add", "remove", "get"], description: "操作类型" }, componentType: { type: "string", description: "组件类型，如 cc.Sprite" }, componentId: { type: "string", description: "组件 ID (用于 remove 操作)" }, properties: { type: "object", description: "组件属性 (用于 add 操作)" }, }, required: ["nodeId", "action"], }, }, { name: "manage_script", description: "管理脚本文件", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "delete", "read", "write"], description: "操作类型" }, path: { type: "string", description: "脚本路径，如 db://assets/scripts/NewScript.js" }, content: { type: "string", description: "脚本内容 (用于 create 和 write 操作)" }, name: { type: "string", description: "脚本名称 (用于 create 操作)" }, }, required: ["action", "path"], }, }, { name: "batch_execute", description: "批处理执行多个操作", inputSchema: { type: "object", properties: { operations: { type: "array", items: { type: "object", properties: { tool: { type: "string", description: "工具名称" }, params: { type: "object", description: "工具参数" }, }, required: ["tool", "params"], }, description: "操作列表", }, }, required: ["operations"], }, }, { name: "manage_asset", description: "管理资源", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "delete", "move", "get_info"], description: "操作类型" }, path: { type: "string", description: "资源路径，如 db://assets/textures" }, targetPath: { type: "string", description: "目标路径 (用于 move 操作)" }, content: { type: "string", description: "资源内容 (用于 create 操作)" }, }, required: ["action", "path"], }, }, { name: "scene_management", description: "场景管理", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "delete", "duplicate", "get_info"], description: "操作类型", }, path: { type: "string", description: "场景路径，如 db://assets/scenes/NewScene.fire" }, targetPath: { type: "string", description: "目标路径 (用于 duplicate 操作)" }, name: { type: "string", description: "场景名称 (用于 create 操作)" }, }, required: ["action", "path"], }, }, { name: "prefab_management", description: "预制体管理", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "update", "instantiate", "get_info"], description: "操作类型", }, path: { type: "string", description: "预制体路径，如 db://assets/prefabs/NewPrefab.prefab" }, nodeId: { type: "string", description: "节点 ID (用于 create 操作)" }, parentId: { type: "string", description: "父节点 ID (用于 instantiate 操作)" }, }, required: ["action", "path"], }, }, { name: "manage_editor", description: "管理编辑器", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["get_selection", "set_selection", "refresh_editor"], description: "操作类型", }, target: { type: "string", enum: ["node", "asset"], description: "目标类型 (用于 set_selection 操作)", }, properties: { type: "object", description: "操作属性" }, }, required: ["action"], }, }, { name: "find_gameobjects", description: "查找游戏对象", inputSchema: { type: "object", properties: { conditions: { type: "object", description: "查找条件" }, recursive: { type: "boolean", default: true, description: "是否递归查找" }, }, required: ["conditions"], }, }, { name: "manage_material", description: "管理材质", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "delete", "get_info"], description: "操作类型" }, path: { type: "string", description: "材质路径，如 db://assets/materials/NewMaterial.mat" }, properties: { type: "object", description: "材质属性" }, }, required: ["action", "path"], }, }, { name: "manage_texture", description: "管理纹理", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "delete", "get_info"], description: "操作类型" }, path: { type: "string", description: "纹理路径，如 db://assets/textures/NewTexture.png" }, properties: { type: "object", description: "纹理属性" }, }, required: ["action", "path"], }, }, { name: "execute_menu_item", description: "执行菜单项", inputSchema: { type: "object", properties: { menuPath: { type: "string", description: "菜单项路径" }, }, required: ["menuPath"], }, }, { name: "apply_text_edits", description: "应用文本编辑", inputSchema: { type: "object", properties: { filePath: { type: "string", description: "文件路径" }, edits: { type: "array", items: { type: "object" }, description: "编辑操作列表" }, }, required: ["filePath", "edits"], }, }, { name: "read_console", description: "读取控制台", inputSchema: { type: "object", properties: { limit: { type: "number", description: "输出限制" }, type: { type: "string", enum: ["log", "error", "warn"], description: "输出类型" }, }, }, }, { name: "validate_script", description: "验证脚本", inputSchema: { type: "object", properties: { filePath: { type: "string", description: "脚本路径" }, }, required: ["filePath"], }, }, { name: "find_in_file", description: "在项目中全局搜索文本内容", inputSchema: { type: "object", properties: { query: { type: "string", description: "搜索关键词" }, extensions: { type: "array", items: { type: "string" }, description: "文件后缀列表 (例如 ['.js', '.ts'])", default: [".js", ".ts", ".json", ".fire", ".prefab", ".xml", ".txt", ".md"] }, includeSubpackages: { type: "boolean", default: true, description: "是否搜索子包 (暂时默认搜索 assets 目录)" } }, required: ["query"] } }, { name: "manage_undo", description: "管理编辑器的撤销和重做历史", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["undo", "redo", "begin_group", "end_group", "cancel_group"], description: "操作类型" }, description: { type: "string", description: "撤销组的描述 (用于 begin_group)" } }, required: ["action"] } }, { name: "manage_vfx", description: "管理全场景特效 (粒子系统)", inputSchema: { type: "object", properties: { action: { type: "string", enum: ["create", "update", "get_info"], description: "操作类型" }, nodeId: { type: "string", description: "节点 UUID (用于 update/get_info)" }, properties: { type: "object", description: "粒子系统属性 (用于 create/update)", properties: { duration: { type: "number", description: "发射时长" }, emissionRate: { type: "number", description: "发射速率" }, life: { type: "number", description: "生命周期" }, lifeVar: { type: "number", description: "生命周期变化" }, startColor: { type: "string", description: "起始颜色 (Hex)" }, endColor: { type: "string", description: "结束颜色 (Hex)" }, startSize: { type: "number", description: "起始大小" }, endSize: { type: "number", description: "结束大小" }, speed: { type: "number", description: "速度" }, angle: { type: "number", description: "角度" }, gravity: { type: "object", properties: { x: { type: "number" }, y: { type: "number" } } }, file: { type: "string", description: "粒子文件路径 (plist) 或 texture 路径" } } }, name: { type: "string", description: "节点名称 (用于 create)" }, parentId: { type: "string", description: "父节点 ID (用于 create)" } }, required: ["action"] } } ]; }; module.exports = { "scene-script": "scene-script.js", load() { addLog("info", "MCP Bridge Plugin Loaded"); // 读取配置 let profile = this.getProfile(); serverConfig.port = profile.get("last-port") || 3456; let autoStart = profile.get("auto-start"); if (autoStart) { addLog("info", "Auto-start is enabled. Initializing server..."); // 延迟一点启动，确保编辑器环境完全就绪 setTimeout(() => { this.startServer(serverConfig.port); }, 1000); } }, // 获取配置文件的辅助函数 getProfile() { // 'local' 表示存储在项目本地（local/mcp-bridge.json） return Editor.Profile.load("profile://local/mcp-bridge.json", "mcp-bridge"); }, unload() { this.stopServer(); }, startServer(port) { if (mcpServer) this.stopServer(); try { mcpServer = http.createServer((req, res) => { res.setHeader("Content-Type", "application/json"); res.setHeader("Access-Control-Allow-Origin", "*"); let body = ""; req.on("data", (chunk) => { body += chunk; }); req.on("end", () => { const url = req.url; if (url === "/list-tools") { const tools = getToolsList(); addLog("info", `AI Client requested tool list`); // 明确返回成功结构 res.writeHead(200); return res.end(JSON.stringify({ tools: tools })); } if (url === "/call-tool") { try { const { name, arguments: args } = JSON.parse(body || "{}"); addLog("mcp", `REQ -> [${name}]`); this.handleMcpCall(name, args, (err, result) => { const response = { content: [ { type: "text", text: err ? `Error: ${err}` : typeof result === "object" ? JSON.stringify(result, null, 2) : result, }, ], }; if (err) { addLog("error", `RES <- [${name}] 失败: ${err}`); } else { // 成功时尝试捕获简单的结果预览（如果是字符串或简短对象） let preview = ""; if (typeof result === 'string') { preview = result.length > 100 ? result.substring(0, 100) + "..." : result; } else if (typeof result === 'object') { try { const jsonStr = JSON.stringify(result); preview = jsonStr.length > 100 ? jsonStr.substring(0, 100) + "..." : jsonStr; } catch (e) { preview = "Object (Circular/Unserializable)"; } } addLog("success", `RES <- [${name}] 成功 : ${preview}`); } res.writeHead(200); res.end(JSON.stringify(response)); }); } catch (e) { if (e instanceof SyntaxError) { addLog("error", `JSON Parse Error: ${e.message}`); res.writeHead(400); res.end(JSON.stringify({ error: "Invalid JSON" })); } else { addLog("error", `Internal Server Error: ${e.message}`); res.writeHead(500); res.end(JSON.stringify({ error: e.message })); } } return; } // --- 兜底处理 (404) --- res.writeHead(404); res.end(JSON.stringify({ error: "Not Found", url: url })); }); }); mcpServer.on("error", (e) => { addLog("error", `Server Error: ${e.message}`); }); mcpServer.listen(port, () => { serverConfig.active = true; addLog("success", `MCP Server running at http://127.0.0.1:${port}`); Editor.Ipc.sendToPanel("mcp-bridge", "mcp-bridge:state-changed", serverConfig); }); // 启动成功后顺便存一下端口 this.getProfile().set("last-port", port); this.getProfile().save(); } catch (e) { addLog("error", `Failed to start server: ${e.message}`); } }, stopServer() { if (mcpServer) { mcpServer.close(); mcpServer = null; serverConfig.active = false; addLog("warn", "MCP Server stopped"); Editor.Ipc.sendToPanel("mcp-bridge", "mcp-bridge:state-changed", serverConfig); } }, // 统一处理逻辑，方便日志记录 handleMcpCall(name, args, callback) { if (isSceneBusy && (name === "save_scene" || name === "create_node")) { return callback("Editor is busy (Processing Scene), please wait a moment."); } switch (name) { case "get_selected_node": const ids = Editor.Selection.curSelection("node"); callback(null, ids); break; case "set_node_name": // 使用 scene:set-property 以支持撤销 Editor.Ipc.sendToPanel("scene", "scene:set-property", { id: args.id, path: "name", type: "String", value: args.newName, isSubProp: false }); callback(null, `Node name updated to ${args.newName}`); break; case "save_scene": isSceneBusy = true; addLog("info", "Preparing to save scene... Waiting for UI sync."); // 强制延迟保存，防止死锁 setTimeout(() => { Editor.Ipc.sendToMain("scene:save-scene"); addLog("info", "Executing Safe Save..."); setTimeout(() => { isSceneBusy = false; addLog("info", "Safe Save completed."); callback(null, "Scene saved successfully."); }, 1000); }, 500); break; case "get_scene_hierarchy": Editor.Scene.callSceneScript("mcp-bridge", "get-hierarchy", callback); break; case "update_node_transform": const { id, x, y, scaleX, scaleY, color } = args; // 将多个属性修改打包到一个 Undo 组中 Editor.Ipc.sendToPanel("scene", "scene:undo-record", "Transform Update"); try { // 注意：Cocos Creator 属性类型通常首字母大写，如 'Float', 'String', 'Boolean' // 也有可能支持 'Number'，但 'Float' 更保险 if (x !== undefined) Editor.Ipc.sendToPanel("scene", "scene:set-property", { id, path: "x", type: "Float", value: x, isSubProp: false }); if (y !== undefined) Editor.Ipc.sendToPanel("scene", "scene:set-property", { id, path: "y", type: "Float", value: y, isSubProp: false }); if (scaleX !== undefined) Editor.Ipc.sendToPanel("scene", "scene:set-property", { id, path: "scaleX", type: "Float", value: scaleX, isSubProp: false }); if (scaleY !== undefined) Editor.Ipc.sendToPanel("scene", "scene:set-property", { id, path: "scaleY", type: "Float", value: scaleY, isSubProp: false }); if (color) { // 颜色稍微复杂，传递 hex 字符串可能需要 Color 对象转换，但 set-property 也许可以直接接受 info // 安全起见，颜色还是走 scene-script 或者尝试直接 set-property // 这里的 color 是 Hex String。尝试传 String 让编辑器解析? // 通常编辑器需要 cc.Color 对象或 {r,g,b,a} // 暂时保留 color 通过 scene-script 处理? 或者跳过? // 为了保持一致性，还是走 scene-script 更新颜色，但这样颜色可能无法 undo。 // 改进：使用 scene script 处理颜色，但尝试手动 record? // 暂且忽略颜色的 Undo，先保证 Transform 的 Undo。 Editor.Scene.callSceneScript("mcp-bridge", "update-node-transform", { id, color }, (err) => { if (err) addLog("warn", "Color update failed or partial"); }); } Editor.Ipc.sendToPanel("scene", "scene:undo-commit"); callback(null, "Transform updated"); } catch (e) { Editor.Ipc.sendToPanel("scene", "scene:undo-cancel"); callback(e); } break; case "create_scene": const sceneUrl = `db://assets/${args.sceneName}.fire`; if (Editor.assetdb.exists(sceneUrl)) { return callback("Scene already exists"); } Editor.assetdb.create(sceneUrl, getNewSceneTemplate(), (err) => { callback(err, err ? null : `Standard Scene created at ${sceneUrl}`); }); break; case "create_prefab": const prefabUrl = `db://assets/${args.prefabName}.prefab`; Editor.Ipc.sendToMain("scene:create-prefab", args.nodeId, prefabUrl); callback(null, `Command sent: Creating prefab '${args.prefabName}'`); break; case "open_scene": isSceneBusy = true; // 锁定 const openUuid = Editor.assetdb.urlToUuid(args.url); if (openUuid) { Editor.Ipc.sendToMain("scene:open-by-uuid", openUuid); setTimeout(() => { isSceneBusy = false; callback(null, `Success: Opening scene ${args.url}`); }, 2000); } else { isSceneBusy = false; callback(`Could not find asset with URL ${args.url}`); } break; case "create_node": Editor.Scene.callSceneScript("mcp-bridge", "create-node", args, callback); break; case "manage_components": Editor.Scene.callSceneScript("mcp-bridge", "manage-components", args, callback); break; case "manage_script": this.manageScript(args, callback); break; case "batch_execute": this.batchExecute(args, callback); break; case "manage_asset": this.manageAsset(args, callback); break; case "scene_management": this.sceneManagement(args, callback); break; case "prefab_management": this.prefabManagement(args, callback); break; case "manage_editor": this.manageEditor(args, callback); break; case "find_gameobjects": Editor.Scene.callSceneScript("mcp-bridge", "find-gameobjects", args, callback); break; case "manage_material": this.manageMaterial(args, callback); break; case "manage_texture": this.manageTexture(args, callback); break; case "execute_menu_item": this.executeMenuItem(args, callback); break; case "apply_text_edits": this.applyTextEdits(args, callback); break; case "read_console": this.readConsole(args, callback); break; case "validate_script": this.validateScript(args, callback); break; case "find_in_file": this.findInFile(args, callback); break; case "manage_undo": this.manageUndo(args, callback); break; case "manage_vfx": // 【修复】在主进程预先解析 URL 为 UUID，因为渲染进程(scene-script)无法访问 Editor.assetdb if (args.properties && args.properties.file) { if (typeof args.properties.file === 'string' && args.properties.file.startsWith("db://")) { const uuid = Editor.assetdb.urlToUuid(args.properties.file); if (uuid) { args.properties.file = uuid; // 替换为 UUID } else { console.warn(`Failed to resolve path to UUID: ${args.properties.file}`); } } } // 预先获取默认贴图 UUID (尝试多个可能的路径) const defaultPaths = [ "db://internal/image/default_sprite_splash", "db://internal/image/default_sprite_splash.png", "db://internal/image/default_particle", "db://internal/image/default_particle.png" ]; for (const path of defaultPaths) { const uuid = Editor.assetdb.urlToUuid(path); if (uuid) { args.defaultSpriteUuid = uuid; addLog("info", `[mcp-bridge] Resolved Default Sprite UUID: ${uuid} from ${path}`); break; } } if (!args.defaultSpriteUuid) { addLog("warn", "[mcp-bridge] Failed to resolve any default sprite UUID."); } Editor.Scene.callSceneScript("mcp-bridge", "manage-vfx", args, callback); break; default: callback(`Unknown tool: ${name}`); break; } }, // 管理脚本文件 manageScript(args, callback) { const { action, path, content } = args; switch (action) { case "create": if (Editor.assetdb.exists(path)) { return callback(`Script already exists at ${path}`); } // 确保父目录存在 const fs = require("fs"); const pathModule = require("path"); const absolutePath = Editor.assetdb.urlToFspath(path); const dirPath = pathModule.dirname(absolutePath); if (!fs.existsSync(dirPath)) { fs.mkdirSync(dirPath, { recursive: true }); } Editor.assetdb.create( path, content || `const { ccclass, property } = cc._decorator; @ccclass export default class NewScript extends cc.Component { @property(cc.Label) label: cc.Label = null; @property text: string = 'hello'; // LIFE-CYCLE CALLBACKS: onLoad () {} start () {} update (dt) {} }`, (err) => { callback(err, err ? null : `Script created at ${path}`); }, ); break; case "delete": if (!Editor.assetdb.exists(path)) { return callback(`Script not found at ${path}`); } Editor.assetdb.delete([path], (err) => { callback(err, err ? null : `Script deleted at ${path}`); }); break; case "read": Editor.assetdb.queryInfoByUuid(Editor.assetdb.urlToUuid(path), (err, info) => { if (err) { return callback(`Failed to get script info: ${err}`); } Editor.assetdb.loadAny(path, (err, content) => { callback(err, err ? null : content); }); }); break; case "write": Editor.assetdb.create(path, content, (err) => { callback(err, err ? null : `Script updated at ${path}`); }); break; default: callback(`Unknown script action: ${action}`); break; } }, // 批处理执行 batchExecute(args, callback) { const { operations } = args; const results = []; let completed = 0; if (!operations || operations.length === 0) { return callback("No operations provided"); } operations.forEach((operation, index) => { this.handleMcpCall(operation.tool, operation.params, (err, result) => { results[index] = { tool: operation.tool, error: err, result: result }; completed++; if (completed === operations.length) { callback(null, results); } }); }); }, // 管理资源 manageAsset(args, callback) { const { action, path, targetPath, content } = args; switch (action) { case "create": if (Editor.assetdb.exists(path)) { return callback(`Asset already exists at ${path}`); } // 确保父目录存在 const fs = require("fs"); const pathModule = require("path"); const absolutePath = Editor.assetdb.urlToFspath(path); const dirPath = pathModule.dirname(absolutePath); if (!fs.existsSync(dirPath)) { fs.mkdirSync(dirPath, { recursive: true }); } Editor.assetdb.create(path, content || "", (err) => { callback(err, err ? null : `Asset created at ${path}`); }); break; case "delete": if (!Editor.assetdb.exists(path)) { return callback(`Asset not found at ${path}`); } Editor.assetdb.delete([path], (err) => { callback(err, err ? null : `Asset deleted at ${path}`); }); break; case "move": if (!Editor.assetdb.exists(path)) { return callback(`Asset not found at ${path}`); } if (Editor.assetdb.exists(targetPath)) { return callback(`Target asset already exists at ${targetPath}`); } Editor.assetdb.move(path, targetPath, (err) => { callback(err, err ? null : `Asset moved from ${path} to ${targetPath}`); }); break; case "get_info": try { if (!Editor.assetdb.exists(path)) { return callback(`Asset not found: ${path}`); } const uuid = Editor.assetdb.urlToUuid(path); // Return basic info constructed manually to avoid API compatibility issues callback(null, { url: path, uuid: uuid, exists: true }); } catch (e) { callback(`Error getting asset info: ${e.message}`); } break; default: callback(`Unknown asset action: ${action}`); break; } }, // 场景管理 sceneManagement(args, callback) { const { action, path, targetPath, name } = args; switch (action) { case "create": if (Editor.assetdb.exists(path)) { return callback(`Scene already exists at ${path}`); } // 确保父目录存在 const fs = require("fs"); const pathModule = require("path"); const absolutePath = Editor.assetdb.urlToFspath(path); const dirPath = pathModule.dirname(absolutePath); if (!fs.existsSync(dirPath)) { fs.mkdirSync(dirPath, { recursive: true }); } Editor.assetdb.create(path, getNewSceneTemplate(), (err) => { callback(err, err ? null : `Scene created at ${path}`); }); break; case "delete": if (!Editor.assetdb.exists(path)) { return callback(`Scene not found at ${path}`); } Editor.assetdb.delete([path], (err) => { callback(err, err ? null : `Scene deleted at ${path}`); }); break; case "duplicate": if (!Editor.assetdb.exists(path)) { return callback(`Scene not found at ${path}`); } if (!targetPath) { return callback(`Target path is required for duplicate operation`); } if (Editor.assetdb.exists(targetPath)) { return callback(`Target scene already exists at ${targetPath}`); } // 读取原场景内容 Editor.assetdb.loadAny(path, (err, content) => { if (err) { return callback(`Failed to read scene: ${err}`); } // 确保目标目录存在 const fs = require("fs"); const pathModule = require("path"); const targetAbsolutePath = Editor.assetdb.urlToFspath(targetPath); const targetDirPath = pathModule.dirname(targetAbsolutePath); if (!fs.existsSync(targetDirPath)) { fs.mkdirSync(targetDirPath, { recursive: true }); } // 创建复制的场景 Editor.assetdb.create(targetPath, content, (err) => { callback(err, err ? null : `Scene duplicated from ${path} to ${targetPath}`); }); }); break; case "get_info": Editor.assetdb.queryInfoByUuid(Editor.assetdb.urlToUuid(path), (err, info) => { callback(err, err ? null : info); }); break; default: callback(`Unknown scene action: ${action}`); break; } }, // 预制体管理 prefabManagement(args, callback) { const { action, path, nodeId, parentId } = args; switch (action) { case "create": if (!nodeId) { return callback(`Node ID is required for create operation`); } if (Editor.assetdb.exists(path)) { return callback(`Prefab already exists at ${path}`); } // 确保父目录存在 const fs = require("fs"); const pathModule = require("path"); const absolutePath = Editor.assetdb.urlToFspath(path); const dirPath = pathModule.dirname(absolutePath); if (!fs.existsSync(dirPath)) { fs.mkdirSync(dirPath, { recursive: true }); } // 从节点创建预制体 Editor.Ipc.sendToMain("scene:create-prefab", nodeId, path); callback(null, `Command sent: Creating prefab from node ${nodeId} at ${path}`); break; case "update": if (!nodeId) { return callback(`Node ID is required for update operation`); } if (!Editor.assetdb.exists(path)) { return callback(`Prefab not found at ${path}`); } // 更新预制体 Editor.Ipc.sendToMain("scene:update-prefab", nodeId, path); callback(null, `Command sent: Updating prefab ${path} from node ${nodeId}`); break; case "instantiate": if (!Editor.assetdb.exists(path)) { return callback(`Prefab not found at ${path}`); } // 实例化预制体 Editor.Scene.callSceneScript( "mcp-bridge", "instantiate-prefab", { prefabPath: path, parentId: parentId, }, callback, ); break; case "get_info": Editor.assetdb.queryInfoByUuid(Editor.assetdb.urlToUuid(path), (err, info) => { callback(err, err ? null : info); }); break; default: callback(`Unknown prefab action: ${action}`); } }, // 管理编辑器 manageEditor(args, callback) { const { action, target, properties } = args; switch (action) { case "get_selection": // 获取当前选中的资源或节点 const nodeSelection = Editor.Selection.curSelection("node"); const assetSelection = Editor.Selection.curSelection("asset"); callback(null, { nodes: nodeSelection, assets: assetSelection, }); break; case "set_selection": // 设置选中状态 if (target === "node" && properties.nodes) { Editor.Selection.select("node", properties.nodes); } else if (target === "asset" && properties.assets) { Editor.Selection.select("asset", properties.assets); } callback(null, "Selection updated"); break; case "refresh_editor": // 刷新编辑器 Editor.assetdb.refresh(); callback(null, "Editor refreshed"); break; default: callback("Unknown action"); break; } }, // 管理材质 manageMaterial(args, callback) { const { action, path, properties } = args; switch (action) { case "create": if (Editor.assetdb.exists(path)) { return callback(`Material already exists at ${path}`); } // 确保父目录存在 const fs = require("fs"); const pathModule = require("path"); const absolutePath = Editor.assetdb.urlToFspath(path); const dirPath = pathModule.dirname(absolutePath); if (!fs.existsSync(dirPath)) { fs.mkdirSync(dirPath, { recursive: true }); } // 创建材质资源 const materialContent = JSON.stringify({ __type__: "cc.Material", _name: "", _objFlags: 0, _native: "", effects: [ { technique: 0, defines: {}, uniforms: properties.uniforms || {}, }, ], }); Editor.assetdb.create(path, materialContent, (err) => { callback(err, err ? null : `Material created at ${path}`); }); break; case "delete": if (!Editor.assetdb.exists(path)) { return callback(`Material not found at ${path}`); } Editor.assetdb.delete([path], (err) => { callback(err, err ? null : `Material deleted at ${path}`); }); break; case "get_info": Editor.assetdb.queryInfoByUuid(Editor.assetdb.urlToUuid(path), (err, info) => { callback(err, err ? null : info); }); break; default: callback(`Unknown material action: ${action}`); break; } }, // 管理纹理 manageTexture(args, callback) { const { action, path, properties } = args; switch (action) { case "create": if (Editor.assetdb.exists(path)) { return callback(`Texture already exists at ${path}`); } // 确保父目录存在 const fs = require("fs"); const pathModule = require("path"); const absolutePath = Editor.assetdb.urlToFspath(path); const dirPath = pathModule.dirname(absolutePath); if (!fs.existsSync(dirPath)) { fs.mkdirSync(dirPath, { recursive: true }); } // 创建纹理资源（简化版，实际需要处理纹理文件） const textureContent = JSON.stringify({ __type__: "cc.Texture2D", _name: "", _objFlags: 0, _native: properties.native || "", width: properties.width || 128, height: properties.height || 128, }); Editor.assetdb.create(path, textureContent, (err) => { callback(err, err ? null : `Texture created at ${path}`); }); break; case "delete": if (!Editor.assetdb.exists(path)) { return callback(`Texture not found at ${path}`); } Editor.assetdb.delete([path], (err) => { callback(err, err ? null : `Texture deleted at ${path}`); }); break; case "get_info": Editor.assetdb.queryInfoByUuid(Editor.assetdb.urlToUuid(path), (err, info) => { callback(err, err ? null : info); }); break; default: callback(`Unknown texture action: ${action}`); break; } }, // 执行菜单项 executeMenuItem(args, callback) { const { menuPath } = args; try { // 执行菜单项 Editor.Ipc.sendToMain("menu:click", menuPath); callback(null, `Menu item executed: ${menuPath}`); } catch (err) { callback(`Failed to execute menu item: ${err.message}`); } }, // 应用文本编辑 applyTextEdits(args, callback) { const { filePath, edits } = args; // 读取文件内容 Editor.assetdb.queryInfoByUrl(filePath, (err, info) => { if (err) { callback(`Failed to get file info: ${err.message}`); return; } Editor.assetdb.loadAny(filePath, (err, content) => { if (err) { callback(`Failed to load file: ${err.message}`); return; } // 应用编辑操作 let updatedContent = content; edits.forEach((edit) => { switch (edit.type) { case "insert": updatedContent = updatedContent.slice(0, edit.position) + edit.text + updatedContent.slice(edit.position); break; case "delete": updatedContent = updatedContent.slice(0, edit.start) + updatedContent.slice(edit.end); break; case "replace": updatedContent = updatedContent.slice(0, edit.start) + edit.text + updatedContent.slice(edit.end); break; } }); // 写回文件 Editor.assetdb.create(filePath, updatedContent, (err) => { callback(err, err ? null : `Text edits applied to ${filePath}`); }); }); }); }, // 读取控制台 readConsole(args, callback) { const { limit, type } = args; let filteredOutput = logBuffer; if (type) { filteredOutput = filteredOutput.filter((item) => item.type === type); } if (limit) { filteredOutput = filteredOutput.slice(-limit); } callback(null, filteredOutput); }, // 验证脚本 validateScript(args, callback) { const { filePath } = args; // 读取脚本内容 Editor.assetdb.queryInfoByUrl(filePath, (err, info) => { if (err) { callback(`Failed to get file info: ${err.message}`); return; } Editor.assetdb.loadAny(filePath, (err, content) => { if (err) { callback(`Failed to load file: ${err.message}`); return; } try { // 对于 JavaScript 脚本，使用 eval 进行简单验证 if (filePath.endsWith(".js")) { // 包装在函数中以避免变量污染 const wrapper = `(function() { ${content} })`; eval(wrapper); } // 对于 TypeScript 脚本，这里可以添加更复杂的验证逻辑 callback(null, { valid: true, message: "Script syntax is valid" }); } catch (err) { callback(null, { valid: false, message: err.message }); } }); }); }, // 暴露给 MCP 或面板的 API 封装 messages: { "open-test-panel"() { Editor.Panel.open("mcp-bridge"); }, "get-server-state"(event) { event.reply(null, { config: serverConfig, logs: logBuffer }); }, "toggle-server"(event, port) { if (serverConfig.active) this.stopServer(); else this.startServer(port); }, "clear-logs"() { logBuffer = []; addLog("info", "Logs cleared"); }, // 修改场景中的节点（需要通过 scene-script） "set-node-property"(event, args) { addLog("mcp", `Creating node: ${args.name} (${args.type})`); // 确保第一个参数 'mcp-bridge' 和 package.json 的 name 一致 Editor.Scene.callSceneScript("mcp-bridge", "set-property", args, (err, result) => { if (err) { Editor.error("Scene Script Error:", err); } if (event && event.reply) { event.reply(err, result); } }); }, "create-node"(event, args) { addLog("mcp", `Creating node: ${args.name} (${args.type})`); Editor.Scene.callSceneScript("mcp-bridge", "create-node", args, (err, result) => { if (err) addLog("error", `CreateNode Failed: ${err}`); else addLog("success", `Node Created: ${result}`); event.reply(err, result); }); }, "get-server-state"(event) { let profile = this.getProfile(); event.reply(null, { config: serverConfig, logs: logBuffer, autoStart: profile.get("auto-start"), // 返回自动启动状态 }); }, "set-auto-start"(event, value) { this.getProfile().set("auto-start", value); this.getProfile().save(); addLog("info", `Auto-start set to: ${value}`); }, }, // 验证脚本 validateScript(args, callback) { const { filePath } = args; // 读取脚本内容 Editor.assetdb.queryInfoByUrl(filePath, (err, info) => { if (err) { callback(`Failed to get file info: ${err.message}`); return; } Editor.assetdb.loadMeta(info.uuid, (err, content) => { if (err) { callback(`Failed to load file: ${err.message}`); return; } try { // 对于 JavaScript 脚本，使用 eval 进行简单验证 if (filePath.endsWith('.js')) { // 包装在函数中以避免变量污染 const wrapper = `(function() { ${content} })`; eval(wrapper); } // 对于 TypeScript 脚本，这里可以添加更复杂的验证逻辑 callback(null, { valid: true, message: 'Script syntax is valid' }); } catch (err) { callback(null, { valid: false, message: err.message }); } }); }); }, // 全局文件搜索 findInFile(args, callback) { const { query, extensions, includeSubpackages } = args; const fs = require('fs'); const path = require('path'); const assetsPath = Editor.assetdb.urlToFspath("db://assets"); const validExtensions = extensions || [".js", ".ts", ".json", ".fire", ".prefab", ".xml", ".txt", ".md"]; const results = []; const MAX_RESULTS = 500; // 限制返回结果数量，防止溢出 try { // 递归遍历函数 const walk = (dir) => { if (results.length >= MAX_RESULTS) return; const list = fs.readdirSync(dir); list.forEach((file) => { if (results.length >= MAX_RESULTS) return; // 忽略隐藏文件和 node_modules if (file.startsWith('.') || file === 'node_modules' || file === 'bin' || file === 'local') return; const filePath = path.join(dir, file); const stat = fs.statSync(filePath); if (stat && stat.isDirectory()) { walk(filePath); } else { // 检查后缀 const ext = path.extname(file).toLowerCase(); if (validExtensions.includes(ext)) { try { const content = fs.readFileSync(filePath, 'utf8'); // 简单的行匹配 const lines = content.split('\n'); lines.forEach((line, index) => { if (results.length >= MAX_RESULTS) return; if (line.includes(query)) { // 转换为项目相对路径 (db://assets/...) const relativePath = path.relative(assetsPath, filePath); // 统一使用 forward slash const dbPath = "db://assets/" + relativePath.split(path.sep).join('/'); results.push({ filePath: dbPath, line: index + 1, content: line.trim() }); } }); } catch (e) { // 读取文件出错，跳过 } } } }); }; walk(assetsPath); callback(null, results); } catch (err) { callback(`Find in file failed: ${err.message}`); } }, // 管理撤销/重做 manageUndo(args, callback) { const { action, description } = args; try { switch (action) { case "undo": Editor.Ipc.sendToPanel("scene", "scene:undo"); callback(null, "Undo command executed"); break; case "redo": Editor.Ipc.sendToPanel("scene", "scene:redo"); callback(null, "Redo command executed"); break; case "begin_group": // scene:undo-record [id] // 这里的 id 好像是可选的，或者用于区分不同的事务 Editor.Ipc.sendToPanel("scene", "scene:undo-record", description || "MCP Action"); callback(null, `Undo group started: ${description || "MCP Action"}`); break; case "end_group": Editor.Ipc.sendToPanel("scene", "scene:undo-commit"); callback(null, "Undo group committed"); break; case "cancel_group": Editor.Ipc.sendToPanel("scene", "scene:undo-cancel"); callback(null, "Undo group cancelled"); break; default: callback(`Unknown undo action: ${action}`); } } catch (err) { callback(`Undo operation failed: ${err.message}`); } }, };