@eggjs/tegg-controller-plugin 3.62.0 → 4.0.0-beta.10

package/lib/impl/mcp/MCPServerHelper.js

@@ -1,122 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MCPServerHelper = void 0;
-const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
-const MCPControllerRegister_1 = require("./MCPControllerRegister");
-const tegg_types_1 = require("@eggjs/tegg-types");
-class MCPServerHelper {
-    constructor(opts) {
-        this.server = new mcp_js_1.McpServer({
-            name: opts.name,
-            version: opts.version,
-        }, { capabilities: { logging: {} } });
-        this.hooks = opts.hooks;
-    }
-    async mcpResourceRegister(getOrCreateEggObject, controllerProto, resourceMeta) {
-        var _a, _b, _c;
-        const handler = async (...args) => {
-            const eggObj = await getOrCreateEggObject(controllerProto, controllerProto.name);
-            const realObj = eggObj.obj;
-            const realMethod = realObj[resourceMeta.name];
-            return Reflect.apply(realMethod, realObj, args);
-        };
-        const name = (_a = resourceMeta.mcpName) !== null && _a !== void 0 ? _a : resourceMeta.name;
-        if (resourceMeta.uri) {
-            this.server.registerResource(name, resourceMeta.uri, (_b = resourceMeta.metadata) !== null && _b !== void 0 ? _b : {}, handler);
-        }
-        else if (resourceMeta.template) {
-            this.server.registerResource(name, resourceMeta.template, (_c = resourceMeta.metadata) !== null && _c !== void 0 ? _c : {}, handler);
-        }
-        else {
-            throw new Error(`MCPResource ${name} must have uri or template`);
-        }
-    }
-    async mcpToolRegister(getOrCreateEggObject, controllerProto, toolMeta) {
-        var _a, _b, _c, _d;
-        const controllerMeta = controllerProto.getMetaData(tegg_types_1.CONTROLLER_META_DATA);
-        const name = (_a = toolMeta.mcpName) !== null && _a !== void 0 ? _a : toolMeta.name;
-        const description = toolMeta.description;
-        let schema;
-        if ((_b = toolMeta.detail) === null || _b === void 0 ? void 0 : _b.argsSchema) {
-            schema = (_c = toolMeta.detail) === null || _c === void 0 ? void 0 : _c.argsSchema;
-        }
-        else if (MCPControllerRegister_1.MCPControllerRegister.hooks.length > 0) {
-            for (const hook of MCPControllerRegister_1.MCPControllerRegister.hooks) {
-                schema = await ((_d = hook.schemaLoader) === null || _d === void 0 ? void 0 : _d.call(hook, controllerMeta, toolMeta));
-                if (schema) {
-                    break;
-                }
-            }
-        }
-        const handler = async (...args) => {
-            const eggObj = await getOrCreateEggObject(controllerProto, controllerProto.name);
-            const realObj = eggObj.obj;
-            const realMethod = realObj[toolMeta.name];
-            let newArgs = [];
-            if (schema && toolMeta.detail) {
-                // 如果有 schema 则证明入参第一个就是 schema
-                newArgs[toolMeta.detail.index] = args[0];
-                // 如果有 schema 则证明入参第二个就是 extra
-                if (toolMeta.extra) {
-                    newArgs[toolMeta.extra] = args[1];
-                }
-            }
-            else if (toolMeta.extra) {
-                // 无 schema, 那么入参第一个就是 extra
-                newArgs[toolMeta.extra] = args[0];
-            }
-            newArgs = [...newArgs, ...args];
-            return Reflect.apply(realMethod, realObj, newArgs);
-        };
-        this.server.registerTool(name, {
-            description,
-            inputSchema: schema,
-            // TODO: outputSchema
-        }, handler);
-    }
-    async mcpPromptRegister(getOrCreateEggObject, controllerProto, promptMeta) {
-        var _a, _b, _c, _d;
-        const controllerMeta = controllerProto.getMetaData(tegg_types_1.CONTROLLER_META_DATA);
-        const name = (_a = promptMeta.mcpName) !== null && _a !== void 0 ? _a : promptMeta.name;
-        const description = promptMeta.description;
-        let schema;
-        if ((_b = promptMeta.detail) === null || _b === void 0 ? void 0 : _b.argsSchema) {
-            schema = (_c = promptMeta.detail) === null || _c === void 0 ? void 0 : _c.argsSchema;
-        }
-        else if (MCPControllerRegister_1.MCPControllerRegister.hooks.length > 0) {
-            for (const hook of MCPControllerRegister_1.MCPControllerRegister.hooks) {
-                schema = await ((_d = hook.schemaLoader) === null || _d === void 0 ? void 0 : _d.call(hook, controllerMeta, promptMeta));
-                if (schema) {
-                    break;
-                }
-            }
-        }
-        const handler = async (...args) => {
-            const eggObj = await getOrCreateEggObject(controllerProto, controllerProto.name);
-            const realObj = eggObj.obj;
-            const realMethod = realObj[promptMeta.name];
-            let newArgs = [];
-            if (schema && promptMeta.detail) {
-                // 如果有 schema 则证明入参第一个就是 schema
-                newArgs[promptMeta.detail.index] = args[0];
-                // 如果有 schema 则证明入参第二个就是 extra
-                if (promptMeta.extra) {
-                    newArgs[promptMeta.extra] = args[1];
-                }
-            }
-            else if (promptMeta.extra) {
-                // 无 schema, 那么入参第一个就是 extra
-                newArgs[promptMeta.extra] = args[0];
-            }
-            newArgs = [...newArgs, ...args];
-            return Reflect.apply(realMethod, realObj, newArgs);
-        };
-        this.server.registerPrompt(name, {
-            title: promptMeta.title,
-            description,
-            argsSchema: schema,
-        }, handler);
-    }
-}
-exports.MCPServerHelper = MCPServerHelper;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTUNQU2VydmVySGVscGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiTUNQU2VydmVySGVscGVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7OztBQUFBLG9FQUtpRDtBQUNqRCxtRUFHaUM7QUFDakMsa0RBQWlHO0FBU2pHLE1BQWEsZUFBZTtJQUcxQixZQUFZLElBQTRCO1FBQ3RDLElBQUksQ0FBQyxNQUFNLEdBQUcsSUFBSSxrQkFBUyxDQUN6QjtZQUNFLElBQUksRUFBRSxJQUFJLENBQUMsSUFBSTtZQUNmLE9BQU8sRUFBRSxJQUFJLENBQUMsT0FBTztTQUN0QixFQUNELEVBQUUsWUFBWSxFQUFFLEVBQUUsT0FBTyxFQUFFLEVBQUUsRUFBRSxFQUFFLENBQ2xDLENBQUM7UUFDRixJQUFJLENBQUMsS0FBSyxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUM7SUFDMUIsQ0FBQztJQUVELEtBQUssQ0FBQyxtQkFBbUIsQ0FDdkIsb0JBQXVGLEVBQ3ZGLGVBQTZCLEVBQzdCLFlBQTZCOztRQUU3QixNQUFNLE9BQU8sR0FBRyxLQUFLLEVBQUUsR0FBRyxJQUFJLEVBQUUsRUFBRTtZQUNoQyxNQUFNLE1BQU0sR0FBRyxNQUFNLG9CQUFvQixDQUN2QyxlQUFlLEVBQ2YsZUFBZSxDQUFDLElBQUksQ0FDckIsQ0FBQztZQUNGLE1BQU0sT0FBTyxHQUFHLE1BQU0sQ0FBQyxHQUFHLENBQUM7WUFDM0IsTUFBTSxVQUFVLEdBQUcsT0FBTyxDQUFDLFlBQVksQ0FBQyxJQUFJLENBQUMsQ0FBQztZQUM5QyxPQUFPLE9BQU8sQ0FBQyxLQUFLLENBQ2xCLFVBQVUsRUFDVixPQUFPLEVBQ1AsSUFBSSxDQUMrQixDQUFDO1FBQ3hDLENBQUMsQ0FBQztRQUNGLE1BQU0sSUFBSSxHQUFHLE1BQUEsWUFBWSxDQUFDLE9BQU8sbUNBQUksWUFBWSxDQUFDLElBQUksQ0FBQztRQUN2RCxJQUFJLFlBQVksQ0FBQyxHQUFHLEVBQUUsQ0FBQztZQUNyQixJQUFJLENBQUMsTUFBTSxDQUFDLGdCQUFnQixDQUFDLElBQUksRUFBRSxZQUFZLENBQUMsR0FBRyxFQUFFLE1BQUEsWUFBWSxDQUFDLFFBQVEsbUNBQUksRUFBRSxFQUFFLE9BQU8sQ0FBQyxDQUFDO1FBQzdGLENBQUM7YUFBTSxJQUFJLFlBQVksQ0FBQyxRQUFRLEVBQUUsQ0FBQztZQUNqQyxJQUFJLENBQUMsTUFBTSxDQUFDLGdCQUFnQixDQUFDLElBQUksRUFBRSxZQUFZLENBQUMsUUFBUSxFQUFFLE1BQUEsWUFBWSxDQUFDLFFBQVEsbUNBQUksRUFBRSxFQUFFLE9BQU8sQ0FBQyxDQUFDO1FBQ2xHLENBQUM7YUFBTSxDQUFDO1lBQ04sTUFBTSxJQUFJLEtBQUssQ0FBQyxlQUFlLElBQUksNEJBQTRCLENBQUMsQ0FBQztRQUNuRSxDQUFDO0lBQ0gsQ0FBQztJQUVELEtBQUssQ0FBQyxlQUFlLENBQ25CLG9CQUF1RixFQUN2RixlQUE2QixFQUM3QixRQUFxQjs7UUFFckIsTUFBTSxjQUFjLEdBQUcsZUFBZSxDQUFDLFdBQVcsQ0FDaEQsaUNBQW9CLENBQ0EsQ0FBQztRQUN2QixNQUFNLElBQUksR0FBVyxNQUFBLFFBQVEsQ0FBQyxPQUFPLG1DQUFJLFFBQVEsQ0FBQyxJQUFJLENBQUM7UUFDdkQsTUFBTSxXQUFXLEdBQXVCLFFBQVEsQ0FBQyxXQUFXLENBQUM7UUFDN0QsSUFBSSxNQUF3RSxDQUFDO1FBQzdFLElBQUksTUFBQSxRQUFRLENBQUMsTUFBTSwwQ0FBRSxVQUFVLEVBQUUsQ0FBQztZQUNoQyxNQUFNLEdBQUcsTUFBQSxRQUFRLENBQUMsTUFBTSwwQ0FBRSxVQUFVLENBQUM7UUFDdkMsQ0FBQzthQUFNLElBQUksNkNBQXFCLENBQUMsS0FBSyxDQUFDLE1BQU0sR0FBRyxDQUFDLEVBQUUsQ0FBQztZQUNsRCxLQUFLLE1BQU0sSUFBSSxJQUFJLDZDQUFxQixDQUFDLEtBQUssRUFBRSxDQUFDO2dCQUMvQyxNQUFNLEdBQUcsTUFBTSxDQUFBLE1BQUEsSUFBSSxDQUFDLFlBQVkscURBQUcsY0FBYyxFQUFFLFFBQVEsQ0FBQyxDQUFBLENBQUM7Z0JBQzdELElBQUksTUFBTSxFQUFFLENBQUM7b0JBQ1gsTUFBTTtnQkFDUixDQUFDO1lBQ0gsQ0FBQztRQUNILENBQUM7UUFDRCxNQUFNLE9BQU8sR0FBRyxLQUFLLEVBQUUsR0FBRyxJQUFJLEVBQUUsRUFBRTtZQUNoQyxNQUFNLE1BQU0sR0FBRyxNQUFNLG9CQUFvQixDQUN2QyxlQUFlLEVBQ2YsZUFBZSxDQUFDLElBQUksQ0FDckIsQ0FBQztZQUNGLE1BQU0sT0FBTyxHQUFHLE1BQU0sQ0FBQyxHQUFHLENBQUM7WUFDM0IsTUFBTSxVQUFVLEdBQUcsT0FBTyxDQUFDLFFBQVEsQ0FBQyxJQUFJLENBQUMsQ0FBQztZQUMxQyxJQUFJLE9BQU8sR0FBVSxFQUFFLENBQUM7WUFDeEIsSUFBSSxNQUFNLElBQUksUUFBUSxDQUFDLE1BQU0sRUFBRSxDQUFDO2dCQUM5QiwrQkFBK0I7Z0JBQy9CLE9BQU8sQ0FBQyxRQUFRLENBQUMsTUFBTSxDQUFDLEtBQUssQ0FBQyxHQUFHLElBQUksQ0FBQyxDQUFDLENBQUMsQ0FBQztnQkFFekMsOEJBQThCO2dCQUM5QixJQUFJLFFBQVEsQ0FBQyxLQUFLLEVBQUUsQ0FBQztvQkFDbkIsT0FBTyxDQUFDLFFBQVEsQ0FBQyxLQUFLLENBQUMsR0FBRyxJQUFJLENBQUMsQ0FBQyxDQUFDLENBQUM7Z0JBQ3BDLENBQUM7WUFDSCxDQUFDO2lCQUFNLElBQUksUUFBUSxDQUFDLEtBQUssRUFBRSxDQUFDO2dCQUMxQiw0QkFBNEI7Z0JBQzVCLE9BQU8sQ0FBQyxRQUFRLENBQUMsS0FBSyxDQUFDLEdBQUcsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDO1lBQ3BDLENBQUM7WUFDRCxPQUFPLEdBQUcsQ0FBRSxHQUFHLE9BQU8sRUFBRSxHQUFHLElBQUksQ0FBRSxDQUFDO1lBQ2xDLE9BQU8sT0FBTyxDQUFDLEtBQUssQ0FBQyxVQUFVLEVBQUUsT0FBTyxFQUFFLE9BQU8sQ0FBNkIsQ0FBQztRQUNqRixDQUFDLENBQUM7UUFDRixJQUFJLENBQUMsTUFBTSxDQUFDLFlBQVksQ0FBQyxJQUFJLEVBQUU7WUFDN0IsV0FBVztZQUNYLFdBQVcsRUFBRSxNQUFNO1lBQ25CLHFCQUFxQjtTQUN0QixFQUFFLE9BQU8sQ0FBQyxDQUFDO0lBQ2QsQ0FBQztJQUVELEtBQUssQ0FBQyxpQkFBaUIsQ0FDckIsb0JBQXVGLEVBQ3ZGLGVBQTZCLEVBQzdCLFVBQXlCOztRQUV6QixNQUFNLGNBQWMsR0FBRyxlQUFlLENBQUMsV0FBVyxDQUFDLGlDQUFvQixDQUFzQixDQUFDO1FBQzlGLE1BQU0sSUFBSSxHQUFXLE1BQUEsVUFBVSxDQUFDLE9BQU8sbUNBQUksVUFBVSxDQUFDLElBQUksQ0FBQztRQUMzRCxNQUFNLFdBQVcsR0FBdUIsVUFBVSxDQUFDLFdBQVcsQ0FBQztRQUMvRCxJQUFJLE1BQTBFLENBQUM7UUFDL0UsSUFBSSxNQUFBLFVBQVUsQ0FBQyxNQUFNLDBDQUFFLFVBQVUsRUFBRSxDQUFDO1lBQ2xDLE1BQU0sR0FBRyxNQUFBLFVBQVUsQ0FBQyxNQUFNLDBDQUFFLFVBQVUsQ0FBQztRQUN6QyxDQUFDO2FBQU0sSUFBSSw2Q0FBcUIsQ0FBQyxLQUFLLENBQUMsTUFBTSxHQUFHLENBQUMsRUFBRSxDQUFDO1lBQ2xELEtBQUssTUFBTSxJQUFJLElBQUksNkNBQXFCLENBQUMsS0FBSyxFQUFFLENBQUM7Z0JBQy9DLE1BQU0sR0FBRyxNQUFNLENBQUEsTUFBQSxJQUFJLENBQUMsWUFBWSxxREFBRyxjQUFjLEVBQUUsVUFBVSxDQUFDLENBQUEsQ0FBQztnQkFDL0QsSUFBSSxNQUFNLEVBQUUsQ0FBQztvQkFDWCxNQUFNO2dCQUNSLENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztRQUNELE1BQU0sT0FBTyxHQUFHLEtBQUssRUFBRSxHQUFHLElBQUksRUFBRSxFQUFFO1lBQ2hDLE1BQU0sTUFBTSxHQUFHLE1BQU0sb0JBQW9CLENBQUMsZUFBZSxFQUFFLGVBQWUsQ0FBQyxJQUFJLENBQUMsQ0FBQztZQUNqRixNQUFNLE9BQU8sR0FBRyxNQUFNLENBQUMsR0FBRyxDQUFDO1lBQzNCLE1BQU0sVUFBVSxHQUFHLE9BQU8sQ0FBQyxVQUFVLENBQUMsSUFBSSxDQUFDLENBQUM7WUFDNUMsSUFBSSxPQUFPLEdBQVUsRUFBRSxDQUFDO1lBQ3hCLElBQUksTUFBTSxJQUFJLFVBQVUsQ0FBQyxNQUFNLEVBQUUsQ0FBQztnQkFDaEMsK0JBQStCO2dCQUMvQixPQUFPLENBQUMsVUFBVSxDQUFDLE1BQU0sQ0FBQyxLQUFLLENBQUMsR0FBRyxJQUFJLENBQUMsQ0FBQyxDQUFDLENBQUM7Z0JBRTNDLDhCQUE4QjtnQkFDOUIsSUFBSSxVQUFVLENBQUMsS0FBSyxFQUFFLENBQUM7b0JBQ3JCLE9BQU8sQ0FBQyxVQUFVLENBQUMsS0FBSyxDQUFDLEdBQUcsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDO2dCQUN0QyxDQUFDO1lBQ0gsQ0FBQztpQkFBTSxJQUFJLFVBQVUsQ0FBQyxLQUFLLEVBQUUsQ0FBQztnQkFDNUIsNEJBQTRCO2dCQUM1QixPQUFPLENBQUMsVUFBVSxDQUFDLEtBQUssQ0FBQyxHQUFHLElBQUksQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUN0QyxDQUFDO1lBQ0QsT0FBTyxHQUFHLENBQUUsR0FBRyxPQUFPLEVBQUUsR0FBRyxJQUFJLENBQUUsQ0FBQztZQUNsQyxPQUFPLE9BQU8sQ0FBQyxLQUFLLENBQUMsVUFBVSxFQUFFLE9BQU8sRUFBRSxPQUFPLENBQStCLENBQUM7UUFDbkYsQ0FBQyxDQUFDO1FBQ0YsSUFBSSxDQUFDLE1BQU0sQ0FBQyxjQUFjLENBQUMsSUFBSSxFQUFFO1lBQy9CLEtBQUssRUFBRSxVQUFVLENBQUMsS0FBSztZQUN2QixXQUFXO1lBQ1gsVUFBVSxFQUFFLE1BQU07U0FDbkIsRUFBRSxPQUFPLENBQUMsQ0FBQztJQUNkLENBQUM7Q0FDRjtBQTFJRCwwQ0EwSUMifQ==

package/typings/index.d.ts

@@ -1,17 +0,0 @@
-import 'egg';
-import '@eggjs/tegg-plugin';
-import '@eggjs/mcp-proxy';
-import { RootProtoManager } from '../lib/RootProtoManager';
-import { ControllerRegisterFactory } from '../lib/ControllerRegisterFactory';
-import { ControllerMetaBuilderFactory } from '@eggjs/tegg';
-
-declare module 'egg' {
-  export interface TEggControllerApp extends MCPProxyApp {
-    rootProtoManager: RootProtoManager;
-    controllerRegisterFactory: ControllerRegisterFactory;
-    controllerMetaBuilderFactory: typeof ControllerMetaBuilderFactory;
-  }
-
-  export interface Application extends TEggControllerApp {
-  }
-}