easy-mcp-nest 0.1.0

package/dist/tooling/tool-registry/tool-registry.service.js

@@ -0,0 +1,122 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolRegistryService = void 0;
+const common_1 = require("@nestjs/common");
+const easy_mcp_error_1 = require("../../core/errors/easy-mcp-error");
+const logger_util_1 = require("../../core/utils/logger.util");
+let ToolRegistryService = class ToolRegistryService {
+    registry = new Map();
+    registerTool(definition) {
+        if (this.registry.has(definition.name)) {
+            throw new Error(`Tool name '${definition.name}' already registered.`);
+        }
+        this.registry.set(definition.name, definition);
+        logger_util_1.logger.info("ToolRegistryService", `Tool registered: ${definition.name}`, {
+            component: "ToolRegistry",
+            toolName: definition.name,
+        });
+    }
+    getTool(name) {
+        return this.registry.get(name);
+    }
+    async executeTool(name, args) {
+        const tool = this.getTool(name);
+        if (!tool) {
+            throw new easy_mcp_error_1.ToolNotFoundError(name);
+        }
+        try {
+            const result = await tool.execute(args);
+            return result;
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            throw new easy_mcp_error_1.ToolExecutionError(`Tool '${name}' execution failed: ${errorMessage}`, name, error instanceof Error ? error : undefined);
+        }
+    }
+    registerToolFromConfig(toolInput) {
+        const parameters = {};
+        for (const [key, prop] of Object.entries(toolInput.inputSchema.properties)) {
+            const typeMap = {
+                'STRING': 'string',
+                'NUMBER': 'number',
+                'INTEGER': 'integer',
+                'BOOLEAN': 'boolean',
+                'ARRAY': 'array',
+                'OBJECT': 'object',
+            };
+            const lowerType = typeMap[prop.type];
+            if (!lowerType) {
+                throw new Error(`Tool '${toolInput.name}': Unsupported property type '${prop.type}' for property '${key}'. Must be one of: STRING, NUMBER, INTEGER, BOOLEAN, ARRAY, OBJECT`);
+            }
+            parameters[key] = {
+                type: lowerType,
+                description: prop.description || '',
+                ...(prop.enum && { enum: prop.enum }),
+            };
+        }
+        const required = toolInput.inputSchema.required || [];
+        const toolDefinition = {
+            name: toolInput.name,
+            description: toolInput.description,
+            execute: toolInput.function,
+            parameters,
+            required,
+        };
+        this.registerTool(toolDefinition);
+    }
+    getToolSchemasForLLM() {
+        return Array.from(this.registry.values()).map((tool) => ({
+            type: "function",
+            function: {
+                name: tool.name,
+                description: tool.description,
+                parameters: {
+                    type: "object",
+                    properties: tool.parameters,
+                    required: tool.required,
+                },
+            },
+        }));
+    }
+    getToolsAsRegistrationInput() {
+        return Array.from(this.registry.values()).map((tool) => {
+            const properties = {};
+            for (const [key, param] of Object.entries(tool.parameters)) {
+                const typeMap = {
+                    'string': 'STRING',
+                    'number': 'NUMBER',
+                    'integer': 'INTEGER',
+                    'boolean': 'BOOLEAN',
+                    'array': 'ARRAY',
+                    'object': 'OBJECT',
+                };
+                properties[key] = {
+                    type: typeMap[param.type] || 'STRING',
+                    description: param.description,
+                    ...(param.enum && { enum: param.enum }),
+                };
+            }
+            return {
+                name: tool.name,
+                description: tool.description,
+                function: tool.execute,
+                inputSchema: {
+                    type: 'OBJECT',
+                    properties,
+                    required: tool.required,
+                },
+            };
+        });
+    }
+};
+exports.ToolRegistryService = ToolRegistryService;
+exports.ToolRegistryService = ToolRegistryService = __decorate([
+    (0, common_1.Injectable)()
+], ToolRegistryService);
+//# sourceMappingURL=tool-registry.service.js.map

package/dist/tooling/tool-registry/tool-registry.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-registry.service.js","sourceRoot":"","sources":["../../../src/tooling/tool-registry/tool-registry.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAG5C,qEAAyF;AACzF,8DAAsD;AAoB/C,IAAM,mBAAmB,GAAzB,MAAM,mBAAmB;IACb,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IAMvD,YAAY,CAAC,UAA0B;QAC5C,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,cAAc,UAAU,CAAC,IAAI,uBAAuB,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC/C,oBAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,oBAAoB,UAAU,CAAC,IAAI,EAAE,EAAE;YACxE,SAAS,EAAE,cAAc;YACzB,QAAQ,EAAE,UAAU,CAAC,IAAI;SAC1B,CAAC,CAAC;IACL,CAAC;IAKM,OAAO,CAAC,IAAY;QACzB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAUM,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,IAAyB;QAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,kCAAiB,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACxC,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,IAAI,mCAAkB,CAC1B,SAAS,IAAI,uBAAuB,YAAY,EAAE,EAClD,IAAI,EACJ,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;QACJ,CAAC;IACH,CAAC;IAOM,sBAAsB,CAAC,SAAgC;QAE5D,MAAM,UAAU,GAAkC,EAAE,CAAC;QAErD,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;YAE3E,MAAM,OAAO,GAA0C;gBACrD,QAAQ,EAAE,QAAQ;gBAClB,QAAQ,EAAE,QAAQ;gBAClB,SAAS,EAAE,SAAS;gBACpB,SAAS,EAAE,SAAS;gBACpB,OAAO,EAAE,OAAO;gBAChB,QAAQ,EAAE,QAAQ;aACnB,CAAC;YAEF,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC;YAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,SAAS,SAAS,CAAC,IAAI,iCAAiC,IAAI,CAAC,IAAI,mBAAmB,GAAG,oEAAoE,CAAC,CAAC;YAC/K,CAAC;YAED,UAAU,CAAC,GAAG,CAAC,GAAG;gBAChB,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gBACnC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;aACtC,CAAC;QACJ,CAAC;QAGD,MAAM,QAAQ,GAAG,SAAS,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;QAGtD,MAAM,cAAc,GAAmB;YACrC,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,WAAW,EAAE,SAAS,CAAC,WAAW;YAClC,OAAO,EAAE,SAAS,CAAC,QAAQ;YAC3B,UAAU;YACV,QAAQ;SACT,CAAC;QAGF,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;IACpC,CAAC;IAOM,oBAAoB;QACzB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACvD,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE;oBACV,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;iBACxB;aACF;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAQM,2BAA2B;QAChC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAErD,MAAM,UAAU,GAAwB,EAAE,CAAC;YAC3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBAE3D,MAAM,OAAO,GAA0C;oBACrD,QAAQ,EAAE,QAAQ;oBAClB,QAAQ,EAAE,QAAQ;oBAClB,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,SAAS;oBACpB,OAAO,EAAE,OAAO;oBAChB,QAAQ,EAAE,QAAQ;iBACnB,CAAC;gBACF,UAAU,CAAC,GAAG,CAAC,GAAG;oBAChB,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,QAAQ;oBACrC,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;iBACxC,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,QAAQ,EAAE,IAAI,CAAC,OAAO;gBACtB,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU;oBACV,QAAQ,EAAE,IAAI,CAAC,QAAQ;iBACxB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AA/JY,kDAAmB;8BAAnB,mBAAmB;IAD/B,IAAA,mBAAU,GAAE;GACA,mBAAmB,CA+J/B"}

package/dist/tooling/tool.interface.d.ts

@@ -0,0 +1,13 @@
+export interface ToolParameter {
+    type: "string" | "number" | "integer" | "boolean" | "array" | "object";
+    description: string;
+    enum?: string[];
+}
+export type ToolFunction = (args: Record<string, any>) => Promise<any>;
+export interface ToolDefinition {
+    name: string;
+    description: string;
+    execute: ToolFunction;
+    parameters: Record<string, ToolParameter>;
+    required: string[];
+}

package/dist/tooling/tool.interface.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=tool.interface.js.map

package/dist/tooling/tool.interface.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.interface.js","sourceRoot":"","sources":["../../src/tooling/tool.interface.ts"],"names":[],"mappings":""}