import { getAppState } from '@features/app-state/getAppState';
import { isError } from '@shared/core/isError';
import type { McpTool } from '@shared/mcp-tool/McpTool';
import type { Express, Request, Response } from 'express';
import { z } from 'zod';
import { createContentArraySchema } from './ContentSchemas';
/**
* Parameters for setting up request handlers
*/
type SetupRequestHandlersParams = {
/** Express application instance */
app: Express;
/** Tools enabled for REST mode */
restEnabledTools: McpTool[];
};
/**
* Configures request handlers for each enabled tool
*
* @param params - Configuration parameters
*/
export function setupRequestHandlers({ app, restEnabledTools }: SetupRequestHandlersParams): void {
restEnabledTools.forEach((tool) => {
app.post(`/api/${tool.name}`, (req: Request, res: Response) => {
void (async (): Promise<void> => {
try {
// Validate request body against schema
const params = tool.inputSchema
? (tool.inputSchema as z.ZodObject<z.ZodRawShape>).parse(req.body)
: undefined;
const result = await tool.handler({ params, appState: getAppState() });
if (isError(result)) {
// Return error response
res.status(500).json({
isError: true,
content: [
{
type: 'text',
text: result.message,
},
],
});
return;
}
// Check if the tool has a JSON output schema for validation
if (tool.outputTypes.includes('json') && tool.jsonOutputSchema) {
const schema = createContentArraySchema(tool.jsonOutputSchema as z.ZodTypeAny);
// Validate response against schema
const validatedResponse = schema.parse(result);
// Return successful response
res.status(201).json({
isError: false,
content: validatedResponse,
});
} else {
// No JSON schema validation needed
res.status(201).json({
isError: false,
content: result,
});
}
} catch (error: unknown) {
// Fallback error handling
console.error(`Error in ${tool.name} endpoint:`, error);
if (error instanceof z.ZodError) {
// Handle validation errors
res.status(400).json({
error: 'Validation error',
details: error.errors,
});
return;
}
if (error instanceof Error) {
// Handle known errors with message
res.status(500).json({
error: 'Server error',
message: error.message,
});
return;
}
// Handle unknown errors
res.status(500).json({
error: 'Internal server error',
message: 'An unexpected error occurred',
});
}
})();
});
});
}
