@kubb/mcp 5.0.0-beta.5 → 5.0.0-beta.7

package/dist/index.js

@@ -1,30 +1,21 @@
 import "./chunk--u3MIqq1.js";
-import process$1 from "node:process";
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { z } from "zod";
+import http from "node:http";
+import { createRequestListener } from "@remix-run/node-fetch-server";
+import { ValibotJsonSchemaAdapter } from "@tmcp/adapter-valibot";
+import { HttpTransport } from "@tmcp/transport-http";
+import { StdioTransport } from "@tmcp/transport-stdio";
+import { McpServer } from "tmcp";
 import { EventEmitter } from "node:events";
-import { existsSync } from "node:fs";
+import fs, { existsSync } from "node:fs";
 import path from "node:path";
 import { createKubb } from "@kubb/core";
+import { defineTool } from "tmcp/tool";
+import { tool } from "tmcp/utils";
+import * as v from "valibot";
 import { createJiti } from "jiti";
+import process$1 from "node:process";
 //#region package.json
-var version = "5.0.0-beta.5";
-//#endregion
-//#region src/schemas/generateSchema.ts
-const generateSchema = z.object({
-	config: z.string().optional().describe("Path to kubb.config file (supports .ts, .js, .cjs). If not provided, will look for kubb.config.{ts,js,cjs} in current directory"),
-	input: z.string().optional().describe("Path to OpenAPI/Swagger spec file (overrides config)"),
-	output: z.string().optional().describe("Output directory path (overrides config)"),
-	logLevel: z.enum([
-		"silent",
-		"error",
-		"warn",
-		"info",
-		"verbose",
-		"debug"
-	]).optional().default("info").describe("Log level for build output")
-});
+var version = "5.0.0-beta.7";
 //#endregion
 //#region ../../internals/utils/src/errors.ts
 /**
@@ -162,6 +153,21 @@ function isPromise(result) {
 	return result !== null && result !== void 0 && typeof result["then"] === "function";
 }
 //#endregion
+//#region src/schemas/generateSchema.ts
+const generateSchema = v.object({
+	config: v.optional(v.pipe(v.string(), v.minLength(1), v.description("Path to kubb.config file (supports .ts, .js, .cjs). If not provided, will look for kubb.config.{ts,js,cjs} in current directory"))),
+	input: v.optional(v.pipe(v.string(), v.minLength(1), v.description("Path to OpenAPI/Swagger spec file (overrides config)"))),
+	output: v.optional(v.pipe(v.string(), v.minLength(1), v.description("Output directory path (overrides config)"))),
+	logLevel: v.optional(v.pipe(v.picklist([
+		"silent",
+		"error",
+		"warn",
+		"info",
+		"verbose",
+		"debug"
+	]), v.description("Log level for build output")), "info")
+});
+//#endregion
 //#region src/types.ts
 const NotifyTypes = {
 	INFO: "INFO",
@@ -215,8 +221,6 @@ async function loadModule(filePath) {
 	return mod;
 }
 async function loadUserConfig(configPath, { notify }) {
-	let userConfig;
-	let cwd;
 	if (configPath) {
 		const ext = path.extname(configPath);
 		if (!ALLOWED_CONFIG_EXTENSIONS.has(ext)) {
@@ -232,41 +236,44 @@ async function loadUserConfig(configPath, { notify }) {
 			await notify(NotifyTypes.CONFIG_ERROR, msg);
 			throw new Error(msg);
 		}
-		cwd = path.dirname(resolvedConfigPath);
+		const cwd = path.dirname(resolvedConfigPath);
 		try {
-			userConfig = await loadModule(resolvedConfigPath);
+			const userConfig = await loadModule(resolvedConfigPath);
 			await notify(NotifyTypes.CONFIG_LOADED, `Loaded config from ${resolvedConfigPath}`);
+			return {
+				userConfig,
+				cwd
+			};
 		} catch (error) {
-			await notify(NotifyTypes.CONFIG_ERROR, `Failed to load config: ${error instanceof Error ? error.message : String(error)}`);
-			throw new Error(`Failed to load config: ${error instanceof Error ? error.message : String(error)}`);
-		}
-	} else {
-		cwd = process.cwd();
-		const configFileNames = [
-			"kubb.config.ts",
-			"kubb.config.mts",
-			"kubb.config.cts",
-			"kubb.config.js",
-			"kubb.config.cjs"
-		];
-		for (const configFileName of configFileNames) {
-			const configFilePath = path.resolve(process.cwd(), configFileName);
-			if (!existsSync(configFilePath)) continue;
-			try {
-				userConfig = await loadModule(configFilePath);
-				await notify(NotifyTypes.CONFIG_LOADED, `Loaded ${configFileName} from current directory`);
-				break;
-			} catch {}
+			const msg = `Failed to load config: ${error instanceof Error ? error.message : String(error)}`;
+			await notify(NotifyTypes.CONFIG_ERROR, msg);
+			throw new Error(msg);
 		}
-		if (!userConfig) {
-			await notify(NotifyTypes.CONFIG_ERROR, "No config file found");
-			throw new Error(`No config file found. Please provide a config path or create one of: ${configFileNames.join(", ")}`);
+	}
+	const cwd = process.cwd();
+	const configFileNames = [
+		"kubb.config.ts",
+		"kubb.config.mts",
+		"kubb.config.cts",
+		"kubb.config.js",
+		"kubb.config.cjs"
+	];
+	for (const configFileName of configFileNames) {
+		const configFilePath = path.resolve(process.cwd(), configFileName);
+		if (!existsSync(configFilePath)) continue;
+		try {
+			const userConfig = await loadModule(configFilePath);
+			await notify(NotifyTypes.CONFIG_LOADED, `Loaded ${configFileName} from current directory`);
+			return {
+				userConfig,
+				cwd
+			};
+		} catch (err) {
+			await notify(NotifyTypes.CONFIG_ERROR, `Failed to load ${configFileName}: ${err instanceof Error ? err.message : String(err)}`);
 		}
 	}
-	return {
-		userConfig,
-		cwd
-	};
+	await notify(NotifyTypes.CONFIG_ERROR, "No config file found");
+	throw new Error(`No config file found. Please provide a config path or create one of: ${configFileNames.join(", ")}`);
 }
 //#endregion
 //#region src/utils/resolveCwd.ts
@@ -285,40 +292,27 @@ function resolveCwd(userConfig, cwd) {
 }
 //#endregion
 //#region src/utils/resolveUserConfig.ts
-/**
-* Resolve the config by handling function configs and returning the final configuration
-*/
 async function resolveUserConfig(config, options) {
-	let kubbUserConfig = Promise.resolve(config);
-	if (typeof config === "function") {
-		const possiblePromise = config({
-			logLevel: options.logLevel,
-			config: options.configPath
-		});
-		if (isPromise(possiblePromise)) kubbUserConfig = possiblePromise;
-		else kubbUserConfig = Promise.resolve(possiblePromise);
-	}
-	return await kubbUserConfig;
+	const result = typeof config === "function" ? config({
+		logLevel: options.logLevel,
+		config: options.configPath
+	}) : config;
+	const resolved = isPromise(result) ? await result : result;
+	return Array.isArray(resolved) ? resolved[0] : resolved;
 }
 //#endregion
 //#region src/tools/generate.ts
-/**
-* Build tool that generates code from OpenAPI specs using Kubb.
-* Sends real-time notifications of build progress and events.
-*/
-async function generate(schema, handler) {
+const generateTool = defineTool({
+	name: "generate",
+	description: "Generate OpenAPI spec helpers using Kubb configuration",
+	schema: generateSchema
+}, async function generate(schema) {
 	const { config: configPath, input, output, logLevel } = schema;
 	try {
 		const hooks = new AsyncEventEmitter();
 		const messages = [];
-		const notify = async (type, message, data) => {
+		const notify = async (type, message, _data) => {
 			messages.push(`${type}: ${message}`);
-			await handler.sendNotification("kubb/progress", {
-				type,
-				message,
-				timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-				...data
-			});
 		};
 		hooks.on("kubb:info", async ({ message }) => {
 			await notify(NotifyTypes.INFO, message);
@@ -327,7 +321,7 @@ async function generate(schema, handler) {
 			await notify(NotifyTypes.SUCCESS, message);
 		});
 		hooks.on("kubb:error", async ({ error }) => {
-			await notify(NotifyTypes.ERROR, error.message, { stack: error.stack });
+			await notify(NotifyTypes.ERROR, error.message);
 		});
 		hooks.on("kubb:warn", async ({ message }) => {
 			await notify(NotifyTypes.WARN, message);
@@ -359,7 +353,7 @@ async function generate(schema, handler) {
 			const configResult = await loadUserConfig(configPath, { notify });
 			userConfig = configResult.userConfig;
 			cwd = configResult.cwd;
-			if (Array.isArray(userConfig) && userConfig.length) throw new Error("Array type in kubb.config.ts is not supported in this tool. Please provide a single configuration object.");
+			if (Array.isArray(userConfig)) throw new Error("Array type in kubb.config.ts is not supported in this tool. Please provide a single configuration object.");
 			userConfig = await resolveUserConfig(userConfig, {
 				configPath,
 				logLevel
@@ -367,13 +361,7 @@ async function generate(schema, handler) {
 		} catch (error) {
 			const errorMessage = error instanceof Error ? error.message : String(error);
 			await notify(NotifyTypes.CONFIG_ERROR, errorMessage);
-			return {
-				content: [{
-					type: "text",
-					text: errorMessage
-				}],
-				isError: true
-			};
+			return tool.error(errorMessage);
 		}
 		const inputPath = input ?? (userConfig.input && "path" in userConfig.input ? userConfig.input.path : void 0);
 		const config = {
@@ -388,7 +376,7 @@ async function generate(schema, handler) {
 				path: output
 			} : userConfig.output
 		};
-		await notify(NotifyTypes.CONFIG_READY, "Configuration ready", { root: config.root });
+		await notify(NotifyTypes.CONFIG_READY, "Configuration ready");
 		await notify(NotifyTypes.SETUP_START, "Setting up Kubb");
 		const kubb = createKubb(config, { hooks });
 		await kubb.setup();
@@ -398,79 +386,234 @@ async function generate(schema, handler) {
 		await notify(NotifyTypes.BUILD_END, `Build complete - Generated ${files.length} files`);
 		if (error || failedPlugins.size > 0) {
 			const allErrors = [error, ...Array.from(failedPlugins).filter((it) => it.error).map((it) => it.error)].filter(Boolean);
-			await notify(NotifyTypes.BUILD_FAILED, `Build failed with ${allErrors.length} error(s)`, {
-				errorCount: allErrors.length,
-				errors: allErrors.map((err) => err.message)
-			});
-			return {
-				content: [{
-					type: "text",
-					text: `Build failed:\n${allErrors.map((err) => err.message).join("\n")}\n\n${messages.join("\n")}`
-				}],
-				isError: true
-			};
+			await notify(NotifyTypes.BUILD_FAILED, `Build failed with ${allErrors.length} error(s)`);
+			return tool.error(`Build failed:\n${allErrors.map((err) => err.message).join("\n")}\n\n${messages.join("\n")}`);
 		}
-		await notify(NotifyTypes.BUILD_SUCCESS, `Build completed successfully - Generated ${files.length} files`, { filesCount: files.length });
-		return { content: [{
-			type: "text",
-			text: `Build completed successfully!\n\nGenerated ${files.length} files\n\n${messages.join("\n")}`
-		}] };
+		await notify(NotifyTypes.BUILD_SUCCESS, `Build completed successfully - Generated ${files.length} files`);
+		return tool.text(`Build completed successfully!\n\nGenerated ${files.length} files\n\n${messages.join("\n")}`);
 	} catch (caughtError) {
-		const error = caughtError;
-		await handler.sendNotification("kubb/progress", {
-			type: NotifyTypes.FATAL_ERROR,
-			message: error.message,
-			stack: error.stack,
-			timestamp: (/* @__PURE__ */ new Date()).toISOString()
-		});
-		return {
-			content: [{
-				type: "text",
-				text: `Build error: ${error.message}\n${error.stack || ""}`
-			}],
-			isError: true
-		};
+		const error = toError(caughtError);
+		return tool.error(`Build error: ${error.message}\n${error.stack ?? ""}`);
+	}
+});
+//#endregion
+//#region ../../internals/shared/src/constants.ts
+const KUBB_CONFIG_FILENAME = "kubb.config.ts";
+const availablePlugins = [
+	{
+		value: "plugin-ts",
+		label: "TypeScript",
+		hint: "Recommended",
+		packageName: "@kubb/plugin-ts",
+		importName: "pluginTs",
+		category: "types"
+	},
+	{
+		value: "plugin-client",
+		label: "Client (Fetch/Axios)",
+		packageName: "@kubb/plugin-client",
+		importName: "pluginClient",
+		category: "client"
+	},
+	{
+		value: "plugin-react-query",
+		label: "React Query / TanStack Query",
+		packageName: "@kubb/plugin-react-query",
+		importName: "pluginReactQuery",
+		category: "framework"
+	},
+	{
+		value: "plugin-vue-query",
+		label: "Vue Query",
+		packageName: "@kubb/plugin-vue-query",
+		importName: "pluginVueQuery",
+		category: "framework"
+	},
+	{
+		value: "plugin-zod",
+		label: "Zod Schemas",
+		packageName: "@kubb/plugin-zod",
+		importName: "pluginZod",
+		category: "validation"
+	},
+	{
+		value: "plugin-faker",
+		label: "Faker.js Mocks",
+		packageName: "@kubb/plugin-faker",
+		importName: "pluginFaker",
+		category: "mocks"
+	},
+	{
+		value: "plugin-msw",
+		label: "MSW Handlers",
+		packageName: "@kubb/plugin-msw",
+		importName: "pluginMsw",
+		category: "mocks"
+	},
+	{
+		value: "plugin-cypress",
+		label: "Cypress Tests",
+		packageName: "@kubb/plugin-cypress",
+		importName: "pluginCypress",
+		category: "testing"
+	},
+	{
+		value: "plugin-mcp",
+		label: "MCP Server (AI / Model Context Protocol)",
+		packageName: "@kubb/plugin-mcp",
+		importName: "pluginMcp",
+		category: "ai"
+	},
+	{
+		value: "plugin-redoc",
+		label: "ReDoc Documentation",
+		packageName: "@kubb/plugin-redoc",
+		importName: "pluginRedoc",
+		category: "documentation"
 	}
+];
+const pluginDefaultConfigs = {
+	"plugin-ts": `pluginTs({
+      output: { path: 'models' },
+    })`,
+	"plugin-client": `pluginClient({
+      output: { path: 'clients' },
+    })`,
+	"plugin-react-query": `pluginReactQuery({
+      output: { path: 'hooks' },
+    })`,
+	"plugin-vue-query": `pluginVueQuery({
+      output: { path: 'hooks' },
+    })`,
+	"plugin-zod": `pluginZod({
+      output: { path: 'zod' },
+    })`,
+	"plugin-faker": `pluginFaker({
+      output: { path: 'mocks' },
+    })`,
+	"plugin-msw": `pluginMsw({
+      output: { path: 'msw' },
+    })`,
+	"plugin-cypress": `pluginCypress({
+      output: { path: 'cypress' },
+    })`,
+	"plugin-mcp": `pluginMcp({
+      output: { path: 'mcp' },
+    })`,
+	"plugin-redoc": `pluginRedoc({
+      output: { path: 'redoc' },
+    })`
+};
+//#endregion
+//#region ../../internals/shared/src/init.ts
+function generateConfigFile({ selectedPlugins, inputPath, outputPath }) {
+	return `import { defineConfig } from 'kubb'
+${selectedPlugins.map((plugin) => `import { ${plugin.importName} } from '${plugin.packageName}'`).join("\n")}
+
+export default defineConfig({
+  root: '.',
+  input: {
+    path: '${inputPath}',
+  },
+  output: {
+    path: '${outputPath}',
+    clean: true,
+  },
+  plugins: [
+${selectedPlugins.map((plugin) => {
+		return `    ${pluginDefaultConfigs[plugin.value] ?? `${plugin.importName}()`},`;
+	}).join("\n")}
+  ],
+})
+`;
 }
 //#endregion
-//#region src/server.ts
-/**
-* Kubb MCP Server
-*
-* Provides tools for building OpenAPI specifications using Kubb.
-*/
-async function startServer() {
+//#region src/schemas/initSchema.ts
+const initSchema = v.object({
+	input: v.optional(v.pipe(v.string(), v.minLength(1), v.description("Path to OpenAPI spec (default: ./openapi.yaml)"))),
+	output: v.optional(v.pipe(v.string(), v.minLength(1), v.description("Output directory (default: ./src/gen)"))),
+	plugins: v.optional(v.pipe(v.string(), v.minLength(1), v.description("Comma-separated list of plugins: plugin-ts,plugin-zod,...")))
+});
+//#endregion
+//#region src/tools/init.ts
+function resolvePlugins(pluginsFlag) {
+	if (!pluginsFlag) return [];
+	const requested = pluginsFlag.split(",").map((v) => v.trim()).filter(Boolean);
+	return availablePlugins.filter((p) => requested.includes(p.value));
+}
+const initTool = defineTool({
+	name: "init",
+	description: "Scaffold a kubb.config.ts in the current directory (non-interactive). Does not install packages.",
+	schema: initSchema
+}, async ({ input = "./openapi.yaml", output = "./src/gen", plugins }) => {
+	const selected = resolvePlugins(plugins);
+	const content = generateConfigFile({
+		selectedPlugins: selected,
+		inputPath: input,
+		outputPath: output
+	});
+	const dest = path.join(process$1.cwd(), KUBB_CONFIG_FILENAME);
+	if (fs.existsSync(dest)) return tool.error(`${KUBB_CONFIG_FILENAME} already exists at ${dest}. Delete it first before running init again.`);
+	fs.writeFileSync(dest, content, "utf-8");
+	const packageList = ["kubb", ...selected.map((p) => p.packageName)].join(" ");
+	return tool.text(`Created kubb.config.ts\n\nInstall packages:\n  npm install ${packageList}\n\nThen run:\n  npx kubb generate`);
+});
+//#endregion
+//#region src/tools/validate.ts
+const validateTool = defineTool({
+	name: "validate",
+	description: "Validate an OpenAPI/Swagger specification file or URL",
+	schema: v.object({ input: v.pipe(v.string(), v.minLength(1), v.description("Path or URL to the OpenAPI/Swagger specification")) })
+}, async ({ input }) => {
+	let mod;
 	try {
-		const transport = new StdioServerTransport();
-		const server = new McpServer({
-			name: "Kubb",
-			version
-		});
-		server.tool("generate", "Generate OpenAPI spec helpers using Kubb configuration", generateSchema.shape, async (args) => {
-			return generate(args, { sendNotification: async (method, params) => {
-				try {
-					await transport.send({
-						jsonrpc: "2.0",
-						method,
-						params
-					});
-				} catch (error) {
-					console.error("Failed to send notification:", error);
-				}
-			} });
-		});
-		await server.connect(transport);
-	} catch (error) {
-		console.error("Failed to start MCP server:", error);
-		process$1.exit(1);
+		mod = await import("@kubb/adapter-oas");
+	} catch {
+		return tool.error("The validate tool requires @kubb/adapter-oas.\nInstall: npm install @kubb/adapter-oas");
+	}
+	try {
+		await mod.adapterOas().validate(input, { throwOnError: true });
+		return tool.text(`Validation successful: ${input}`);
+	} catch (err) {
+		return tool.error(`Validation failed:\n${err instanceof Error ? err.message : String(err)}`);
+	}
+});
+//#endregion
+//#region src/server.ts
+function createMcpServer() {
+	const server = new McpServer({
+		name: "Kubb",
+		version
+	}, {
+		adapter: new ValibotJsonSchemaAdapter(),
+		capabilities: { tools: {} }
+	});
+	server.tools([
+		generateTool,
+		validateTool,
+		initTool
+	]);
+	return server;
+}
+async function startServer({ port, host = "localhost" } = {}) {
+	const server = createMcpServer();
+	if (port === void 0) {
+		new StdioTransport(server).listen();
+		return;
 	}
+	const transport = new HttpTransport(server, { path: "/mcp" });
+	http.createServer(createRequestListener(async (request) => {
+		return await transport.respond(request) ?? new Response("Not Found", { status: 404 });
+	})).listen(port, host, () => {
+		console.log(`Kubb MCP server on http://${host}:${port}`);
+	});
 }
 //#endregion
 //#region src/index.ts
-async function run(_argv) {
-	await startServer();
+async function run(_argv, options) {
+	await startServer(options);
 }
 //#endregion
-export { run };
+export { createMcpServer, run };
 
 //# sourceMappingURL=index.js.map