import { z } from 'zod';
import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
import { componentNames, getComponentSchema } from '@patternfly/patternfly-component-schemas/json';
import { type McpTool } from './server';
import { getOptions } from './options.context';
import { memo } from './server.caching';
import { fuzzySearch } from './server.search';
/**
* Derive the component schema type from @patternfly/patternfly-component-schemas
*/
type ComponentSchema = Awaited<ReturnType<typeof getComponentSchema>>;
/**
* componentSchemas tool function (tuple pattern)
*
* Creates an MCP tool that retrieves JSON Schema for PatternFly React components.
* Uses fuzzy search to handle typos and case variations, with related fallback suggestions.
*
* @param options - Optional configuration options (defaults to OPTIONS)
* @returns {McpTool} MCP tool tuple [name, schema, callback]
*/
const componentSchemasTool = (options = getOptions()): McpTool => {
const memoGetComponentSchema = memo(
async (componentName: string): Promise<ComponentSchema> => getComponentSchema(componentName),
options?.toolMemoOptions?.fetchDocs // Use the same memo options as fetchDocs
);
const callback = async (args: any = {}) => {
const { componentName } = args;
if (typeof componentName !== 'string') {
throw new McpError(
ErrorCode.InvalidParams,
`Missing required parameter: componentName (must be a string): ${componentName}`
);
}
// Use fuzzySearch with `isFuzzyMatch` to handle exact and intentional suggestions in one pass
const results = fuzzySearch(componentName, componentNames, {
maxDistance: 3,
maxResults: 5,
isFuzzyMatch: true,
deduplicateByNormalized: true
});
const exact = results.find(result => result.matchType === 'exact');
if (exact) {
let componentSchema: ComponentSchema;
try {
componentSchema = await memoGetComponentSchema(exact.item);
} catch (error) {
throw new McpError(
ErrorCode.InternalError,
`Failed to fetch component schema: ${error}`
);
}
return {
content: [
{
type: 'text',
text: JSON.stringify(componentSchema, null, 2)
}
]
};
}
const suggestions = results.map(result => result.item).slice(0, 3);
const suggestionMessage = suggestions.length
? `Did you mean ${suggestions.map(suggestion => `"${suggestion}"`).join(', ')}?`
: 'No similar components found.';
throw new McpError(
ErrorCode.InvalidParams,
`Component "${componentName.trim()}" not found. ${suggestionMessage}`
);
};
return [
'componentSchemas',
{
description: 'Get JSON Schema for a PatternFly React component. Returns prop definitions, types, and validation rules. Use this for structured component metadata, not documentation.',
inputSchema: {
componentName: z.string().describe('Name of the PatternFly component (e.g., "Button", "Table")')
}
},
callback
];
};
/**
* A tool name, typically the first entry in the tuple. Used in logging and deduplication.
*/
componentSchemasTool.toolName = 'componentSchemas';
export { componentSchemasTool };
