@structured-world/gitlab-mcp 6.31.0 → 6.31.2

package/dist/src/session-manager.js

@@ -0,0 +1,125 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SessionManager = void 0;
+exports.getSessionManager = getSessionManager;
+exports.resetSessionManager = resetSessionManager;
+const index_js_1 = require("@modelcontextprotocol/sdk/server/index.js");
+const config_1 = require("./config");
+const handlers_1 = require("./handlers");
+const schema_utils_1 = require("./utils/schema-utils");
+const logger_1 = require("./logger");
+const DEFAULT_SESSION_TIMEOUT_MS = 30 * 60 * 1000;
+class SessionManager {
+    sessions = new Map();
+    cleanupInterval = null;
+    sessionTimeoutMs;
+    schemaModeDetected = false;
+    constructor(sessionTimeoutMs) {
+        this.sessionTimeoutMs = sessionTimeoutMs ?? DEFAULT_SESSION_TIMEOUT_MS;
+    }
+    start() {
+        this.cleanupInterval = setInterval(() => {
+            this.cleanupStaleSessions();
+        }, 60_000);
+        this.cleanupInterval.unref();
+        logger_1.logger.info({ sessionTimeoutMs: this.sessionTimeoutMs }, "Session manager started");
+    }
+    async createSession(sessionId, transport) {
+        if (this.sessions.has(sessionId)) {
+            logger_1.logger.warn({ sessionId }, "Duplicate sessionId detected — closing existing session");
+            await this.removeSession(sessionId);
+        }
+        const server = new index_js_1.Server({ name: config_1.packageName, version: config_1.packageVersion }, { capabilities: { tools: { listChanged: true } } });
+        server.oninitialized = () => {
+            if (!this.schemaModeDetected) {
+                this.schemaModeDetected = true;
+                const clientVersion = server.getClientVersion();
+                (0, schema_utils_1.setDetectedSchemaMode)(clientVersion?.name);
+            }
+        };
+        await (0, handlers_1.setupHandlers)(server);
+        await server.connect(transport);
+        const now = Date.now();
+        this.sessions.set(sessionId, {
+            server,
+            sessionId,
+            createdAt: now,
+            lastActivityAt: now,
+        });
+        logger_1.logger.info({ sessionId, activeSessions: this.sessions.size }, "Session created");
+        return server;
+    }
+    touchSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (session) {
+            session.lastActivityAt = Date.now();
+        }
+    }
+    async removeSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return;
+        this.sessions.delete(sessionId);
+        try {
+            await session.server.close();
+        }
+        catch (error) {
+            logger_1.logger.debug({ err: error, sessionId }, "Error closing session server (may already be closed)");
+        }
+        logger_1.logger.info({ sessionId, activeSessions: this.sessions.size }, "Session removed");
+    }
+    async broadcastToolsListChanged() {
+        const promises = [];
+        for (const [sessionId, session] of this.sessions) {
+            promises.push(session.server
+                .notification({ method: "notifications/tools/list_changed" })
+                .then(() => {
+                logger_1.logger.debug({ sessionId }, "Sent tools/list_changed to session");
+            })
+                .catch((error) => {
+                logger_1.logger.debug({ err: error, sessionId }, "Failed to send tools/list_changed to session");
+            }));
+        }
+        await Promise.allSettled(promises);
+        logger_1.logger.info({ sessionCount: this.sessions.size }, "Broadcast tools/list_changed to all sessions");
+    }
+    get activeSessionCount() {
+        return this.sessions.size;
+    }
+    cleanupStaleSessions() {
+        const now = Date.now();
+        const stale = [];
+        for (const [sessionId, session] of this.sessions) {
+            if (now - session.lastActivityAt > this.sessionTimeoutMs) {
+                stale.push(sessionId);
+            }
+        }
+        if (stale.length === 0)
+            return;
+        logger_1.logger.info({ staleCount: stale.length, activeSessions: this.sessions.size }, "Cleaning up stale sessions");
+        for (const sessionId of stale) {
+            this.removeSession(sessionId).catch((error) => {
+                logger_1.logger.error({ err: error, sessionId }, "Error during stale session cleanup");
+            });
+        }
+    }
+    async shutdown() {
+        if (this.cleanupInterval) {
+            clearInterval(this.cleanupInterval);
+            this.cleanupInterval = null;
+        }
+        const sessionIds = [...this.sessions.keys()];
+        await Promise.allSettled(sessionIds.map(id => this.removeSession(id)));
+        logger_1.logger.info("Session manager shut down");
+    }
+}
+exports.SessionManager = SessionManager;
+let sessionManagerInstance = null;
+function getSessionManager() {
+    sessionManagerInstance ??= new SessionManager();
+    return sessionManagerInstance;
+}
+function resetSessionManager() {
+    sessionManagerInstance = null;
+}
+//# sourceMappingURL=session-manager.js.map

package/dist/src/session-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-manager.js","sourceRoot":"","sources":["../../src/session-manager.ts"],"names":[],"mappings":";;;AAiNA,8CAGC;AAKD,kDAEC;AA3ND,wEAAmE;AAEnE,qCAAuD;AACvD,yCAA2C;AAC3C,uDAA6D;AAC7D,qCAAkC;AAGlC,MAAM,0BAA0B,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAgBlD,MAAa,cAAc;IACjB,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC7C,eAAe,GAA0C,IAAI,CAAC;IACrD,gBAAgB,CAAS;IAClC,kBAAkB,GAAG,KAAK,CAAC;IAEnC,YAAY,gBAAyB;QACnC,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,IAAI,0BAA0B,CAAC;IACzE,CAAC;IAMD,KAAK;QAEH,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC,EAAE,MAAM,CAAC,CAAC;QAGX,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,eAAM,CAAC,IAAI,CAAC,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,EAAE,yBAAyB,CAAC,CAAC;IACtF,CAAC;IAMD,KAAK,CAAC,aAAa,CAAC,SAAiB,EAAE,SAAoB;QAEzD,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACjC,eAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,yDAAyD,CAAC,CAAC;YACtF,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,iBAAM,CACvB,EAAE,IAAI,EAAE,oBAAW,EAAE,OAAO,EAAE,uBAAc,EAAE,EAC9C,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,CACnD,CAAC;QAIF,MAAM,CAAC,aAAa,GAAG,GAAG,EAAE;YAC1B,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC7B,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBAC/B,MAAM,aAAa,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;gBAChD,IAAA,oCAAqB,EAAC,aAAa,EAAE,IAAI,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC,CAAC;QAGF,MAAM,IAAA,wBAAa,EAAC,MAAM,CAAC,CAAC;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE;YAC3B,MAAM;YACN,SAAS;YACT,SAAS,EAAE,GAAG;YACd,cAAc,EAAE,GAAG;SACpB,CAAC,CAAC;QAEH,eAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,iBAAiB,CAAC,CAAC;QAElF,OAAO,MAAM,CAAC;IAChB,CAAC;IAKD,YAAY,CAAC,SAAiB;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,OAAO;QAErB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEhC,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CACV,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,EACzB,sDAAsD,CACvD,CAAC;QACJ,CAAC;QAED,eAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,iBAAiB,CAAC,CAAC;IACpF,CAAC;IAKD,KAAK,CAAC,yBAAyB;QAC7B,MAAM,QAAQ,GAAoB,EAAE,CAAC;QAErC,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjD,QAAQ,CAAC,IAAI,CACX,OAAO,CAAC,MAAM;iBACX,YAAY,CAAC,EAAE,MAAM,EAAE,kCAAkC,EAAE,CAAC;iBAC5D,IAAI,CAAC,GAAG,EAAE;gBACT,eAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,EAAE,oCAAoC,CAAC,CAAC;YACpE,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBACxB,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,8CAA8C,CAAC,CAAC;YAC1F,CAAC,CAAC,CACL,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,eAAM,CAAC,IAAI,CACT,EAAE,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EACpC,8CAA8C,CAC/C,CAAC;IACJ,CAAC;IAKD,IAAI,kBAAkB;QACpB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC5B,CAAC;IAKO,oBAAoB;QAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjD,IAAI,GAAG,GAAG,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACzD,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE/B,eAAM,CAAC,IAAI,CACT,EAAE,UAAU,EAAE,KAAK,CAAC,MAAM,EAAE,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAChE,4BAA4B,CAC7B,CAAC;QAEF,KAAK,MAAM,SAAS,IAAI,KAAK,EAAE,CAAC;YAE9B,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBACrD,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,oCAAoC,CAAC,CAAC;YAChF,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7C,MAAM,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAEvE,eAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAC3C,CAAC;CACF;AAjLD,wCAiLC;AAMD,IAAI,sBAAsB,GAA0B,IAAI,CAAC;AAEzD,SAAgB,iBAAiB;IAC/B,sBAAsB,KAAK,IAAI,cAAc,EAAE,CAAC;IAChD,OAAO,sBAAsB,CAAC;AAChC,CAAC;AAKD,SAAgB,mBAAmB;IACjC,sBAAsB,GAAG,IAAI,CAAC;AAChC,CAAC"}