fix(scripts): validate mock webview bridge port (#3073)

Co-authored-by: Steven Enamakel <enamakel@tinyhumans.ai>
This commit is contained in:
Alexzhu
2026-06-01 22:15:41 -07:00
committed by GitHub
co-authored by Steven Enamakel
parent c6f4c65fe3
commit 2a67e6ab3f
2 changed files with 55 additions and 1 deletions
@@ -0,0 +1,31 @@
import assert from 'node:assert/strict';
import { spawnSync } from 'node:child_process';
import { dirname, resolve } from 'node:path';
import { fileURLToPath } from 'node:url';
import { test } from 'node:test';
const HERE = dirname(fileURLToPath(import.meta.url));
const SCRIPT = resolve(HERE, '..', 'mock-webview-bridge.mjs');
function run(args) {
return spawnSync(process.execPath, [SCRIPT, ...args], {
encoding: 'utf8',
});
}
test('mock-webview-bridge --help prints usage without opening a listener', () => {
const result = run(['--help']);
assert.equal(result.status, 0, result.stderr);
assert.match(result.stdout, /Usage: node scripts\/mock-webview-bridge\.mjs \[port\]/);
assert.equal(result.stderr, '');
});
test('mock-webview-bridge rejects invalid ports before WebSocket startup', () => {
for (const args of [['nope'], ['0'], ['65536'], ['9826', 'extra'], ['9826', '']]) {
const result = run(args);
assert.equal(result.status, 2, result.stdout);
assert.doesNotMatch(result.stderr, /ERR_SOCKET_BAD_PORT/);
}
});
+24 -1
View File
@@ -6,7 +6,30 @@
import { WebSocketServer } from 'ws';
const port = Number(process.argv[2] ?? 9826);
function usage() {
return 'Usage: node scripts/mock-webview-bridge.mjs [port]';
}
function readPortArg() {
const args = process.argv.slice(2);
const [rawPort] = args;
if (rawPort === '--help' || rawPort === '-h') {
console.log(usage());
process.exit(0);
}
if (args.length > 1) {
console.error(usage());
process.exit(2);
}
const port = Number(rawPort ?? 9826);
if (!Number.isInteger(port) || port <= 0 || port >= 65536) {
console.error('[mock-bridge] port must be an integer between 1 and 65535');
process.exit(2);
}
return port;
}
const port = readPortArg();
const wss = new WebSocketServer({ host: '127.0.0.1', port });
console.log(`[mock-bridge] listening on ws://127.0.0.1:${port}`);