@@ -3,20 +3,21 @@ import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
33import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js' ;
44import { McpUnity } from './unity/mcpUnity.js' ;
55import { Logger , LogLevel } from './utils/logger.js' ;
6- import { createMenuItemTool } from './tools/menuItemTool.js' ;
7- import { createSelectGameObjectTool } from './tools/selectGameObjectTool.js' ;
8- import { createAddPackageTool } from './tools/addPackageTool.js' ;
9- import { createRunTestsTool } from './tools/runTestsTool.js' ;
10- import { createSendConsoleLogTool } from './tools/sendConsoleLogTool.js' ;
11- import { createUpdateComponentTool } from './tools/updateComponentTool.js' ;
12- import { createAddAssetToSceneTool } from './tools/addAssetToSceneTool.js' ;
13- import { createGetMenuItemsResource } from './resources/getMenuItemResource.js' ;
14- import { createGetConsoleLogsResource } from './resources/getConsoleLogResource.js' ;
15- import { createGetHierarchyResource } from './resources/getHierarchyResource.js' ;
16- import { createGetPackagesResource } from './resources/getPackagesResource.js' ;
17- import { createGetAssetsResource } from './resources/getAssetsResource.js' ;
18- import { createGetTestsResource } from './resources/getTestsResource.js' ;
19- import { createGetGameObjectResource } from './resources/getGameObjectResource.js' ;
6+ import { registerMenuItemTool } from './tools/menuItemTool.js' ;
7+ import { registerSelectGameObjectTool } from './tools/selectGameObjectTool.js' ;
8+ import { registerAddPackageTool } from './tools/addPackageTool.js' ;
9+ import { registerRunTestsTool } from './tools/runTestsTool.js' ;
10+ import { registerSendConsoleLogTool } from './tools/sendConsoleLogTool.js' ;
11+ import { registerUpdateComponentTool } from './tools/updateComponentTool.js' ;
12+ import { registerAddAssetToSceneTool } from './tools/addAssetToSceneTool.js' ;
13+ import { registerGetMenuItemsResource } from './resources/getMenuItemResource.js' ;
14+ import { registerGetConsoleLogsResource } from './resources/getConsoleLogResource.js' ;
15+ import { registerGetHierarchyResource } from './resources/getHierarchyResource.js' ;
16+ import { registerGetPackagesResource } from './resources/getPackagesResource.js' ;
17+ import { registerGetAssetsResource } from './resources/getAssetsResource.js' ;
18+ import { registerGetTestsResource } from './resources/getTestsResource.js' ;
19+ import { registerGetGameObjectResource } from './resources/getGameObjectResource.js' ;
20+ import { registerGameObjectHandlingStrategyPrompt } from './prompts/gameobjectHandlingStrategyPrompt.js' ;
2021// Initialize loggers
2122const serverLogger = new Logger ( 'Server' , LogLevel . INFO ) ;
2223const unityLogger = new Logger ( 'Unity' , LogLevel . INFO ) ;
@@ -30,26 +31,29 @@ const server = new McpServer({
3031 capabilities : {
3132 tools : { } ,
3233 resources : { } ,
34+ prompts : { } ,
3335 } ,
3436} ) ;
3537// Initialize MCP HTTP bridge with Unity editor
3638const mcpUnity = new McpUnity ( unityLogger ) ;
37- // Add all tools to the registry
38- createMenuItemTool ( server , mcpUnity , toolLogger ) ;
39- createSelectGameObjectTool ( server , mcpUnity , toolLogger ) ;
40- createAddPackageTool ( server , mcpUnity , toolLogger ) ;
41- createRunTestsTool ( server , mcpUnity , toolLogger ) ;
42- createSendConsoleLogTool ( server , mcpUnity , toolLogger ) ;
43- createUpdateComponentTool ( server , mcpUnity , toolLogger ) ;
44- createAddAssetToSceneTool ( server , mcpUnity , toolLogger ) ;
45- // Create and register all resources with the MCP server
46- createGetTestsResource ( server , mcpUnity , resourceLogger ) ;
47- createGetGameObjectResource ( server , mcpUnity , resourceLogger ) ;
48- createGetMenuItemsResource ( server , mcpUnity , resourceLogger ) ;
49- createGetConsoleLogsResource ( server , mcpUnity , resourceLogger ) ;
50- createGetHierarchyResource ( server , mcpUnity , resourceLogger ) ;
51- createGetPackagesResource ( server , mcpUnity , resourceLogger ) ;
52- createGetAssetsResource ( server , mcpUnity , resourceLogger ) ;
39+ // Register all tools into the MCP server
40+ registerMenuItemTool ( server , mcpUnity , toolLogger ) ;
41+ registerSelectGameObjectTool ( server , mcpUnity , toolLogger ) ;
42+ registerAddPackageTool ( server , mcpUnity , toolLogger ) ;
43+ registerRunTestsTool ( server , mcpUnity , toolLogger ) ;
44+ registerSendConsoleLogTool ( server , mcpUnity , toolLogger ) ;
45+ registerUpdateComponentTool ( server , mcpUnity , toolLogger ) ;
46+ registerAddAssetToSceneTool ( server , mcpUnity , toolLogger ) ;
47+ // Register all resources into the MCP server
48+ registerGetTestsResource ( server , mcpUnity , resourceLogger ) ;
49+ registerGetGameObjectResource ( server , mcpUnity , resourceLogger ) ;
50+ registerGetMenuItemsResource ( server , mcpUnity , resourceLogger ) ;
51+ registerGetConsoleLogsResource ( server , mcpUnity , resourceLogger ) ;
52+ registerGetHierarchyResource ( server , mcpUnity , resourceLogger ) ;
53+ registerGetPackagesResource ( server , mcpUnity , resourceLogger ) ;
54+ registerGetAssetsResource ( server , mcpUnity , resourceLogger ) ;
55+ // Register all prompts into the MCP server
56+ registerGameObjectHandlingStrategyPrompt ( server ) ;
5357// Server startup function
5458async function startServer ( ) {
5559 try {
0 commit comments