@karashiiro/my-cool-proxy 1.0.0

package/dist/e2e/helpers/test-config-generator.d.ts

@@ -0,0 +1,14 @@
+import type { ServerConfig } from "../../types/interfaces.js";
+export declare function generateTestConfig(config: ServerConfig): {
+    configPath: string;
+    cleanup: () => void;
+};
+export declare function generateHttpTestConfig(overrides?: Partial<ServerConfig>): {
+    configPath: string;
+    cleanup: () => void;
+};
+export declare function generateStdioTestConfig(overrides?: Partial<ServerConfig>): {
+    configPath: string;
+    cleanup: () => void;
+};
+//# sourceMappingURL=test-config-generator.d.ts.map

package/dist/e2e/helpers/test-config-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-config-generator.d.ts","sourceRoot":"","sources":["../../../src/e2e/helpers/test-config-generator.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAS9D,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,YAAY,GAAG;IACxD,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB,CAmBA;AAKD,wBAAgB,sBAAsB,CAAC,SAAS,GAAE,OAAO,CAAC,YAAY,CAAM;;;EAS3E;AAKD,wBAAgB,uBAAuB,CAAC,SAAS,GAAE,OAAO,CAAC,YAAY,CAAM;;;EAO5E"}

package/dist/e2e/helpers/test-config-generator.js

@@ -0,0 +1,39 @@
+import { writeFileSync, unlinkSync, mkdtempSync } from "node:fs";
+import { join } from "node:path";
+import { tmpdir } from "node:os";
+export function generateTestConfig(config) {
+    const tempDir = mkdtempSync(join(tmpdir(), "mcp-test-"));
+    const configPath = join(tempDir, "config.json");
+    writeFileSync(configPath, JSON.stringify(config, null, 2), "utf-8");
+    return {
+        configPath,
+        cleanup: () => {
+            try {
+                unlinkSync(configPath);
+                // Note: We don't remove the directory itself to avoid issues if other files exist
+            }
+            catch {
+                // Ignore cleanup errors
+            }
+        },
+    };
+}
+export function generateHttpTestConfig(overrides = {}) {
+    const config = {
+        transport: "http",
+        port: 3000,
+        host: "localhost",
+        mcpClients: {},
+        ...overrides,
+    };
+    return generateTestConfig(config);
+}
+export function generateStdioTestConfig(overrides = {}) {
+    const config = {
+        transport: "stdio",
+        mcpClients: {},
+        ...overrides,
+    };
+    return generateTestConfig(config);
+}
+//# sourceMappingURL=test-config-generator.js.map

package/dist/e2e/helpers/test-config-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-config-generator.js","sourceRoot":"","sources":["../../../src/e2e/helpers/test-config-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACjE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAUjC,MAAM,UAAU,kBAAkB,CAAC,MAAoB,EAGrD;IAEA,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;IACzD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;IAGhD,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IAEpE,OAAO;QACL,UAAU;QACV,OAAO,EAAE,GAAG,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,UAAU,CAAC,UAAU,CAAC,CAAC;gBACvB,kFAAkF;YACpF,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QAAA,CACF;KACF,CAAC;AAAA,CACH;AAKD,MAAM,UAAU,sBAAsB,CAAC,SAAS,GAA0B,EAAE,EAAE;IAC5E,MAAM,MAAM,GAAiB;QAC3B,SAAS,EAAE,MAAM;QACjB,IAAI,EAAE,IAAI;QACV,IAAI,EAAE,WAAW;QACjB,UAAU,EAAE,EAAE;QACd,GAAG,SAAS;KACb,CAAC;IACF,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC;AAAA,CACnC;AAKD,MAAM,UAAU,uBAAuB,CAAC,SAAS,GAA0B,EAAE,EAAE;IAC7E,MAAM,MAAM,GAAiB;QAC3B,SAAS,EAAE,OAAO;QAClB,UAAU,EAAE,EAAE;QACd,GAAG,SAAS;KACb,CAAC;IACF,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC;AAAA,CACnC"}

package/dist/e2e/helpers/toy-server-manager.d.ts

@@ -0,0 +1,10 @@
+type ToyServerType = "calculator" | "data" | "sampling" | "elicitation";
+export declare class ToyServerManager {
+    private servers;
+    startHttp(serverType: ToyServerType, port: number): Promise<string>;
+    stopAll(): Promise<void>;
+    stop(key: string): Promise<void>;
+    count(): number;
+}
+export {};
+//# sourceMappingURL=toy-server-manager.d.ts.map

package/dist/e2e/helpers/toy-server-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toy-server-manager.d.ts","sourceRoot":"","sources":["../../../src/e2e/helpers/toy-server-manager.ts"],"names":[],"mappings":"AAOA,KAAK,aAAa,GAAG,YAAY,GAAG,MAAM,GAAG,UAAU,GAAG,aAAa,CAAC;AAWxE,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,OAAO,CAA6C;IAStD,SAAS,CAAC,UAAU,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CA4CxE;IAKK,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAO7B;IAKK,IAAI,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAMrC;IAKD,KAAK,IAAI,MAAM,CAEd;CACF"}

package/dist/e2e/helpers/toy-server-manager.js

@@ -0,0 +1,58 @@
+import { startHttpCalculatorServer, startHttpDataServer, startHttpSamplingServer, startHttpElicitationServer, } from "../fixtures/toy-servers/index.js";
+export class ToyServerManager {
+    servers = new Map();
+    async startHttp(serverType, port) {
+        const key = `${serverType}-${port}`;
+        if (this.servers.has(key)) {
+            throw new Error(`Server ${key} is already running`);
+        }
+        let closeFunc;
+        switch (serverType) {
+            case "calculator": {
+                const server = await startHttpCalculatorServer(port);
+                closeFunc = server.close;
+                break;
+            }
+            case "data": {
+                const server = await startHttpDataServer(port);
+                closeFunc = server.close;
+                break;
+            }
+            case "sampling": {
+                const server = await startHttpSamplingServer(port);
+                closeFunc = server.close;
+                break;
+            }
+            case "elicitation": {
+                const server = await startHttpElicitationServer(port);
+                closeFunc = server.close;
+                break;
+            }
+            default:
+                throw new Error(`Unknown server type: ${serverType}`);
+        }
+        this.servers.set(key, {
+            type: serverType,
+            port,
+            close: closeFunc,
+        });
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        return key;
+    }
+    async stopAll() {
+        const closePromises = Array.from(this.servers.values()).map((server) => server.close());
+        await Promise.all(closePromises);
+        this.servers.clear();
+    }
+    async stop(key) {
+        const server = this.servers.get(key);
+        if (server) {
+            await server.close();
+            this.servers.delete(key);
+        }
+    }
+    count() {
+        return this.servers.size;
+    }
+}
+//# sourceMappingURL=toy-server-manager.js.map

package/dist/e2e/helpers/toy-server-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toy-server-manager.js","sourceRoot":"","sources":["../../../src/e2e/helpers/toy-server-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,mBAAmB,EACnB,uBAAuB,EACvB,0BAA0B,GAC3B,MAAM,kCAAkC,CAAC;AAa1C,MAAM,OAAO,gBAAgB;IACnB,OAAO,GAAmC,IAAI,GAAG,EAAE,CAAC;IAS5D,KAAK,CAAC,SAAS,CAAC,UAAyB,EAAE,IAAY,EAAmB;QACxE,MAAM,GAAG,GAAG,GAAG,UAAU,IAAI,IAAI,EAAE,CAAC;QAEpC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,UAAU,GAAG,qBAAqB,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,SAA8B,CAAC;QAEnC,QAAQ,UAAU,EAAE,CAAC;YACnB,KAAK,YAAY,EAAE,CAAC;gBAClB,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAAC,IAAI,CAAC,CAAC;gBACrD,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;gBACzB,MAAM;YACR,CAAC;YACD,KAAK,MAAM,EAAE,CAAC;gBACZ,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAC/C,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;gBACzB,MAAM;YACR,CAAC;YACD,KAAK,UAAU,EAAE,CAAC;gBAChB,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,IAAI,CAAC,CAAC;gBACnD,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;gBACzB,MAAM;YACR,CAAC;YACD,KAAK,aAAa,EAAE,CAAC;gBACnB,MAAM,MAAM,GAAG,MAAM,0BAA0B,CAAC,IAAI,CAAC,CAAC;gBACtD,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;gBACzB,MAAM;YACR,CAAC;YACD;gBACE,MAAM,IAAI,KAAK,CAAC,wBAAwB,UAAU,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,IAAI,EAAE,UAAU;YAChB,IAAI;YACJ,KAAK,EAAE,SAAS;SACjB,CAAC,CAAC;QAGH,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAEzD,OAAO,GAAG,CAAC;IAAA,CACZ;IAKD,KAAK,CAAC,OAAO,GAAkB;QAC7B,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrE,MAAM,CAAC,KAAK,EAAE,CACf,CAAC;QAEF,MAAM,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IAAA,CACtB;IAKD,KAAK,CAAC,IAAI,CAAC,GAAW,EAAiB;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;IAAA,CACF;IAKD,KAAK,GAAW;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAAA,CAC1B;CACF"}

package/dist/handlers/shutdown-handler.d.ts

@@ -0,0 +1,8 @@
+import type { IMCPClientManager, ILogger, IShutdownHandler } from "../types/interfaces.js";
+export declare class ShutdownHandler implements IShutdownHandler {
+    private clientPool;
+    private logger;
+    constructor(clientPool: IMCPClientManager, logger: ILogger);
+    shutdown(): Promise<void>;
+}
+//# sourceMappingURL=shutdown-handler.d.ts.map

package/dist/handlers/shutdown-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdown-handler.d.ts","sourceRoot":"","sources":["../../src/handlers/shutdown-handler.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,iBAAiB,EACjB,OAAO,EACP,gBAAgB,EACjB,MAAM,wBAAwB,CAAC;AAmBhC,qBACa,eAAgB,YAAW,gBAAgB;IAEnB,OAAO,CAAC,UAAU;IAC5B,OAAO,CAAC,MAAM;IAFvC,YAC2C,UAAU,EAAE,iBAAiB,EACvC,MAAM,EAAE,OAAO,EAC5C;IAQE,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAS9B;CACF"}

package/dist/handlers/shutdown-handler.js

@@ -0,0 +1,37 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+import { injectable } from "inversify";
+import { $inject } from "../container/decorators.js";
+import { TYPES } from "../types/index.js";
+let ShutdownHandler = class ShutdownHandler {
+    clientPool;
+    logger;
+    constructor(clientPool, logger) {
+        this.clientPool = clientPool;
+        this.logger = logger;
+    }
+    async shutdown() {
+        this.logger.info("Shutting down...");
+        await this.clientPool.close();
+        this.logger.info("Shutdown complete");
+        process.exit(0);
+    }
+};
+ShutdownHandler = __decorate([
+    injectable(),
+    __param(0, $inject(TYPES.MCPClientManager)),
+    __param(1, $inject(TYPES.Logger)),
+    __metadata("design:paramtypes", [Object, Object])
+], ShutdownHandler);
+export { ShutdownHandler };
+//# sourceMappingURL=shutdown-handler.js.map

package/dist/handlers/shutdown-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdown-handler.js","sourceRoot":"","sources":["../../src/handlers/shutdown-handler.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAyBnC,IAAM,eAAe,GAArB,MAAM,eAAe;IAEiB,UAAU;IACpB,MAAM;IAFvC,YAC2C,UAA6B,EACvC,MAAe,EAC9C;0BAFyC,UAAU;sBACpB,MAAM;IACpC,CAAC;IAQJ,KAAK,CAAC,QAAQ,GAAkB;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAGrC,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAE9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAEtC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAAA,CACjB;CACF,CAAA;AAtBY,eAAe;IAD3B,UAAU,EAAE;IAGR,WAAA,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;IAC/B,WAAA,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;;GAHb,eAAe,CAsB3B"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+import "reflect-metadata";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,kBAAkB,CAAC"}

package/dist/index.js

@@ -0,0 +1,235 @@
+import "reflect-metadata";
+import { createContainer } from "./container/inversify.config.js";
+import { TYPES } from "./types/index.js";
+import { serveHttp } from "@karashiiro/mcp/http";
+import { serveStdio } from "@karashiiro/mcp/stdio";
+import { loadConfig, mergeEnvConfig } from "./utils/config-loader.js";
+import { MCPGatewayServer } from "./mcp/gateway-server.js";
+import { CreateMessageRequestSchema, ElicitRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { parseArgs } from "./utils/cli-args.js";
+import { getConfigPaths, getPlatformConfigDir } from "./utils/config-paths.js";
+function registerProxyHandlers(sessionId, clientManager, gatewayServer, logger, capabilities) {
+    const clients = clientManager.getClientsBySession(sessionId);
+    for (const [serverName, clientSession] of clients) {
+        if (capabilities.sampling) {
+            clientSession.setRequestHandler(CreateMessageRequestSchema, async (request) => {
+                logger.debug(`Received sampling request from upstream server '${serverName}', forwarding to downstream`);
+                try {
+                    const result = await gatewayServer.forwardSamplingRequest(request.params);
+                    return result;
+                }
+                catch (error) {
+                    logger.error(`Failed to forward sampling request from '${serverName}'`, error instanceof Error ? error : new Error(String(error)));
+                    throw error;
+                }
+            });
+            logger.debug(`Registered sampling request handler for upstream server '${serverName}'`);
+        }
+        if (capabilities.elicitation) {
+            clientSession.setRequestHandler(ElicitRequestSchema, async (request) => {
+                logger.debug(`Received elicitation request from upstream server '${serverName}', forwarding to downstream`);
+                try {
+                    const result = await gatewayServer.forwardElicitationRequest(request.params);
+                    return result;
+                }
+                catch (error) {
+                    logger.error(`Failed to forward elicitation request from '${serverName}'`, error instanceof Error ? error : new Error(String(error)));
+                    throw error;
+                }
+            });
+            logger.debug(`Registered elicitation request handler for upstream server '${serverName}'`);
+        }
+    }
+    const clientCount = clients.size;
+    if (capabilities.sampling || capabilities.elicitation) {
+        logger.info(`Registered proxy handlers on ${clientCount} upstream client(s): ` +
+            `sampling=${!!capabilities.sampling}, elicitation=${!!capabilities.elicitation}`);
+    }
+}
+async function initializeClientsForSession(sessionId, config, clientManager, clientCapabilities) {
+    const connectionPromises = Object.entries(config.mcpClients).map(async ([name, clientConfig]) => {
+        if (clientConfig.type === "http") {
+            return clientManager.addHttpClient(name, clientConfig.url, sessionId, clientConfig.headers, clientConfig.allowedTools, clientCapabilities);
+        }
+        else if (clientConfig.type === "stdio") {
+            return clientManager.addStdioClient(name, clientConfig.command, sessionId, clientConfig.args, clientConfig.env, clientConfig.allowedTools, clientCapabilities);
+        }
+        else {
+            const _exhaustiveCheck = clientConfig;
+            return {
+                name,
+                success: false,
+                error: `Unknown client type: ${_exhaustiveCheck.type}`,
+            };
+        }
+    });
+    const results = await Promise.allSettled(connectionPromises);
+    const successful = [];
+    const failed = [];
+    for (const result of results) {
+        if (result.status === "fulfilled") {
+            if (result.value.success) {
+                successful.push(result.value.name);
+            }
+            else {
+                failed.push({
+                    name: result.value.name,
+                    error: result.value.error || "Unknown error",
+                });
+            }
+        }
+        else {
+            const errorMessage = result.reason instanceof Error
+                ? result.reason.message
+                : String(result.reason ?? "Unknown rejection");
+            failed.push({
+                name: "unknown",
+                error: errorMessage,
+            });
+        }
+    }
+    return { successful, failed };
+}
+async function startHttpMode(container, config) {
+    const logger = container.get(TYPES.Logger);
+    if (config.port === undefined || config.host === undefined) {
+        throw new Error("Port and host are required for HTTP mode");
+    }
+    const clientManager = container.get(TYPES.MCPClientManager);
+    const toolRegistry = container.get(TYPES.ToolRegistry);
+    const resourceAggregation = container.get(TYPES.ResourceAggregationService);
+    const promptAggregation = container.get(TYPES.PromptAggregationService);
+    const shutdownHandler = container.get(TYPES.ShutdownHandler);
+    const capabilityStore = container.get(TYPES.CapabilityStore);
+    const serverInfoPreloader = container.get(TYPES.ServerInfoPreloader);
+    logger.info("Preloading upstream server info...");
+    const preloadedServers = await serverInfoPreloader.preloadServerInfo(config);
+    const aggregatedInstructions = serverInfoPreloader.buildAggregatedInstructions(preloadedServers);
+    logger.info(`Preloaded info from ${preloadedServers.length} server(s) for gateway instructions`);
+    const handle = await serveHttp(async (sessionId) => {
+        logger.info(`Creating gateway server for session ${sessionId}`);
+        const gatewayServer = new MCPGatewayServer(toolRegistry, clientManager, logger, resourceAggregation, promptAggregation, aggregatedInstructions);
+        gatewayServer.setOnDownstreamInitialized(async (capabilities) => {
+            logger.info(`Session ${sessionId}: Downstream client initialized with capabilities: ` +
+                `sampling=${!!capabilities.sampling}, elicitation=${!!capabilities.elicitation}`);
+            capabilityStore.setCapabilities(sessionId, capabilities);
+            const initResult = await initializeClientsForSession(sessionId, config, clientManager, capabilities);
+            if (initResult.failed.length > 0) {
+                logger.warn(`Session ${sessionId}: ${initResult.failed.length} server(s) failed to connect: ` +
+                    initResult.failed.map((f) => `${f.name} (${f.error})`).join(", "));
+            }
+            if (initResult.successful.length === 0 &&
+                Object.keys(config.mcpClients).length > 0) {
+                logger.warn(`Session ${sessionId}: All configured servers failed to connect.`);
+            }
+            logger.info(`Session ${sessionId}: ${initResult.successful.length} server(s) connected successfully`);
+            registerProxyHandlers(sessionId, clientManager, gatewayServer, logger, capabilities);
+        });
+        return gatewayServer.getServer();
+    }, {
+        port: config.port,
+        host: config.host,
+        sessions: {
+            onSessionClosed: async (sessionId) => {
+                logger.debug(`Session ${sessionId} closed, cleaning up...`);
+                try {
+                    await clientManager.closeSession(sessionId);
+                    capabilityStore.deleteCapabilities(sessionId);
+                }
+                catch (error) {
+                    logger.error(`Failed to close session ${sessionId}`, error instanceof Error ? error : new Error(String(error)));
+                }
+            },
+        },
+    });
+    logger.info(`MCP Lua Gateway listening on http://${config.host}:${config.port}`);
+    process.on("SIGINT", async () => {
+        await handle.close();
+        await shutdownHandler.shutdown();
+    });
+}
+async function startStdioMode(container, config) {
+    const logger = container.get(TYPES.Logger);
+    const clientManager = container.get(TYPES.MCPClientManager);
+    const toolRegistry = container.get(TYPES.ToolRegistry);
+    const resourceAggregation = container.get(TYPES.ResourceAggregationService);
+    const promptAggregation = container.get(TYPES.PromptAggregationService);
+    const capabilityStore = container.get(TYPES.CapabilityStore);
+    const serverInfoPreloader = container.get(TYPES.ServerInfoPreloader);
+    const SESSION_ID = "default";
+    logger.info("Preloading upstream server info...");
+    const preloadedServers = await serverInfoPreloader.preloadServerInfo(config);
+    const aggregatedInstructions = serverInfoPreloader.buildAggregatedInstructions(preloadedServers);
+    logger.info(`Preloaded info from ${preloadedServers.length} server(s) for gateway instructions`);
+    const handle = await serveStdio(() => {
+        const gatewayServer = new MCPGatewayServer(toolRegistry, clientManager, logger, resourceAggregation, promptAggregation, aggregatedInstructions);
+        gatewayServer.setOnDownstreamInitialized(async (capabilities) => {
+            logger.info(`Downstream client initialized with capabilities: ` +
+                `sampling=${!!capabilities.sampling}, elicitation=${!!capabilities.elicitation}`);
+            capabilityStore.setCapabilities(SESSION_ID, capabilities);
+            const initResult = await initializeClientsForSession(SESSION_ID, config, clientManager, capabilities);
+            if (initResult.failed.length > 0) {
+                logger.warn(`${initResult.failed.length} server(s) failed to connect: ` +
+                    initResult.failed.map((f) => `${f.name} (${f.error})`).join(", "));
+            }
+            if (initResult.successful.length === 0 &&
+                Object.keys(config.mcpClients).length > 0) {
+                logger.warn(`All configured servers failed to connect. Gateway running but no servers available.`);
+            }
+            logger.info(`${initResult.successful.length} server(s) connected successfully`);
+            registerProxyHandlers(SESSION_ID, clientManager, gatewayServer, logger, capabilities);
+        });
+        return gatewayServer.getServer();
+    });
+    logger.info("MCP Lua Gateway running in stdio mode");
+    process.on("SIGINT", async () => {
+        await handle.close();
+        await clientManager.close();
+        logger.info("Shutdown complete");
+        process.exit(0);
+    });
+}
+function printHelp() {
+    console.log("MCP Lua Gateway - Proxy for multiple MCP servers with Lua scripting\n");
+    console.log("Usage: my-cool-proxy [options]\n");
+    console.log("Options:");
+    console.log("  -c, --config-path    Show config file search paths and exit");
+    console.log("  -h, --help           Show this help message and exit\n");
+    console.log("Environment variables:");
+    console.log("  CONFIG_PATH          Override config file location");
+    console.log("  PORT                 Override server port (HTTP mode)");
+    console.log("  HOST                 Override server host (HTTP mode)\n");
+    console.log("See CONFIG.md for full configuration reference.");
+}
+function printConfigPaths() {
+    console.log("Config file search order:\n");
+    const paths = getConfigPaths();
+    for (const p of paths) {
+        const status = p.exists ? "[EXISTS]" : "[NOT FOUND]";
+        const label = p.source === "env" ? "ENV: CONFIG_PATH" : "Platform config";
+        console.log(`  ${status} ${label}`);
+        console.log(`          ${p.path}\n`);
+    }
+    console.log(`Platform config directory: ${getPlatformConfigDir()}`);
+}
+async function main() {
+    const args = parseArgs(process.argv.slice(2));
+    if (args.help) {
+        printHelp();
+        process.exit(0);
+    }
+    if (args.showConfigPath) {
+        printConfigPaths();
+        process.exit(0);
+    }
+    const config = mergeEnvConfig(loadConfig());
+    const container = createContainer(config);
+    if (config.transport === "stdio") {
+        await startStdioMode(container, config);
+    }
+    else {
+        await startHttpMode(container, config);
+    }
+}
+main().catch(console.error);
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,kBAAkB,CAAC;AAE1B,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAWzC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EACL,0BAA0B,EAC1B,mBAAmB,GACpB,MAAM,oCAAoC,CAAC;AAI5C,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAY/E,SAAS,qBAAqB,CAC5B,SAAiB,EACjB,aAAgC,EAChC,aAA+B,EAC/B,MAAe,EACf,YAAoC,EAC9B;IACN,MAAM,OAAO,GAAG,aAAa,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAE7D,KAAK,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,IAAI,OAAO,EAAE,CAAC;QAElD,IAAI,YAAY,CAAC,QAAQ,EAAE,CAAC;YAC1B,aAAa,CAAC,iBAAiB,CAC7B,0BAA0B,EAC1B,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC;gBACjB,MAAM,CAAC,KAAK,CACV,mDAAmD,UAAU,6BAA6B,CAC3F,CAAC;gBACF,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,sBAAsB,CACvD,OAAO,CAAC,MAAM,CACf,CAAC;oBACF,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CACV,4CAA4C,UAAU,GAAG,EACzD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAC1D,CAAC;oBACF,MAAM,KAAK,CAAC;gBACd,CAAC;YAAA,CACF,CACF,CAAC;YACF,MAAM,CAAC,KAAK,CACV,4DAA4D,UAAU,GAAG,CAC1E,CAAC;QACJ,CAAC;QAGD,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC;YAC7B,aAAa,CAAC,iBAAiB,CAAC,mBAAmB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC;gBACtE,MAAM,CAAC,KAAK,CACV,sDAAsD,UAAU,6BAA6B,CAC9F,CAAC;gBACF,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,yBAAyB,CAC1D,OAAO,CAAC,MAAM,CACf,CAAC;oBACF,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CACV,+CAA+C,UAAU,GAAG,EAC5D,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAC1D,CAAC;oBACF,MAAM,KAAK,CAAC;gBACd,CAAC;YAAA,CACF,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CACV,+DAA+D,UAAU,GAAG,CAC7E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IACjC,IAAI,YAAY,CAAC,QAAQ,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC;QACtD,MAAM,CAAC,IAAI,CACT,gCAAgC,WAAW,uBAAuB;YAChE,YAAY,CAAC,CAAC,YAAY,CAAC,QAAQ,iBAAiB,CAAC,CAAC,YAAY,CAAC,WAAW,EAAE,CACnF,CAAC;IACJ,CAAC;AAAA,CACF;AAYD,KAAK,UAAU,2BAA2B,CACxC,SAAiB,EACjB,MAAoB,EACpB,aAAgC,EAChC,kBAA2C,EACZ;IAC/B,MAAM,kBAAkB,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAC9D,KAAK,EAAE,CAAC,IAAI,EAAE,YAAY,CAAC,EAAmC,EAAE,CAAC;QAC/D,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACjC,OAAO,aAAa,CAAC,aAAa,CAChC,IAAI,EACJ,YAAY,CAAC,GAAG,EAChB,SAAS,EACT,YAAY,CAAC,OAAO,EACpB,YAAY,CAAC,YAAY,EACzB,kBAAkB,CACnB,CAAC;QACJ,CAAC;aAAM,IAAI,YAAY,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACzC,OAAO,aAAa,CAAC,cAAc,CACjC,IAAI,EACJ,YAAY,CAAC,OAAO,EACpB,SAAS,EACT,YAAY,CAAC,IAAI,EACjB,YAAY,CAAC,GAAG,EAChB,YAAY,CAAC,YAAY,EACzB,kBAAkB,CACnB,CAAC;QACJ,CAAC;aAAM,CAAC;YAGN,MAAM,gBAAgB,GAAU,YAAY,CAAC;YAC7C,OAAO;gBACL,IAAI;gBACJ,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,wBAAyB,gBAAqC,CAAC,IAAI,EAAE;aAC7E,CAAC;QACJ,CAAC;IAAA,CACF,CACF,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IAE7D,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,MAAM,MAAM,GAA2C,EAAE,CAAC;IAE1D,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBACzB,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;oBACvB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,eAAe;iBAC7C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,CAAC;YAGN,MAAM,YAAY,GAChB,MAAM,CAAC,MAAM,YAAY,KAAK;gBAC5B,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO;gBACvB,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,IAAI,mBAAmB,CAAC,CAAC;YACnD,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;AAAA,CAC/B;AAED,KAAK,UAAU,aAAa,CAC1B,SAA8C,EAC9C,MAAoB,EACpB;IACA,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAU,KAAK,CAAC,MAAM,CAAC,CAAC;IAGpD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAC3D,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IAGD,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CACjC,KAAK,CAAC,gBAAgB,CACvB,CAAC;IACF,MAAM,YAAY,GAAG,SAAS,CAAC,GAAG,CAAgB,KAAK,CAAC,YAAY,CAAC,CAAC;IACtE,MAAM,mBAAmB,GAAG,SAAS,CAAC,GAAG,CACvC,KAAK,CAAC,0BAA0B,CACjC,CAAC;IACF,MAAM,iBAAiB,GAAG,SAAS,CAAC,GAAG,CACrC,KAAK,CAAC,wBAAwB,CAC/B,CAAC;IACF,MAAM,eAAe,GAAG,SAAS,CAAC,GAAG,CACnC,KAAK,CAAC,eAAe,CACtB,CAAC;IACF,MAAM,eAAe,GAAG,SAAS,CAAC,GAAG,CACnC,KAAK,CAAC,eAAe,CACtB,CAAC;IACF,MAAM,mBAAmB,GAAG,SAAS,CAAC,GAAG,CACvC,KAAK,CAAC,mBAAmB,CAC1B,CAAC;IAGF,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IAClD,MAAM,gBAAgB,GAAG,MAAM,mBAAmB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,sBAAsB,GAC1B,mBAAmB,CAAC,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IACpE,MAAM,CAAC,IAAI,CACT,uBAAuB,gBAAgB,CAAC,MAAM,qCAAqC,CACpF,CAAC;IAGF,MAAM,MAAM,GAAG,MAAM,SAAS,CAC5B,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC,uCAAuC,SAAS,EAAE,CAAC,CAAC;QAKhE,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,YAAY,EACZ,aAAa,EACb,MAAM,EACN,mBAAmB,EACnB,iBAAiB,EACjB,sBAAsB,CACvB,CAAC;QAIF,aAAa,CAAC,0BAA0B,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC;YAC/D,MAAM,CAAC,IAAI,CACT,WAAW,SAAS,qDAAqD;gBACvE,YAAY,CAAC,CAAC,YAAY,CAAC,QAAQ,iBAAiB,CAAC,CAAC,YAAY,CAAC,WAAW,EAAE,CACnF,CAAC;YAGF,eAAe,CAAC,eAAe,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAIzD,MAAM,UAAU,GAAG,MAAM,2BAA2B,CAClD,SAAS,EACT,MAAM,EACN,aAAa,EACb,YAAY,CACb,CAAC;YAEF,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CACT,WAAW,SAAS,KAAK,UAAU,CAAC,MAAM,CAAC,MAAM,gCAAgC;oBAC/E,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CACpE,CAAC;YACJ,CAAC;YAED,IACE,UAAU,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;gBAClC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EACzC,CAAC;gBACD,MAAM,CAAC,IAAI,CACT,WAAW,SAAS,6CAA6C,CAClE,CAAC;YACJ,CAAC;YAED,MAAM,CAAC,IAAI,CACT,WAAW,SAAS,KAAK,UAAU,CAAC,UAAU,CAAC,MAAM,mCAAmC,CACzF,CAAC;YAGF,qBAAqB,CACnB,SAAS,EACT,aAAa,EACb,aAAa,EACb,MAAM,EACN,YAAY,CACb,CAAC;QAAA,CACH,CAAC,CAAC;QAEH,OAAO,aAAa,CAAC,SAAS,EAAE,CAAC;IAAA,CAClC,EACD;QACE,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,QAAQ,EAAE;YAER,eAAe,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,WAAW,SAAS,yBAAyB,CAAC,CAAC;gBAC5D,IAAI,CAAC;oBACH,MAAM,aAAa,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;oBAC5C,eAAe,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;gBAChD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAEf,MAAM,CAAC,KAAK,CACV,2BAA2B,SAAS,EAAE,EACtC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAC1D,CAAC;gBACJ,CAAC;YAAA,CACF;SACF;KACF,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,uCAAuC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,EAAE,CACpE,CAAC;IAGF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,eAAe,CAAC,QAAQ,EAAE,CAAC;IAAA,CAClC,CAAC,CAAC;AAAA,CACJ;AAED,KAAK,UAAU,cAAc,CAC3B,SAA8C,EAC9C,MAAoB,EACpB;IACA,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAU,KAAK,CAAC,MAAM,CAAC,CAAC;IACpD,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CACjC,KAAK,CAAC,gBAAgB,CACvB,CAAC;IACF,MAAM,YAAY,GAAG,SAAS,CAAC,GAAG,CAAgB,KAAK,CAAC,YAAY,CAAC,CAAC;IACtE,MAAM,mBAAmB,GAAG,SAAS,CAAC,GAAG,CACvC,KAAK,CAAC,0BAA0B,CACjC,CAAC;IACF,MAAM,iBAAiB,GAAG,SAAS,CAAC,GAAG,CACrC,KAAK,CAAC,wBAAwB,CAC/B,CAAC;IACF,MAAM,eAAe,GAAG,SAAS,CAAC,GAAG,CACnC,KAAK,CAAC,eAAe,CACtB,CAAC;IACF,MAAM,mBAAmB,GAAG,SAAS,CAAC,GAAG,CACvC,KAAK,CAAC,mBAAmB,CAC1B,CAAC;IAGF,MAAM,UAAU,GAAG,SAAS,CAAC;IAG7B,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IAClD,MAAM,gBAAgB,GAAG,MAAM,mBAAmB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,sBAAsB,GAC1B,mBAAmB,CAAC,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IACpE,MAAM,CAAC,IAAI,CACT,uBAAuB,gBAAgB,CAAC,MAAM,qCAAqC,CACpF,CAAC;IAGF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,GAAG,EAAE,CAAC;QAGpC,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,YAAY,EACZ,aAAa,EACb,MAAM,EACN,mBAAmB,EACnB,iBAAiB,EACjB,sBAAsB,CACvB,CAAC;QAGF,aAAa,CAAC,0BAA0B,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC;YAC/D,MAAM,CAAC,IAAI,CACT,mDAAmD;gBACjD,YAAY,CAAC,CAAC,YAAY,CAAC,QAAQ,iBAAiB,CAAC,CAAC,YAAY,CAAC,WAAW,EAAE,CACnF,CAAC;YAGF,eAAe,CAAC,eAAe,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YAG1D,MAAM,UAAU,GAAG,MAAM,2BAA2B,CAClD,UAAU,EACV,MAAM,EACN,aAAa,EACb,YAAY,CACb,CAAC;YAEF,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CACT,GAAG,UAAU,CAAC,MAAM,CAAC,MAAM,gCAAgC;oBACzD,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CACpE,CAAC;YACJ,CAAC;YAED,IACE,UAAU,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;gBAClC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EACzC,CAAC;gBACD,MAAM,CAAC,IAAI,CACT,qFAAqF,CACtF,CAAC;YACJ,CAAC;YAED,MAAM,CAAC,IAAI,CACT,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,mCAAmC,CACnE,CAAC;YAGF,qBAAqB,CACnB,UAAU,EACV,aAAa,EACb,aAAa,EACb,MAAM,EACN,YAAY,CACb,CAAC;QAAA,CACH,CAAC,CAAC;QAEH,OAAO,aAAa,CAAC,SAAS,EAAE,CAAC;IAAA,CAClC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;IAGrD,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,aAAa,CAAC,KAAK,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAAA,CACjB,CAAC,CAAC;AAAA,CACJ;AAUD,SAAS,SAAS,GAAS;IACzB,OAAO,CAAC,GAAG,CACT,uEAAuE,CACxE,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACxB,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;IAC7E,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IACxE,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IACtC,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;IACpE,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,CAAC;IACzE,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;AAAA,CAChE;AAQD,SAAS,gBAAgB,GAAS;IAChC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC3C,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;IAC/B,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC;QACrD,MAAM,KAAK,GAAG,CAAC,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,iBAAiB,CAAC;QAC1E,OAAO,CAAC,GAAG,CAAC,KAAK,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC;QACpC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,oBAAoB,EAAE,EAAE,CAAC,CAAC;AAAA,CACrE;AAED,KAAK,UAAU,IAAI,GAAG;IAEpB,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAE9C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,SAAS,EAAE,CAAC;QACZ,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,gBAAgB,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAGD,MAAM,MAAM,GAAG,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC;IAC5C,MAAM,SAAS,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IAG1C,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;QACjC,MAAM,cAAc,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;SAAM,CAAC;QAEN,MAAM,aAAa,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACzC,CAAC;AAAA,CACF;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC"}

package/dist/lua/runtime-infinite-loop-worker.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=runtime-infinite-loop-worker.d.ts.map

package/dist/lua/runtime-infinite-loop-worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-infinite-loop-worker.d.ts","sourceRoot":"","sources":["../../src/lua/runtime-infinite-loop-worker.ts"],"names":[],"mappings":""}

package/dist/lua/runtime-infinite-loop-worker.js

@@ -0,0 +1,33 @@
+import { parentPort } from "node:worker_threads";
+import { WasmoonRuntime } from "./runtime.js";
+const logger = {
+    info: (msg) => console.log(`[Worker] ${msg}`),
+    error: (msg, err) => console.error(`[Worker] ${msg}`, err || ""),
+    debug: () => {
+    },
+};
+async function runInfiniteLoopTest() {
+    const runtime = new WasmoonRuntime(logger);
+    const infiniteLoopScript = `
+    while true do
+      -- This should hang forever
+    end
+  `;
+    try {
+        await runtime.executeScript(infiniteLoopScript, new Map());
+        parentPort?.postMessage({ type: "completed" });
+    }
+    catch (error) {
+        parentPort?.postMessage({
+            type: "error",
+            message: error.message,
+        });
+    }
+}
+runInfiniteLoopTest().catch((error) => {
+    parentPort?.postMessage({
+        type: "error",
+        message: error.message,
+    });
+});
+//# sourceMappingURL=runtime-infinite-loop-worker.js.map

package/dist/lua/runtime-infinite-loop-worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-infinite-loop-worker.js","sourceRoot":"","sources":["../../src/lua/runtime-infinite-loop-worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAG9C,MAAM,MAAM,GAAG;IACb,IAAI,EAAE,CAAC,GAAW,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,EAAE,CAAC;IACrD,KAAK,EAAE,CAAC,GAAW,EAAE,GAAW,EAAE,EAAE,CAClC,OAAO,CAAC,KAAK,CAAC,YAAY,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;IAC7C,KAAK,EAAE,GAAG,EAAE,CAAC;IAAC,CAEb;CACF,CAAC;AAEF,KAAK,UAAU,mBAAmB,GAAG;IACnC,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;IAE3C,MAAM,kBAAkB,GAAG;;;;GAI1B,CAAC;IAEF,IAAI,CAAC;QAEH,MAAM,OAAO,CAAC,aAAa,CAAC,kBAAkB,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QAC3D,UAAU,EAAE,WAAW,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;IACjD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,UAAU,EAAE,WAAW,CAAC;YACtB,IAAI,EAAE,OAAO;YACb,OAAO,EAAG,KAAe,CAAC,OAAO;SAClC,CAAC,CAAC;IACL,CAAC;AAAA,CACF;AAGD,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;IACrC,UAAU,EAAE,WAAW,CAAC;QACtB,IAAI,EAAE,OAAO;QACb,OAAO,EAAG,KAAe,CAAC,OAAO;KAClC,CAAC,CAAC;AAAA,CACJ,CAAC,CAAC"}

package/dist/lua/runtime.d.ts

@@ -0,0 +1,11 @@
+import type { ILuaRuntime, ILogger } from "../types/interfaces.js";
+import type { MCPClientSession } from "../mcp/client-session.js";
+export declare class WasmoonRuntime implements ILuaRuntime {
+    private logger;
+    private factory;
+    constructor(logger: ILogger);
+    executeScript(script: string, mcpServers: Map<string, MCPClientSession>): Promise<unknown>;
+    private createEngine;
+    private injectMCPServers;
+}
+//# sourceMappingURL=runtime.d.ts.map

package/dist/lua/runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../../src/lua/runtime.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAQnE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAQjE,qBACa,cAAe,YAAW,WAAW;IAGb,OAAO,CAAC,MAAM;IAFjD,OAAO,CAAC,OAAO,CAAa;IAE5B,YAA2C,MAAM,EAAE,OAAO,EAEzD;IAEK,aAAa,CACjB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,GACxC,OAAO,CAAC,OAAO,CAAC,CAqClB;YAEa,YAAY;YA4BZ,gBAAgB;CAsG/B"}