kumo-cli 1.0.3 → 1.0.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +13 -5
- package/dist/auth/credentialsStore.js +61 -14
- package/dist/auth/credentialsStore.js.map +1 -1
- package/dist/auth/credentialsStore.test.js +60 -0
- package/dist/auth/credentialsStore.test.js.map +1 -0
- package/dist/claude/fileHistory.js +79 -0
- package/dist/claude/fileHistory.js.map +1 -0
- package/dist/claude/sessionScanner.js +7 -95
- package/dist/claude/sessionScanner.js.map +1 -1
- package/dist/claude/sessionScanner.test.js +24 -0
- package/dist/claude/sessionScanner.test.js.map +1 -0
- package/dist/claude/transcriptParser.js +83 -0
- package/dist/claude/transcriptParser.js.map +1 -0
- package/dist/claude/transcriptParser.test.js +78 -0
- package/dist/claude/transcriptParser.test.js.map +1 -0
- package/dist/commands/agentPrompt.js +18 -0
- package/dist/commands/agentPrompt.js.map +1 -0
- package/dist/commands/cliArgs.js +56 -0
- package/dist/commands/cliArgs.js.map +1 -0
- package/dist/commands/cliArgs.test.js +25 -0
- package/dist/commands/cliArgs.test.js.map +1 -0
- package/dist/core/logger.js +7 -2
- package/dist/core/logger.js.map +1 -1
- package/dist/core/projectFileWatcher.js +56 -0
- package/dist/core/projectFileWatcher.js.map +1 -0
- package/dist/handlers/fileHandlerPayloads.js +72 -0
- package/dist/handlers/fileHandlerPayloads.js.map +1 -0
- package/dist/handlers/fileHandlerPayloads.test.js +55 -0
- package/dist/handlers/fileHandlerPayloads.test.js.map +1 -0
- package/dist/handlers/fileHandlers.js +74 -0
- package/dist/handlers/fileHandlers.js.map +1 -0
- package/dist/handlers/messageRouter.js +8 -205
- package/dist/handlers/messageRouter.js.map +1 -1
- package/dist/handlers/promptImages.js +113 -0
- package/dist/handlers/promptImages.js.map +1 -0
- package/dist/handlers/promptImages.test.js +52 -0
- package/dist/handlers/promptImages.test.js.map +1 -0
- package/dist/handlers/router.types.js +2 -0
- package/dist/handlers/router.types.js.map +1 -0
- package/dist/handlers/runtimeHandlers.js +67 -0
- package/dist/handlers/runtimeHandlers.js.map +1 -0
- package/dist/handlers/sessionHandlers.js +119 -0
- package/dist/handlers/sessionHandlers.js.map +1 -0
- package/dist/handlers/terminalHandlers.js +40 -0
- package/dist/handlers/terminalHandlers.js.map +1 -0
- package/dist/index.js +60 -145
- package/dist/index.js.map +1 -1
- package/dist/pty/notificationDetector.js +29 -0
- package/dist/pty/notificationDetector.js.map +1 -0
- package/dist/pty/notificationDetector.test.js +24 -0
- package/dist/pty/notificationDetector.test.js.map +1 -0
- package/dist/pty/ptyMenuParser.js +11 -277
- package/dist/pty/ptyMenuParser.js.map +1 -1
- package/dist/pty/ptyMenuParser.test.js +26 -0
- package/dist/pty/ptyMenuParser.test.js.map +1 -0
- package/dist/pty/selectionMenuDetector.js +49 -0
- package/dist/pty/selectionMenuDetector.js.map +1 -0
- package/dist/pty/selectionMenuDetector.test.js +62 -0
- package/dist/pty/selectionMenuDetector.test.js.map +1 -0
- package/dist/pty/selectionMenuMeta.js +54 -0
- package/dist/pty/selectionMenuMeta.js.map +1 -0
- package/dist/pty/selectionMenuMeta.test.js +26 -0
- package/dist/pty/selectionMenuMeta.test.js.map +1 -0
- package/dist/pty/selectionMenuParser.js +83 -0
- package/dist/pty/selectionMenuParser.js.map +1 -0
- package/dist/pty/selectionMenuParser.test.js +30 -0
- package/dist/pty/selectionMenuParser.test.js.map +1 -0
- package/dist/pty/spawn.js +31 -5
- package/dist/pty/spawn.js.map +1 -1
- package/dist/pty/spawn.test.js +16 -0
- package/dist/pty/spawn.test.js.map +1 -0
- package/dist/pty/statusSessionDetector.js +28 -0
- package/dist/pty/statusSessionDetector.js.map +1 -0
- package/dist/pty/statusSessionDetector.test.js +23 -0
- package/dist/pty/statusSessionDetector.test.js.map +1 -0
- package/dist/pty/terminalManager.js +11 -6
- package/dist/pty/terminalManager.js.map +1 -1
- package/dist/pty/textInputDetector.js +60 -0
- package/dist/pty/textInputDetector.js.map +1 -0
- package/dist/pty/textInputDetector.test.js +32 -0
- package/dist/pty/textInputDetector.test.js.map +1 -0
- package/dist/services/bootstrapService.js +28 -0
- package/dist/services/bootstrapService.js.map +1 -0
- package/dist/services/checkpointService.js +113 -0
- package/dist/services/checkpointService.js.map +1 -0
- package/dist/services/checkpointService.test.js +61 -0
- package/dist/services/checkpointService.test.js.map +1 -0
- package/dist/services/claudeMenuBridge.js +116 -0
- package/dist/services/claudeMenuBridge.js.map +1 -0
- package/dist/services/claudeMenuBridge.test.js +90 -0
- package/dist/services/claudeMenuBridge.test.js.map +1 -0
- package/dist/services/claudeService.js +5 -98
- package/dist/services/claudeService.js.map +1 -1
- package/dist/services/cliRuntime.js +49 -0
- package/dist/services/cliRuntime.js.map +1 -0
- package/dist/services/cliRuntime.test.js +55 -0
- package/dist/services/cliRuntime.test.js.map +1 -0
- package/dist/services/fileService.js +19 -23
- package/dist/services/fileService.js.map +1 -1
- package/dist/services/modelAdapter.test.js +47 -0
- package/dist/services/modelAdapter.test.js.map +1 -0
- package/dist/services/modelArgs.test.js +64 -0
- package/dist/services/modelArgs.test.js.map +1 -0
- package/dist/services/modelBaseUrl.test.js +32 -0
- package/dist/services/modelBaseUrl.test.js.map +1 -0
- package/dist/services/modelService.js +20 -9
- package/dist/services/modelService.js.map +1 -1
- package/dist/services/modelService.test.js +60 -0
- package/dist/services/modelService.test.js.map +1 -1
- package/dist/services/notificationDedupe.js +23 -0
- package/dist/services/notificationDedupe.js.map +1 -0
- package/dist/services/notificationDedupe.test.js +20 -0
- package/dist/services/notificationDedupe.test.js.map +1 -0
- package/dist/services/pairingService.js +17 -8
- package/dist/services/pairingService.js.map +1 -1
- package/dist/services/sessionListService.js +19 -0
- package/dist/services/sessionListService.js.map +1 -0
- package/dist/services/sessionListService.test.js +22 -0
- package/dist/services/sessionListService.test.js.map +1 -0
- package/dist/services/sessionRollbackService.js +195 -0
- package/dist/services/sessionRollbackService.js.map +1 -0
- package/dist/services/sessionRollbackService.test.js +142 -0
- package/dist/services/sessionRollbackService.test.js.map +1 -0
- package/dist/services/sessionService.js +41 -52
- package/dist/services/sessionService.js.map +1 -1
- package/dist/services/tunnelHelpers.js +69 -0
- package/dist/services/tunnelHelpers.js.map +1 -0
- package/dist/services/tunnelHelpers.test.js +42 -0
- package/dist/services/tunnelHelpers.test.js.map +1 -0
- package/dist/services/tunnelService.js +69 -25
- package/dist/services/tunnelService.js.map +1 -1
- package/dist/services/tunnelService.test.js +91 -0
- package/dist/services/tunnelService.test.js.map +1 -0
- package/dist/transport/webrtcTransport.js +145 -0
- package/dist/transport/webrtcTransport.js.map +1 -0
- package/dist/transport/webrtcTransport.test.js +104 -0
- package/dist/transport/webrtcTransport.test.js.map +1 -0
- package/dist/transport/webrtcTypes.js +29 -0
- package/dist/transport/webrtcTypes.js.map +1 -0
- package/dist/transport/webrtcTypes.test.js +20 -0
- package/dist/transport/webrtcTypes.test.js.map +1 -0
- package/dist/transport/wsClient.js +16 -3
- package/dist/transport/wsClient.js.map +1 -1
- package/dist/utils/safePath.js +18 -6
- package/dist/utils/safePath.js.map +1 -1
- package/package.json +5 -2
- package/scripts/run-tests.mjs +46 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tunnelHelpers.js","sourceRoot":"","sources":["../../src/services/tunnelHelpers.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;AAE/C,0DAA0D;AAC1D,MAAM,UAAU,uBAAuB,CAAC,IAAY,EAAE,aAAsB;IAC1E,MAAM,WAAW,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAClF,IAAI,CAAC,WAAW;QAAE,OAAO,SAAS,CAAC;IACnC,OAAO,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,IAAI,EAAE,CAAC;AAC5E,CAAC;AAED,2EAA2E;AAC3E,MAAM,UAAU,mBAAmB,CAAC,SAAiB;IACnD,OAAO,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC;AACrH,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,gBAAgB,CAAC,OAAyB;IACxD,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,MAAM,KAAK,KAAK,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QAC3E,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AAC7C,CAAC;AAED,gFAAgF;AAChF,MAAM,UAAU,0BAA0B,CACxC,OAA+B,EAC/B,UAAkB,EAClB,IAAY,EACZ,UAAmB;IAEnB,MAAM,eAAe,GAAG,MAAM,CAAC,WAAW,CACxC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;SACpB,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE;QAChB,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;QACnC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,YAAY,EAAE,iBAAiB,EAAE,2BAA2B,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACpI,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QACpB,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;QACnC,IAAI,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YACpD,OAAO,CAAC,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,0CAA0C,EAAE,UAAU,UAAU,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC;QAC1G,CAAC;QACD,OAAO,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACtB,CAAC,CAAC,CACL,CAAC;IAEF,IAAI,UAAU;QAAE,eAAe,CAAC,IAAI,GAAG,UAAU,CAAC;IAClD,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,mFAAmF;AACnF,MAAM,UAAU,4BAA4B,CAC1C,OAA+B,EAC/B,UAAkB,EAClB,IAAY,EACZ,UAAmB;IAEnB,MAAM,eAAe,GAAG,MAAM,CAAC,WAAW,CACxC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,kBAAkB,EAAE,mBAAmB,EAAE,gBAAgB,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CACpI,CAAC;IAEF,MAAM,YAAY,GAAG,UAAU,UAAU,IAAI,IAAI,EAAE,CAAC;IACpD,IAAI,eAAe,CAAC,QAAQ,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACpH,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAClF,MAAM,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC;YACpD,MAAM,cAAc,GAAG,IAAI,GAAG,CAAS;gBACrC,UAAU,UAAU,IAAI,IAAI,EAAE;gBAC9B,oBAAoB,IAAI,EAAE;gBAC1B,UAAU,CAAC,CAAC,CAAC,UAAU,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE;gBACxC,UAAU,CAAC,CAAC,CAAC,WAAW,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE;aAC1C,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YACnB,IAAI,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/B,eAAe,CAAC,QAAQ,GAAG,GAAG,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;YAChF,CAAC;QACH,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,IAAI,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC;QAClC,eAAe,CAAC,YAAY,CAAC,GAAG,eAAe,CAAC,YAAY,CAAC;aAC1D,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC;aACjC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,eAAe,CAAC;AACzB,CAAC"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import test from 'node:test';
|
|
2
|
+
import assert from 'node:assert/strict';
|
|
3
|
+
import { decodeTunnelBody, filterTunnelRequestHeaders, normalizeTunnelPath, resolveTunnelHostHeader, rewriteTunnelResponseHeaders, } from './tunnelHelpers.js';
|
|
4
|
+
test('normalizeTunnelPath falls back to root for malformed paths', () => {
|
|
5
|
+
assert.equal(normalizeTunnelPath('/api/health'), '/api/health');
|
|
6
|
+
assert.equal(normalizeTunnelPath('//evil'), '/');
|
|
7
|
+
});
|
|
8
|
+
test('resolveTunnelHostHeader appends port when missing', () => {
|
|
9
|
+
assert.equal(resolveTunnelHostHeader(3000, 'localhost'), 'localhost:3000');
|
|
10
|
+
assert.equal(resolveTunnelHostHeader(3000, 'localhost:4000'), 'localhost:4000');
|
|
11
|
+
});
|
|
12
|
+
test('decodeTunnelBody skips get requests and decodes post payloads', () => {
|
|
13
|
+
assert.equal(decodeTunnelBody({ requestId: '1', method: 'GET', path: '/', headers: {}, port: 3000, body: 'aGk=' }), undefined);
|
|
14
|
+
assert.equal(decodeTunnelBody({ requestId: '1', method: 'POST', path: '/', headers: {}, port: 3000, body: Buffer.from('hi').toString('base64') })?.toString('utf-8'), 'hi');
|
|
15
|
+
});
|
|
16
|
+
test('filterTunnelRequestHeaders strips hop-by-hop headers and rewrites local origins', () => {
|
|
17
|
+
const headers = filterTunnelRequestHeaders({
|
|
18
|
+
Host: 'example.com',
|
|
19
|
+
Connection: 'keep-alive',
|
|
20
|
+
Origin: 'http://localhost:8080',
|
|
21
|
+
Referer: 'http://127.0.0.1:8080/app',
|
|
22
|
+
Accept: 'application/json',
|
|
23
|
+
}, '127.0.0.1', 3000, 'localhost:3000');
|
|
24
|
+
assert.deepEqual(headers, {
|
|
25
|
+
Origin: 'http://127.0.0.1:3000',
|
|
26
|
+
Referer: 'http://127.0.0.1:3000/app',
|
|
27
|
+
Accept: 'application/json',
|
|
28
|
+
host: 'localhost:3000',
|
|
29
|
+
});
|
|
30
|
+
});
|
|
31
|
+
test('rewriteTunnelResponseHeaders strips local redirect origins and cookie domain', () => {
|
|
32
|
+
const headers = rewriteTunnelResponseHeaders({
|
|
33
|
+
location: 'http://127.0.0.1:3000/login?next=%2F',
|
|
34
|
+
'set-cookie': 'sid=abc; Domain=localhost; Path=/; Secure',
|
|
35
|
+
'content-length': '10',
|
|
36
|
+
}, '127.0.0.1', 3000, 'localhost:3000');
|
|
37
|
+
assert.deepEqual(headers, {
|
|
38
|
+
location: '/login?next=%2F',
|
|
39
|
+
'set-cookie': 'sid=abc; Path=/',
|
|
40
|
+
});
|
|
41
|
+
});
|
|
42
|
+
//# sourceMappingURL=tunnelHelpers.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tunnelHelpers.test.js","sourceRoot":"","sources":["../../src/services/tunnelHelpers.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EACL,gBAAgB,EAChB,0BAA0B,EAC1B,mBAAmB,EACnB,uBAAuB,EACvB,4BAA4B,GAC7B,MAAM,oBAAoB,CAAC;AAE5B,IAAI,CAAC,4DAA4D,EAAE,GAAG,EAAE;IACtE,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,aAAa,CAAC,EAAE,aAAa,CAAC,CAAC;IAChE,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAC;AACnD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mDAAmD,EAAE,GAAG,EAAE;IAC7D,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC3E,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,EAAE,gBAAgB,CAAC,EAAE,gBAAgB,CAAC,CAAC;AAClF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACzE,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IAC/H,MAAM,CAAC,KAAK,CACV,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,OAAO,CAAC,EACvJ,IAAI,CACL,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iFAAiF,EAAE,GAAG,EAAE;IAC3F,MAAM,OAAO,GAAG,0BAA0B,CACxC;QACE,IAAI,EAAE,aAAa;QACnB,UAAU,EAAE,YAAY;QACxB,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,2BAA2B;QACpC,MAAM,EAAE,kBAAkB;KAC3B,EACD,WAAW,EACX,IAAI,EACJ,gBAAgB,CACjB,CAAC;IAEF,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE;QACxB,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,2BAA2B;QACpC,MAAM,EAAE,kBAAkB;QAC1B,IAAI,EAAE,gBAAgB;KACvB,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8EAA8E,EAAE,GAAG,EAAE;IACxF,MAAM,OAAO,GAAG,4BAA4B,CAC1C;QACE,QAAQ,EAAE,sCAAsC;QAChD,YAAY,EAAE,2CAA2C;QACzD,gBAAgB,EAAE,IAAI;KACvB,EACD,WAAW,EACX,IAAI,EACJ,gBAAgB,CACjB,CAAC;IAEF,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE;QACxB,QAAQ,EAAE,iBAAiB;QAC3B,YAAY,EAAE,iBAAiB;KAChC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,37 +1,81 @@
|
|
|
1
|
+
import http from 'node:http';
|
|
1
2
|
import { log } from '../core/logger.js';
|
|
3
|
+
import { decodeTunnelBody, DEFAULT_TUNNEL_HOST, filterTunnelRequestHeaders, normalizeTunnelPath, resolveTunnelHostHeader, rewriteTunnelResponseHeaders, } from './tunnelHelpers.js';
|
|
4
|
+
const MAX_TUNNEL_BODY_BYTES = Number(process.env.KUMO_TUNNEL_BODY_BYTES ?? 10 * 1024 * 1024);
|
|
5
|
+
const MAX_TUNNEL_RESPONSE_BYTES = Number(process.env.KUMO_TUNNEL_RESPONSE_BYTES ?? 10 * 1024 * 1024);
|
|
6
|
+
const TUNNEL_TIMEOUT_MS = Number(process.env.KUMO_TUNNEL_TIMEOUT_MS ?? 30000);
|
|
7
|
+
const allowedPorts = new Set((process.env.KUMO_ALLOWED_TUNNEL_PORTS ?? '')
|
|
8
|
+
.split(',')
|
|
9
|
+
.map((port) => Number(port.trim()))
|
|
10
|
+
.filter((port) => Number.isInteger(port) && port > 0 && port <= 65535));
|
|
11
|
+
function tunnelError(requestId, status, message) {
|
|
12
|
+
return { type: 'tunnel_response', requestId, status, headers: {}, body: Buffer.from(message).toString('base64') };
|
|
13
|
+
}
|
|
14
|
+
async function proxyHttpRequest(input) {
|
|
15
|
+
return await new Promise((resolve, reject) => {
|
|
16
|
+
const req = http.request({
|
|
17
|
+
protocol: 'http:',
|
|
18
|
+
hostname: input.hostname,
|
|
19
|
+
port: input.port,
|
|
20
|
+
method: input.method,
|
|
21
|
+
path: input.path,
|
|
22
|
+
headers: input.headers,
|
|
23
|
+
}, (res) => {
|
|
24
|
+
const status = res.statusCode ?? 502;
|
|
25
|
+
const headers = {};
|
|
26
|
+
for (const [k, v] of Object.entries(res.headers)) {
|
|
27
|
+
if (v == null)
|
|
28
|
+
continue;
|
|
29
|
+
headers[k] = Array.isArray(v) ? v.join(',') : String(v);
|
|
30
|
+
}
|
|
31
|
+
const chunks = [];
|
|
32
|
+
let total = 0;
|
|
33
|
+
res.on('data', (chunk) => {
|
|
34
|
+
total += chunk.length;
|
|
35
|
+
if (total > MAX_TUNNEL_RESPONSE_BYTES) {
|
|
36
|
+
req.destroy(new Error('response body too large'));
|
|
37
|
+
return;
|
|
38
|
+
}
|
|
39
|
+
chunks.push(chunk);
|
|
40
|
+
});
|
|
41
|
+
res.on('end', () => resolve({ status, headers, body: Buffer.concat(chunks) }));
|
|
42
|
+
res.on('error', reject);
|
|
43
|
+
});
|
|
44
|
+
req.setTimeout(TUNNEL_TIMEOUT_MS, () => req.destroy(new Error('timeout')));
|
|
45
|
+
req.on('error', reject);
|
|
46
|
+
if (input.body)
|
|
47
|
+
req.write(input.body);
|
|
48
|
+
req.end();
|
|
49
|
+
});
|
|
50
|
+
}
|
|
2
51
|
// proxy a http request from the server through this cli to a local service
|
|
3
52
|
export async function handleTunnelRequest(req) {
|
|
4
|
-
const
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
}));
|
|
53
|
+
const port = Number(req.port);
|
|
54
|
+
if (!Number.isInteger(port) || port < 1 || port > 65535)
|
|
55
|
+
return tunnelError(req.requestId, 400, 'invalid port');
|
|
56
|
+
if (allowedPorts.size > 0 && !allowedPorts.has(port))
|
|
57
|
+
return tunnelError(req.requestId, 403, 'port not allowed');
|
|
58
|
+
const safePath = normalizeTunnelPath(req.path);
|
|
59
|
+
const targetHost = DEFAULT_TUNNEL_HOST;
|
|
60
|
+
const url = `http://${targetHost}:${port}${safePath}`;
|
|
61
|
+
const hostHeader = resolveTunnelHostHeader(port, req.host);
|
|
62
|
+
const body = decodeTunnelBody(req);
|
|
63
|
+
if (body && body.length > MAX_TUNNEL_BODY_BYTES)
|
|
64
|
+
return tunnelError(req.requestId, 413, 'request body too large');
|
|
65
|
+
const filteredHeaders = filterTunnelRequestHeaders(req.headers, targetHost, port, hostHeader);
|
|
18
66
|
log(`[tunnel] ${req.method} ${url}`);
|
|
19
67
|
try {
|
|
20
|
-
const res = await
|
|
68
|
+
const res = await proxyHttpRequest({
|
|
69
|
+
hostname: targetHost,
|
|
70
|
+
port,
|
|
71
|
+
path: safePath,
|
|
21
72
|
method: req.method,
|
|
22
73
|
headers: filteredHeaders,
|
|
23
|
-
body
|
|
24
|
-
redirect: 'manual',
|
|
74
|
+
body,
|
|
25
75
|
});
|
|
26
|
-
const buf =
|
|
76
|
+
const buf = res.body;
|
|
27
77
|
log(`[tunnel] response ${res.status} ${url} (${buf.length} bytes)`);
|
|
28
|
-
const respHeaders =
|
|
29
|
-
const remoteOrigin = `http://${targetHost}:${req.port}`;
|
|
30
|
-
if (respHeaders['location'])
|
|
31
|
-
respHeaders['location'] = respHeaders['location'].replace(remoteOrigin, '');
|
|
32
|
-
if (respHeaders['set-cookie']) {
|
|
33
|
-
respHeaders['set-cookie'] = respHeaders['set-cookie'].replace(/;\s*domain=[^;]*/gi, '').replace(/;\s*secure/gi, '');
|
|
34
|
-
}
|
|
78
|
+
const respHeaders = rewriteTunnelResponseHeaders(res.headers, targetHost, port, hostHeader);
|
|
35
79
|
return { type: 'tunnel_response', requestId: req.requestId, status: res.status, headers: respHeaders, body: buf.toString('base64') };
|
|
36
80
|
}
|
|
37
81
|
catch (err) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tunnelService.js","sourceRoot":"","sources":["../../src/services/tunnelService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"tunnelService.js","sourceRoot":"","sources":["../../src/services/tunnelService.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AACxC,OAAO,EACL,gBAAgB,EAChB,mBAAmB,EACnB,0BAA0B,EAC1B,mBAAmB,EACnB,uBAAuB,EACvB,4BAA4B,GAC7B,MAAM,oBAAoB,CAAC;AAoB5B,MAAM,qBAAqB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;AAC7F,MAAM,yBAAyB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;AACrG,MAAM,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,KAAK,CAAC,CAAC;AAC9E,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,EAAE,CAAC;KAC1C,KAAK,CAAC,GAAG,CAAC;KACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;KAClC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,IAAI,KAAK,CAAC,CACzE,CAAC;AAEF,SAAS,WAAW,CAAC,SAAiB,EAAE,MAAc,EAAE,OAAe;IACrE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;AACpH,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,KAO/B;IACC,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CACtB;YACE,QAAQ,EAAE,OAAO;YACjB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO;SACvB,EACD,CAAC,GAAG,EAAE,EAAE;YACN,MAAM,MAAM,GAAG,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC;YACrC,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjD,IAAI,CAAC,IAAI,IAAI;oBAAE,SAAS;gBACxB,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBAC/B,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC;gBACtB,IAAI,KAAK,GAAG,yBAAyB,EAAE,CAAC;oBACtC,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAC;oBAClD,OAAO;gBACT,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC,CAAC;YACH,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/E,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC1B,CAAC,CACF,CAAC;QAEF,GAAG,CAAC,UAAU,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC3E,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxB,IAAI,KAAK,CAAC,IAAI;YAAE,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtC,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,2EAA2E;AAC3E,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,GAAqB;IAC7D,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,KAAK;QAAE,OAAO,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC;IAChH,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;QAAE,OAAO,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACjH,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC/C,MAAM,UAAU,GAAG,mBAAmB,CAAC;IACvC,MAAM,GAAG,GAAG,UAAU,UAAU,IAAI,IAAI,GAAG,QAAQ,EAAE,CAAC;IACtD,MAAM,UAAU,GAAG,uBAAuB,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IAC3D,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;IACnC,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,qBAAqB;QAAE,OAAO,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,wBAAwB,CAAC,CAAC;IAClH,MAAM,eAAe,GAAG,0BAA0B,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IAC9F,GAAG,CAAC,YAAY,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;IACrC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC;YACjC,QAAQ,EAAE,UAAU;YACpB,IAAI;YACJ,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,OAAO,EAAE,eAAe;YACxB,IAAI;SACL,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC;QACrB,GAAG,CAAC,qBAAqB,GAAG,CAAC,MAAM,IAAI,GAAG,KAAK,GAAG,CAAC,MAAM,SAAS,CAAC,CAAC;QACpE,MAAM,WAAW,GAAG,4BAA4B,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5F,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;IACvI,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO;YACL,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,EAAE;YACX,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,GAAI,GAAa,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;SAC/E,CAAC;IACJ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import test from 'node:test';
|
|
2
|
+
import assert from 'node:assert/strict';
|
|
3
|
+
import http from 'node:http';
|
|
4
|
+
import { handleTunnelRequest } from './tunnelService.js';
|
|
5
|
+
test('preserves browser client hint headers through tunnel proxy', async () => {
|
|
6
|
+
const seenHeaders = await new Promise((resolve, reject) => {
|
|
7
|
+
const server = http.createServer((req, res) => {
|
|
8
|
+
resolve(req.headers);
|
|
9
|
+
res.end('ok');
|
|
10
|
+
server.close();
|
|
11
|
+
});
|
|
12
|
+
server.on('error', reject);
|
|
13
|
+
server.listen(0, '127.0.0.1', async () => {
|
|
14
|
+
const address = server.address();
|
|
15
|
+
if (!address || typeof address === 'string')
|
|
16
|
+
return reject(new Error('server address unavailable'));
|
|
17
|
+
const response = await handleTunnelRequest({
|
|
18
|
+
requestId: 'req-1',
|
|
19
|
+
method: 'GET',
|
|
20
|
+
path: '/',
|
|
21
|
+
port: address.port,
|
|
22
|
+
headers: {
|
|
23
|
+
'user-agent': 'Mozilla/5.0 Chrome/120.0.0.0',
|
|
24
|
+
'sec-ch-ua': '"Chromium";v="120"',
|
|
25
|
+
'sec-ch-ua-mobile': '?0',
|
|
26
|
+
'sec-ch-ua-platform': '"Windows"',
|
|
27
|
+
'upgrade-insecure-requests': '1',
|
|
28
|
+
connection: 'keep-alive',
|
|
29
|
+
host: 'provider.ecologynet.local:3000',
|
|
30
|
+
},
|
|
31
|
+
});
|
|
32
|
+
assert.equal(response.status, 200);
|
|
33
|
+
});
|
|
34
|
+
});
|
|
35
|
+
assert.equal(seenHeaders['user-agent'], 'Mozilla/5.0 Chrome/120.0.0.0');
|
|
36
|
+
assert.equal(seenHeaders['sec-ch-ua'], '"Chromium";v="120"');
|
|
37
|
+
assert.equal(seenHeaders['sec-ch-ua-mobile'], '?0');
|
|
38
|
+
assert.equal(seenHeaders['sec-ch-ua-platform'], '"Windows"');
|
|
39
|
+
assert.equal(seenHeaders['upgrade-insecure-requests'], '1');
|
|
40
|
+
assert.equal(seenHeaders.connection, 'keep-alive');
|
|
41
|
+
assert.notEqual(seenHeaders.host, 'provider.ecologynet.local:3000');
|
|
42
|
+
});
|
|
43
|
+
test('replaces non-browser user agents for browser-gated local apps', async () => {
|
|
44
|
+
const seenHeaders = await new Promise((resolve, reject) => {
|
|
45
|
+
const server = http.createServer((req, res) => {
|
|
46
|
+
resolve(req.headers);
|
|
47
|
+
res.end('ok');
|
|
48
|
+
server.close();
|
|
49
|
+
});
|
|
50
|
+
server.on('error', reject);
|
|
51
|
+
server.listen(0, '127.0.0.1', async () => {
|
|
52
|
+
const address = server.address();
|
|
53
|
+
if (!address || typeof address === 'string')
|
|
54
|
+
return reject(new Error('server address unavailable'));
|
|
55
|
+
const response = await handleTunnelRequest({
|
|
56
|
+
requestId: 'req-2',
|
|
57
|
+
method: 'GET',
|
|
58
|
+
path: '/',
|
|
59
|
+
port: address.port,
|
|
60
|
+
headers: { 'user-agent': 'Dart/3.0 (dart:io)' },
|
|
61
|
+
});
|
|
62
|
+
assert.equal(response.status, 200);
|
|
63
|
+
});
|
|
64
|
+
});
|
|
65
|
+
assert.match(seenHeaders['user-agent'] ?? '', /Chrome\/120/);
|
|
66
|
+
});
|
|
67
|
+
test('modernizes old browser user agents for Rails allow_browser modern gate', async () => {
|
|
68
|
+
const seenHeaders = await new Promise((resolve, reject) => {
|
|
69
|
+
const server = http.createServer((req, res) => {
|
|
70
|
+
resolve(req.headers);
|
|
71
|
+
res.end('ok');
|
|
72
|
+
server.close();
|
|
73
|
+
});
|
|
74
|
+
server.on('error', reject);
|
|
75
|
+
server.listen(0, '127.0.0.1', async () => {
|
|
76
|
+
const address = server.address();
|
|
77
|
+
if (!address || typeof address === 'string')
|
|
78
|
+
return reject(new Error('server address unavailable'));
|
|
79
|
+
const response = await handleTunnelRequest({
|
|
80
|
+
requestId: 'req-3',
|
|
81
|
+
method: 'GET',
|
|
82
|
+
path: '/',
|
|
83
|
+
port: address.port,
|
|
84
|
+
headers: { 'user-agent': 'Mozilla/5.0 AppleWebKit/537.36 Chrome/119.0.0.0 Safari/537.36' },
|
|
85
|
+
});
|
|
86
|
+
assert.equal(response.status, 200);
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
assert.equal(seenHeaders['user-agent'], 'Mozilla/5.0 AppleWebKit/537.36 Chrome/120.0.0.0 Safari/537.36');
|
|
90
|
+
});
|
|
91
|
+
//# sourceMappingURL=tunnelService.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tunnelService.test.js","sourceRoot":"","sources":["../../src/services/tunnelService.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEzD,IAAI,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;IAC5E,MAAM,WAAW,GAAG,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpG,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC;gBACzC,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE;oBACP,YAAY,EAAE,8BAA8B;oBAC5C,WAAW,EAAE,oBAAoB;oBACjC,kBAAkB,EAAE,IAAI;oBACxB,oBAAoB,EAAE,WAAW;oBACjC,2BAA2B,EAAE,GAAG;oBAChC,UAAU,EAAE,YAAY;oBACxB,IAAI,EAAE,gCAAgC;iBACvC;aACF,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,8BAA8B,CAAC,CAAC;IACxE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,oBAAoB,CAAC,CAAC;IAC7D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE,IAAI,CAAC,CAAC;IACpD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,oBAAoB,CAAC,EAAE,WAAW,CAAC,CAAC;IAC7D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,2BAA2B,CAAC,EAAE,GAAG,CAAC,CAAC;IAC5D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IACnD,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,gCAAgC,CAAC,CAAC;AACtE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;IAC/E,MAAM,WAAW,GAAG,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpG,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC;gBACzC,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE,EAAE,YAAY,EAAE,oBAAoB,EAAE;aAChD,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,EAAE,EAAE,aAAa,CAAC,CAAC;AAC/D,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;IACxF,MAAM,WAAW,GAAG,MAAM,IAAI,OAAO,CAA2B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpG,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC;gBACzC,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE,EAAE,YAAY,EAAE,+DAA+D,EAAE;aAC3F,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,+DAA+D,CAAC,CAAC;AAC3G,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,145 @@
|
|
|
1
|
+
import { log } from '../core/logger.js';
|
|
2
|
+
import { domainForType, unwrapDomainEnvelope, wrapDomainEnvelope } from './webrtcTypes.js';
|
|
3
|
+
const routedDomains = new Set(['prompt', 'session', 'transcript', 'file']);
|
|
4
|
+
export class CliWebRtcTransport {
|
|
5
|
+
terminals;
|
|
6
|
+
send;
|
|
7
|
+
createPeerConnection;
|
|
8
|
+
hasPermission;
|
|
9
|
+
routeMessage;
|
|
10
|
+
peers = new Map();
|
|
11
|
+
channels = new Map();
|
|
12
|
+
terminalToSession = new Map();
|
|
13
|
+
constructor(terminals, send, createPeerConnection, hasPermission = () => true, routeMessage) {
|
|
14
|
+
this.terminals = terminals;
|
|
15
|
+
this.send = send;
|
|
16
|
+
this.createPeerConnection = createPeerConnection;
|
|
17
|
+
this.hasPermission = hasPermission;
|
|
18
|
+
this.routeMessage = routeMessage;
|
|
19
|
+
}
|
|
20
|
+
setMessageRouter(routeMessage) { this.routeMessage = routeMessage; }
|
|
21
|
+
async handleSignal(msg) {
|
|
22
|
+
if (!msg.sessionId)
|
|
23
|
+
return;
|
|
24
|
+
if (msg.signalType === 'offer') {
|
|
25
|
+
await this.acceptOffer(msg.sessionId, msg.payload);
|
|
26
|
+
return;
|
|
27
|
+
}
|
|
28
|
+
if (msg.signalType === 'ice') {
|
|
29
|
+
const peer = this.peers.get(msg.sessionId);
|
|
30
|
+
if (peer && msg.payload)
|
|
31
|
+
await peer.addIceCandidate(msg.payload);
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
attachTerminal(terminalId, sessionId) {
|
|
35
|
+
this.terminalToSession.set(terminalId, sessionId);
|
|
36
|
+
}
|
|
37
|
+
sendTerminalOutput(terminalId, data) {
|
|
38
|
+
const sessionId = this.terminalToSession.get(terminalId);
|
|
39
|
+
if (!sessionId)
|
|
40
|
+
return false;
|
|
41
|
+
return this.sendEnvelope(sessionId, { type: 'terminal_output', terminalId, data });
|
|
42
|
+
}
|
|
43
|
+
sendTerminalClosed(terminalId, exitCode) {
|
|
44
|
+
const sessionId = this.terminalToSession.get(terminalId);
|
|
45
|
+
this.terminalToSession.delete(terminalId);
|
|
46
|
+
if (!sessionId)
|
|
47
|
+
return false;
|
|
48
|
+
return this.sendEnvelope(sessionId, { type: 'terminal_closed', terminalId, exitCode });
|
|
49
|
+
}
|
|
50
|
+
sendMessage(sessionId, payload) {
|
|
51
|
+
return this.sendEnvelope(sessionId, payload);
|
|
52
|
+
}
|
|
53
|
+
close(sessionId) {
|
|
54
|
+
const ids = sessionId ? [sessionId] : Array.from(this.peers.keys());
|
|
55
|
+
for (const id of ids) {
|
|
56
|
+
try {
|
|
57
|
+
this.channels.get(id)?.close();
|
|
58
|
+
}
|
|
59
|
+
catch { }
|
|
60
|
+
try {
|
|
61
|
+
this.peers.get(id)?.close();
|
|
62
|
+
}
|
|
63
|
+
catch { }
|
|
64
|
+
this.channels.delete(id);
|
|
65
|
+
this.peers.delete(id);
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
async acceptOffer(sessionId, offer) {
|
|
69
|
+
if (!this.hasPermission(sessionId)) {
|
|
70
|
+
this.sendState(sessionId, 'permission_denied');
|
|
71
|
+
return;
|
|
72
|
+
}
|
|
73
|
+
this.close(sessionId);
|
|
74
|
+
const peer = this.createPeerConnection();
|
|
75
|
+
this.peers.set(sessionId, peer);
|
|
76
|
+
peer.onicecandidate = (event) => {
|
|
77
|
+
if (event.candidate)
|
|
78
|
+
this.send({ type: 'transport_signal', sessionId, signalType: 'ice', payload: event.candidate });
|
|
79
|
+
};
|
|
80
|
+
peer.ondatachannel = (event) => this.attachChannel(sessionId, event.channel);
|
|
81
|
+
peer.onconnectionstatechange = () => this.sendState(sessionId, peer.connectionState ?? 'unknown');
|
|
82
|
+
peer.oniceconnectionstatechange = () => this.sendState(sessionId, peer.iceConnectionState ?? 'unknown');
|
|
83
|
+
await peer.setRemoteDescription(offer);
|
|
84
|
+
const answer = await peer.createAnswer();
|
|
85
|
+
await peer.setLocalDescription(answer);
|
|
86
|
+
this.send({ type: 'transport_signal', sessionId, signalType: 'answer', payload: peer.localDescription ?? answer });
|
|
87
|
+
}
|
|
88
|
+
attachChannel(sessionId, channel) {
|
|
89
|
+
this.channels.set(sessionId, channel);
|
|
90
|
+
channel.onopen = () => this.sendState(sessionId, 'datachannel_open');
|
|
91
|
+
channel.onclose = () => this.sendState(sessionId, 'datachannel_closed');
|
|
92
|
+
channel.onerror = (event) => log(`[webrtc] datachannel error session=${sessionId}: ${String(event)}`);
|
|
93
|
+
channel.onmessage = (event) => this.handleChannelMessage(sessionId, event.data);
|
|
94
|
+
}
|
|
95
|
+
handleChannelMessage(sessionId, raw) {
|
|
96
|
+
let msg;
|
|
97
|
+
try {
|
|
98
|
+
msg = JSON.parse(String(raw));
|
|
99
|
+
}
|
|
100
|
+
catch {
|
|
101
|
+
return;
|
|
102
|
+
}
|
|
103
|
+
const payload = unwrapDomainEnvelope(msg);
|
|
104
|
+
if (!payload)
|
|
105
|
+
return;
|
|
106
|
+
if (!this.hasPermission(sessionId)) {
|
|
107
|
+
this.sendState(sessionId, 'permission_denied');
|
|
108
|
+
return;
|
|
109
|
+
}
|
|
110
|
+
if (routedDomains.has(domainForType(payload.type))) {
|
|
111
|
+
this.routeMessage?.(sessionId, { ...payload, transportSessionId: sessionId });
|
|
112
|
+
return;
|
|
113
|
+
}
|
|
114
|
+
if (payload.type === 'terminal_create') {
|
|
115
|
+
const terminalId = this.terminals.create(payload.cols ?? 120, payload.rows ?? 30, payload.cwd);
|
|
116
|
+
this.attachTerminal(terminalId, sessionId);
|
|
117
|
+
this.sendEnvelope(sessionId, { type: 'terminal_created', terminalId });
|
|
118
|
+
return;
|
|
119
|
+
}
|
|
120
|
+
if (payload.type === 'heartbeat' || payload.type === 'terminal_created' || payload.type === 'terminal_output' || payload.type === 'terminal_closed')
|
|
121
|
+
return;
|
|
122
|
+
if (!this.isSessionTerminal(sessionId, payload.terminalId))
|
|
123
|
+
return;
|
|
124
|
+
if (payload.type === 'terminal_input')
|
|
125
|
+
this.terminals.write(payload.terminalId, payload.data);
|
|
126
|
+
if (payload.type === 'terminal_resize')
|
|
127
|
+
this.terminals.resize(payload.terminalId, payload.cols, payload.rows);
|
|
128
|
+
if (payload.type === 'terminal_close')
|
|
129
|
+
this.terminals.close(payload.terminalId);
|
|
130
|
+
}
|
|
131
|
+
isSessionTerminal(sessionId, terminalId) {
|
|
132
|
+
return typeof terminalId === 'string' && this.terminalToSession.get(terminalId) === sessionId;
|
|
133
|
+
}
|
|
134
|
+
sendEnvelope(sessionId, payload) {
|
|
135
|
+
const channel = this.channels.get(sessionId);
|
|
136
|
+
if (!channel || channel.readyState !== 'open')
|
|
137
|
+
return false;
|
|
138
|
+
channel.send(JSON.stringify(wrapDomainEnvelope(payload)));
|
|
139
|
+
return true;
|
|
140
|
+
}
|
|
141
|
+
sendState(sessionId, state) {
|
|
142
|
+
this.send({ type: 'transport_signal', sessionId, signalType: 'state', payload: { state, at: Date.now() } });
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
//# sourceMappingURL=webrtcTransport.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"webrtcTransport.js","sourceRoot":"","sources":["../../src/transport/webrtcTransport.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,GAAG,EAAE,MAAM,mBAAmB,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,kBAAkB,EAA2H,MAAM,kBAAkB,CAAC;AAMpN,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC;AAE3E,MAAM,OAAO,kBAAkB;IAMV;IACA;IACA;IACA;IACT;IATF,KAAK,GAAG,IAAI,GAAG,EAA8B,CAAC;IAC9C,QAAQ,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC9C,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEtD,YACmB,SAA0B,EAC1B,IAAY,EACZ,oBAA2C,EAC3C,gBAA8B,GAAG,EAAE,CAAC,IAAI,EACjD,YAAwB;QAJf,cAAS,GAAT,SAAS,CAAiB;QAC1B,SAAI,GAAJ,IAAI,CAAQ;QACZ,yBAAoB,GAApB,oBAAoB,CAAuB;QAC3C,kBAAa,GAAb,aAAa,CAA2B;QACjD,iBAAY,GAAZ,YAAY,CAAY;IAC/B,CAAC;IAEJ,gBAAgB,CAAC,YAAuB,IAAU,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC;IAErF,KAAK,CAAC,YAAY,CAAC,GAAqB;QACtC,IAAI,CAAC,GAAG,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,GAAG,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YACnD,OAAO;QACT,CAAC;QACD,IAAI,GAAG,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC3C,IAAI,IAAI,IAAI,GAAG,CAAC,OAAO;gBAAE,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,cAAc,CAAC,UAAkB,EAAE,SAAiB;QAClD,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IACpD,CAAC;IAED,kBAAkB,CAAC,UAAkB,EAAE,IAAY;QACjD,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS;YAAE,OAAO,KAAK,CAAC;QAC7B,OAAO,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;IACrF,CAAC;IAED,kBAAkB,CAAC,UAAkB,EAAE,QAAgB;QACrD,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACzD,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC,SAAS;YAAE,OAAO,KAAK,CAAC;QAC7B,OAAO,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;IACzF,CAAC;IAED,WAAW,CAAC,SAAiB,EAAE,OAAmD;QAChF,OAAO,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,MAAM,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACpE,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;YACrB,IAAI,CAAC;gBAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;YAChD,IAAI,CAAC;gBAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;YAC7C,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,SAAiB,EAAE,KAAc;QACzD,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;YAC/C,OAAO;QACT,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACtB,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACzC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,CAAC,KAAK,EAAE,EAAE;YAC9B,IAAI,KAAK,CAAC,SAAS;gBAAE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,SAAS,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;QACvH,CAAC,CAAC;QACF,IAAI,CAAC,aAAa,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7E,IAAI,CAAC,uBAAuB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,eAAe,IAAI,SAAS,CAAC,CAAC;QAClG,IAAI,CAAC,0BAA0B,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,kBAAkB,IAAI,SAAS,CAAC,CAAC;QACxG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QACzC,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,gBAAgB,IAAI,MAAM,EAAE,CAAC,CAAC;IACrH,CAAC;IAEO,aAAa,CAAC,SAAiB,EAAE,OAAwB;QAC/D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACtC,OAAO,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;QACrE,OAAO,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC;QACxE,OAAO,CAAC,OAAO,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,sCAAsC,SAAS,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtG,OAAO,CAAC,SAAS,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAClF,CAAC;IAEO,oBAAoB,CAAC,SAAiB,EAAE,GAAY;QAC1D,IAAI,GAAqB,CAAC;QAC1B,IAAI,CAAC;YAAC,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAqB,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,OAAO;QAAC,CAAC;QAC5E,MAAM,OAAO,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;YAC/C,OAAO;QACT,CAAC;QACD,IAAI,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;YACnD,IAAI,CAAC,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,GAAG,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,CAAC,CAAC;YAC9E,OAAO;QACT,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAE,OAAO,CAAC,IAAe,IAAI,GAAG,EAAG,OAAO,CAAC,IAAe,IAAI,EAAE,EAAE,OAAO,CAAC,GAAyB,CAAC,CAAC;YAC7I,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;YAC3C,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC,CAAC;YACvE,OAAO;QACT,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,IAAI,KAAK,kBAAkB,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB;YAAE,OAAO;QAC5J,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,OAAO,CAAC,UAAgC,CAAC;YAAE,OAAO;QACzF,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB;YAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,UAAoB,EAAE,OAAO,CAAC,IAAc,CAAC,CAAC;QAClH,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB;YAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,UAAoB,EAAE,OAAO,CAAC,IAAc,EAAE,OAAO,CAAC,IAAc,CAAC,CAAC;QAC5I,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB;YAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,UAAoB,CAAC,CAAC;IAC5F,CAAC;IAEO,iBAAiB,CAAC,SAAiB,EAAE,UAAmB;QAC9D,OAAO,OAAO,UAAU,KAAK,QAAQ,IAAI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,SAAS,CAAC;IAChG,CAAC;IAEO,YAAY,CAAC,SAAiB,EAAE,OAAmD;QACzF,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,UAAU,KAAK,MAAM;YAAE,OAAO,KAAK,CAAC;QAC5D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,SAAS,CAAC,SAAiB,EAAE,KAAa;QAChD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;IAC9G,CAAC;CACF"}
|
|
@@ -0,0 +1,104 @@
|
|
|
1
|
+
import test from 'node:test';
|
|
2
|
+
import assert from 'node:assert/strict';
|
|
3
|
+
import { CliWebRtcTransport } from './webrtcTransport.js';
|
|
4
|
+
class FakeChannel {
|
|
5
|
+
readyState = 'open';
|
|
6
|
+
sent = [];
|
|
7
|
+
onopen = null;
|
|
8
|
+
onmessage = null;
|
|
9
|
+
onclose = null;
|
|
10
|
+
onerror = null;
|
|
11
|
+
send(data) { this.sent.push(data); }
|
|
12
|
+
close() { this.readyState = 'closed'; }
|
|
13
|
+
}
|
|
14
|
+
class FakePeer {
|
|
15
|
+
localDescription = null;
|
|
16
|
+
connectionState = 'new';
|
|
17
|
+
iceConnectionState = 'new';
|
|
18
|
+
channel = new FakeChannel();
|
|
19
|
+
remoteDescription = null;
|
|
20
|
+
candidates = [];
|
|
21
|
+
onicecandidate = null;
|
|
22
|
+
ondatachannel = null;
|
|
23
|
+
onconnectionstatechange = null;
|
|
24
|
+
oniceconnectionstatechange = null;
|
|
25
|
+
async setRemoteDescription(description) { this.remoteDescription = description; }
|
|
26
|
+
async createAnswer() { return { type: 'answer', sdp: 'answer-sdp' }; }
|
|
27
|
+
async setLocalDescription(description) { this.localDescription = description; }
|
|
28
|
+
async addIceCandidate(candidate) { this.candidates.push(candidate); }
|
|
29
|
+
close() { }
|
|
30
|
+
}
|
|
31
|
+
function makeTerminals() {
|
|
32
|
+
const writes = [];
|
|
33
|
+
const resizes = [];
|
|
34
|
+
const closes = [];
|
|
35
|
+
return {
|
|
36
|
+
writes,
|
|
37
|
+
resizes,
|
|
38
|
+
closes,
|
|
39
|
+
create: () => 'term-1',
|
|
40
|
+
write: (id, data) => { writes.push([id, data]); return true; },
|
|
41
|
+
resize: (id, cols, rows) => { resizes.push([id, cols, rows]); return true; },
|
|
42
|
+
close: (id) => { closes.push(id); return true; },
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
test('accepts offer, answers, attaches datachannel and maps terminal io', async () => {
|
|
46
|
+
const sent = [];
|
|
47
|
+
const peer = new FakePeer();
|
|
48
|
+
const terminals = makeTerminals();
|
|
49
|
+
const transport = new CliWebRtcTransport(terminals, (data) => sent.push(data), () => peer);
|
|
50
|
+
await transport.handleSignal({ type: 'webrtc_signal', sessionId: 's1', signalType: 'offer', payload: { type: 'offer' } });
|
|
51
|
+
peer.ondatachannel?.({ channel: peer.channel });
|
|
52
|
+
peer.channel.onmessage?.({ data: JSON.stringify({ v: 1, type: 'terminal_create', cols: 80, rows: 24 }) });
|
|
53
|
+
peer.channel.onmessage?.({ data: JSON.stringify({ v: 1, type: 'terminal_input', terminalId: 'term-1', data: 'pwd\r' }) });
|
|
54
|
+
peer.channel.onmessage?.({ data: JSON.stringify({ v: 1, type: 'terminal_resize', terminalId: 'term-1', cols: 100, rows: 30 }) });
|
|
55
|
+
assert.deepEqual(peer.remoteDescription, { type: 'offer' });
|
|
56
|
+
assert.deepEqual(sent.at(0), { type: 'transport_signal', sessionId: 's1', signalType: 'answer', payload: { type: 'answer', sdp: 'answer-sdp' } });
|
|
57
|
+
assert.deepEqual(JSON.parse(peer.channel.sent[0]), { v: 1, domain: 'terminal', payload: { type: 'terminal_created', terminalId: 'term-1' } });
|
|
58
|
+
assert.deepEqual(terminals.writes, [['term-1', 'pwd\r']]);
|
|
59
|
+
assert.deepEqual(terminals.resizes, [['term-1', 100, 30]]);
|
|
60
|
+
});
|
|
61
|
+
test('denies offer and channel messages without local permission', async () => {
|
|
62
|
+
const sent = [];
|
|
63
|
+
const peer = new FakePeer();
|
|
64
|
+
const terminals = makeTerminals();
|
|
65
|
+
const transport = new CliWebRtcTransport(terminals, (data) => sent.push(data), () => peer, () => false);
|
|
66
|
+
await transport.handleSignal({ type: 'webrtc_signal', sessionId: 's1', signalType: 'offer', payload: { type: 'offer' } });
|
|
67
|
+
transport.attachTerminal('term-1', 's1');
|
|
68
|
+
peer.ondatachannel?.({ channel: peer.channel });
|
|
69
|
+
peer.channel.onmessage?.({ data: JSON.stringify({ v: 1, type: 'terminal_input', terminalId: 'term-1', data: 'blocked' }) });
|
|
70
|
+
assert.equal(peer.remoteDescription, null);
|
|
71
|
+
assert.deepEqual(terminals.writes, []);
|
|
72
|
+
assert.equal(sent.filter((msg) => JSON.stringify(msg).includes('permission_denied')).length, 1);
|
|
73
|
+
});
|
|
74
|
+
test('ignores terminal messages for another session', async () => {
|
|
75
|
+
const peer = new FakePeer();
|
|
76
|
+
const terminals = makeTerminals();
|
|
77
|
+
const transport = new CliWebRtcTransport(terminals, () => { }, () => peer);
|
|
78
|
+
await transport.handleSignal({ type: 'webrtc_signal', sessionId: 's2', signalType: 'offer', payload: { type: 'offer' } });
|
|
79
|
+
transport.attachTerminal('term-1', 's1');
|
|
80
|
+
peer.ondatachannel?.({ channel: peer.channel });
|
|
81
|
+
peer.channel.onmessage?.({ data: JSON.stringify({ v: 1, type: 'terminal_input', terminalId: 'term-1', data: 'bad' }) });
|
|
82
|
+
assert.deepEqual(terminals.writes, []);
|
|
83
|
+
});
|
|
84
|
+
test('routes safe domain datachannel messages through router', async () => {
|
|
85
|
+
const peer = new FakePeer();
|
|
86
|
+
const routed = [];
|
|
87
|
+
const terminals = makeTerminals();
|
|
88
|
+
const transport = new CliWebRtcTransport(terminals, () => { }, () => peer);
|
|
89
|
+
transport.setMessageRouter((_sessionId, msg) => routed.push(msg));
|
|
90
|
+
await transport.handleSignal({ type: 'webrtc_signal', sessionId: 's3', signalType: 'offer', payload: { type: 'offer' } });
|
|
91
|
+
peer.ondatachannel?.({ channel: peer.channel });
|
|
92
|
+
peer.channel.onmessage?.({ data: JSON.stringify({ v: 1, domain: 'prompt', payload: { type: 'send_prompt', text: 'hello' } }) });
|
|
93
|
+
assert.deepEqual(routed, [{ type: 'send_prompt', text: 'hello', transportSessionId: 's3' }]);
|
|
94
|
+
});
|
|
95
|
+
test('sends safe domain response through open datachannel envelope', async () => {
|
|
96
|
+
const peer = new FakePeer();
|
|
97
|
+
const terminals = makeTerminals();
|
|
98
|
+
const transport = new CliWebRtcTransport(terminals, () => { }, () => peer);
|
|
99
|
+
await transport.handleSignal({ type: 'webrtc_signal', sessionId: 's4', signalType: 'offer', payload: { type: 'offer' } });
|
|
100
|
+
peer.ondatachannel?.({ channel: peer.channel });
|
|
101
|
+
assert.equal(transport.sendMessage('s4', { type: 'file_read_result', path: 'a.ts', content: 'ok' }), true);
|
|
102
|
+
assert.deepEqual(JSON.parse(peer.channel.sent[0]), { v: 1, domain: 'file', payload: { type: 'file_read_result', path: 'a.ts', content: 'ok' } });
|
|
103
|
+
});
|
|
104
|
+
//# sourceMappingURL=webrtcTransport.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"webrtcTransport.test.js","sourceRoot":"","sources":["../../src/transport/webrtcTransport.test.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAG1D,MAAM,WAAW;IACf,UAAU,GAAG,MAAM,CAAC;IACpB,IAAI,GAAa,EAAE,CAAC;IACpB,MAAM,GAAwB,IAAI,CAAC;IACnC,SAAS,GAAgD,IAAI,CAAC;IAC9D,OAAO,GAAwB,IAAI,CAAC;IACpC,OAAO,GAAsC,IAAI,CAAC;IAElD,IAAI,CAAC,IAAY,IAAU,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClD,KAAK,KAAW,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,CAAC,CAAC;CAC9C;AAED,MAAM,QAAQ;IACZ,gBAAgB,GAAY,IAAI,CAAC;IACjC,eAAe,GAAG,KAAK,CAAC;IACxB,kBAAkB,GAAG,KAAK,CAAC;IAC3B,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,iBAAiB,GAAY,IAAI,CAAC;IAClC,UAAU,GAAc,EAAE,CAAC;IAC3B,cAAc,GAAqD,IAAI,CAAC;IACxE,aAAa,GAA2D,IAAI,CAAC;IAC7E,uBAAuB,GAAwB,IAAI,CAAC;IACpD,0BAA0B,GAAwB,IAAI,CAAC;IAEvD,KAAK,CAAC,oBAAoB,CAAC,WAAoB,IAAmB,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,CAAC,CAAC;IACzG,KAAK,CAAC,YAAY,KAAuB,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;IACxF,KAAK,CAAC,mBAAmB,CAAC,WAAoB,IAAmB,IAAI,CAAC,gBAAgB,GAAG,WAAW,CAAC,CAAC,CAAC;IACvG,KAAK,CAAC,eAAe,CAAC,SAAkB,IAAmB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC7F,KAAK,KAAU,CAAC;CACjB;AAED,SAAS,aAAa;IACpB,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,MAAM,OAAO,GAAoC,EAAE,CAAC;IACpD,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,OAAO;QACL,MAAM;QACN,OAAO;QACP,MAAM;QACN,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ;QACtB,KAAK,EAAE,CAAC,EAAU,EAAE,IAAY,EAAE,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC9E,MAAM,EAAE,CAAC,EAAU,EAAE,IAAY,EAAE,IAAY,EAAE,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QACpG,KAAK,EAAE,CAAC,EAAU,EAAE,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;KACzD,CAAC;AACJ,CAAC;AAED,IAAI,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;IACnF,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,SAAkB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAEpG,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1H,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1G,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1H,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;IAEjI,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5D,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC;IAClJ,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC9I,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC1D,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;AAC7D,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;IAC5E,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,SAAkB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;IAEjH,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1H,SAAS,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACzC,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IAE5H,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACvC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AAClG,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;IAC/D,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,SAAkB,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAEnF,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1H,SAAS,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACzC,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IAExH,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;IACxE,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,SAAkB,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IACnF,SAAS,CAAC,gBAAgB,CAAC,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAElE,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1H,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;IAEhI,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC/F,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;IAC9E,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,SAAkB,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAEnF,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1H,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAEhD,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IAC3G,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;AACnJ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
export function domainForType(type) {
|
|
2
|
+
if (!type)
|
|
3
|
+
return 'unknown';
|
|
4
|
+
if (type.startsWith('file_'))
|
|
5
|
+
return 'file';
|
|
6
|
+
if (['send_prompt', 'send_key', 'select_option', 'submit_input', 'prompt_options', 'text_input'].includes(type))
|
|
7
|
+
return 'prompt';
|
|
8
|
+
if (type.startsWith('session_') || type.startsWith('request_session_') || type === 'request_new_session' || type === 'delete_session')
|
|
9
|
+
return 'session';
|
|
10
|
+
if (['transcript', 'jsonl_message', 'assistant_done', 'thinking'].includes(type))
|
|
11
|
+
return 'transcript';
|
|
12
|
+
if (type.startsWith('terminal_'))
|
|
13
|
+
return 'terminal';
|
|
14
|
+
return 'unknown';
|
|
15
|
+
}
|
|
16
|
+
export function wrapDomainEnvelope(msg) {
|
|
17
|
+
return { v: 1, domain: domainForType(msg.type), payload: msg };
|
|
18
|
+
}
|
|
19
|
+
export function unwrapDomainEnvelope(msg) {
|
|
20
|
+
if ('domain' in msg) {
|
|
21
|
+
const payload = msg.payload;
|
|
22
|
+
if (typeof payload?.type === 'string')
|
|
23
|
+
return payload;
|
|
24
|
+
}
|
|
25
|
+
if ('type' in msg && typeof msg.type === 'string')
|
|
26
|
+
return msg;
|
|
27
|
+
return null;
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=webrtcTypes.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"webrtcTypes.js","sourceRoot":"","sources":["../../src/transport/webrtcTypes.ts"],"names":[],"mappings":"AAQA,MAAM,UAAU,aAAa,CAAC,IAAwB;IACpD,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,MAAM,CAAC;IAC5C,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,eAAe,EAAE,cAAc,EAAE,gBAAgB,EAAE,YAAY,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IACjI,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,IAAI,IAAI,KAAK,qBAAqB,IAAI,IAAI,KAAK,gBAAgB;QAAE,OAAO,SAAS,CAAC;IACxJ,IAAI,CAAC,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,YAAY,CAAC;IACtG,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,UAAU,CAAC;IACpD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,GAA+C;IAChF,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAqB;IACxD,IAAI,QAAQ,IAAI,GAAG,EAAE,CAAC;QACpB,MAAM,OAAO,GAAI,GAAsB,CAAC,OAAO,CAAC;QAChD,IAAI,OAAO,OAAO,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC;IACxD,CAAC;IACD,IAAI,MAAM,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,GAAG,CAAC;IAC9D,OAAO,IAAI,CAAC;AACd,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import test from 'node:test';
|
|
2
|
+
import assert from 'node:assert/strict';
|
|
3
|
+
import { domainForType, unwrapDomainEnvelope, wrapDomainEnvelope } from './webrtcTypes.js';
|
|
4
|
+
test('domain envelope classifies low-risk domains', () => {
|
|
5
|
+
assert.equal(domainForType('send_prompt'), 'prompt');
|
|
6
|
+
assert.equal(domainForType('request_session_history'), 'session');
|
|
7
|
+
assert.equal(domainForType('jsonl_message'), 'transcript');
|
|
8
|
+
assert.equal(domainForType('file_read'), 'file');
|
|
9
|
+
});
|
|
10
|
+
test('domain envelope wraps and unwraps payloads', () => {
|
|
11
|
+
const msg = { type: 'file_read', path: 'src/index.ts' };
|
|
12
|
+
const envelope = wrapDomainEnvelope(msg);
|
|
13
|
+
assert.equal(envelope.domain, 'file');
|
|
14
|
+
assert.deepEqual(unwrapDomainEnvelope(envelope), msg);
|
|
15
|
+
});
|
|
16
|
+
test('domain envelope accepts legacy terminal payloads', () => {
|
|
17
|
+
const msg = { v: 1, type: 'terminal_input', terminalId: 't1', data: 'ls' };
|
|
18
|
+
assert.deepEqual(unwrapDomainEnvelope(msg), msg);
|
|
19
|
+
});
|
|
20
|
+
//# sourceMappingURL=webrtcTypes.test.js.map
|