@cloudcli-ai/cloudcli 1.29.5 → 1.31.0

package/dist-server/server/modules/providers/tests/mcp.test.js

@@ -0,0 +1,250 @@
+import assert from 'node:assert/strict';
+import fs from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import TOML from '@iarna/toml';
+import { providerMcpService } from '../../../modules/providers/services/mcp.service.js';
+import { AppError } from '../../../shared/utils.js';
+const patchHomeDir = (nextHomeDir) => {
+    const original = os.homedir;
+    os.homedir = () => nextHomeDir;
+    return () => {
+        os.homedir = original;
+    };
+};
+const readJson = async (filePath) => {
+    const content = await fs.readFile(filePath, 'utf8');
+    return JSON.parse(content);
+};
+/**
+ * This test covers Claude MCP support for all scopes (user/local/project) and all transports (stdio/http/sse),
+ * including add, update/list, and remove operations.
+ */
+test('providerMcpService handles claude MCP scopes/transports with file-backed persistence', { concurrency: false }, async () => {
+    const tempRoot = await fs.mkdtemp(path.join(os.tmpdir(), 'llm-mcp-claude-'));
+    const workspacePath = path.join(tempRoot, 'workspace');
+    await fs.mkdir(workspacePath, { recursive: true });
+    const restoreHomeDir = patchHomeDir(tempRoot);
+    try {
+        await providerMcpService.upsertProviderMcpServer('claude', {
+            name: 'claude-user-stdio',
+            scope: 'user',
+            transport: 'stdio',
+            command: 'npx',
+            args: ['-y', 'my-server'],
+            env: { API_KEY: 'secret' },
+        });
+        await providerMcpService.upsertProviderMcpServer('claude', {
+            name: 'claude-local-http',
+            scope: 'local',
+            transport: 'http',
+            url: 'https://example.com/mcp',
+            headers: { Authorization: 'Bearer token' },
+            workspacePath,
+        });
+        await providerMcpService.upsertProviderMcpServer('claude', {
+            name: 'claude-project-sse',
+            scope: 'project',
+            transport: 'sse',
+            url: 'https://example.com/sse',
+            headers: { 'X-API-Key': 'abc' },
+            workspacePath,
+        });
+        const grouped = await providerMcpService.listProviderMcpServers('claude', { workspacePath });
+        assert.ok(grouped.user.some((server) => server.name === 'claude-user-stdio' && server.transport === 'stdio'));
+        assert.ok(grouped.local.some((server) => server.name === 'claude-local-http' && server.transport === 'http'));
+        assert.ok(grouped.project.some((server) => server.name === 'claude-project-sse' && server.transport === 'sse'));
+        // update behavior is the same upsert route with same name
+        await providerMcpService.upsertProviderMcpServer('claude', {
+            name: 'claude-project-sse',
+            scope: 'project',
+            transport: 'sse',
+            url: 'https://example.com/sse-updated',
+            headers: { 'X-API-Key': 'updated' },
+            workspacePath,
+        });
+        const projectConfig = await readJson(path.join(workspacePath, '.mcp.json'));
+        const projectServers = projectConfig.mcpServers;
+        const projectServer = projectServers['claude-project-sse'];
+        assert.equal(projectServer.url, 'https://example.com/sse-updated');
+        const removeResult = await providerMcpService.removeProviderMcpServer('claude', {
+            name: 'claude-local-http',
+            scope: 'local',
+            workspacePath,
+        });
+        assert.equal(removeResult.removed, true);
+    }
+    finally {
+        restoreHomeDir();
+        await fs.rm(tempRoot, { recursive: true, force: true });
+    }
+});
+/**
+ * This test covers Codex MCP support for user/project scopes, stdio/http formats,
+ * and validation for unsupported scope/transport combinations.
+ */
+test('providerMcpService handles codex MCP TOML config and capability validation', { concurrency: false }, async () => {
+    const tempRoot = await fs.mkdtemp(path.join(os.tmpdir(), 'llm-mcp-codex-'));
+    const workspacePath = path.join(tempRoot, 'workspace');
+    await fs.mkdir(workspacePath, { recursive: true });
+    const restoreHomeDir = patchHomeDir(tempRoot);
+    try {
+        await providerMcpService.upsertProviderMcpServer('codex', {
+            name: 'codex-user-stdio',
+            scope: 'user',
+            transport: 'stdio',
+            command: 'python',
+            args: ['server.py'],
+            env: { API_KEY: 'x' },
+            envVars: ['API_KEY'],
+            cwd: '/tmp',
+        });
+        await providerMcpService.upsertProviderMcpServer('codex', {
+            name: 'codex-project-http',
+            scope: 'project',
+            transport: 'http',
+            url: 'https://codex.example.com/mcp',
+            headers: { 'X-Custom-Header': 'value' },
+            envHttpHeaders: { 'X-API-Key': 'MY_API_KEY_ENV' },
+            bearerTokenEnvVar: 'MY_API_TOKEN',
+            workspacePath,
+        });
+        const userTomlPath = path.join(tempRoot, '.codex', 'config.toml');
+        const userConfig = TOML.parse(await fs.readFile(userTomlPath, 'utf8'));
+        const userServers = userConfig.mcp_servers;
+        const userStdio = userServers['codex-user-stdio'];
+        assert.equal(userStdio.command, 'python');
+        const projectTomlPath = path.join(workspacePath, '.codex', 'config.toml');
+        const projectConfig = TOML.parse(await fs.readFile(projectTomlPath, 'utf8'));
+        const projectServers = projectConfig.mcp_servers;
+        const projectHttp = projectServers['codex-project-http'];
+        assert.equal(projectHttp.url, 'https://codex.example.com/mcp');
+        await assert.rejects(providerMcpService.upsertProviderMcpServer('codex', {
+            name: 'codex-local',
+            scope: 'local',
+            transport: 'stdio',
+            command: 'node',
+        }), (error) => error instanceof AppError &&
+            error.code === 'MCP_SCOPE_NOT_SUPPORTED' &&
+            error.statusCode === 400);
+        await assert.rejects(providerMcpService.upsertProviderMcpServer('codex', {
+            name: 'codex-sse',
+            scope: 'project',
+            transport: 'sse',
+            url: 'https://example.com/sse',
+            workspacePath,
+        }), (error) => error instanceof AppError &&
+            error.code === 'MCP_TRANSPORT_NOT_SUPPORTED' &&
+            error.statusCode === 400);
+    }
+    finally {
+        restoreHomeDir();
+        await fs.rm(tempRoot, { recursive: true, force: true });
+    }
+});
+/**
+ * This test covers Gemini/Cursor MCP JSON formats and user/project scope persistence.
+ */
+test('providerMcpService handles gemini and cursor MCP JSON config formats', { concurrency: false }, async () => {
+    const tempRoot = await fs.mkdtemp(path.join(os.tmpdir(), 'llm-mcp-gc-'));
+    const workspacePath = path.join(tempRoot, 'workspace');
+    await fs.mkdir(workspacePath, { recursive: true });
+    const restoreHomeDir = patchHomeDir(tempRoot);
+    try {
+        await providerMcpService.upsertProviderMcpServer('gemini', {
+            name: 'gemini-stdio',
+            scope: 'user',
+            transport: 'stdio',
+            command: 'node',
+            args: ['server.js'],
+            env: { TOKEN: '$TOKEN' },
+            cwd: './server',
+        });
+        await providerMcpService.upsertProviderMcpServer('gemini', {
+            name: 'gemini-http',
+            scope: 'project',
+            transport: 'http',
+            url: 'https://gemini.example.com/mcp',
+            headers: { Authorization: 'Bearer token' },
+            workspacePath,
+        });
+        await providerMcpService.upsertProviderMcpServer('cursor', {
+            name: 'cursor-stdio',
+            scope: 'project',
+            transport: 'stdio',
+            command: 'npx',
+            args: ['-y', 'mcp-server'],
+            env: { API_KEY: 'value' },
+            workspacePath,
+        });
+        await providerMcpService.upsertProviderMcpServer('cursor', {
+            name: 'cursor-http',
+            scope: 'user',
+            transport: 'http',
+            url: 'http://localhost:3333/mcp',
+            headers: { API_KEY: 'value' },
+        });
+        const geminiUserConfig = await readJson(path.join(tempRoot, '.gemini', 'settings.json'));
+        const geminiUserServer = geminiUserConfig.mcpServers['gemini-stdio'];
+        assert.equal(geminiUserServer.command, 'node');
+        assert.equal(geminiUserServer.type, undefined);
+        const geminiProjectConfig = await readJson(path.join(workspacePath, '.gemini', 'settings.json'));
+        const geminiProjectServer = geminiProjectConfig.mcpServers['gemini-http'];
+        assert.equal(geminiProjectServer.type, 'http');
+        const cursorUserConfig = await readJson(path.join(tempRoot, '.cursor', 'mcp.json'));
+        const cursorHttpServer = cursorUserConfig.mcpServers['cursor-http'];
+        assert.equal(cursorHttpServer.url, 'http://localhost:3333/mcp');
+        assert.equal(cursorHttpServer.type, undefined);
+    }
+    finally {
+        restoreHomeDir();
+        await fs.rm(tempRoot, { recursive: true, force: true });
+    }
+});
+/**
+ * This test covers the global MCP adder requirement: only http/stdio are allowed and
+ * one payload is written to all providers.
+ */
+test('providerMcpService global adder writes to all providers and rejects unsupported transports', { concurrency: false }, async () => {
+    const tempRoot = await fs.mkdtemp(path.join(os.tmpdir(), 'llm-mcp-global-'));
+    const workspacePath = path.join(tempRoot, 'workspace');
+    await fs.mkdir(workspacePath, { recursive: true });
+    const restoreHomeDir = patchHomeDir(tempRoot);
+    try {
+        const globalResult = await providerMcpService.addMcpServerToAllProviders({
+            name: 'global-http',
+            scope: 'project',
+            transport: 'http',
+            url: 'https://global.example.com/mcp',
+            workspacePath,
+        });
+        const expectCursorGlobal = process.platform !== 'win32';
+        assert.equal(globalResult.length, expectCursorGlobal ? 4 : 3);
+        assert.ok(globalResult.every((entry) => entry.created === true));
+        const claudeProject = await readJson(path.join(workspacePath, '.mcp.json'));
+        assert.ok(claudeProject.mcpServers['global-http']);
+        const codexProject = TOML.parse(await fs.readFile(path.join(workspacePath, '.codex', 'config.toml'), 'utf8'));
+        assert.ok(codexProject.mcp_servers['global-http']);
+        const geminiProject = await readJson(path.join(workspacePath, '.gemini', 'settings.json'));
+        assert.ok(geminiProject.mcpServers['global-http']);
+        if (expectCursorGlobal) {
+            const cursorProject = await readJson(path.join(workspacePath, '.cursor', 'mcp.json'));
+            assert.ok(cursorProject.mcpServers['global-http']);
+        }
+        await assert.rejects(providerMcpService.addMcpServerToAllProviders({
+            name: 'global-sse',
+            scope: 'project',
+            transport: 'sse',
+            url: 'https://example.com/sse',
+            workspacePath,
+        }), (error) => error instanceof AppError &&
+            error.code === 'INVALID_GLOBAL_MCP_TRANSPORT' &&
+            error.statusCode === 400);
+    }
+    finally {
+        restoreHomeDir();
+        await fs.rm(tempRoot, { recursive: true, force: true });
+    }
+});
+//# sourceMappingURL=mcp.test.js.map

package/dist-server/server/modules/providers/tests/mcp.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.test.js","sourceRoot":"","sources":["../../../../../server/modules/providers/tests/mcp.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,IAAI,MAAM,aAAa,CAAC;AAE/B,OAAO,EAAE,kBAAkB,EAAE,MAAM,6CAA6C,CAAC;AACjF,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,MAAM,YAAY,GAAG,CAAC,WAAmB,EAAE,EAAE;IAC3C,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC;IAC3B,EAAU,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;IACxC,OAAO,GAAG,EAAE;QACT,EAAU,CAAC,OAAO,GAAG,QAAQ,CAAC;IACjC,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAgB,EAAoC,EAAE;IAC5E,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACpD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;AACxD,CAAC,CAAC;AAEF;;;GAGG;AACH,IAAI,CAAC,sFAAsF,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,IAAI,EAAE;IAC9H,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAC7E,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACvD,MAAM,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEnD,MAAM,cAAc,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,CAAC;QACH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,IAAI,EAAE,WAAW,CAAC;YACzB,GAAG,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;SAC3B,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,OAAO;YACd,SAAS,EAAE,MAAM;YACjB,GAAG,EAAE,yBAAyB;YAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,cAAc,EAAE;YAC1C,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,oBAAoB;YAC1B,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,yBAAyB;YAC9B,OAAO,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE;YAC/B,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,sBAAsB,CAAC,QAAQ,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;QAC7F,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,mBAAmB,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,CAAC,CAAC,CAAC;QAC9G,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,mBAAmB,IAAI,MAAM,CAAC,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC;QAC9G,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,oBAAoB,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC;QAEhH,0DAA0D;QAC1D,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,oBAAoB;YAC1B,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,iCAAiC;YACtC,OAAO,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE;YACnC,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC,CAAC;QAC5E,MAAM,cAAc,GAAG,aAAa,CAAC,UAAqC,CAAC;QAC3E,MAAM,aAAa,GAAG,cAAc,CAAC,oBAAoB,CAA4B,CAAC;QACtF,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,EAAE,iCAAiC,CAAC,CAAC;QAEnE,MAAM,YAAY,GAAG,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YAC9E,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,OAAO;YACd,aAAa;SACd,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;YAAS,CAAC;QACT,cAAc,EAAE,CAAC;QACjB,MAAM,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH;;;GAGG;AACH,IAAI,CAAC,4EAA4E,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,IAAI,EAAE;IACpH,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAC5E,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACvD,MAAM,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEnD,MAAM,cAAc,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,CAAC;QACH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,OAAO,EAAE;YACxD,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,QAAQ;YACjB,IAAI,EAAE,CAAC,WAAW,CAAC;YACnB,GAAG,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE;YACrB,OAAO,EAAE,CAAC,SAAS,CAAC;YACpB,GAAG,EAAE,MAAM;SACZ,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,OAAO,EAAE;YACxD,IAAI,EAAE,oBAAoB;YAC1B,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,MAAM;YACjB,GAAG,EAAE,+BAA+B;YACpC,OAAO,EAAE,EAAE,iBAAiB,EAAE,OAAO,EAAE;YACvC,cAAc,EAAE,EAAE,WAAW,EAAE,gBAAgB,EAAE;YACjD,iBAAiB,EAAE,cAAc;YACjC,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;QAClE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAA4B,CAAC;QAClG,MAAM,WAAW,GAAG,UAAU,CAAC,WAAsC,CAAC;QACtE,MAAM,SAAS,GAAG,WAAW,CAAC,kBAAkB,CAA4B,CAAC;QAC7E,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAE1C,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;QAC1E,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC,CAA4B,CAAC;QACxG,MAAM,cAAc,GAAG,aAAa,CAAC,WAAsC,CAAC;QAC5E,MAAM,WAAW,GAAG,cAAc,CAAC,oBAAoB,CAA4B,CAAC;QACpF,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,+BAA+B,CAAC,CAAC;QAE/D,MAAM,MAAM,CAAC,OAAO,CAClB,kBAAkB,CAAC,uBAAuB,CAAC,OAAO,EAAE;YAClD,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,OAAO;YACd,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,MAAM;SAChB,CAAC,EACF,CAAC,KAAc,EAAE,EAAE,CACjB,KAAK,YAAY,QAAQ;YACzB,KAAK,CAAC,IAAI,KAAK,yBAAyB;YACxC,KAAK,CAAC,UAAU,KAAK,GAAG,CAC3B,CAAC;QAEF,MAAM,MAAM,CAAC,OAAO,CAClB,kBAAkB,CAAC,uBAAuB,CAAC,OAAO,EAAE;YAClD,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,yBAAyB;YAC9B,aAAa;SACd,CAAC,EACF,CAAC,KAAc,EAAE,EAAE,CACjB,KAAK,YAAY,QAAQ;YACzB,KAAK,CAAC,IAAI,KAAK,6BAA6B;YAC5C,KAAK,CAAC,UAAU,KAAK,GAAG,CAC3B,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,cAAc,EAAE,CAAC;QACjB,MAAM,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH;;GAEG;AACH,IAAI,CAAC,sEAAsE,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,IAAI,EAAE;IAC9G,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;IACzE,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACvD,MAAM,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEnD,MAAM,cAAc,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,CAAC;QACH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,cAAc;YACpB,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,CAAC,WAAW,CAAC;YACnB,GAAG,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;YACxB,GAAG,EAAE,UAAU;SAChB,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,MAAM;YACjB,GAAG,EAAE,gCAAgC;YACrC,OAAO,EAAE,EAAE,aAAa,EAAE,cAAc,EAAE;YAC1C,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,cAAc;YACpB,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,IAAI,EAAE,YAAY,CAAC;YAC1B,GAAG,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;YACzB,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,uBAAuB,CAAC,QAAQ,EAAE;YACzD,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,MAAM;YACjB,GAAG,EAAE,2BAA2B;YAChC,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC9B,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC;QACzF,MAAM,gBAAgB,GAAI,gBAAgB,CAAC,UAAsC,CAAC,cAAc,CAA4B,CAAC;QAC7H,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAE/C,MAAM,mBAAmB,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC;QACjG,MAAM,mBAAmB,GAAI,mBAAmB,CAAC,UAAsC,CAAC,aAAa,CAA4B,CAAC;QAClI,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAE/C,MAAM,gBAAgB,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;QACpF,MAAM,gBAAgB,GAAI,gBAAgB,CAAC,UAAsC,CAAC,aAAa,CAA4B,CAAC;QAC5H,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAAG,EAAE,2BAA2B,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC;YAAS,CAAC;QACT,cAAc,EAAE,CAAC;QACjB,MAAM,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH;;;GAGG;AACH,IAAI,CAAC,4FAA4F,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,KAAK,IAAI,EAAE;IACpI,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAC7E,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACvD,MAAM,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEnD,MAAM,cAAc,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,MAAM,kBAAkB,CAAC,0BAA0B,CAAC;YACvE,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,MAAM;YACjB,GAAG,EAAE,gCAAgC;YACrC,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,kBAAkB,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC;QAEjE,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC,CAAC;QAC5E,MAAM,CAAC,EAAE,CAAE,aAAa,CAAC,UAAsC,CAAC,aAAa,CAAC,CAAC,CAAC;QAEhF,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,CAA4B,CAAC;QACzI,MAAM,CAAC,EAAE,CAAE,YAAY,CAAC,WAAuC,CAAC,aAAa,CAAC,CAAC,CAAC;QAEhF,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC;QAC3F,MAAM,CAAC,EAAE,CAAE,aAAa,CAAC,UAAsC,CAAC,aAAa,CAAC,CAAC,CAAC;QAEhF,IAAI,kBAAkB,EAAE,CAAC;YACvB,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;YACtF,MAAM,CAAC,EAAE,CAAE,aAAa,CAAC,UAAsC,CAAC,aAAa,CAAC,CAAC,CAAC;QAClF,CAAC;QAED,MAAM,MAAM,CAAC,OAAO,CAClB,kBAAkB,CAAC,0BAA0B,CAAC;YAC5C,IAAI,EAAE,YAAY;YAClB,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,yBAAyB;YAC9B,aAAa;SACd,CAAC,EACF,CAAC,KAAc,EAAE,EAAE,CACjB,KAAK,YAAY,QAAQ;YACzB,KAAK,CAAC,IAAI,KAAK,8BAA8B;YAC7C,KAAK,CAAC,UAAU,KAAK,GAAG,CAC3B,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,cAAc,EAAE,CAAC;QACjB,MAAM,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist-server/server/modules/websocket/index.js

@@ -0,0 +1,3 @@
+export { WS_OPEN_STATE, connectedClients } from './services/websocket-state.service.js';
+export { createWebSocketServer } from './services/websocket-server.service.js';
+//# sourceMappingURL=index.js.map

package/dist-server/server/modules/websocket/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../server/modules/websocket/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,uCAAuC,CAAC;AACxF,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC"}

package/dist-server/server/modules/websocket/services/chat-websocket.service.js

@@ -0,0 +1,188 @@
+import { connectedClients } from '../../../modules/websocket/services/websocket-state.service.js';
+import { WebSocketWriter } from '../../../modules/websocket/services/websocket-writer.service.js';
+import { createNormalizedMessage, parseIncomingJsonObject } from '../../../shared/utils.js';
+const DEFAULT_PROVIDER = 'claude';
+/**
+ * Normalizes potentially invalid provider names coming from websocket payloads.
+ */
+function readProvider(value) {
+    if (value === 'claude' || value === 'cursor' || value === 'codex' || value === 'gemini') {
+        return value;
+    }
+    return DEFAULT_PROVIDER;
+}
+/**
+ * Extracts the authenticated request user id in the formats currently produced
+ * by platform and OSS auth code paths.
+ */
+function readRequestUserId(request) {
+    const user = request?.user;
+    if (!user) {
+        return null;
+    }
+    if (typeof user.id === 'string' || typeof user.id === 'number') {
+        return user.id;
+    }
+    if (typeof user.userId === 'string' || typeof user.userId === 'number') {
+        return user.userId;
+    }
+    return null;
+}
+/**
+ * Handles authenticated chat websocket messages used by the main chat panel.
+ */
+export function handleChatConnection(ws, request, dependencies) {
+    console.log('[INFO] Chat WebSocket connected');
+    connectedClients.add(ws);
+    const writer = new WebSocketWriter(ws, readRequestUserId(request));
+    ws.on('message', async (rawMessage) => {
+        try {
+            const parsed = parseIncomingJsonObject(rawMessage);
+            if (!parsed) {
+                throw new Error('Invalid websocket payload');
+            }
+            const data = parsed;
+            const messageType = data.type;
+            if (!messageType) {
+                throw new Error('Message type is required');
+            }
+            if (messageType === 'claude-command') {
+                await dependencies.queryClaudeSDK(data.command ?? '', data.options, writer);
+                return;
+            }
+            if (messageType === 'cursor-command') {
+                await dependencies.spawnCursor(data.command ?? '', data.options, writer);
+                return;
+            }
+            if (messageType === 'codex-command') {
+                await dependencies.queryCodex(data.command ?? '', data.options, writer);
+                return;
+            }
+            if (messageType === 'gemini-command') {
+                await dependencies.spawnGemini(data.command ?? '', data.options, writer);
+                return;
+            }
+            if (messageType === 'cursor-resume') {
+                await dependencies.spawnCursor('', {
+                    sessionId: data.sessionId,
+                    resume: true,
+                    cwd: data.options?.cwd,
+                }, writer);
+                return;
+            }
+            if (messageType === 'abort-session') {
+                const provider = readProvider(data.provider);
+                const sessionId = typeof data.sessionId === 'string' ? data.sessionId : '';
+                let success = false;
+                if (provider === 'cursor') {
+                    success = dependencies.abortCursorSession(sessionId);
+                }
+                else if (provider === 'codex') {
+                    success = dependencies.abortCodexSession(sessionId);
+                }
+                else if (provider === 'gemini') {
+                    success = dependencies.abortGeminiSession(sessionId);
+                }
+                else {
+                    success = await dependencies.abortClaudeSDKSession(sessionId);
+                }
+                writer.send(createNormalizedMessage({
+                    kind: 'complete',
+                    exitCode: success ? 0 : 1,
+                    aborted: true,
+                    success,
+                    sessionId,
+                    provider,
+                }));
+                return;
+            }
+            if (messageType === 'claude-permission-response') {
+                if (typeof data.requestId === 'string' && data.requestId.length > 0) {
+                    dependencies.resolveToolApproval(data.requestId, {
+                        allow: Boolean(data.allow),
+                        updatedInput: data.updatedInput,
+                        message: typeof data.message === 'string' ? data.message : undefined,
+                        rememberEntry: data.rememberEntry,
+                    });
+                }
+                return;
+            }
+            if (messageType === 'cursor-abort') {
+                const sessionId = typeof data.sessionId === 'string' ? data.sessionId : '';
+                const success = dependencies.abortCursorSession(sessionId);
+                writer.send(createNormalizedMessage({
+                    kind: 'complete',
+                    exitCode: success ? 0 : 1,
+                    aborted: true,
+                    success,
+                    sessionId,
+                    provider: 'cursor',
+                }));
+                return;
+            }
+            if (messageType === 'check-session-status') {
+                const provider = readProvider(data.provider);
+                const sessionId = typeof data.sessionId === 'string' ? data.sessionId : '';
+                let isActive = false;
+                if (provider === 'cursor') {
+                    isActive = dependencies.isCursorSessionActive(sessionId);
+                }
+                else if (provider === 'codex') {
+                    isActive = dependencies.isCodexSessionActive(sessionId);
+                }
+                else if (provider === 'gemini') {
+                    isActive = dependencies.isGeminiSessionActive(sessionId);
+                }
+                else {
+                    isActive = dependencies.isClaudeSDKSessionActive(sessionId);
+                    if (isActive) {
+                        dependencies.reconnectSessionWriter(sessionId, ws);
+                    }
+                }
+                writer.send({
+                    type: 'session-status',
+                    sessionId,
+                    provider,
+                    isProcessing: isActive,
+                });
+                return;
+            }
+            if (messageType === 'get-pending-permissions') {
+                const sessionId = typeof data.sessionId === 'string' ? data.sessionId : '';
+                if (sessionId && dependencies.isClaudeSDKSessionActive(sessionId)) {
+                    const pending = dependencies.getPendingApprovalsForSession(sessionId);
+                    writer.send({
+                        type: 'pending-permissions-response',
+                        sessionId,
+                        data: pending,
+                    });
+                }
+                return;
+            }
+            if (messageType === 'get-active-sessions') {
+                writer.send({
+                    type: 'active-sessions',
+                    sessions: {
+                        claude: dependencies.getActiveClaudeSDKSessions(),
+                        cursor: dependencies.getActiveCursorSessions(),
+                        codex: dependencies.getActiveCodexSessions(),
+                        gemini: dependencies.getActiveGeminiSessions(),
+                    },
+                });
+            }
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            console.error('[ERROR] Chat WebSocket error:', message);
+            writer.send({
+                type: 'error',
+                error: message,
+            });
+        }
+    });
+    ws.on('close', () => {
+        console.log('[INFO] Chat client disconnected');
+        connectedClients.delete(ws);
+    });
+}
+//# sourceMappingURL=chat-websocket.service.js.map

package/dist-server/server/modules/websocket/services/chat-websocket.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-websocket.service.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/services/chat-websocket.service.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,yDAAyD,CAAC;AAC3F,OAAO,EAAE,eAAe,EAAE,MAAM,0DAA0D,CAAC;AAM3F,OAAO,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAerF,MAAM,gBAAgB,GAAgB,QAAQ,CAAC;AAgC/C;;GAEG;AACH,SAAS,YAAY,CAAC,KAAc;IAClC,IAAI,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,OAAO,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxF,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CACxB,OAAkD;IAElD,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,CAAC;IAC3B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ,IAAI,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;QAC/D,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACvE,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,EAAa,EACb,OAAsC,EACtC,YAAuC;IAEvC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IAC/C,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEzB,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,EAAE,EAAE,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC;IAEnE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;QACpC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACnD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YAED,MAAM,IAAI,GAAG,MAA6B,CAAC;YAC3C,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC;YAC9B,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC9C,CAAC;YAED,IAAI,WAAW,KAAK,gBAAgB,EAAE,CAAC;gBACrC,MAAM,YAAY,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC5E,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,gBAAgB,EAAE,CAAC;gBACrC,MAAM,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACzE,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,eAAe,EAAE,CAAC;gBACpC,MAAM,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACxE,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,gBAAgB,EAAE,CAAC;gBACrC,MAAM,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACzE,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,eAAe,EAAE,CAAC;gBACpC,MAAM,YAAY,CAAC,WAAW,CAC5B,EAAE,EACF;oBACE,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,MAAM,EAAE,IAAI;oBACZ,GAAG,EAAE,IAAI,CAAC,OAAO,EAAE,GAAG;iBACvB,EACD,MAAM,CACP,CAAC;gBACF,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,eAAe,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC7C,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3E,IAAI,OAAO,GAAG,KAAK,CAAC;gBAEpB,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBAC1B,OAAO,GAAG,YAAY,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;gBACvD,CAAC;qBAAM,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;oBAChC,OAAO,GAAG,YAAY,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBACtD,CAAC;qBAAM,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACjC,OAAO,GAAG,YAAY,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;gBACvD,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG,MAAM,YAAY,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;gBAChE,CAAC;gBAED,MAAM,CAAC,IAAI,CACT,uBAAuB,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBACzB,OAAO,EAAE,IAAI;oBACb,OAAO;oBACP,SAAS;oBACT,QAAQ;iBACT,CAAC,CACH,CAAC;gBACF,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,4BAA4B,EAAE,CAAC;gBACjD,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpE,YAAY,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,EAAE;wBAC/C,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;wBAC1B,YAAY,EAAE,IAAI,CAAC,YAAY;wBAC/B,OAAO,EAAE,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;wBACpE,aAAa,EAAE,IAAI,CAAC,aAAa;qBAClC,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,cAAc,EAAE,CAAC;gBACnC,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3E,MAAM,OAAO,GAAG,YAAY,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;gBAC3D,MAAM,CAAC,IAAI,CACT,uBAAuB,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBACzB,OAAO,EAAE,IAAI;oBACb,OAAO;oBACP,SAAS;oBACT,QAAQ,EAAE,QAAQ;iBACnB,CAAC,CACH,CAAC;gBACF,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,sBAAsB,EAAE,CAAC;gBAC3C,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC7C,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3E,IAAI,QAAQ,GAAG,KAAK,CAAC;gBAErB,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBAC1B,QAAQ,GAAG,YAAY,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;gBAC3D,CAAC;qBAAM,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;oBAChC,QAAQ,GAAG,YAAY,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;gBAC1D,CAAC;qBAAM,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACjC,QAAQ,GAAG,YAAY,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;gBAC3D,CAAC;qBAAM,CAAC;oBACN,QAAQ,GAAG,YAAY,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;oBAC5D,IAAI,QAAQ,EAAE,CAAC;wBACb,YAAY,CAAC,sBAAsB,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;oBACrD,CAAC;gBACH,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,gBAAgB;oBACtB,SAAS;oBACT,QAAQ;oBACR,YAAY,EAAE,QAAQ;iBACvB,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,yBAAyB,EAAE,CAAC;gBAC9C,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3E,IAAI,SAAS,IAAI,YAAY,CAAC,wBAAwB,CAAC,SAAS,CAAC,EAAE,CAAC;oBAClE,MAAM,OAAO,GAAG,YAAY,CAAC,6BAA6B,CAAC,SAAS,CAAC,CAAC;oBACtE,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,8BAA8B;wBACpC,SAAS;wBACT,IAAI,EAAE,OAAO;qBACd,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO;YACT,CAAC;YAED,IAAI,WAAW,KAAK,qBAAqB,EAAE,CAAC;gBAC1C,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,iBAAiB;oBACvB,QAAQ,EAAE;wBACR,MAAM,EAAE,YAAY,CAAC,0BAA0B,EAAE;wBACjD,MAAM,EAAE,YAAY,CAAC,uBAAuB,EAAE;wBAC9C,KAAK,EAAE,YAAY,CAAC,sBAAsB,EAAE;wBAC5C,MAAM,EAAE,YAAY,CAAC,uBAAuB,EAAE;qBAC/C;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,OAAO;aACf,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QAClB,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QAC/C,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist-server/server/modules/websocket/services/plugin-websocket-proxy.service.js

@@ -0,0 +1,52 @@
+import { WebSocket } from 'ws';
+/**
+ * Proxies an authenticated client websocket to a plugin websocket endpoint.
+ */
+export function handlePluginWsProxy(clientWs, pathname, getPluginPort) {
+    const pluginName = pathname.replace('/plugin-ws/', '');
+    if (!pluginName || /[^a-zA-Z0-9_-]/.test(pluginName)) {
+        clientWs.close(4400, 'Invalid plugin name');
+        return;
+    }
+    const port = getPluginPort(pluginName);
+    if (!port) {
+        clientWs.close(4404, 'Plugin not running');
+        return;
+    }
+    const upstream = new WebSocket(`ws://127.0.0.1:${port}/ws`);
+    upstream.on('open', () => {
+        console.log(`[Plugins] WS proxy connected to "${pluginName}" on port ${port}`);
+    });
+    upstream.on('message', (data) => {
+        if (clientWs.readyState === WebSocket.OPEN) {
+            clientWs.send(data);
+        }
+    });
+    clientWs.on('message', (data) => {
+        if (upstream.readyState === WebSocket.OPEN) {
+            upstream.send(data);
+        }
+    });
+    upstream.on('close', () => {
+        if (clientWs.readyState === WebSocket.OPEN) {
+            clientWs.close();
+        }
+    });
+    clientWs.on('close', () => {
+        if (upstream.readyState === WebSocket.OPEN) {
+            upstream.close();
+        }
+    });
+    upstream.on('error', (error) => {
+        console.error(`[Plugins] WS proxy error for "${pluginName}":`, error.message);
+        if (clientWs.readyState === WebSocket.OPEN) {
+            clientWs.close(4502, 'Upstream error');
+        }
+    });
+    clientWs.on('error', () => {
+        if (upstream.readyState === WebSocket.OPEN) {
+            upstream.close();
+        }
+    });
+}
+//# sourceMappingURL=plugin-websocket-proxy.service.js.map

package/dist-server/server/modules/websocket/services/plugin-websocket-proxy.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin-websocket-proxy.service.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/services/plugin-websocket-proxy.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAE/B;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,QAAmB,EACnB,QAAgB,EAChB,aAAoD;IAEpD,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU,IAAI,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;QACrD,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;QAC5C,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;IACvC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;QAC3C,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC,kBAAkB,IAAI,KAAK,CAAC,CAAC;IAE5D,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;QACvB,OAAO,CAAC,GAAG,CAAC,oCAAoC,UAAU,aAAa,IAAI,EAAE,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;QAC9B,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;QAC9B,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QACxB,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QACxB,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;QAC7B,OAAO,CAAC,KAAK,CAAC,iCAAiC,UAAU,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9E,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QACxB,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}