@kubb/plugin-mcp 5.0.0-alpha.29 → 5.0.0-alpha.30

package/dist/index.js

@@ -1,10 +1,10 @@
 import "./chunk--u3MIqq1.js";
 import path from "node:path";
-import { caseParams, createFunctionParameter, createFunctionParameters, createOperationParams, createParameterGroup, transform, walk } from "@kubb/ast";
+import { caseParams, createFunctionParameter, createFunctionParameters, createOperationParams, createParameterGroup } from "@kubb/ast";
 import { functionPrinter, pluginTsName } from "@kubb/plugin-ts";
 import { Const, File, Function as Function$1 } from "@kubb/react-fabric";
 import { Fragment, jsx, jsxs } from "@kubb/react-fabric/jsx-runtime";
-import { createPlugin, defineGenerator, definePresets, defineResolver, getBarrelFiles, getPreset, runGeneratorOperation, runGeneratorOperations, runGeneratorSchema } from "@kubb/core";
+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";
@@ -480,42 +480,40 @@ server.registerTool(${JSON.stringify(tool.name)}, {
 //#region src/generators/mcpGenerator.tsx
 const mcpGenerator = defineGenerator({
 	name: "mcp",
-	type: "react",
-	Operation({ node, options, config, driver, resolver, plugin }) {
+	operation(node, options) {
+		const { resolver, driver, root } = this;
 		const { output, client, paramsCasing, group } = options;
-		const root = path.resolve(config.root, config.output.path);
 		const pluginTs = driver.getPlugin(pluginTsName);
 		if (!pluginTs?.resolver) return null;
-		const transformedNode = plugin.transformer ? transform(node, plugin.transformer) : node;
-		const casedParams = caseParams(transformedNode.parameters, paramsCasing);
+		const casedParams = 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");
 		const importedTypeNames = [
-			...pathParams.map((p) => pluginTs.resolver.resolvePathParamsName(transformedNode, p)),
-			...queryParams.map((p) => pluginTs.resolver.resolveQueryParamsName(transformedNode, p)),
-			...headerParams.map((p) => pluginTs.resolver.resolveHeaderParamsName(transformedNode, p)),
-			transformedNode.requestBody?.schema ? pluginTs.resolver.resolveDataName(transformedNode) : void 0,
-			pluginTs.resolver.resolveResponseName(transformedNode),
-			...transformedNode.responses.filter((r) => Number(r.statusCode) >= 400).map((r) => pluginTs.resolver.resolveResponseStatusName(transformedNode, r.statusCode))
+			...pathParams.map((p) => pluginTs.resolver.resolvePathParamsName(node, p)),
+			...queryParams.map((p) => pluginTs.resolver.resolveQueryParamsName(node, p)),
+			...headerParams.map((p) => pluginTs.resolver.resolveHeaderParamsName(node, p)),
+			node.requestBody?.schema ? pluginTs.resolver.resolveDataName(node) : void 0,
+			pluginTs.resolver.resolveResponseName(node),
+			...node.responses.filter((r) => Number(r.statusCode) >= 400).map((r) => pluginTs.resolver.resolveResponseStatusName(node, r.statusCode))
 		].filter(Boolean);
 		const meta = {
-			name: resolver.resolveName(transformedNode.operationId),
+			name: resolver.resolveName(node.operationId),
 			file: resolver.resolveFile({
-				name: transformedNode.operationId,
+				name: node.operationId,
 				extname: ".ts",
-				tag: transformedNode.tags[0] ?? "default",
-				path: transformedNode.path
+				tag: node.tags[0] ?? "default",
+				path: node.path
 			}, {
 				root,
 				output,
 				group
 			}),
 			fileTs: pluginTs.resolver.resolveFile({
-				name: transformedNode.operationId,
+				name: node.operationId,
 				extname: ".ts",
-				tag: transformedNode.tags[0] ?? "default",
-				path: transformedNode.path
+				tag: node.tags[0] ?? "default",
+				path: node.path
 			}, {
 				root,
 				output: pluginTs.options?.output ?? output,
@@ -587,7 +585,7 @@ const mcpGenerator = defineGenerator({
 				] }),
 				/* @__PURE__ */ jsx(McpHandler, {
 					name: meta.name,
-					node: transformedNode,
+					node,
 					resolver: pluginTs.resolver,
 					baseURL: client.baseURL,
 					dataReturnType: client.dataReturnType || "data",
@@ -608,10 +606,9 @@ const mcpGenerator = defineGenerator({
 */
 const serverGenerator = defineGenerator({
 	name: "operations",
-	type: "react",
-	Operations({ nodes, adapter, options, config, driver, resolver, plugin }) {
+	operations(nodes, options) {
+		const { adapter, config, resolver, plugin, driver, root } = this;
 		const { output, paramsCasing, group } = options;
-		const root = path.resolve(config.root, config.output.path);
 		const pluginZod = driver.getPlugin(pluginZodName);
 		if (!pluginZod?.resolver) return;
 		const name = "server";
@@ -626,46 +623,45 @@ const serverGenerator = defineGenerator({
 			meta: { pluginName: plugin.name }
 		};
 		const operationsMapped = nodes.map((node) => {
-			const transformedNode = plugin.transformer ? transform(node, plugin.transformer) : node;
-			const casedParams = caseParams(transformedNode.parameters, paramsCasing);
+			const casedParams = 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");
 			const mcpFile = resolver.resolveFile({
-				name: transformedNode.operationId,
+				name: node.operationId,
 				extname: ".ts",
-				tag: transformedNode.tags[0] ?? "default",
-				path: transformedNode.path
+				tag: node.tags[0] ?? "default",
+				path: node.path
 			}, {
 				root,
 				output,
 				group
 			});
 			const zodFile = pluginZod.resolver.resolveFile({
-				name: transformedNode.operationId,
+				name: node.operationId,
 				extname: ".ts",
-				tag: transformedNode.tags[0] ?? "default",
-				path: transformedNode.path
+				tag: node.tags[0] ?? "default",
+				path: node.path
 			}, {
 				root,
 				output: pluginZod.options?.output ?? output,
 				group: pluginZod.options?.group
 			});
-			const requestName = transformedNode.requestBody?.schema ? pluginZod.resolver.resolveDataName(transformedNode) : void 0;
-			const successStatus = findSuccessStatusCode(transformedNode.responses);
-			const responseName = successStatus ? pluginZod.resolver.resolveResponseStatusName(transformedNode, successStatus) : void 0;
+			const requestName = node.requestBody?.schema ? pluginZod.resolver.resolveDataName(node) : void 0;
+			const successStatus = findSuccessStatusCode(node.responses);
+			const responseName = successStatus ? pluginZod.resolver.resolveResponseStatusName(node, successStatus) : void 0;
 			const resolveParams = (params) => params.map((p) => ({
 				name: p.name,
-				schemaName: pluginZod.resolver.resolveParamName(transformedNode, p)
+				schemaName: pluginZod.resolver.resolveParamName(node, p)
 			}));
 			return {
 				tool: {
-					name: transformedNode.operationId,
-					title: transformedNode.summary || void 0,
-					description: transformedNode.description || `Make a ${transformedNode.method.toUpperCase()} request to ${transformedNode.path}`
+					name: node.operationId,
+					title: node.summary || void 0,
+					description: node.description || `Make a ${node.method.toUpperCase()} request to ${node.path}`
 				},
 				mcp: {
-					name: resolver.resolveName(transformedNode.operationId),
+					name: resolver.resolveName(node.operationId),
 					file: mcpFile
 				},
 				zod: {
@@ -676,7 +672,7 @@ const serverGenerator = defineGenerator({
 					responseName,
 					file: zodFile
 				},
-				node: transformedNode
+				node
 			};
 		});
 		const imports = operationsMapped.flatMap(({ mcp, zod }) => {
@@ -764,10 +760,9 @@ const serverGenerator = defineGenerator({
 */
 const serverGeneratorLegacy = defineGenerator({
 	name: "operations",
-	type: "react",
-	Operations({ nodes, adapter, options, config, driver, resolver, plugin }) {
+	operations(nodes, options) {
+		const { adapter, config, resolver, plugin, driver, root } = this;
 		const { output, paramsCasing, group } = options;
-		const root = path.resolve(config.root, config.output.path);
 		const pluginZod = driver.getPlugin(pluginZodName);
 		if (!pluginZod?.resolver) return;
 		const name = "server";
@@ -782,42 +777,41 @@ const serverGeneratorLegacy = defineGenerator({
 			meta: { pluginName: plugin.name }
 		};
 		const operationsMapped = nodes.map((node) => {
-			const transformedNode = plugin.transformer ? transform(node, plugin.transformer) : node;
-			const casedParams = caseParams(transformedNode.parameters, paramsCasing);
+			const casedParams = caseParams(node.parameters, paramsCasing);
 			const queryParams = casedParams.filter((p) => p.in === "query");
 			const headerParams = casedParams.filter((p) => p.in === "header");
 			const mcpFile = resolver.resolveFile({
-				name: transformedNode.operationId,
+				name: node.operationId,
 				extname: ".ts",
-				tag: transformedNode.tags[0] ?? "default",
-				path: transformedNode.path
+				tag: node.tags[0] ?? "default",
+				path: node.path
 			}, {
 				root,
 				output,
 				group
 			});
 			const zodFile = pluginZod?.resolver.resolveFile({
-				name: transformedNode.operationId,
+				name: node.operationId,
 				extname: ".ts",
-				tag: transformedNode.tags[0] ?? "default",
-				path: transformedNode.path
+				tag: node.tags[0] ?? "default",
+				path: node.path
 			}, {
 				root,
 				output: pluginZod?.options?.output ?? output,
 				group: pluginZod?.options?.group
 			});
-			const requestName = transformedNode.requestBody?.schema ? pluginZod?.resolver.resolveDataName(transformedNode) : void 0;
-			const responseName = pluginZod?.resolver.resolveResponseName(transformedNode);
-			const zodQueryParams = queryParams.length ? pluginZod?.resolver.resolveQueryParamsName(transformedNode, queryParams[0]) : void 0;
-			const zodHeaderParams = headerParams.length ? pluginZod?.resolver.resolveHeaderParamsName(transformedNode, headerParams[0]) : void 0;
+			const requestName = node.requestBody?.schema ? pluginZod?.resolver.resolveDataName(node) : void 0;
+			const responseName = pluginZod?.resolver.resolveResponseName(node);
+			const zodQueryParams = queryParams.length ? pluginZod?.resolver.resolveQueryParamsName(node, queryParams[0]) : void 0;
+			const zodHeaderParams = headerParams.length ? pluginZod?.resolver.resolveHeaderParamsName(node, headerParams[0]) : void 0;
 			return {
 				tool: {
-					name: transformedNode.operationId,
-					title: transformedNode.summary || void 0,
-					description: transformedNode.description || `Make a ${transformedNode.method.toUpperCase()} request to ${transformedNode.path}`
+					name: node.operationId,
+					title: node.summary || void 0,
+					description: node.description || `Make a ${node.method.toUpperCase()} request to ${node.path}`
 				},
 				mcp: {
-					name: resolver.resolveName(transformedNode.operationId),
+					name: resolver.resolveName(node.operationId),
 					file: mcpFile
 				},
 				zod: {
@@ -828,7 +822,7 @@ const serverGeneratorLegacy = defineGenerator({
 					responseName,
 					file: zodFile
 				},
-				node: transformedNode
+				node
 			};
 		});
 		const imports = operationsMapped.flatMap(({ mcp, zod }) => {
@@ -904,6 +898,9 @@ const serverGeneratorLegacy = defineGenerator({
 	}
 });
 //#endregion
+//#region package.json
+var version = "5.0.0-alpha.30";
+//#endregion
 //#region src/resolvers/resolverMcp.ts
 /**
 * Resolver for `@kubb/plugin-mcp` that provides the default naming
@@ -965,8 +962,10 @@ const pluginMcp = createPlugin((options) => {
 		transformer: userTransformer,
 		generators: userGenerators
 	});
+	const mergedGenerator = mergeGenerators(preset.generators ?? []);
 	return {
 		name: pluginMcpName,
+		version,
 		get resolver() {
 			return preset.resolver;
 		},
@@ -976,6 +975,9 @@ const pluginMcp = createPlugin((options) => {
 		get options() {
 			return {
 				output,
+				exclude,
+				include,
+				override,
 				group: group ? {
 					...group,
 					name: group.name ? group.name : (ctx) => {
@@ -997,12 +999,19 @@ const pluginMcp = createPlugin((options) => {
 			};
 		},
 		pre: [pluginTsName, pluginZodName].filter(Boolean),
-		async install() {
-			const { config, fabric, plugin, adapter, rootNode, driver } = this;
-			const root = path.resolve(config.root, config.output.path);
-			const resolver = preset.resolver;
-			if (!adapter) throw new Error("Plugin cannot work without adapter being set");
-			const baseURL = adapter.rootNode?.meta?.baseURL;
+		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?.rootNode?.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({
@@ -1029,42 +1038,7 @@ const pluginMcp = createPlugin((options) => {
 				imports: [],
 				exports: []
 			});
-			const collectedOperations = [];
-			const generatorContext = {
-				generators: preset.generators,
-				plugin,
-				resolver,
-				exclude,
-				include,
-				override,
-				fabric,
-				adapter,
-				config,
-				driver
-			};
-			await walk(rootNode, {
-				depth: "shallow",
-				async schema(schemaNode) {
-					await runGeneratorSchema(schemaNode, generatorContext);
-				},
-				async operation(operationNode) {
-					if (resolver.resolveOptions(operationNode, {
-						options: plugin.options,
-						exclude,
-						include,
-						override
-					}) !== null) collectedOperations.push(operationNode);
-					await runGeneratorOperation(operationNode, generatorContext);
-				}
-			});
-			await runGeneratorOperations(collectedOperations, generatorContext);
-			const barrelFiles = await getBarrelFiles(this.fabric.files, {
-				type: output.barrelType ?? "named",
-				root,
-				output,
-				meta: { pluginName: this.plugin.name }
-			});
-			await this.upsertFile(...barrelFiles);
+			await this.openInStudio({ ast: true });
 		}
 	};
 });