/**
* Tools Registration Module
*
* This file registers all available MCP tools with the server.
*
* ADDING A NEW TOOL:
* 1. Create your tool file in src/tools/
* 2. Import it at the top of this file
* 3. Call it in the registerTools function below
*
* See docs/CONTRIBUTING.md for detailed instructions.
* See src/tools/README.md for tool organization.
* See src/tools/metadata/README.md for YAML metadata approach.
*/
import { FastMCP } from 'fastmcp/dist/FastMCP.js';
import log from '../logger.js';
import answerAppium from './documentation/answer-appium.js';
import createSession from './session/create-session.js';
import deleteSession from './session/delete-session.js';
import generateLocators from './test-generation/locators.js';
import selectPlatform from './session/select-platform.js';
import selectDevice from './session/select-device.js';
import bootSimulator from './ios/boot-simulator.js';
import setupWDA from './ios/setup-wda.js';
import installWDA from './ios/install-wda.js';
import generateTest from './test-generation/generate-tests.js';
import scroll from './navigations/scroll.js';
import scrollToElement from './navigations/scroll-to-element.js';
import swipe from './navigations/swipe.js';
import findElement from './interactions/find.js';
import clickElement from './interactions/click.js';
import doubleTap from './interactions/double-tap.js';
import longPress from './interactions/long-press.js';
import dragAndDrop from './interactions/drag-and-drop.js';
import setValue from './interactions/set-value.js';
import getText from './interactions/get-text.js';
import getPageSource from './interactions/get-page-source.js';
import screenshot from './interactions/screenshot.js';
import activateApp from './app-management/activate-app.js';
import installApp from './app-management/install-app.js';
import uninstallApp from './app-management/uninstall-app.js';
import terminateApp from './app-management/terminate-app.js';
import listApps from './app-management/list-apps.js';
import getContexts from './context/get-contexts.js';
import switchContext from './context/switch-context.js';
export default function registerTools(server: FastMCP): void {
// Wrap addTool to inject logging around tool execution
const originalAddTool = (server as any).addTool.bind(server);
(server as any).addTool = (toolDef: any) => {
const toolName = toolDef?.name ?? 'unknown_tool';
const originalExecute = toolDef?.execute;
if (typeof originalExecute !== 'function') {
return originalAddTool(toolDef);
}
const SENSITIVE_KEYS = [
'password',
'token',
'accessToken',
'authorization',
'apiKey',
'apikey',
'secret',
'clientSecret',
];
const redactArgs = (obj: any) => {
try {
return JSON.parse(
JSON.stringify(obj, (key, value) => {
if (
key &&
SENSITIVE_KEYS.some(k => key.toLowerCase().includes(k))
) {
return '[REDACTED]';
}
// Avoid logging extremely large buffers/strings
if (value && typeof value === 'string' && value.length > 2000) {
return `[string:${value.length}]`;
}
if (
value &&
typeof Buffer !== 'undefined' &&
Buffer.isBuffer(value)
) {
return `[buffer:${(value as Buffer).length}]`;
}
return value;
})
);
} catch {
return '[Unserializable args]';
}
};
return originalAddTool({
...toolDef,
execute: async (args: any, context: any) => {
const start = Date.now();
log.info(`[TOOL START] ${toolName}`, redactArgs(args));
try {
const result = await originalExecute(args, context);
const duration = Date.now() - start;
log.info(`[TOOL END] ${toolName} (${duration}ms)`);
return result;
} catch (err: any) {
const duration = Date.now() - start;
const msg = err?.stack || err?.message || String(err);
log.error(`[TOOL ERROR] ${toolName} (${duration}ms): ${msg}`);
throw err;
}
},
});
};
// Session Management
selectPlatform(server);
selectDevice(server);
createSession(server);
deleteSession(server);
// iOS Setup
bootSimulator(server);
setupWDA(server);
installWDA(server);
// Navigation
scroll(server);
scrollToElement(server);
swipe(server);
// Element Interactions
findElement(server);
clickElement(server);
doubleTap(server);
longPress(server);
dragAndDrop(server);
setValue(server);
getText(server);
getPageSource(server);
screenshot(server);
// App Management
activateApp(server);
installApp(server);
uninstallApp(server);
terminateApp(server);
listApps(server);
// Context Management
getContexts(server);
switchContext(server);
// Test Generation
generateLocators(server);
generateTest(server);
// Documentation
answerAppium(server);
log.info('All tools registered');
}
