@kubb/plugin-mcp
Version:
Generator mcp
118 lines (116 loc) • 4.38 kB
JavaScript
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 ? `/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