@mgsoftwarebv/mg-dashboard-mcp 3.4.1 → 3.4.3

package/dist/index.js

@@ -6,8 +6,8 @@ import { Client as Client$1 } from '@modelcontextprotocol/sdk/client/index.js';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { ListToolsRequestSchema, CallToolRequestSchema, isInitializeRequest, ListPromptsRequestSchema, GetPromptRequestSchema, ListResourcesRequestSchema, ReadResourceRequestSchema, ListResourceTemplatesRequestSchema, CompleteRequestSchema, SubscribeRequestSchema, UnsubscribeRequestSchema, ListToolsResultSchema, CallToolResultSchema, ListPromptsResultSchema, GetPromptResultSchema, ListResourcesResultSchema, ReadResourceResultSchema, ListResourceTemplatesResultSchema, CompleteResultSchema, EmptyResultSchema } from '@modelcontextprotocol/sdk/types.js';
 import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
-import { ListToolsRequestSchema, CallToolRequestSchema, isInitializeRequest } from '@modelcontextprotocol/sdk/types.js';
 import { createServer } from 'http';
 import { randomUUID, createHash, randomBytes, createCipheriv, createDecipheriv } from 'crypto';
 import { createClient } from '@supabase/supabase-js';
@@ -155,6 +155,50 @@ async function runBridge(handshake, proxyUrl2, refresh) {
     { name: "mg-dashboard-mcp-proxy", version: "1.0.0" },
     { capabilities: { tools: {}, prompts: {}, resources: {}, logging: {} } }
   );
+  const forwardRequest = (resultSchema) => async (request) => upstream.request(
+    request,
+    resultSchema
+  );
+  stdioServer.setRequestHandler(
+    ListToolsRequestSchema,
+    forwardRequest(ListToolsResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    CallToolRequestSchema,
+    forwardRequest(CallToolResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    ListPromptsRequestSchema,
+    forwardRequest(ListPromptsResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    GetPromptRequestSchema,
+    forwardRequest(GetPromptResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    ListResourcesRequestSchema,
+    forwardRequest(ListResourcesResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    ReadResourceRequestSchema,
+    forwardRequest(ReadResourceResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    ListResourceTemplatesRequestSchema,
+    forwardRequest(ListResourceTemplatesResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    CompleteRequestSchema,
+    forwardRequest(CompleteResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    SubscribeRequestSchema,
+    forwardRequest(EmptyResultSchema)
+  );
+  stdioServer.setRequestHandler(
+    UnsubscribeRequestSchema,
+    forwardRequest(EmptyResultSchema)
+  );
   stdioServer.fallbackRequestHandler = async (request) => {
     return await upstream.request(
       request,
@@ -172,6 +216,17 @@ async function runBridge(handshake, proxyUrl2, refresh) {
   };
   await stdioServer.connect(new StdioServerTransport());
   console.error(`[Proxy] Bridge ready. Forwarding stdio \u2192 ${proxyUrl2}`);
+  await new Promise((resolve) => {
+    const keepAlive = setInterval(() => void 0, 2147483647);
+    const done = () => {
+      clearInterval(keepAlive);
+      resolve();
+    };
+    process.stdin.once("close", done);
+    process.stdin.once("end", done);
+    process.once("SIGINT", done);
+    process.once("SIGTERM", done);
+  });
 }
 async function runProxyMode(args2) {
   const proxyUrl2 = getArg(args2, "proxy-url") || process.env.MG_DASHBOARD_PROXY_URL;