@kya-os/mcp-i 0.1.0-alpha.2.7 → 0.1.0-alpha.2.8

package/dist/auto.js

@@ -9,5 +9,8 @@ const logger_1 = require("./logger");
     catch (error) {
         const logger = (0, logger_1.getLogger)();
         logger.error('[MCP-I] Failed to auto-initialize:', error);
+        if (error.stack) {
+            logger.debug('[MCP-I] Error stack:', error.stack);
+        }
     }
 })();

package/dist/index.js

@@ -421,9 +421,24 @@ async function enableMCPIdentity(options) {
 }
 function createMCPMiddleware(identity) {
     return (server) => {
+        if (!server || typeof server !== 'object') {
+            const logger = (0, logger_1.getLogger)();
+            logger.warn('Invalid MCP Server object passed to middleware');
+            return;
+        }
+        if (!server.setRequestHandler || typeof server.setRequestHandler !== 'function') {
+            const logger = (0, logger_1.getLogger)();
+            logger.warn('MCP Server missing setRequestHandler method');
+            return;
+        }
         const originalSetRequestHandler = server.setRequestHandler.bind(server);
-        const originalConnect = server.connect.bind(server);
+        const originalConnect = server.connect ? server.connect.bind(server) : null;
         server.setRequestHandler = function (method, handler) {
+            if (!method || typeof method !== 'string') {
+                const logger = (0, logger_1.getLogger)();
+                logger.warn('setRequestHandler called with invalid method:', method);
+                return originalSetRequestHandler.call(this, method, handler);
+            }
             const wrappedHandler = async (...args) => {
                 const result = await handler(...args);
                 if (result && typeof result === 'object' && 'content' in result) {
@@ -433,15 +448,17 @@ function createMCPMiddleware(identity) {
             };
             return originalSetRequestHandler(method, wrappedHandler);
         };
-        server.connect = async function (transport) {
-            if (this.serverInfo && this.serverInfo.capabilities) {
-                this.serverInfo.capabilities['mcp-i'] = identity.getCapabilities();
-            }
-            this.setRequestHandler('mcp-i/challenge', async (request) => {
-                return identity.respondToChallenge(request.params);
-            });
-            return originalConnect(transport);
-        };
+        if (originalConnect) {
+            server.connect = async function (transport) {
+                if (this.serverInfo && this.serverInfo.capabilities) {
+                    this.serverInfo.capabilities['mcp-i'] = identity.getCapabilities();
+                }
+                this.setRequestHandler('mcp-i/challenge', async (request) => {
+                    return identity.respondToChallenge(request.params);
+                });
+                return originalConnect.call(this, transport);
+            };
+        }
     };
 }
 function patchMCPServer(identity) {
@@ -455,7 +472,13 @@ function patchMCPServer(identity) {
         const OriginalConstructor = OriginalServer;
         MCPModule.Server = function (...args) {
             const instance = new OriginalConstructor(...args);
-            middleware(instance, identity);
+            try {
+                middleware(instance, identity);
+            }
+            catch (error) {
+                const logger = (0, logger_1.getLogger)();
+                logger.warn('Failed to apply MCP-I middleware:', error);
+            }
             return instance;
         };
         Object.setPrototypeOf(MCPModule.Server, OriginalConstructor);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kya-os/mcp-i",
-  "version": "0.1.0-alpha.2.7",
+  "version": "0.1.0-alpha.2.8",
   "description": "COMING SOON:Production-ready MCP Identity with automatic registration, key rotation, and optimized performance",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",