Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

chore: support multiple websockets

Browse source 
This will allow peer to peer communication (actually peer to proxy to peer, where peers are trace viewer webapp and vscode, and testServer is the proxy)
This commit is contained in:
Rui Figueira 2024-09-14 18:16:12 +01:00
parent aeb4d182f7
commit 77e2624a68
1 changed files with 17 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
packages/playwright-core/src/utils/httpServer.ts
View file

@ -17,7 +17,7 @@
import type http from 'http'; import type http from 'http';
import fs from 'fs'; import fs from 'fs';
import path from 'path'; import path from 'path';
import { mime, wsServer } from '../utilsBundle'; import { mime, wsServer, type WebSocket } from '../utilsBundle';
import { assert } from './debug'; import { assert } from './debug';
import { createHttpServer } from './network'; import { createHttpServer } from './network';
import { ManualPromise } from './manualPromise'; import { ManualPromise } from './manualPromise';
@ -81,9 +81,17 @@ export class HttpServer {
assert(!this._wsGuid, 'can only create one main websocket transport per server'); assert(!this._wsGuid, 'can only create one main websocket transport per server');
this._wsGuid = guid || createGuid(); this._wsGuid = guid || createGuid();
const wss = new wsServer({ server: this._server, path: '/' + this._wsGuid }); const wss = new wsServer({ server: this._server, path: '/' + this._wsGuid });
const openedWebSockets = () => [...wss.clients].filter(ws => ws.readyState === ws.OPEN);
transport.sendEvent = (method, params) => {
const msg = JSON.stringify({ method, params });
for (const ws of openedWebSockets())
ws.send(msg);
};
transport.close = () => {
for (const ws of openedWebSockets())
ws.close();
};
wss.on('connection', ws => { wss.on('connection', ws => {
transport.sendEvent = (method, params) => ws.send(JSON.stringify({ method, params }));
transport.close = () => ws.close();
ws.on('message', async message => { ws.on('message', async message => {
const { id, method, params } = JSON.parse(String(message)); const { id, method, params } = JSON.parse(String(message));
try { try {
@ -93,8 +101,12 @@ export class HttpServer {
ws.send(JSON.stringify({ id, error: String(e) })); ws.send(JSON.stringify({ id, error: String(e) }));
} }
}); });
ws.on('close', () => transport.onclose()); const onclose = () => {
ws.on('error', () => transport.onclose()); if (openedWebSockets().length === 0)
transport.onclose();
};
ws.on('close', onclose);
ws.on('error', onclose);
}); });
} }