cprime-supergateway 3.4.3 → 3.4.5

package/dist/gateways/stdioToSse.js

@@ -8,15 +8,19 @@ import { getVersion } from '../lib/getVersion.js';
 import { onSignals } from '../lib/onSignals.js';
 import { serializeCorsOrigin } from '../lib/serializeCorsOrigin.js';
 import { EncryptionService } from '../services/encryptionService.js';
+import { initMongoClient } from '../lib/initMongoClient.js';
+import { McpServerLogRepository } from '../lib/mcpServerLogRepository.js';
 const encryptionService = new EncryptionService('env');
-const plaintext = await encryptionService.decryptText(process.env.ENCRYPTED_ENV ?? '', process.env.AAD_JSON ? JSON.parse(process.env.AAD_JSON) : {});
 let decryptedEnvs = {};
-try {
-    const asObj = JSON.parse(plaintext);
-    decryptedEnvs = asObj;
-}
-catch {
-    console.error('Failed to parse decrypted envs', plaintext);
+if (process.env.ENCRYPTED_ENV) {
+    const plaintext = await encryptionService.decryptText(process.env.ENCRYPTED_ENV, process.env.AAD_JSON ? JSON.parse(process.env.AAD_JSON) : {});
+    try {
+        const asObj = JSON.parse(plaintext);
+        decryptedEnvs = asObj;
+    }
+    catch {
+        console.error('Failed to parse decrypted envs', plaintext);
+    }
 }
 const setResponseHeaders = ({ res, headers, }) => Object.entries(headers).forEach(([key, value]) => {
     res.setHeader(key, value);
@@ -34,6 +38,8 @@ export async function stdioToSse(args) {
     logger.info(`  - CORS: ${corsOrigin ? `enabled (${serializeCorsOrigin({ corsOrigin })})` : 'disabled'}`);
     logger.info(`  - Health endpoints: ${healthEndpoints.length ? healthEndpoints.join(', ') : '(none)'}`);
     onSignals({ logger });
+    const mongoClient = initMongoClient();
+    const mcpServerLogRepository = new McpServerLogRepository(mongoClient, process.env.MONGO_DB ?? 'localhost', process.env.LOGS_COLLECTION ?? 'mcp_server_logs');
     const child = spawn(stdioCmd, {
         shell: true,
         env: { ...process.env, ...decryptedEnvs },
@@ -72,10 +78,28 @@ export async function stdioToSse(args) {
         await server.connect(sseTransport);
         const sessionId = sseTransport.sessionId;
         if (sessionId) {
-            sessions[sessionId] = { transport: sseTransport, response: res };
+            sessions[sessionId] = {
+                transport: sseTransport,
+                response: res,
+                ip: req.ip ?? '',
+                userId: req.query.userId ?? '',
+            };
         }
         sseTransport.onmessage = (msg) => {
             logger.info(`SSE → Child (session ${sessionId}): ${JSON.stringify(msg)}`);
+            mcpServerLogRepository
+                .insert({
+                ip: req.ip ?? '',
+                userId: req.query.userId ?? '',
+                sessionId,
+                type: 'rpc',
+                data: msg,
+                createdAt: new Date(),
+                updatedAt: new Date(),
+            })
+                .catch((err) => {
+                logger.error(`Failed to insert log:`, JSON.stringify(err));
+            });
             child.stdin.write(JSON.stringify(msg) + '\n');
         };
         sseTransport.onclose = () => {
@@ -129,6 +153,19 @@ export async function stdioToSse(args) {
                 for (const [sid, session] of Object.entries(sessions)) {
                     try {
                         session.transport.send(jsonMsg);
+                        mcpServerLogRepository
+                            .insert({
+                            ip: session.ip,
+                            userId: session.userId,
+                            sessionId: sid,
+                            type: 'system',
+                            data: line,
+                            createdAt: new Date(),
+                            updatedAt: new Date(),
+                        })
+                            .catch((err) => {
+                            logger.error(`Failed to insert log:`, JSON.stringify(err));
+                        });
                     }
                     catch (err) {
                         logger.error(`Failed to send to session ${sid}:`, err);

package/dist/lib/initMongoClient.js

@@ -0,0 +1,8 @@
+import { MongoClient } from 'mongodb';
+export function initMongoClient(uri = process.env.MONGO_URI) {
+    if (!uri) {
+        throw new Error('MONGO_URI is not set');
+    }
+    const client = new MongoClient(uri);
+    return client;
+}

package/dist/lib/mcpServerLogRepository.js

@@ -0,0 +1,20 @@
+export class McpServerLogRepository {
+    client;
+    collection;
+    constructor(client, dbName = 'local', collectionName = 'mcp_server_logs') {
+        this.client = client;
+        this.collection = this.client.db(dbName).collection(collectionName);
+    }
+    insert(data) {
+        return this.collection.insertOne(data);
+    }
+    update(sessionId, update) {
+        return this.collection.updateOne({ sessionId, 'data.id': update.id }, {
+            $set: {
+                'data.result': update.result,
+                'data.error': update.error,
+                updatedAt: new Date(),
+            },
+        });
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cprime-supergateway",
-  "version": "3.4.3",
+  "version": "3.4.5",
   "description": "Run MCP stdio servers over SSE, Streamable HTTP or visa versa",
   "repository": {
     "type": "git",
@@ -16,7 +16,7 @@
   ],
   "type": "module",
   "bin": {
-    "supergateway": "dist/index.js"
+    "cprime-supergateway": "dist/index.js"
   },
   "scripts": {
     "build": "tsc -p tsconfig.build.json",

package/src/gateways/stdioToSse.ts

@@ -26,16 +26,18 @@ export interface StdioToSseArgs {
 }
 
 const encryptionService = new EncryptionService('env')
-const plaintext = await encryptionService.decryptText(
-  process.env.ENCRYPTED_ENV ?? '',
-  process.env.AAD_JSON ? JSON.parse(process.env.AAD_JSON) : {},
-)
 let decryptedEnvs = {}
-try {
-  const asObj = JSON.parse(plaintext)
-  decryptedEnvs = asObj
-} catch {
-  console.error('Failed to parse decrypted envs', plaintext)
+if (process.env.ENCRYPTED_ENV) {
+  const plaintext = await encryptionService.decryptText(
+    process.env.ENCRYPTED_ENV,
+    process.env.AAD_JSON ? JSON.parse(process.env.AAD_JSON) : {},
+  )
+  try {
+    const asObj = JSON.parse(plaintext)
+    decryptedEnvs = asObj
+  } catch {
+    console.error('Failed to parse decrypted envs', plaintext)
+  }
 }
 
 const setResponseHeaders = ({