@gavdi/cap-mcp 0.9.4 → 0.9.7

package/lib/utils.js

@@ -1,136 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MCP_SESSION_HEADER = exports.LOGGER = void 0;
-exports.createMcpServer = createMcpServer;
-exports.handleMcpSessionRequest = handleMcpSessionRequest;
-exports.parseEntityElements = parseEntityElements;
-const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
-const zod_1 = require("zod");
-const structures_1 = require("./annotations/structures");
-/* @ts-ignore */
-const cds = global.cds || require("@sap/cds"); // This is a work around for missing cds context
-exports.LOGGER = cds.log("cds-mcp");
-exports.MCP_SESSION_HEADER = "mcp-session-id";
-function createMcpServer(annotations) {
-    const packageInfo = require("../package.json");
-    const server = new mcp_js_1.McpServer({
-        name: packageInfo.name,
-        version: packageInfo.version,
-        capabilities: {
-            tools: { listChanged: true },
-            resources: { listChanged: true },
-            prompts: { listChanged: true },
-        },
-    });
-    exports.LOGGER.debug("Annotations found for server = ", annotations);
-    if (!annotations)
-        return server;
-    // TODO: Handle the parsed annotations
-    // TODO: Error handling
-    // TODO: This should only be mapped once, not per each server instance. Maybe this should be pre-packaged on load?
-    for (const [_, v] of annotations.entries()) {
-        switch (v.constructor) {
-            case structures_1.McpToolAnnotation:
-                const model = v;
-                const parameters = buildParameters(model.parameters);
-                exports.LOGGER.debug("Adding tool", model);
-                if (model.entityKey) {
-                    const keys = buildParameters(model.keyTypeMap);
-                    server.tool(model.name, { ...keys, ...parameters }, async (data) => {
-                        const service = cds.services[model.serviceName];
-                        const received = data;
-                        const receivedKeys = {};
-                        const receivedParams = {};
-                        for (const [k, v] of Object.entries(received)) {
-                            if (model.keyTypeMap?.has(k)) {
-                                receivedKeys[k] = v;
-                            }
-                            if (!model.parameters?.has(k))
-                                continue;
-                            receivedParams[k] = v;
-                        }
-                        const response = await service.send({
-                            event: model.target,
-                            entity: model.entityKey,
-                            data: receivedParams,
-                            params: [receivedKeys],
-                        });
-                        return {
-                            content: Array.isArray(response)
-                                ? response.map((el) => ({ type: "text", text: String(el) }))
-                                : [{ type: "text", text: String(response) }], // TODO: This should be dynamic based on the return type
-                        };
-                    });
-                    continue;
-                }
-                server.tool(model.name, parameters, async (data) => {
-                    exports.LOGGER.debug("Tool call received, targeting service: ", model.serviceName, model.target);
-                    const service = cds.services[model.serviceName];
-                    const response = await service.send(model.target, data);
-                    exports.LOGGER.debug("MCP Tool response received and being packaged");
-                    return {
-                        content: Array.isArray(response)
-                            ? response.map((el) => ({ type: "text", text: String(el) }))
-                            : [{ type: "text", text: String(response) }], // TODO: This should be dynamic based on the return type
-                    };
-                });
-                continue;
-            case structures_1.McpResourceAnnotation:
-                exports.LOGGER.debug("This is a resource");
-                continue;
-            case structures_1.McpPromptAnnotation:
-                exports.LOGGER.debug("This is a prompt");
-                continue;
-            default:
-                exports.LOGGER.error("Invalid annotation data type");
-                throw new Error("Invalid annotation");
-        }
-    }
-    return server;
-}
-async function handleMcpSessionRequest(req, res, sessions) {
-    const sessionIdHeader = req.headers[exports.MCP_SESSION_HEADER];
-    if (!sessionIdHeader || !sessions.has(sessionIdHeader)) {
-        res.status(400).send("Invalid or missing session ID");
-        return;
-    }
-    const session = sessions.get(sessionIdHeader);
-    if (!session) {
-        res.status(400).send("Invalid session");
-        return;
-    }
-    await session.transport.handleRequest(req, res);
-}
-function parseEntityElements(entity) {
-    const elements = entity.elements;
-    if (!elements) {
-        exports.LOGGER.error("Invalid object - cannot be parsed", entity);
-        throw new Error("Failed to parse entity object");
-    }
-    const result = new Map();
-    for (const el of elements) {
-        if (!el.type)
-            continue;
-        result.set(el.name, el.type?.replace("cds.", ""));
-    }
-    return result;
-}
-function buildParameters(params) {
-    if (!params || params.size <= 0)
-        return {};
-    const result = {};
-    for (const [k, v] of params.entries()) {
-        result[k] = determineParameterType(v);
-    }
-    return result;
-}
-function determineParameterType(paramType) {
-    switch (paramType) {
-        case "String":
-            return zod_1.z.string();
-        case "Integer":
-            return zod_1.z.number();
-        default:
-            return zod_1.z.number();
-    }
-}