workflows.ts•3.01 kB
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
import { z } from 'zod';
import { validateXcodeInstallation } from '../utils/validation.js';
import { getDescription } from './types.js';
import {
workflowTapElementTool,
WORKFLOW_TAP_ELEMENT_DOCS,
WORKFLOW_TAP_ELEMENT_DOCS_MINI,
} from '../tools/workflows/tap-element.js';
import {
workflowFreshInstallTool,
WORKFLOW_FRESH_INSTALL_DOCS,
WORKFLOW_FRESH_INSTALL_DOCS_MINI,
} from '../tools/workflows/fresh-install.js';
const ENABLE_DEFER_LOADING = process.env.XC_MCP_DEFER_LOADING !== 'false';
const DEFER_LOADING_CONFIG = ENABLE_DEFER_LOADING
? ({ defer_loading: true } as Record<string, unknown>)
: {};
export function registerWorkflowTools(server: McpServer): void {
// workflow-tap-element
server.registerTool(
'workflow-tap-element',
{
description: getDescription(WORKFLOW_TAP_ELEMENT_DOCS, WORKFLOW_TAP_ELEMENT_DOCS_MINI),
inputSchema: {
elementQuery: z.string().describe('Search term for element (e.g., "Login", "Submit")'),
inputText: z.string().optional().describe('Text to type after tapping'),
verifyResult: z.boolean().default(false).describe('Take screenshot after action'),
udid: z.string().optional().describe('Target device'),
screenContext: z.string().optional().describe('Screen name for tracking'),
},
...DEFER_LOADING_CONFIG,
},
async args => {
try {
await validateXcodeInstallation();
return await workflowTapElementTool(args);
} catch (error) {
if (error instanceof McpError) throw error;
throw new McpError(
ErrorCode.InternalError,
`Tool execution failed: ${error instanceof Error ? error.message : String(error)}`
);
}
}
);
// workflow-fresh-install
server.registerTool(
'workflow-fresh-install',
{
description: getDescription(WORKFLOW_FRESH_INSTALL_DOCS, WORKFLOW_FRESH_INSTALL_DOCS_MINI),
inputSchema: {
projectPath: z.string().describe('Path to .xcodeproj or .xcworkspace'),
scheme: z.string().describe('Build scheme name'),
simulatorUdid: z.string().optional().describe('Target simulator'),
eraseSimulator: z.boolean().default(false).describe('Wipe simulator data'),
configuration: z.enum(['Debug', 'Release']).default('Debug'),
launchArguments: z.array(z.string()).optional(),
environmentVariables: z.record(z.string()).optional(),
},
...DEFER_LOADING_CONFIG,
},
async args => {
try {
await validateXcodeInstallation();
return await workflowFreshInstallTool(args);
} catch (error) {
if (error instanceof McpError) throw error;
throw new McpError(
ErrorCode.InternalError,
`Tool execution failed: ${error instanceof Error ? error.message : String(error)}`
);
}
}
);
}