import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { McpUnity } from '../unity/mcpUnity.js';
import { McpUnityError, ErrorType } from '../utils/errors.js';
import * as z from 'zod';
import { Logger } from '../utils/logger.js';
const toolName = 'unload_scene';
const toolDescription = 'Unloads a scene by path or name (does not delete the scene asset, just closes it from the hierarchy)';
const paramsSchema = z.object({
scenePath: z.string().optional().describe("Full asset path to the scene (e.g., 'Assets/Scenes/MyScene.unity')"),
sceneName: z.string().optional().describe('Scene name without extension (used if scenePath not provided)'),
saveIfDirty: z.boolean().optional().describe('If true, saves the scene before unloading if it has unsaved changes. Default: true'),
removeScene: z.boolean().optional().describe('If true, removes the scene from the hierarchy. If false, keeps it but unloaded. Default: true')
});
export function registerUnloadSceneTool(server: McpServer, mcpUnity: McpUnity, logger: Logger) {
logger.info(`Registering tool: ${toolName}`);
server.tool(
toolName,
toolDescription,
paramsSchema.shape,
async (params: any) => {
try {
logger.info(`Executing tool: ${toolName}`, params);
const result = await toolHandler(mcpUnity, params);
logger.info(`Tool execution successful: ${toolName}`);
return result;
} catch (error) {
logger.error(`Tool execution failed: ${toolName}`, error);
throw error;
}
}
);
}
async function toolHandler(mcpUnity: McpUnity, params: any) {
if (!params.scenePath && !params.sceneName) {
throw new McpUnityError(
ErrorType.VALIDATION,
"Either 'scenePath' or 'sceneName' must be provided"
);
}
const response = await mcpUnity.sendRequest({
method: toolName,
params
});
if (!response.success) {
throw new McpUnityError(
ErrorType.TOOL_EXECUTION,
response.message || 'Failed to unload scene'
);
}
return {
content: [{
type: response.type,
text: response.message || 'Successfully unloaded scene'
}],
data: {
sceneName: response.sceneName,
scenePath: response.scenePath,
wasDirty: response.wasDirty,
removed: response.removed
}
};
}
