@kubb/plugin-mcp 5.0.0-alpha.33 → 5.0.0-alpha.35

package/dist/index.js

@@ -1,10 +1,9 @@
 import "./chunk--u3MIqq1.js";
 import path from "node:path";
-import { caseParams, createFile, createFunctionParameter, createFunctionParameters, createOperationParams, createParameterGroup, createSource, createText } from "@kubb/ast";
+import { ast, defineGenerator, definePlugin, defineResolver } from "@kubb/core";
 import { functionPrinter, pluginTsName } from "@kubb/plugin-ts";
-import { Const, File, Function as Function$1 } from "@kubb/renderer-jsx";
+import { Const, File, Function as Function$1, jsxRenderer } from "@kubb/renderer-jsx";
 import { Fragment, jsx, jsxs } from "@kubb/renderer-jsx/jsx-runtime";
-import { createPlugin, defineGenerator, definePresets, defineResolver, getPreset, mergeGenerators } from "@kubb/core";
 import { pluginZodName } from "@kubb/plugin-zod";
 import { pluginClientName } from "@kubb/plugin-client";
 import { source } from "@kubb/plugin-client/templates/clients/axios.source";
@@ -270,6 +269,15 @@ function getParamsMapping(params) {
 function zodExprFromSchemaNode(schema) {
 	let expr;
 	switch (schema.type) {
+		case "enum": {
+			const rawValues = schema.namedEnumValues?.length ? schema.namedEnumValues.map((v) => v.value) : (schema.enumValues ?? []).filter((v) => v !== null);
+			if (rawValues.length > 0 && rawValues.every((v) => typeof v === "string")) expr = `z.enum([${rawValues.map((v) => JSON.stringify(v)).join(", ")}])`;
+			else if (rawValues.length > 0) {
+				const literals = rawValues.map((v) => `z.literal(${JSON.stringify(v)})`);
+				expr = literals.length === 1 ? literals[0] : `z.union([${literals.join(", ")}])`;
+			} else expr = "z.string()";
+			break;
+		}
 		case "integer":
 			expr = "z.coerce.number()";
 			break;
@@ -300,7 +308,7 @@ function McpHandler({ name, node, resolver, baseURL, dataReturnType, paramsCasin
 	const urlPath = new URLPath(node.path);
 	const contentType = node.requestBody?.contentType;
 	const isFormData = contentType === "multipart/form-data";
-	const casedParams = caseParams(node.parameters, paramsCasing);
+	const casedParams = ast.caseParams(node.parameters, paramsCasing);
 	const queryParams = casedParams.filter((p) => p.in === "query");
 	const headerParams = casedParams.filter((p) => p.in === "header");
 	const originalPathParams = node.parameters.filter((p) => p.in === "path");
@@ -314,7 +322,7 @@ function McpHandler({ name, node, resolver, baseURL, dataReturnType, paramsCasin
 		`ResponseErrorConfig<${errorResponses.length > 0 ? errorResponses.join(" | ") : "Error"}>`,
 		requestName || "unknown"
 	].filter(Boolean);
-	const paramsNode = createOperationParams(node, {
+	const paramsNode = ast.createOperationParams(node, {
 		paramsType: "object",
 		pathParamsType: "inline",
 		resolver,
@@ -408,7 +416,7 @@ function Server({ name, serverName, serverVersion, paramsCasing, operations }) {
           `
 			}),
 			operations.map(({ tool, mcp, zod, node }) => {
-				const pathParams = caseParams(node.parameters, paramsCasing).filter((p) => p.in === "path");
+				const pathParams = ast.caseParams(node.parameters, paramsCasing).filter((p) => p.in === "path");
 				const pathEntries = [];
 				const otherEntries = [];
 				for (const p of pathParams) {
@@ -432,7 +440,7 @@ function Server({ name, serverName, serverVersion, paramsCasing, operations }) {
 				});
 				otherEntries.sort((a, b) => a.key.localeCompare(b.key));
 				const entries = [...pathEntries, ...otherEntries];
-				const paramsNode = entries.length ? createFunctionParameters({ params: [createParameterGroup({ properties: entries.map((e) => createFunctionParameter({
+				const paramsNode = entries.length ? ast.createFunctionParameters({ params: [ast.createParameterGroup({ properties: entries.map((e) => ast.createFunctionParameter({
 					name: e.key,
 					optional: false
 				})) })] }) : void 0;
@@ -480,12 +488,13 @@ server.registerTool(${JSON.stringify(tool.name)}, {
 //#region src/generators/mcpGenerator.tsx
 const mcpGenerator = defineGenerator({
 	name: "mcp",
-	operation(node, options) {
-		const { resolver, driver, root } = this;
-		const { output, client, paramsCasing, group } = options;
+	renderer: jsxRenderer,
+	operation(node, ctx) {
+		const { resolver, driver, root } = ctx;
+		const { output, client, paramsCasing, group } = ctx.options;
 		const pluginTs = driver.getPlugin(pluginTsName);
 		if (!pluginTs?.resolver) return null;
-		const casedParams = caseParams(node.parameters, paramsCasing);
+		const casedParams = ast.caseParams(node.parameters, paramsCasing);
 		const pathParams = casedParams.filter((p) => p.in === "path");
 		const queryParams = casedParams.filter((p) => p.in === "query");
 		const headerParams = casedParams.filter((p) => p.in === "header");
@@ -606,9 +615,10 @@ const mcpGenerator = defineGenerator({
 */
 const serverGenerator = defineGenerator({
 	name: "operations",
-	operations(nodes, options) {
-		const { adapter, config, resolver, plugin, driver, root } = this;
-		const { output, paramsCasing, group } = options;
+	renderer: jsxRenderer,
+	operations(nodes, ctx) {
+		const { adapter, config, resolver, plugin, driver, root } = ctx;
+		const { output, paramsCasing, group } = ctx.options;
 		const pluginZod = driver.getPlugin(pluginZodName);
 		if (!pluginZod?.resolver) return;
 		const name = "server";
@@ -623,7 +633,7 @@ const serverGenerator = defineGenerator({
 			meta: { pluginName: plugin.name }
 		};
 		const operationsMapped = nodes.map((node) => {
-			const casedParams = caseParams(node.parameters, paramsCasing);
+			const casedParams = ast.caseParams(node.parameters, paramsCasing);
 			const pathParams = casedParams.filter((p) => p.in === "path");
 			const queryParams = casedParams.filter((p) => p.in === "query");
 			const headerParams = casedParams.filter((p) => p.in === "header");
@@ -760,9 +770,10 @@ const serverGenerator = defineGenerator({
 */
 const serverGeneratorLegacy = defineGenerator({
 	name: "operations",
-	operations(nodes, options) {
-		const { adapter, config, resolver, plugin, driver, root } = this;
-		const { output, paramsCasing, group } = options;
+	renderer: jsxRenderer,
+	operations(nodes, ctx) {
+		const { adapter, config, resolver, plugin, driver, root } = ctx;
+		const { output, paramsCasing, group } = ctx.options;
 		const pluginZod = driver.getPlugin(pluginZodName);
 		if (!pluginZod?.resolver) return;
 		const name = "server";
@@ -777,7 +788,7 @@ const serverGeneratorLegacy = defineGenerator({
 			meta: { pluginName: plugin.name }
 		};
 		const operationsMapped = nodes.map((node) => {
-			const casedParams = caseParams(node.parameters, paramsCasing);
+			const casedParams = ast.caseParams(node.parameters, paramsCasing);
 			const queryParams = casedParams.filter((p) => p.in === "query");
 			const headerParams = casedParams.filter((p) => p.in === "header");
 			const mcpFile = resolver.resolveFile({
@@ -898,9 +909,6 @@ const serverGeneratorLegacy = defineGenerator({
 	}
 });
 //#endregion
-//#region package.json
-var version = "5.0.0-alpha.33";
-//#endregion
 //#region src/resolvers/resolverMcp.ts
 /**
 * Resolver for `@kubb/plugin-mcp` that provides the default naming
@@ -926,65 +934,38 @@ const resolverMcp = defineResolver(() => ({
 	}
 }));
 //#endregion
-//#region src/presets.ts
-/**
-* Built-in preset registry for `@kubb/plugin-mcp`.
-*
-* - `default` — v5 naming with individual zod schemas and per-status responses.
-* - `kubbV4`  — legacy naming with grouped zod schemas and combined responses.
-*/
-const presets = definePresets({
-	default: {
-		name: "default",
-		resolver: resolverMcp,
-		generators: [mcpGenerator, serverGenerator]
-	},
-	kubbV4: {
-		name: "kubbV4",
-		resolver: resolverMcp,
-		generators: [mcpGenerator, serverGeneratorLegacy]
-	}
-});
-//#endregion
 //#region src/plugin.ts
 const pluginMcpName = "plugin-mcp";
-const pluginMcp = createPlugin((options) => {
+const pluginMcp = definePlugin((options) => {
 	const { output = {
 		path: "mcp",
 		barrelType: "named"
-	}, group, exclude = [], include, override = [], paramsCasing, client, compatibilityPreset = "default", resolver: userResolver, transformer: userTransformer, generators: userGenerators = [] } = options;
+	}, group, exclude = [], include, override = [], paramsCasing, client, resolver: userResolver, transformer: userTransformer, generators: userGenerators = [], compatibilityPreset = "default" } = options;
+	const defaultServerGenerator = compatibilityPreset === "kubbV4" ? serverGeneratorLegacy : serverGenerator;
 	const clientName = client?.client ?? "axios";
 	const clientImportPath = client?.importPath ?? (!client?.bundle ? `@kubb/plugin-client/clients/${clientName}` : void 0);
-	const preset = getPreset({
-		preset: compatibilityPreset,
-		presets,
-		resolver: userResolver,
-		transformer: userTransformer,
-		generators: userGenerators
-	});
-	const mergedGenerator = mergeGenerators(preset.generators ?? []);
+	const groupConfig = group ? {
+		...group,
+		name: group.name ? group.name : (ctx) => {
+			if (group.type === "path") return `${ctx.group.split("/")[1]}`;
+			return `${camelCase(ctx.group)}Requests`;
+		}
+	} : void 0;
 	return {
 		name: pluginMcpName,
-		version,
-		get resolver() {
-			return preset.resolver;
-		},
-		get transformer() {
-			return preset.transformer;
-		},
-		get options() {
-			return {
+		options,
+		dependencies: [pluginTsName, pluginZodName],
+		hooks: { "kubb:plugin:setup"(ctx) {
+			const resolver = userResolver ? {
+				...resolverMcp,
+				...userResolver
+			} : resolverMcp;
+			ctx.setOptions({
 				output,
 				exclude,
 				include,
 				override,
-				group: group ? {
-					...group,
-					name: group.name ? group.name : (ctx) => {
-						if (group.type === "path") return `${ctx.group.split("/")[1]}`;
-						return `${camelCase(ctx.group)}Requests`;
-					}
-				} : void 0,
+				group: groupConfig,
 				paramsCasing,
 				client: {
 					client: clientName,
@@ -995,51 +976,36 @@ const pluginMcp = createPlugin((options) => {
 					baseURL: client?.baseURL,
 					paramsCasing: client?.paramsCasing
 				},
-				resolver: preset.resolver
-			};
-		},
-		pre: [pluginTsName, pluginZodName].filter(Boolean),
-		async schema(node, options) {
-			return mergedGenerator.schema?.call(this, node, options);
-		},
-		async operation(node, options) {
-			return mergedGenerator.operation?.call(this, node, options);
-		},
-		async operations(nodes, options) {
-			return mergedGenerator.operations?.call(this, nodes, options);
-		},
-		async buildStart() {
-			const { adapter, driver } = this;
-			const root = this.root;
-			const baseURL = adapter?.inputNode?.meta?.baseURL;
-			if (baseURL) this.plugin.options.client.baseURL = this.plugin.options.client.baseURL || baseURL;
-			const hasClientPlugin = !!driver.getPlugin(pluginClientName);
-			if (this.plugin.options.client.bundle && !hasClientPlugin && !this.plugin.options.client.importPath) await this.addFile(createFile({
+				resolver
+			});
+			ctx.setResolver(resolver);
+			if (userTransformer) ctx.setTransformer(userTransformer);
+			ctx.addGenerator(mcpGenerator);
+			ctx.addGenerator(defaultServerGenerator);
+			for (const gen of userGenerators) ctx.addGenerator(gen);
+			const root = path.resolve(ctx.config.root, ctx.config.output.path);
+			const hasClientPlugin = ctx.config.plugins?.some((p) => p.name === pluginClientName);
+			if (client?.bundle && !hasClientPlugin && !clientImportPath) ctx.injectFile({
 				baseName: "fetch.ts",
 				path: path.resolve(root, ".kubb/fetch.ts"),
-				sources: [createSource({
+				sources: [ast.createSource({
 					name: "fetch",
-					nodes: [createText(this.plugin.options.client.client === "fetch" ? source$1 : source)],
+					nodes: [ast.createText(clientName === "fetch" ? source$1 : source)],
 					isExportable: true,
 					isIndexable: true
-				})],
-				imports: [],
-				exports: []
-			}));
-			if (!hasClientPlugin) await this.addFile(createFile({
+				})]
+			});
+			if (!hasClientPlugin) ctx.injectFile({
 				baseName: "config.ts",
 				path: path.resolve(root, ".kubb/config.ts"),
-				sources: [createSource({
+				sources: [ast.createSource({
 					name: "config",
-					nodes: [createText(source$2)],
+					nodes: [ast.createText(source$2)],
 					isExportable: false,
 					isIndexable: false
-				})],
-				imports: [],
-				exports: []
-			}));
-			await this.openInStudio({ ast: true });
-		}
+				})]
+			});
+		} }
 	};
 });
 //#endregion