UNPKG

@kubb/plugin-mcp

Version:
118 lines (116 loc) 4.38 kB
import { t as __name } from "./chunk-eQyhnF5A.js"; import { n as mcpGenerator, t as serverGenerator } from "./generators-4PG9gQkq.js"; import path from "node:path"; import { definePlugin, getBarrelFiles, getMode } from "@kubb/core"; import { camelCase } from "@kubb/core/transformers"; import { resolveModuleSource } from "@kubb/core/utils"; import { pluginClientName } from "@kubb/plugin-client"; import { OperationGenerator, pluginOasName } from "@kubb/plugin-oas"; import { pluginTsName } from "@kubb/plugin-ts"; import { pluginZodName } from "@kubb/plugin-zod"; //#region src/plugin.ts const pluginMcpName = "plugin-mcp"; const pluginMcp = definePlugin((options) => { const { output = { path: "mcp", barrelType: "named" }, group, exclude = [], include, override = [], transformers = {}, generators = [mcpGenerator, serverGenerator].filter(Boolean), contentType, client } = options; const clientName = client?.client ?? "axios"; const clientImportPath = client?.importPath ?? (!client?.bundle ? `@kubb/plugin-client/clients/${clientName}` : void 0); return { name: pluginMcpName, options: { output, group, client: { client: clientName, clientType: client?.clientType ?? "function", importPath: clientImportPath, dataReturnType: "data", ...options.client } }, pre: [ pluginOasName, pluginTsName, pluginZodName ].filter(Boolean), resolvePath(baseName, pathMode, options$1) { const root = path.resolve(this.config.root, this.config.output.path); if ((pathMode ?? getMode(path.resolve(root, output.path))) === "single") /** * when output is a file then we will always append to the same file(output file), see fileManager.addOrAppend * Other plugins then need to call addOrAppend instead of just add from the fileManager class */ return path.resolve(root, output.path); if (group && (options$1?.group?.path || options$1?.group?.tag)) { const groupName = group?.name ? group.name : (ctx) => { if (group?.type === "path") return `${ctx.group.split("/")[1]}`; return `${camelCase(ctx.group)}Requests`; }; return path.resolve(root, output.path, groupName({ group: group.type === "path" ? options$1.group.path : options$1.group.tag }), baseName); } return path.resolve(root, output.path, baseName); }, resolveName(name, type) { const resolvedName = camelCase(name, { isFile: type === "file" }); if (type) return transformers?.name?.(resolvedName, type) || resolvedName; return resolvedName; }, async install() { const root = path.resolve(this.config.root, this.config.output.path); const mode = getMode(path.resolve(root, output.path)); const oas = await this.getOas(); const baseURL = await this.getBaseURL(); if (baseURL) this.plugin.options.client.baseURL = baseURL; const hasClientPlugin = !!this.pluginManager.getPluginByKey([pluginClientName]); if (this.plugin.options.client.bundle && !hasClientPlugin && !this.plugin.options.client.importPath) await this.addFile({ baseName: "fetch.ts", path: path.resolve(root, ".kubb/fetch.ts"), sources: [{ name: "fetch", value: resolveModuleSource(this.plugin.options.client.client === "fetch" ? "@kubb/plugin-client/templates/clients/fetch" : "@kubb/plugin-client/templates/clients/axios").source, isExportable: true, isIndexable: true }], imports: [], exports: [] }); if (!hasClientPlugin) await this.addFile({ baseName: "config.ts", path: path.resolve(root, ".kubb/config.ts"), sources: [{ name: "config", value: resolveModuleSource("@kubb/plugin-client/templates/config").source, isExportable: false, isIndexable: false }], imports: [], exports: [] }); const files = await new OperationGenerator(this.plugin.options, { fabric: this.fabric, oas, pluginManager: this.pluginManager, events: this.events, plugin: this.plugin, contentType, exclude, include, override, mode }).build(...generators); await this.upsertFile(...files); const barrelFiles = await getBarrelFiles(this.fabric.files, { type: output.barrelType ?? "named", root, output, meta: { pluginKey: this.plugin.key } }); await this.upsertFile(...barrelFiles); } }; }); //#endregion export { pluginMcp, pluginMcpName }; //# sourceMappingURL=index.js.map