server.cjs 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647
  1. const crypto = require('crypto');
  2. const http = require('http');
  3. const fs = require('fs');
  4. const path = require('path');
  5. // ========== WebSocket Protocol (RFC 6455) ==========
  6. const OPCODES = { TEXT: 0x01, CLOSE: 0x08, PING: 0x09, PONG: 0x0A };
  7. const WS_MAGIC = '258EAFA5-E914-47DA-95CA-C5AB0DC85B11';
  8. const MAX_FRAME_PAYLOAD_BYTES = 10 * 1024 * 1024;
  9. function computeAcceptKey(clientKey) {
  10. return crypto.createHash('sha1').update(clientKey + WS_MAGIC).digest('base64');
  11. }
  12. function encodeFrame(opcode, payload) {
  13. const fin = 0x80;
  14. const len = payload.length;
  15. let header;
  16. if (len < 126) {
  17. header = Buffer.alloc(2);
  18. header[0] = fin | opcode;
  19. header[1] = len;
  20. } else if (len < 65536) {
  21. header = Buffer.alloc(4);
  22. header[0] = fin | opcode;
  23. header[1] = 126;
  24. header.writeUInt16BE(len, 2);
  25. } else {
  26. header = Buffer.alloc(10);
  27. header[0] = fin | opcode;
  28. header[1] = 127;
  29. header.writeBigUInt64BE(BigInt(len), 2);
  30. }
  31. return Buffer.concat([header, payload]);
  32. }
  33. function decodeFrame(buffer) {
  34. if (buffer.length < 2) return null;
  35. const secondByte = buffer[1];
  36. const opcode = buffer[0] & 0x0F;
  37. const masked = (secondByte & 0x80) !== 0;
  38. let payloadLen = secondByte & 0x7F;
  39. let offset = 2;
  40. if (!masked) throw new Error('Client frames must be masked');
  41. if (payloadLen === 126) {
  42. if (buffer.length < 4) return null;
  43. payloadLen = buffer.readUInt16BE(2);
  44. offset = 4;
  45. } else if (payloadLen === 127) {
  46. if (buffer.length < 10) return null;
  47. const extendedLen = buffer.readBigUInt64BE(2);
  48. if (extendedLen > BigInt(MAX_FRAME_PAYLOAD_BYTES)) {
  49. throw new Error('WebSocket frame payload exceeds maximum allowed size');
  50. }
  51. payloadLen = Number(extendedLen);
  52. offset = 10;
  53. }
  54. if (payloadLen > MAX_FRAME_PAYLOAD_BYTES) {
  55. throw new Error('WebSocket frame payload exceeds maximum allowed size');
  56. }
  57. const maskOffset = offset;
  58. const dataOffset = offset + 4;
  59. const totalLen = dataOffset + payloadLen;
  60. if (buffer.length < totalLen) return null;
  61. const mask = buffer.slice(maskOffset, dataOffset);
  62. const data = Buffer.alloc(payloadLen);
  63. for (let i = 0; i < payloadLen; i++) {
  64. data[i] = buffer[dataOffset + i] ^ mask[i % 4];
  65. }
  66. return { opcode, payload: data, bytesConsumed: totalLen };
  67. }
  68. // ========== Configuration ==========
  69. const PORT_FILE = process.env.BRAINSTORM_PORT_FILE || null;
  70. const randomPort = () => 49152 + Math.floor(Math.random() * 16383);
  71. // Prefer an explicit port, else the port this session last bound (so a restart
  72. // reuses it and an already-open browser tab reconnects), else a random high port.
  73. function preferredPort() {
  74. if (process.env.BRAINSTORM_PORT) return Number(process.env.BRAINSTORM_PORT);
  75. if (PORT_FILE) {
  76. try {
  77. const p = Number(fs.readFileSync(PORT_FILE, 'utf-8').trim());
  78. if (Number.isInteger(p) && p > 1023 && p < 65536) return p;
  79. } catch (e) { /* no prior port recorded */ }
  80. }
  81. return randomPort();
  82. }
  83. let PORT = preferredPort();
  84. const HOST = process.env.BRAINSTORM_HOST || '127.0.0.1';
  85. const URL_HOST = process.env.BRAINSTORM_URL_HOST || (HOST === '127.0.0.1' ? 'localhost' : HOST);
  86. const SESSION_DIR = process.env.BRAINSTORM_DIR || '/tmp/brainstorm';
  87. const CONTENT_DIR = path.join(SESSION_DIR, 'content');
  88. const STATE_DIR = path.join(SESSION_DIR, 'state');
  89. let ownerPid = process.env.BRAINSTORM_OWNER_PID ? Number(process.env.BRAINSTORM_OWNER_PID) : null;
  90. // Per-session secret key. The companion is reachable by any local browser tab
  91. // and, when bound to a non-loopback host, by any host that can route to it.
  92. // The key authenticates the real client uniformly across loopback, tunnel, and
  93. // remote binds — and defeats DNS rebinding — where a Host/Origin allowlist
  94. // cannot. It rides the served URL as ?key= and is mirrored into a cookie on
  95. // first load so same-origin subresources and the WebSocket carry it for free.
  96. // Persisted alongside the port (BRAINSTORM_TOKEN_FILE) so a restart keeps the
  97. // same key and an already-open tab's cookie still validates.
  98. const TOKEN_FILE = process.env.BRAINSTORM_TOKEN_FILE || null;
  99. function generateToken() {
  100. return crypto.randomBytes(32).toString('hex');
  101. }
  102. function initialToken() {
  103. if (process.env.BRAINSTORM_TOKEN) {
  104. return { value: process.env.BRAINSTORM_TOKEN, source: 'env' };
  105. }
  106. if (TOKEN_FILE) {
  107. try {
  108. const t = fs.readFileSync(TOKEN_FILE, 'utf-8').trim();
  109. if (/^[0-9a-f]{32,}$/i.test(t)) return { value: t, source: 'file' };
  110. } catch (e) { /* no prior token recorded */ }
  111. }
  112. return { value: generateToken(), source: 'generated' };
  113. }
  114. const tokenInfo = initialToken();
  115. let TOKEN = tokenInfo.value;
  116. let tokenSource = tokenInfo.source;
  117. let COOKIE_NAME = 'brainstorm-key-' + PORT; // refined to the actual bound port in onListen
  118. const MIME_TYPES = {
  119. '.html': 'text/html', '.css': 'text/css', '.js': 'application/javascript',
  120. '.json': 'application/json', '.png': 'image/png', '.jpg': 'image/jpeg',
  121. '.jpeg': 'image/jpeg', '.gif': 'image/gif', '.svg': 'image/svg+xml'
  122. };
  123. // ========== Templates and Constants ==========
  124. const WAITING_PAGE = `<!DOCTYPE html>
  125. <html>
  126. <head><meta charset="utf-8"><title>Brainstorm Companion</title>
  127. <style>body { font-family: system-ui, sans-serif; padding: 2rem; max-width: 800px; margin: 0 auto; }
  128. h1 { color: #333; } p { color: #666; }</style>
  129. </head>
  130. <body><h1>Brainstorm Companion</h1>
  131. <p>Waiting for the agent to push a screen...</p></body></html>`;
  132. const FORBIDDEN_PAGE = `<!DOCTYPE html>
  133. <html>
  134. <head><meta charset="utf-8"><title>Session key required</title>
  135. <style>body { font-family: system-ui, sans-serif; padding: 2rem; max-width: 800px; margin: 0 auto; }
  136. h1 { color: #333; } p { color: #666; } code { background: #f0f0f0; padding: 0.1em 0.3em; border-radius: 4px; }</style>
  137. </head>
  138. <body><h1>Session key required</h1>
  139. <p>This page needs the full URL your coding agent gave you, including the
  140. <code>?key=&hellip;</code> part. Copy the complete URL and open it again.</p></body></html>`;
  141. function bootstrapPage(key) {
  142. const jsonKey = JSON.stringify(String(key));
  143. return `<!DOCTYPE html>
  144. <html>
  145. <head><meta charset="utf-8"><title>Opening Brainstorm Companion</title></head>
  146. <body>
  147. <script>
  148. try { sessionStorage.setItem('brainstorm-session-key', ${jsonKey}); } catch (e) {}
  149. location.replace('/');
  150. </script>
  151. </body>
  152. </html>`;
  153. }
  154. const frameTemplate = fs.readFileSync(path.join(__dirname, 'frame-template.html'), 'utf-8');
  155. const helperScript = fs.readFileSync(path.join(__dirname, 'helper.js'), 'utf-8');
  156. const helperInjection = '<script>\n' + helperScript + '\n</script>';
  157. // ========== Helper Functions ==========
  158. function isFullDocument(html) {
  159. const trimmed = html.trimStart().toLowerCase();
  160. return trimmed.startsWith('<!doctype') || trimmed.startsWith('<html');
  161. }
  162. function wrapInFrame(content) {
  163. return frameTemplate.replace('<!-- CONTENT -->', content);
  164. }
  165. function getNewestScreen() {
  166. const files = fs.readdirSync(CONTENT_DIR)
  167. .filter(f => !f.startsWith('.') && f.endsWith('.html'))
  168. .map(f => {
  169. const fp = path.join(CONTENT_DIR, f);
  170. if (!isRegularFileInsideContentDir(fp)) return null;
  171. return { path: fp, mtime: fs.statSync(fp).mtime.getTime() };
  172. })
  173. .filter(Boolean)
  174. .sort((a, b) => b.mtime - a.mtime);
  175. return files.length > 0 ? files[0].path : null;
  176. }
  177. function urlHostForHttp(host) {
  178. const h = String(host);
  179. if (h.startsWith('[') && h.endsWith(']')) return h;
  180. return h.includes(':') ? '[' + h + ']' : h;
  181. }
  182. function companionUrl() {
  183. return 'http://' + urlHostForHttp(URL_HOST) + ':' + PORT + '/?key=' + TOKEN;
  184. }
  185. function browserLauncherForPlatform(url, {
  186. platform = process.platform,
  187. osRelease = require('os').release(),
  188. env = process.env
  189. } = {}) {
  190. const isWSL = platform === 'linux' && /microsoft/i.test(osRelease);
  191. if (platform === 'darwin') return { bin: 'open', args: [url] };
  192. if (platform === 'win32' || isWSL) {
  193. return { bin: 'rundll32.exe', args: ['url.dll,FileProtocolHandler', url] };
  194. }
  195. if (env.DISPLAY || env.WAYLAND_DISPLAY) return { bin: 'xdg-open', args: [url] };
  196. return null;
  197. }
  198. function isRegularFileInsideContentDir(filePath) {
  199. let stat, realContentDir, realFilePath;
  200. try {
  201. stat = fs.lstatSync(filePath);
  202. if (stat.isSymbolicLink()) return false;
  203. if (!stat.isFile()) return false;
  204. if (stat.nlink !== 1) return false;
  205. realContentDir = fs.realpathSync(CONTENT_DIR);
  206. realFilePath = fs.realpathSync(filePath);
  207. } catch (e) {
  208. return false;
  209. }
  210. return realFilePath.startsWith(realContentDir + path.sep);
  211. }
  212. // ========== Authentication ==========
  213. function timingSafeEqualStr(a, b) {
  214. const ab = Buffer.from(String(a));
  215. const bb = Buffer.from(String(b));
  216. if (ab.length !== bb.length) return false;
  217. return crypto.timingSafeEqual(ab, bb);
  218. }
  219. function parseCookies(header) {
  220. const out = {};
  221. if (!header) return out;
  222. for (const part of header.split(';')) {
  223. const eq = part.indexOf('=');
  224. if (eq < 0) continue;
  225. out[part.slice(0, eq).trim()] = part.slice(eq + 1).trim();
  226. }
  227. return out;
  228. }
  229. // A request is authorized if it carries the session key as ?key= or as the
  230. // session cookie. Both are compared in constant time.
  231. function isAuthorized(req) {
  232. const q = req.url.indexOf('?');
  233. if (q >= 0) {
  234. const params = new URLSearchParams(req.url.slice(q + 1));
  235. if (params.has('key')) {
  236. const key = params.get('key');
  237. return Boolean(key && timingSafeEqualStr(key, TOKEN));
  238. }
  239. }
  240. const cookie = parseCookies(req.headers['cookie'])[COOKIE_NAME];
  241. if (cookie && timingSafeEqualStr(cookie, TOKEN)) return true;
  242. return false;
  243. }
  244. function pathnameOf(url) {
  245. const q = url.indexOf('?');
  246. return q >= 0 ? url.slice(0, q) : url;
  247. }
  248. function queryKey(url) {
  249. const q = url.indexOf('?');
  250. if (q < 0) return null;
  251. return new URLSearchParams(url.slice(q + 1)).get('key');
  252. }
  253. function securityHeaders(headers = {}) {
  254. return {
  255. 'Referrer-Policy': 'no-referrer',
  256. 'Cache-Control': 'no-store',
  257. 'X-Frame-Options': 'DENY',
  258. 'Content-Security-Policy': "frame-ancestors 'none'",
  259. 'Cross-Origin-Resource-Policy': 'same-origin',
  260. ...headers
  261. };
  262. }
  263. function isAllowedWebSocketOrigin(req) {
  264. const origin = req.headers.origin;
  265. if (!origin) return true;
  266. const host = req.headers.host;
  267. if (!host) return false;
  268. return origin === 'http://' + host;
  269. }
  270. // ========== HTTP Request Handler ==========
  271. function handleRequest(req, res) {
  272. if (!isAuthorized(req)) {
  273. res.writeHead(403, securityHeaders({ 'Content-Type': 'text/html; charset=utf-8' }));
  274. res.end(FORBIDDEN_PAGE);
  275. return;
  276. }
  277. touchActivity(); // only authorized requests count as activity
  278. // Mirror the key into a cookie so same-origin subresources (/files/*) can
  279. // authenticate after bootstrap. HttpOnly keeps it away from page scripts; the
  280. // WebSocket Origin check below is what blocks cross-origin localhost injection.
  281. res.setHeader('Set-Cookie',
  282. COOKIE_NAME + '=' + TOKEN + '; HttpOnly; SameSite=Strict; Path=/');
  283. const pathname = pathnameOf(req.url);
  284. const keyFromQuery = queryKey(req.url);
  285. if (req.method === 'GET' && pathname === '/' && keyFromQuery && timingSafeEqualStr(keyFromQuery, TOKEN)) {
  286. res.writeHead(200, securityHeaders({ 'Content-Type': 'text/html; charset=utf-8' }));
  287. res.end(bootstrapPage(keyFromQuery));
  288. } else if (req.method === 'GET' && pathname === '/') {
  289. const screenFile = getNewestScreen();
  290. let html = screenFile
  291. ? (raw => isFullDocument(raw) ? raw : wrapInFrame(raw))(fs.readFileSync(screenFile, 'utf-8'))
  292. : WAITING_PAGE;
  293. if (html.includes('</body>')) {
  294. html = html.replace('</body>', helperInjection + '\n</body>');
  295. } else {
  296. html += helperInjection;
  297. }
  298. res.writeHead(200, securityHeaders({ 'Content-Type': 'text/html; charset=utf-8' }));
  299. res.end(html);
  300. } else if (req.method === 'GET' && pathname.startsWith('/files/')) {
  301. const fileName = path.basename(pathname.slice(7));
  302. const filePath = path.join(CONTENT_DIR, fileName);
  303. // Reject empty/dotfile names and anything that isn't a regular file —
  304. // `/files/` would otherwise resolve to CONTENT_DIR and crash readFileSync (EISDIR).
  305. if (!fileName || fileName.startsWith('.') || !isRegularFileInsideContentDir(filePath)) {
  306. res.writeHead(404, securityHeaders());
  307. res.end('Not found');
  308. return;
  309. }
  310. const ext = path.extname(filePath).toLowerCase();
  311. const contentType = MIME_TYPES[ext] || 'application/octet-stream';
  312. res.writeHead(200, securityHeaders({ 'Content-Type': contentType }));
  313. res.end(fs.readFileSync(filePath));
  314. } else {
  315. res.writeHead(404, securityHeaders());
  316. res.end('Not found');
  317. }
  318. }
  319. // ========== WebSocket Connection Handling ==========
  320. const clients = new Set();
  321. function handleUpgrade(req, socket) {
  322. if (!isAuthorized(req) || !isAllowedWebSocketOrigin(req)) { socket.destroy(); return; }
  323. const key = req.headers['sec-websocket-key'];
  324. if (!key) { socket.destroy(); return; }
  325. const accept = computeAcceptKey(key);
  326. socket.write(
  327. 'HTTP/1.1 101 Switching Protocols\r\n' +
  328. 'Upgrade: websocket\r\n' +
  329. 'Connection: Upgrade\r\n' +
  330. 'Sec-WebSocket-Accept: ' + accept + '\r\n\r\n'
  331. );
  332. let buffer = Buffer.alloc(0);
  333. clients.add(socket);
  334. socket.on('data', (chunk) => {
  335. buffer = Buffer.concat([buffer, chunk]);
  336. while (buffer.length > 0) {
  337. let result;
  338. try {
  339. result = decodeFrame(buffer);
  340. } catch (e) {
  341. socket.end(encodeFrame(OPCODES.CLOSE, Buffer.alloc(0)));
  342. clients.delete(socket);
  343. return;
  344. }
  345. if (!result) break;
  346. buffer = buffer.slice(result.bytesConsumed);
  347. switch (result.opcode) {
  348. case OPCODES.TEXT:
  349. handleMessage(result.payload.toString());
  350. break;
  351. case OPCODES.CLOSE:
  352. socket.end(encodeFrame(OPCODES.CLOSE, Buffer.alloc(0)));
  353. clients.delete(socket);
  354. return;
  355. case OPCODES.PING:
  356. socket.write(encodeFrame(OPCODES.PONG, result.payload));
  357. break;
  358. case OPCODES.PONG:
  359. break;
  360. default: {
  361. const closeBuf = Buffer.alloc(2);
  362. closeBuf.writeUInt16BE(1003);
  363. socket.end(encodeFrame(OPCODES.CLOSE, closeBuf));
  364. clients.delete(socket);
  365. return;
  366. }
  367. }
  368. }
  369. });
  370. socket.on('close', () => clients.delete(socket));
  371. socket.on('error', () => clients.delete(socket));
  372. }
  373. function handleMessage(text) {
  374. let event;
  375. try {
  376. event = JSON.parse(text);
  377. } catch (e) {
  378. console.error('Failed to parse WebSocket message:', e.message);
  379. return;
  380. }
  381. touchActivity();
  382. console.log(JSON.stringify({ source: 'user-event', ...event }));
  383. if (event && event.choice) {
  384. const eventsFile = path.join(STATE_DIR, 'events');
  385. fs.appendFileSync(eventsFile, JSON.stringify(event) + '\n');
  386. }
  387. }
  388. function broadcast(msg) {
  389. const frame = encodeFrame(OPCODES.TEXT, Buffer.from(JSON.stringify(msg)));
  390. for (const socket of clients) {
  391. try { socket.write(frame); } catch (e) { clients.delete(socket); }
  392. }
  393. }
  394. // Best-effort: open the user's browser the first time a screen is actually ready
  395. // to show. Skips when disabled, on a non-loopback (remote) bind, or when a
  396. // browser is already connected. Override the launcher with BRAINSTORM_OPEN_CMD.
  397. let browserOpened = false;
  398. function maybeOpenBrowser() {
  399. if (browserOpened) return;
  400. browserOpened = true;
  401. if (!process.env.BRAINSTORM_OPEN) return; // opt-in: only after the user approves the companion
  402. if (HOST !== '127.0.0.1' && HOST !== 'localhost') return;
  403. if (clients.size > 0) return; // the user already opened it
  404. const url = companionUrl(); // must carry the key or the gate 403s it
  405. const cp = require('child_process');
  406. // Operator-provided launcher: run as given (this env var is trusted operator input).
  407. if (process.env.BRAINSTORM_OPEN_CMD) {
  408. try { cp.exec(process.env.BRAINSTORM_OPEN_CMD + ' ' + JSON.stringify(url), () => {}); } catch (e) { /* best effort */ }
  409. return;
  410. }
  411. // Platform launchers: pass the URL as an argv element via execFile (no shell),
  412. // so a url-host containing shell metacharacters can't inject a command.
  413. const launcher = browserLauncherForPlatform(url);
  414. if (!launcher) return; // headless: nothing to open
  415. try { cp.execFile(launcher.bin, launcher.args, () => {}); } catch (e) { /* best effort */ }
  416. }
  417. // ========== Activity Tracking ==========
  418. // Idle timeout: shut down after this long with no activity. Default 4 hours;
  419. // override with BRAINSTORM_IDLE_TIMEOUT_MS (start-server.sh: --idle-timeout-minutes).
  420. const IDLE_TIMEOUT_MS = (() => {
  421. const ms = Number(process.env.BRAINSTORM_IDLE_TIMEOUT_MS);
  422. return Number.isFinite(ms) && ms > 0 ? ms : 4 * 60 * 60 * 1000;
  423. })();
  424. // How often the watchdog checks for owner-death / idleness. Configurable mainly
  425. // so tests can run fast; production default is 60s.
  426. const LIFECYCLE_CHECK_MS = (() => {
  427. const ms = Number(process.env.BRAINSTORM_LIFECYCLE_CHECK_MS);
  428. return Number.isFinite(ms) && ms > 0 ? ms : 60 * 1000;
  429. })();
  430. let lastActivity = Date.now();
  431. function touchActivity() {
  432. lastActivity = Date.now();
  433. }
  434. // ========== File Watching ==========
  435. const debounceTimers = new Map();
  436. // ========== Server Startup ==========
  437. function startServer() {
  438. if (!fs.existsSync(CONTENT_DIR)) fs.mkdirSync(CONTENT_DIR, { recursive: true });
  439. if (!fs.existsSync(STATE_DIR)) fs.mkdirSync(STATE_DIR, { recursive: true });
  440. // Track known files to distinguish new screens from updates.
  441. // macOS fs.watch reports 'rename' for both new files and overwrites,
  442. // so we can't rely on eventType alone.
  443. const knownFiles = new Set(
  444. fs.readdirSync(CONTENT_DIR).filter(f => !f.startsWith('.') && f.endsWith('.html'))
  445. );
  446. const server = http.createServer(handleRequest);
  447. server.on('upgrade', handleUpgrade);
  448. const watcher = fs.watch(CONTENT_DIR, (eventType, filename) => {
  449. if (!filename || filename.startsWith('.') || !filename.endsWith('.html')) return;
  450. if (debounceTimers.has(filename)) clearTimeout(debounceTimers.get(filename));
  451. debounceTimers.set(filename, setTimeout(() => {
  452. debounceTimers.delete(filename);
  453. const filePath = path.join(CONTENT_DIR, filename);
  454. if (!fs.existsSync(filePath)) return; // file was deleted
  455. touchActivity();
  456. if (!knownFiles.has(filename)) {
  457. knownFiles.add(filename);
  458. const eventsFile = path.join(STATE_DIR, 'events');
  459. if (fs.existsSync(eventsFile)) fs.unlinkSync(eventsFile);
  460. console.log(JSON.stringify({ type: 'screen-added', file: filePath }));
  461. maybeOpenBrowser();
  462. } else {
  463. console.log(JSON.stringify({ type: 'screen-updated', file: filePath }));
  464. }
  465. broadcast({ type: 'reload' });
  466. }, 100));
  467. });
  468. watcher.on('error', (err) => console.error('fs.watch error:', err.message));
  469. function shutdown(reason) {
  470. console.log(JSON.stringify({ type: 'server-stopped', reason }));
  471. const infoFile = path.join(STATE_DIR, 'server-info');
  472. if (fs.existsSync(infoFile)) fs.unlinkSync(infoFile);
  473. fs.writeFileSync(
  474. path.join(STATE_DIR, 'server-stopped'),
  475. JSON.stringify({ reason, timestamp: Date.now() }) + '\n'
  476. );
  477. watcher.close();
  478. clearInterval(lifecycleCheck);
  479. // Close any upgraded WebSocket sockets so server.close() can complete and
  480. // the process actually exits instead of lingering on an open connection.
  481. for (const socket of clients) {
  482. try { socket.destroy(); } catch (e) { /* already gone */ }
  483. }
  484. server.close(() => process.exit(0));
  485. }
  486. function ownerAlive() {
  487. if (!ownerPid) return true;
  488. try { process.kill(ownerPid, 0); return true; } catch (e) { return e.code === 'EPERM'; }
  489. }
  490. // Periodically exit if the owner process died or we've been idle too long.
  491. const lifecycleCheck = setInterval(() => {
  492. if (!ownerAlive()) shutdown('owner process exited');
  493. else if (Date.now() - lastActivity > IDLE_TIMEOUT_MS) shutdown('idle timeout');
  494. }, LIFECYCLE_CHECK_MS);
  495. lifecycleCheck.unref();
  496. // Validate owner PID at startup. If it's already dead, the PID resolution
  497. // was wrong (common on WSL, Tailscale SSH, and cross-user scenarios).
  498. // Disable monitoring and rely on the idle timeout instead.
  499. if (ownerPid) {
  500. try { process.kill(ownerPid, 0); }
  501. catch (e) {
  502. if (e.code !== 'EPERM') {
  503. console.log(JSON.stringify({ type: 'owner-pid-invalid', pid: ownerPid, reason: 'dead at startup' }));
  504. ownerPid = null;
  505. }
  506. }
  507. }
  508. // If the preferred port is already taken (e.g. a previous server is still
  509. // alive), fall back to a random port once instead of failing.
  510. let triedFallback = false;
  511. function onListen() {
  512. // Cookie name keys on the ACTUAL bound port (may differ from the preferred
  513. // one after an EADDRINUSE fallback) so it can't collide with another server's
  514. // cookie in the shared localhost jar.
  515. COOKIE_NAME = 'brainstorm-key-' + PORT;
  516. // Record the bound port AND token so the next restart of this session reuses
  517. // them — but ONLY when we got our preferred port. On a fallback we bound a
  518. // *different* port because someone else holds the preferred one; persisting
  519. // would overwrite the shared files and strand that other session's open tab.
  520. if (PORT_FILE && !triedFallback) {
  521. try { fs.writeFileSync(PORT_FILE, String(PORT)); } catch (e) { /* best effort */ }
  522. if (TOKEN_FILE) {
  523. try { fs.writeFileSync(TOKEN_FILE, TOKEN, { mode: 0o600 }); } catch (e) { /* best effort */ }
  524. }
  525. }
  526. const info = JSON.stringify({
  527. type: 'server-started', port: Number(PORT), host: HOST,
  528. url_host: URL_HOST, url: companionUrl(),
  529. screen_dir: CONTENT_DIR, state_dir: STATE_DIR, idle_timeout_ms: IDLE_TIMEOUT_MS
  530. });
  531. console.log(info);
  532. // server-info embeds the key — keep it owner-only.
  533. fs.writeFileSync(path.join(STATE_DIR, 'server-info'), info + '\n', { mode: 0o600 });
  534. }
  535. server.on('error', (err) => {
  536. if (err.code === 'EADDRINUSE' && !triedFallback) {
  537. if (tokenSource === 'env') {
  538. console.error('Server failed to bind: preferred port is in use and BRAINSTORM_TOKEN is set; refusing fallback with explicit token');
  539. process.exit(1);
  540. }
  541. triedFallback = true;
  542. PORT = randomPort();
  543. if (tokenSource === 'file') {
  544. TOKEN = generateToken();
  545. tokenSource = 'generated-fallback';
  546. }
  547. server.listen(PORT, HOST, onListen);
  548. } else {
  549. console.error('Server failed to bind:', err.message);
  550. process.exit(1);
  551. }
  552. });
  553. server.listen(PORT, HOST, onListen);
  554. }
  555. if (require.main === module) {
  556. startServer();
  557. }
  558. module.exports = {
  559. computeAcceptKey,
  560. encodeFrame,
  561. decodeFrame,
  562. browserLauncherForPlatform,
  563. OPCODES,
  564. MAX_FRAME_PAYLOAD_BYTES
  565. };