#!/usr/bin/env node
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import {
CallToolRequestSchema,
ListToolsRequestSchema,
ListResourcesRequestSchema,
ReadResourceRequestSchema,
ListPromptsRequestSchema,
GetPromptRequestSchema,
ErrorCode,
McpError,
} from "@modelcontextprotocol/sdk/types.js";
import { allTools } from "./tools/index.js";
import { listResources, readResource } from "./resources/index.js";
import { listPrompts, getPrompt } from "./prompts/index.js";
// Get working directory from environment or use current
const PROJECT_PATH = process.env.DETOX_PROJECT_PATH || process.cwd();
// Create MCP server
const server = new Server(
{
name: "detox-mcp",
version: "1.0.0",
},
{
capabilities: {
tools: {},
resources: {},
prompts: {},
},
}
);
// ============ Tool Handlers ============
server.setRequestHandler(ListToolsRequestSchema, async () => {
return {
tools: allTools.map((tool) => ({
name: tool.name,
description: tool.description,
inputSchema: tool.inputSchema,
})),
};
});
server.setRequestHandler(CallToolRequestSchema, async (request) => {
const { name, arguments: args } = request.params;
const tool = allTools.find((t) => t.name === name);
if (!tool) {
throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`);
}
try {
// Add project path to args if not specified
const argsWithPath = {
...args,
cwd: (args as any)?.cwd || PROJECT_PATH,
projectPath: (args as any)?.projectPath || PROJECT_PATH,
};
const result = await tool.handler(argsWithPath);
return {
content: [
{
type: "text",
text: JSON.stringify(result, null, 2),
},
],
};
} catch (error: any) {
return {
content: [
{
type: "text",
text: JSON.stringify(
{
success: false,
error: error.message,
},
null,
2
),
},
],
isError: true,
};
}
});
// ============ Resource Handlers ============
server.setRequestHandler(ListResourcesRequestSchema, async () => {
const resources = await listResources(PROJECT_PATH);
return { resources };
});
server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
const { uri } = request.params;
try {
const content = await readResource(uri, PROJECT_PATH);
return {
contents: [content],
};
} catch (error: any) {
throw new McpError(
ErrorCode.InvalidRequest,
`Failed to read resource: ${error.message}`
);
}
});
// ============ Prompt Handlers ============
server.setRequestHandler(ListPromptsRequestSchema, async () => {
return {
prompts: listPrompts(),
};
});
server.setRequestHandler(GetPromptRequestSchema, async (request) => {
const { name, arguments: args } = request.params;
try {
const messages = getPrompt(name, args || {});
return { messages };
} catch (error: any) {
throw new McpError(
ErrorCode.InvalidRequest,
`Failed to get prompt: ${error.message}`
);
}
});
// ============ Main ============
async function main() {
const transport = new StdioServerTransport();
await server.connect(transport);
// Log to stderr so it doesn't interfere with MCP protocol
console.error("Detox MCP server running");
console.error(`Project path: ${PROJECT_PATH}`);
}
main().catch((error) => {
console.error("Fatal error:", error);
process.exit(1);
});
