@rocicorp/zero 1.12.0-canary.8 → 1.12.0-canary.9

package/out/zero-cache/src/types/websocket-handoff.js.map

@@ -1 +1 @@
-{"version":3,"file":"websocket-handoff.js","names":[],"sources":["../../../../../zero-cache/src/types/websocket-handoff.ts"],"sourcesContent":["import type {IncomingMessage} from 'node:http';\nimport {Server} from 'node:http';\nimport type {Socket} from 'node:net';\nimport type {LogContext} from '@rocicorp/logger';\nimport {WebSocketServer, type ServerOptions, type WebSocket} from 'ws';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport {serializableSubset, type IncomingMessageSubset} from './http.ts';\nimport type {MESSAGE_TYPES} from './processes.ts';\nimport {type Receiver, type Sender, type Worker} from './processes.ts';\nimport {closeWithError, PROTOCOL_ERROR} from './ws.ts';\n\nexport type HandoffSpec<P> = {\n  payload: P;\n  sender: Sender;\n};\n\n/**\n * The WebSocketHandoff is a function that either returns the payload and\n * receiver, or invokes the specified `callback` with the payload and\n * receiver. It must not do both.\n *\n * Similarly, an error can be handled by throwing synchronously from the\n * function, or invoking the `onerror` callback.\n */\nexport type WebSocketHandoff<P> = (\n  message: IncomingMessageSubset,\n  callback: (h: HandoffSpec<P>) => void,\n  onerror: (reason: unknown) => void,\n) => HandoffSpec<P> | void;\n\nexport type WebSocketReceiver<P> = (\n  ws: WebSocket,\n  payload: P,\n  msg: IncomingMessageSubset,\n) => void;\n\nexport type WebSocketHandoffHandler = (\n  message: IncomingMessageSubset,\n  socket: Socket,\n  head: ArrayBuffer,\n) => void;\n\n/**\n * Installs websocket handoff logic from either an http.Server\n * receiving requests, or a parent Worker process\n * that is handing off requests to this process.\n */\nexport function createWebSocketHandoffHandler<P>(\n  lc: LogContext,\n  handoff: WebSocketHandoff<P>,\n  serverOptions?: ServerOptions,\n): WebSocketHandoffHandler {\n  const wss = new WebSocketServer(\n    serverOptions ?? {\n      noServer: true,\n    },\n  );\n  return (\n    message: IncomingMessageSubset,\n    socket: Socket,\n    head: ArrayBuffer,\n  ) => {\n    let sent = false;\n\n    function send({payload, sender}: HandoffSpec<P>) {\n      assert(!sent, 'Handoff callback already invoked');\n      sent = true;\n\n      const data = [\n        'handoff',\n        {\n          message: serializableSubset(message),\n          head,\n          payload,\n        },\n      ] satisfies Handoff<P>;\n\n      // \"This event is guaranteed to be passed an instance of the <net.Socket> class\"\n      // https://nodejs.org/api/http.html#event-upgrade\n      sender.send(data, socket);\n    }\n\n    function onError(error: unknown) {\n      // Returning an error on the HTTP handshake looks like a hanging connection\n      // (at least from Chrome) and doesn't report any meaningful error in the browser.\n      // Instead, finish the upgrade to a websocket and then close it with an error.\n      wss.handleUpgrade(\n        message as IncomingMessage,\n        socket,\n        Buffer.from(head),\n        ws => closeWithError(lc, ws, error, PROTOCOL_ERROR),\n      );\n    }\n\n    try {\n      const spec = handoff(message, send, onError);\n      if (spec) {\n        send(spec);\n      }\n    } catch (error) {\n      onError(error);\n    }\n  };\n}\n\n/**\n * Installs websocket handoff logic from either an http.Server\n * receiving requests, or a parent Worker process\n * that is handing off requests to this process.\n */\nexport function installWebSocketHandoff<P>(\n  lc: LogContext,\n  handoff: WebSocketHandoff<P>,\n  source: Server | Worker,\n  serverOptions?: ServerOptions,\n) {\n  const handle = createWebSocketHandoffHandler(lc, handoff, serverOptions);\n\n  if (source instanceof Server) {\n    // handoff messages from an HTTP server\n    source.on('upgrade', handle);\n  } else {\n    // handoff messages from this worker's parent.\n    source.onMessageType<Handoff<P>>('handoff', (msg, socket) => {\n      const {message, head} = msg;\n      handle(message, socket as Socket, head);\n    });\n  }\n}\n\nexport function installWebSocketReceiver<P>(\n  lc: LogContext,\n  server: WebSocketServer,\n  receive: WebSocketReceiver<P>,\n  receiver: Receiver,\n  onAbort?: ((payload: P) => void) | undefined,\n  onReceived?: ((payload: P) => void) | undefined,\n) {\n  receiver.onMessageType<Handoff<P>>('handoff', (msg, socket) => {\n    const {message, head, payload} = msg;\n    onReceived?.(payload);\n    // Per https://nodejs.org/api/child_process.html#subprocesssendmessage-sendhandle-options-callback\n    //\n    // > Any 'message' handlers in the subprocess should verify that socket\n    // > exists, as the connection may have been closed during the time it\n    // > takes to send the connection to the child.\n    if (!socket) {\n      lc.warn?.('websocket closed during handoff');\n      onAbort?.(payload);\n      return;\n    }\n    try {\n      server.handleUpgrade(\n        message as IncomingMessage,\n        socket as Socket,\n        Buffer.from(head),\n        ws => {\n          // Guard against WebSocket being closed during handoff.\n          // This can happen due to network issues or client disconnection\n          // between the time the socket was sent and when handleUpgrade completes.\n          if (ws.readyState === ws.CLOSED || ws.readyState === ws.CLOSING) {\n            lc.warn?.('websocket closed during upgrade, skipping receive');\n            onAbort?.(payload);\n            return;\n          }\n          receive(ws, payload, message);\n        },\n      );\n    } catch (e) {\n      lc.warn?.('error during websocket handleUpgrade', e);\n      (socket as Socket)?.destroy?.();\n      onAbort?.(payload);\n    }\n  });\n}\n\nexport type Handoff<P> = [\n  typeof MESSAGE_TYPES.handoff,\n  {\n    message: IncomingMessageSubset;\n    head: ArrayBuffer;\n    payload: P;\n  },\n];\n"],"mappings":";;;;;;;;;;;;AA+CA,SAAgB,8BACd,IACA,SACA,eACyB;CACzB,MAAM,MAAM,IAAI,gBACd,iBAAiB,EACf,UAAU,KACZ,CACF;CACA,QACE,SACA,QACA,SACG;EACH,IAAI,OAAO;EAEX,SAAS,KAAK,EAAC,SAAS,UAAyB;GAC/C,OAAO,CAAC,MAAM,kCAAkC;GAChD,OAAO;GAEP,MAAM,OAAO,CACX,WACA;IACE,SAAS,mBAAmB,OAAO;IACnC;IACA;GACF,CACF;GAIA,OAAO,KAAK,MAAM,MAAM;EAC1B;EAEA,SAAS,QAAQ,OAAgB;GAI/B,IAAI,cACF,SACA,QACA,OAAO,KAAK,IAAI,IAChB,OAAM,eAAe,IAAI,IAAI,OAAO,cAAc,CACpD;EACF;EAEA,IAAI;GACF,MAAM,OAAO,QAAQ,SAAS,MAAM,OAAO;GAC3C,IAAI,MACF,KAAK,IAAI;EAEb,SAAS,OAAO;GACd,QAAQ,KAAK;EACf;CACF;AACF;;;;;;AAOA,SAAgB,wBACd,IACA,SACA,QACA,eACA;CACA,MAAM,SAAS,8BAA8B,IAAI,SAAS,aAAa;CAEvE,IAAI,kBAAkB,QAEpB,OAAO,GAAG,WAAW,MAAM;MAG3B,OAAO,cAA0B,YAAY,KAAK,WAAW;EAC3D,MAAM,EAAC,SAAS,SAAQ;EACxB,OAAO,SAAS,QAAkB,IAAI;CACxC,CAAC;AAEL;AAEA,SAAgB,yBACd,IACA,QACA,SACA,UACA,SACA,YACA;CACA,SAAS,cAA0B,YAAY,KAAK,WAAW;EAC7D,MAAM,EAAC,SAAS,MAAM,YAAW;EACjC,aAAa,OAAO;EAMpB,IAAI,CAAC,QAAQ;GACX,GAAG,OAAO,iCAAiC;GAC3C,UAAU,OAAO;GACjB;EACF;EACA,IAAI;GACF,OAAO,cACL,SACA,QACA,OAAO,KAAK,IAAI,IAChB,OAAM;IAIJ,IAAI,GAAG,eAAe,GAAG,UAAU,GAAG,eAAe,GAAG,SAAS;KAC/D,GAAG,OAAO,mDAAmD;KAC7D,UAAU,OAAO;KACjB;IACF;IACA,QAAQ,IAAI,SAAS,OAAO;GAC9B,CACF;EACF,SAAS,GAAG;GACV,GAAG,OAAO,wCAAwC,CAAC;GACnD,QAAoB,UAAU;GAC9B,UAAU,OAAO;EACnB;CACF,CAAC;AACH"}
+{"version":3,"file":"websocket-handoff.js","names":[],"sources":["../../../../../zero-cache/src/types/websocket-handoff.ts"],"sourcesContent":["import type {IncomingMessage} from 'node:http';\nimport {Server} from 'node:http';\nimport type {Socket} from 'node:net';\nimport type {LogContext} from '@rocicorp/logger';\nimport {WebSocketServer, type ServerOptions, type WebSocket} from 'ws';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport {serializableSubset, type IncomingMessageSubset} from './http.ts';\nimport type {MESSAGE_TYPES} from './processes.ts';\nimport {type Receiver, type Sender, type Worker} from './processes.ts';\nimport {closeWithError, PROTOCOL_ERROR} from './ws.ts';\n\nexport type HandoffSpec<P> = {\n  payload: P;\n  sender: Sender;\n};\n\n/**\n * The WebSocketHandoff is a function that either returns the payload and\n * receiver, or invokes the specified `callback` with the payload and\n * receiver. It must not do both.\n *\n * Similarly, an error can be handled by throwing synchronously from the\n * function, or invoking the `onerror` callback.\n */\nexport type WebSocketHandoff<P> = (\n  message: IncomingMessageSubset,\n  callback: (h: HandoffSpec<P>) => void,\n  onerror: (reason: unknown) => void,\n) => HandoffSpec<P> | void;\n\nexport type WebSocketReceiver<P> = (\n  ws: WebSocket,\n  payload: P,\n  msg: IncomingMessageSubset,\n) => void;\n\nexport type WebSocketHandoffHandler = (\n  message: IncomingMessageSubset,\n  socket: Socket,\n  head: ArrayBuffer,\n) => void;\n\n/**\n * Installs websocket handoff logic from either an http.Server\n * receiving requests, or a parent Worker process\n * that is handing off requests to this process.\n */\nfunction createWebSocketHandoffHandler<P>(\n  lc: LogContext,\n  handoff: WebSocketHandoff<P>,\n  serverOptions?: ServerOptions,\n): WebSocketHandoffHandler {\n  const wss = new WebSocketServer(\n    serverOptions ?? {\n      noServer: true,\n    },\n  );\n  return (\n    message: IncomingMessageSubset,\n    socket: Socket,\n    head: ArrayBuffer,\n  ) => {\n    let sent = false;\n\n    function send({payload, sender}: HandoffSpec<P>) {\n      assert(!sent, 'Handoff callback already invoked');\n      sent = true;\n\n      const data = [\n        'handoff',\n        {\n          message: serializableSubset(message),\n          head,\n          payload,\n        },\n      ] satisfies Handoff<P>;\n\n      // \"This event is guaranteed to be passed an instance of the <net.Socket> class\"\n      // https://nodejs.org/api/http.html#event-upgrade\n      sender.send(data, socket);\n    }\n\n    function onError(error: unknown) {\n      // Returning an error on the HTTP handshake looks like a hanging connection\n      // (at least from Chrome) and doesn't report any meaningful error in the browser.\n      // Instead, finish the upgrade to a websocket and then close it with an error.\n      wss.handleUpgrade(\n        message as IncomingMessage,\n        socket,\n        Buffer.from(head),\n        ws => {\n          ws.on('error', error => {\n            lc.warn?.('error while rejecting websocket handoff', error);\n            ws.terminate();\n          });\n          closeWithError(lc, ws, error, PROTOCOL_ERROR);\n        },\n      );\n    }\n\n    try {\n      const spec = handoff(message, send, onError);\n      if (spec) {\n        send(spec);\n      }\n    } catch (error) {\n      onError(error);\n    }\n  };\n}\n\n/**\n * Installs websocket handoff logic from either an http.Server\n * receiving requests, or a parent Worker process\n * that is handing off requests to this process.\n */\nexport function installWebSocketHandoff<P>(\n  lc: LogContext,\n  handoff: WebSocketHandoff<P>,\n  source: Server | Worker,\n  serverOptions?: ServerOptions,\n) {\n  const handle = createWebSocketHandoffHandler(lc, handoff, serverOptions);\n\n  if (source instanceof Server) {\n    // handoff messages from an HTTP server\n    source.on('upgrade', handle);\n  } else {\n    // handoff messages from this worker's parent.\n    source.onMessageType<Handoff<P>>('handoff', (msg, socket) => {\n      const {message, head} = msg;\n      handle(message, socket as Socket, head);\n    });\n  }\n}\n\nexport function installWebSocketReceiver<P>(\n  lc: LogContext,\n  server: WebSocketServer,\n  receive: WebSocketReceiver<P>,\n  receiver: Receiver,\n  onAbort?: ((payload: P) => void) | undefined,\n  onReceived?: ((payload: P) => void) | undefined,\n) {\n  receiver.onMessageType<Handoff<P>>('handoff', (msg, socket) => {\n    const {message, head, payload} = msg;\n    onReceived?.(payload);\n    // Per https://nodejs.org/api/child_process.html#subprocesssendmessage-sendhandle-options-callback\n    //\n    // > Any 'message' handlers in the subprocess should verify that socket\n    // > exists, as the connection may have been closed during the time it\n    // > takes to send the connection to the child.\n    if (!socket) {\n      lc.warn?.('websocket closed during handoff');\n      onAbort?.(payload);\n      return;\n    }\n    try {\n      server.handleUpgrade(\n        message as IncomingMessage,\n        socket as Socket,\n        Buffer.from(head),\n        ws => {\n          // Guard against WebSocket being closed during handoff.\n          // This can happen due to network issues or client disconnection\n          // between the time the socket was sent and when handleUpgrade completes.\n          if (ws.readyState === ws.CLOSED || ws.readyState === ws.CLOSING) {\n            lc.warn?.('websocket closed during upgrade, skipping receive');\n            onAbort?.(payload);\n            return;\n          }\n          receive(ws, payload, message);\n        },\n      );\n    } catch (e) {\n      lc.warn?.('error during websocket handleUpgrade', e);\n      (socket as Socket)?.destroy?.();\n      onAbort?.(payload);\n    }\n  });\n}\n\nexport type Handoff<P> = [\n  typeof MESSAGE_TYPES.handoff,\n  {\n    message: IncomingMessageSubset;\n    head: ArrayBuffer;\n    payload: P;\n  },\n];\n"],"mappings":";;;;;;;;;;;;AA+CA,SAAS,8BACP,IACA,SACA,eACyB;CACzB,MAAM,MAAM,IAAI,gBACd,iBAAiB,EACf,UAAU,KACZ,CACF;CACA,QACE,SACA,QACA,SACG;EACH,IAAI,OAAO;EAEX,SAAS,KAAK,EAAC,SAAS,UAAyB;GAC/C,OAAO,CAAC,MAAM,kCAAkC;GAChD,OAAO;GAEP,MAAM,OAAO,CACX,WACA;IACE,SAAS,mBAAmB,OAAO;IACnC;IACA;GACF,CACF;GAIA,OAAO,KAAK,MAAM,MAAM;EAC1B;EAEA,SAAS,QAAQ,OAAgB;GAI/B,IAAI,cACF,SACA,QACA,OAAO,KAAK,IAAI,IAChB,OAAM;IACJ,GAAG,GAAG,UAAS,UAAS;KACtB,GAAG,OAAO,2CAA2C,KAAK;KAC1D,GAAG,UAAU;IACf,CAAC;IACD,eAAe,IAAI,IAAI,OAAO,cAAc;GAC9C,CACF;EACF;EAEA,IAAI;GACF,MAAM,OAAO,QAAQ,SAAS,MAAM,OAAO;GAC3C,IAAI,MACF,KAAK,IAAI;EAEb,SAAS,OAAO;GACd,QAAQ,KAAK;EACf;CACF;AACF;;;;;;AAOA,SAAgB,wBACd,IACA,SACA,QACA,eACA;CACA,MAAM,SAAS,8BAA8B,IAAI,SAAS,aAAa;CAEvE,IAAI,kBAAkB,QAEpB,OAAO,GAAG,WAAW,MAAM;MAG3B,OAAO,cAA0B,YAAY,KAAK,WAAW;EAC3D,MAAM,EAAC,SAAS,SAAQ;EACxB,OAAO,SAAS,QAAkB,IAAI;CACxC,CAAC;AAEL;AAEA,SAAgB,yBACd,IACA,QACA,SACA,UACA,SACA,YACA;CACA,SAAS,cAA0B,YAAY,KAAK,WAAW;EAC7D,MAAM,EAAC,SAAS,MAAM,YAAW;EACjC,aAAa,OAAO;EAMpB,IAAI,CAAC,QAAQ;GACX,GAAG,OAAO,iCAAiC;GAC3C,UAAU,OAAO;GACjB;EACF;EACA,IAAI;GACF,OAAO,cACL,SACA,QACA,OAAO,KAAK,IAAI,IAChB,OAAM;IAIJ,IAAI,GAAG,eAAe,GAAG,UAAU,GAAG,eAAe,GAAG,SAAS;KAC/D,GAAG,OAAO,mDAAmD;KAC7D,UAAU,OAAO;KACjB;IACF;IACA,QAAQ,IAAI,SAAS,OAAO;GAC9B,CACF;EACF,SAAS,GAAG;GACV,GAAG,OAAO,wCAAwC,CAAC;GACnD,QAAoB,UAAU;GAC9B,UAAU,OAAO;EACnB;CACF,CAAC;AACH"}

package/out/zero-cache/src/types/ws.d.ts

@@ -1,9 +1,10 @@
 import type { LogContext } from '@rocicorp/logger';
 import type { WebSocket } from 'ws';
 export declare const PROTOCOL_ERROR = 1002;
-export declare const INTERNAL_ERROR = 1011;
+declare const INTERNAL_ERROR = 1011;
 export type ErrorCode = typeof PROTOCOL_ERROR | typeof INTERNAL_ERROR;
 export declare function closeWithError(lc: LogContext, ws: WebSocket, err: unknown, code?: ErrorCode): void;
 export declare function sendPingsForLiveness(lc: LogContext, ws: WebSocket, intervalMs: number): void;
 export declare function expectPingsForLiveness(lc: LogContext, ws: WebSocket, intervalMs: number, timeoutBufferMs?: number): void;
+export {};
 //# sourceMappingURL=ws.d.ts.map

package/out/zero-cache/src/types/ws.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ws.d.ts","sourceRoot":"","sources":["../../../../../zero-cache/src/types/ws.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,kBAAkB,CAAC;AACjD,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,IAAI,CAAC;AAIlC,eAAO,MAAM,cAAc,OAAO,CAAC;AACnC,eAAO,MAAM,cAAc,OAAO,CAAC;AAEnC,MAAM,MAAM,SAAS,GAAG,OAAO,cAAc,GAAG,OAAO,cAAc,CAAC;AAEtE,wBAAgB,cAAc,CAC5B,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,SAAS,EACb,GAAG,EAAE,OAAO,EACZ,IAAI,GAAE,SAA0B,QASjC;AAED,wBAAgB,oBAAoB,CAClC,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,SAAS,EACb,UAAU,EAAE,MAAM,QAkCnB;AAED,wBAAgB,sBAAsB,CACpC,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,SAAS,EACb,UAAU,EAAE,MAAM,EAClB,eAAe,SAAQ,QA+CxB"}
+{"version":3,"file":"ws.d.ts","sourceRoot":"","sources":["../../../../../zero-cache/src/types/ws.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,kBAAkB,CAAC;AACjD,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,IAAI,CAAC;AAIlC,eAAO,MAAM,cAAc,OAAO,CAAC;AACnC,QAAA,MAAM,cAAc,OAAO,CAAC;AAE5B,MAAM,MAAM,SAAS,GAAG,OAAO,cAAc,GAAG,OAAO,cAAc,CAAC;AAEtE,wBAAgB,cAAc,CAC5B,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,SAAS,EACb,GAAG,EAAE,OAAO,EACZ,IAAI,GAAE,SAA0B,QASjC;AAED,wBAAgB,oBAAoB,CAClC,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,SAAS,EACb,UAAU,EAAE,MAAM,QAkCnB;AAED,wBAAgB,sBAAsB,CACpC,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,SAAS,EACb,UAAU,EAAE,MAAM,EAClB,eAAe,SAAQ,QA+CxB"}

package/out/zero-cache/src/types/ws.js

@@ -30,6 +30,6 @@ function sendPingsForLiveness(lc, ws, intervalMs) {
 	ws.once("close", () => clearInterval(livenessTimer));
 }
 //#endregion
-export { INTERNAL_ERROR, PROTOCOL_ERROR, closeWithError, sendPingsForLiveness };
+export { PROTOCOL_ERROR, closeWithError, sendPingsForLiveness };
 
 //# sourceMappingURL=ws.js.map

package/out/zero-cache/src/types/ws.js.map

@@ -1 +1 @@
-{"version":3,"file":"ws.js","names":[],"sources":["../../../../../zero-cache/src/types/ws.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport type {WebSocket} from 'ws';\nimport {elide} from './strings.ts';\n\n// https://github.com/Luka967/websocket-close-codes\nexport const PROTOCOL_ERROR = 1002;\nexport const INTERNAL_ERROR = 1011;\n\nexport type ErrorCode = typeof PROTOCOL_ERROR | typeof INTERNAL_ERROR;\n\nexport function closeWithError(\n  lc: LogContext,\n  ws: WebSocket,\n  err: unknown,\n  code: ErrorCode = INTERNAL_ERROR,\n) {\n  const endpoint = ws.url ?? 'client';\n  const errMsg = String(err);\n  lc.warn?.(`closing connection to ${endpoint} with error`, errMsg);\n\n  // close messages must be less than or equal to 123 bytes:\n  // https://developer.mozilla.org/en-US/docs/Web/API/WebSocket/close#reason\n  ws.close(code, elide(errMsg, 123));\n}\n\nexport function sendPingsForLiveness(\n  lc: LogContext,\n  ws: WebSocket,\n  intervalMs: number,\n) {\n  let gotLivenessSignal = true;\n\n  let livenessTimer: NodeJS.Timeout | undefined;\n  function startHeartBeats() {\n    livenessTimer = setInterval(() => {\n      if (!gotLivenessSignal) {\n        lc.warn?.(\n          `socket@${ws.url} did not respond to heartbeat. Terminating...`,\n        );\n        ws.terminate();\n        return;\n      }\n      // Reset gotLivenessSignal and expect another pong or message to arrive\n      // before the next interval elapses.\n      gotLivenessSignal = false;\n      ws.ping();\n    }, intervalMs);\n  }\n\n  if (ws.readyState === ws.CONNECTING) {\n    ws.once('open', () => startHeartBeats());\n  } else if (ws.readyState === ws.OPEN) {\n    startHeartBeats();\n  }\n\n  // Both pongs and messages are accepted as signs of liveness.\n  // Checking for pongs only risks false positives as pongs may be backed\n  // up behind a large stream of messages.\n  const signalAlive = () => (gotLivenessSignal = true);\n  ws.on('pong', signalAlive);\n  ws.on('message', signalAlive);\n  ws.once('close', () => clearInterval(livenessTimer));\n}\n\nexport function expectPingsForLiveness(\n  lc: LogContext,\n  ws: WebSocket,\n  intervalMs: number,\n  timeoutBufferMs = 3_000,\n) {\n  const timeoutMs = intervalMs + timeoutBufferMs;\n  let gotLivenessSignal = false;\n\n  let livenessTimer: NodeJS.Timeout | undefined;\n  function startLivenessChecks() {\n    livenessTimer = setInterval(() => {\n      if (!gotLivenessSignal) {\n        lc.warn?.(\n          `socket@${ws.url} did not send heartbeat or messages. Terminating...`,\n        );\n        ws.terminate();\n        return;\n      }\n      // Reset gotLivenessSignal and expect another ping or message to arrive\n      // before the next interval elapses.\n      gotLivenessSignal = false;\n    }, timeoutMs);\n  }\n\n  if (ws.readyState === ws.CONNECTING) {\n    // Heartbeats only start once the connection is open, so time out the\n    // handshake separately. Otherwise a slow handshake eats into the first\n    // heartbeat interval, and a stuck one is reported as a missed heartbeat.\n    const connectTimer = setTimeout(() => {\n      lc.warn?.(\n        `socket@${ws.url} did not connect within ${timeoutMs} ms. Terminating...`,\n      );\n      ws.terminate();\n    }, timeoutMs);\n    ws.once('open', () => {\n      clearTimeout(connectTimer);\n      startLivenessChecks();\n    });\n    ws.once('close', () => clearTimeout(connectTimer));\n  } else if (ws.readyState === ws.OPEN) {\n    startLivenessChecks();\n  }\n\n  // Both pings and messages are accepted as signs of liveness.\n  // Checking for pings only risks false positives as pings may be backed\n  // up behind a large stream of messages.\n  const signalAlive = () => (gotLivenessSignal = true);\n  ws.on('ping', signalAlive);\n  ws.on('message', signalAlive);\n  ws.once('close', () => clearInterval(livenessTimer));\n}\n"],"mappings":";;AAKA,IAAa,iBAAiB;AAC9B,IAAa,iBAAiB;AAI9B,SAAgB,eACd,IACA,IACA,KACA,OAAkB,gBAClB;CACA,MAAM,WAAW,GAAG,OAAO;CAC3B,MAAM,SAAS,OAAO,GAAG;CACzB,GAAG,OAAO,yBAAyB,SAAS,cAAc,MAAM;CAIhE,GAAG,MAAM,MAAM,MAAM,QAAQ,GAAG,CAAC;AACnC;AAEA,SAAgB,qBACd,IACA,IACA,YACA;CACA,IAAI,oBAAoB;CAExB,IAAI;CACJ,SAAS,kBAAkB;EACzB,gBAAgB,kBAAkB;GAChC,IAAI,CAAC,mBAAmB;IACtB,GAAG,OACD,UAAU,GAAG,IAAI,8CACnB;IACA,GAAG,UAAU;IACb;GACF;GAGA,oBAAoB;GACpB,GAAG,KAAK;EACV,GAAG,UAAU;CACf;CAEA,IAAI,GAAG,eAAe,GAAG,YACvB,GAAG,KAAK,cAAc,gBAAgB,CAAC;MAClC,IAAI,GAAG,eAAe,GAAG,MAC9B,gBAAgB;CAMlB,MAAM,oBAAqB,oBAAoB;CAC/C,GAAG,GAAG,QAAQ,WAAW;CACzB,GAAG,GAAG,WAAW,WAAW;CAC5B,GAAG,KAAK,eAAe,cAAc,aAAa,CAAC;AACrD"}
+{"version":3,"file":"ws.js","names":[],"sources":["../../../../../zero-cache/src/types/ws.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport type {WebSocket} from 'ws';\nimport {elide} from './strings.ts';\n\n// https://github.com/Luka967/websocket-close-codes\nexport const PROTOCOL_ERROR = 1002;\nconst INTERNAL_ERROR = 1011;\n\nexport type ErrorCode = typeof PROTOCOL_ERROR | typeof INTERNAL_ERROR;\n\nexport function closeWithError(\n  lc: LogContext,\n  ws: WebSocket,\n  err: unknown,\n  code: ErrorCode = INTERNAL_ERROR,\n) {\n  const endpoint = ws.url ?? 'client';\n  const errMsg = String(err);\n  lc.warn?.(`closing connection to ${endpoint} with error`, errMsg);\n\n  // close messages must be less than or equal to 123 bytes:\n  // https://developer.mozilla.org/en-US/docs/Web/API/WebSocket/close#reason\n  ws.close(code, elide(errMsg, 123));\n}\n\nexport function sendPingsForLiveness(\n  lc: LogContext,\n  ws: WebSocket,\n  intervalMs: number,\n) {\n  let gotLivenessSignal = true;\n\n  let livenessTimer: NodeJS.Timeout | undefined;\n  function startHeartBeats() {\n    livenessTimer = setInterval(() => {\n      if (!gotLivenessSignal) {\n        lc.warn?.(\n          `socket@${ws.url} did not respond to heartbeat. Terminating...`,\n        );\n        ws.terminate();\n        return;\n      }\n      // Reset gotLivenessSignal and expect another pong or message to arrive\n      // before the next interval elapses.\n      gotLivenessSignal = false;\n      ws.ping();\n    }, intervalMs);\n  }\n\n  if (ws.readyState === ws.CONNECTING) {\n    ws.once('open', () => startHeartBeats());\n  } else if (ws.readyState === ws.OPEN) {\n    startHeartBeats();\n  }\n\n  // Both pongs and messages are accepted as signs of liveness.\n  // Checking for pongs only risks false positives as pongs may be backed\n  // up behind a large stream of messages.\n  const signalAlive = () => (gotLivenessSignal = true);\n  ws.on('pong', signalAlive);\n  ws.on('message', signalAlive);\n  ws.once('close', () => clearInterval(livenessTimer));\n}\n\nexport function expectPingsForLiveness(\n  lc: LogContext,\n  ws: WebSocket,\n  intervalMs: number,\n  timeoutBufferMs = 3_000,\n) {\n  const timeoutMs = intervalMs + timeoutBufferMs;\n  let gotLivenessSignal = false;\n\n  let livenessTimer: NodeJS.Timeout | undefined;\n  function startLivenessChecks() {\n    livenessTimer = setInterval(() => {\n      if (!gotLivenessSignal) {\n        lc.warn?.(\n          `socket@${ws.url} did not send heartbeat or messages. Terminating...`,\n        );\n        ws.terminate();\n        return;\n      }\n      // Reset gotLivenessSignal and expect another ping or message to arrive\n      // before the next interval elapses.\n      gotLivenessSignal = false;\n    }, timeoutMs);\n  }\n\n  if (ws.readyState === ws.CONNECTING) {\n    // Heartbeats only start once the connection is open, so time out the\n    // handshake separately. Otherwise a slow handshake eats into the first\n    // heartbeat interval, and a stuck one is reported as a missed heartbeat.\n    const connectTimer = setTimeout(() => {\n      lc.warn?.(\n        `socket@${ws.url} did not connect within ${timeoutMs} ms. Terminating...`,\n      );\n      ws.terminate();\n    }, timeoutMs);\n    ws.once('open', () => {\n      clearTimeout(connectTimer);\n      startLivenessChecks();\n    });\n    ws.once('close', () => clearTimeout(connectTimer));\n  } else if (ws.readyState === ws.OPEN) {\n    startLivenessChecks();\n  }\n\n  // Both pings and messages are accepted as signs of liveness.\n  // Checking for pings only risks false positives as pings may be backed\n  // up behind a large stream of messages.\n  const signalAlive = () => (gotLivenessSignal = true);\n  ws.on('ping', signalAlive);\n  ws.on('message', signalAlive);\n  ws.once('close', () => clearInterval(livenessTimer));\n}\n"],"mappings":";;AAKA,IAAa,iBAAiB;AAC9B,IAAM,iBAAiB;AAIvB,SAAgB,eACd,IACA,IACA,KACA,OAAkB,gBAClB;CACA,MAAM,WAAW,GAAG,OAAO;CAC3B,MAAM,SAAS,OAAO,GAAG;CACzB,GAAG,OAAO,yBAAyB,SAAS,cAAc,MAAM;CAIhE,GAAG,MAAM,MAAM,MAAM,QAAQ,GAAG,CAAC;AACnC;AAEA,SAAgB,qBACd,IACA,IACA,YACA;CACA,IAAI,oBAAoB;CAExB,IAAI;CACJ,SAAS,kBAAkB;EACzB,gBAAgB,kBAAkB;GAChC,IAAI,CAAC,mBAAmB;IACtB,GAAG,OACD,UAAU,GAAG,IAAI,8CACnB;IACA,GAAG,UAAU;IACb;GACF;GAGA,oBAAoB;GACpB,GAAG,KAAK;EACV,GAAG,UAAU;CACf;CAEA,IAAI,GAAG,eAAe,GAAG,YACvB,GAAG,KAAK,cAAc,gBAAgB,CAAC;MAClC,IAAI,GAAG,eAAe,GAAG,MAC9B,gBAAgB;CAMlB,MAAM,oBAAqB,oBAAoB;CAC/C,GAAG,GAAG,QAAQ,WAAW;CACzB,GAAG,GAAG,WAAW,WAAW;CAC5B,GAAG,KAAK,eAAe,cAAc,aAAa,CAAC;AACrD"}

package/out/zero-cache/src/workers/connection.js

@@ -94,6 +94,9 @@ var Connection = class {
 		}
 		return false;
 	}
+	get wsID() {
+		return this.#wsID;
+	}
 	close(reason, ...args) {
 		if (this.#closed) return;
 		this.#closed = true;
@@ -381,6 +384,6 @@ function isTransientSocketMessage(message) {
 	return TRANSIENT_SOCKET_MESSAGE_PATTERNS.some((pattern) => lower.includes(pattern));
 }
 //#endregion
-export { Connection, DownstreamSender, WEBSOCKET_SEND_TIMEOUT_MS, findProtocolError, parseUpstreamMessage, send, sendError };
+export { Connection, DownstreamSender, WEBSOCKET_SEND_TIMEOUT_MS, parseUpstreamMessage, send, sendError };
 
 //# sourceMappingURL=connection.js.map

package/out/zero-cache/src/workers/connection.js.map

@@ -1 +1 @@
-{"version":3,"file":"connection.js","names":["valita.parse","ErrorKind.VersionNotSupported","ErrorOrigin.ZeroCache","ErrorKind.InvalidMessage","WebSocket","ErrorKind.Internal"],"sources":["../../../../../zero-cache/src/workers/connection.ts"],"sourcesContent":["import {pipeline, Readable, Writable} from 'node:stream';\nimport type {LogContext, LogLevel} from '@rocicorp/logger';\nimport type {CloseEvent, Data, ErrorEvent} from 'ws';\nimport WebSocket, {createWebSocketStream} from 'ws';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport * as valita from '../../../shared/src/valita.ts';\nimport type {ConnectedMessage} from '../../../zero-protocol/src/connect.ts';\nimport type {Downstream} from '../../../zero-protocol/src/down.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\nimport type {ErrorBody} from '../../../zero-protocol/src/error.ts';\nimport {\n  isProtocolError,\n  type ProtocolError,\n} from '../../../zero-protocol/src/error.ts';\nimport {\n  POKE_CHUNK_PROTOCOL_VERSION,\n  type PokeChunk,\n  type PokeEndMessage,\n  type PokePartMessage,\n} from '../../../zero-protocol/src/poke.ts';\nimport {\n  MIN_SERVER_SUPPORTED_SYNC_PROTOCOL,\n  PROTOCOL_VERSION,\n} from '../../../zero-protocol/src/protocol-version.ts';\nimport {\n  upstreamSchemaWithUnparsedAnalyzeQuery,\n  type UpstreamWithUnparsedAnalyzeQuery,\n} from '../../../zero-protocol/src/up.ts';\nimport {getOrCreateCounter} from '../observability/metrics.ts';\nimport type {ViewSyncerDownstream} from '../types/downstream.ts';\nimport {\n  ProtocolErrorWithLevel,\n  getLogLevel,\n  wrapWithProtocolError,\n} from '../types/error-with-level.ts';\nimport {PokeChunkEncoder} from '../types/poke-chunk.ts';\nimport type {Source} from '../types/streams.ts';\nimport type {ConnectParams} from './connect-params.ts';\n\nexport type HandlerResult =\n  | {\n      type: 'ok';\n    }\n  | {\n      type: 'fatal';\n      error: ErrorBody;\n    }\n  | {\n      type: 'transient';\n      errors: ErrorBody[];\n    }\n  | StreamResult;\n\nexport type StreamResult =\n  | {\n      type: 'stream';\n      source: 'viewSyncer';\n      stream: Source<ViewSyncerDownstream>;\n    }\n  | {\n      type: 'stream';\n      source: 'pusher';\n      stream: Source<Downstream>;\n    };\n\nexport interface MessageHandler {\n  handleMessage(\n    msg: UpstreamWithUnparsedAnalyzeQuery,\n  ): Promise<HandlerResult[]>;\n}\n\nfunction hasOwn(value: unknown, property: string): boolean {\n  return (\n    typeof value === 'object' &&\n    value !== null &&\n    Object.hasOwn(value, property)\n  );\n}\n\nfunction containsLegacyQuery(message: unknown): boolean {\n  if (!Array.isArray(message)) {\n    return false;\n  }\n\n  const body = message[1];\n  if (typeof body !== 'object' || body === null) {\n    return false;\n  }\n\n  if (\n    message[0] !== 'initConnection' &&\n    message[0] !== 'changeDesiredQueries'\n  ) {\n    return false;\n  }\n\n  const patch = body['desiredQueriesPatch'];\n  return (\n    Array.isArray(patch) && patch.some(operation => hasOwn(operation, 'ast'))\n  );\n}\n\nconst LEGACY_QUERIES_DISABLED_MESSAGE =\n  'Legacy queries are disabled by this Zero server. This client must use custom queries instead of sending query ASTs directly. To temporarily restore compatibility, set ZERO_ALLOW_LEGACY_QUERIES=true on zero-cache.';\n\n// Exported for testing purposes.\n//\n// Client messages are parsed leniently, so a newer client can add message\n// kinds and fields:\n// - A message of a kind this server doesn't know parses as an\n//   UnrecognizedTuple, for the connection to drop. Malformed messages still\n//   throw.\n// - Messages are parsed in strip mode, so fields this server doesn't know are\n//   dropped. Strip rather than passthrough: nothing this server doesn't know\n//   reaches the CVR or the API server, and valita's passthrough drops unknown\n//   fields anyway on any object with a transformed field below it (such as\n//   jsonSchema args). ASTs stay strict in the schema, because a server that\n//   ignored an AST field would run a different query.\nexport function parseUpstreamMessage(\n  value: unknown,\n  allowLegacyQueries: boolean,\n): UpstreamWithUnparsedAnalyzeQuery | valita.UnrecognizedTuple {\n  if (!allowLegacyQueries && containsLegacyQuery(value)) {\n    throw new Error(LEGACY_QUERIES_DISABLED_MESSAGE);\n  }\n  return valita.parse(value, upstreamSchemaWithUnparsedAnalyzeQuery, 'strip');\n}\n\n// Ensures that a downstream message is sent at least every interval, sending a\n// 'pong' if necessary. This is set to be slightly longer than the client-side\n// PING_INTERVAL of 5 seconds, so that in the common case, 'pong's are sent in\n// response to client-initiated 'ping's. However, if the inbound stream is\n// backed up because a command is taking a long time to process, the pings\n// will be stuck in the queue (i.e. back-pressured), in which case pongs will\n// be manually sent to notify the client of server liveness.\n//\n// This is equivalent to what is done for Postgres keepalives on the\n// replication stream (which can similarly be back-pressured):\n// https://github.com/rocicorp/mono/blob/f98cb369a2dbb15650328859c732db358f187ef0/packages/zero-cache/src/services/change-source/pg/logical-replication/stream.ts#L21\nconst DOWNSTREAM_MSG_INTERVAL_MS = 6_000;\nexport const WEBSOCKET_SEND_TIMEOUT_MS = 10_000;\nconst PROTOCOL_VERSION_ATTRIBUTE = 'protocol.version';\nconst EVENT_TYPE_ATTRIBUTE = 'event.type';\n\n/**\n * Represents a connection between the client and server.\n *\n * Handles incoming messages on the connection and dispatches\n * them to the correct service.\n *\n * Listens to the ViewSyncer and sends messages to the client.\n */\nexport class Connection {\n  readonly #ws: WebSocket;\n  readonly #wsID: string;\n  readonly #protocolVersion: number;\n  readonly #lc: LogContext;\n  readonly #onClose: () => void;\n  readonly #messageHandler: MessageHandler;\n  readonly #allowLegacyQueries: boolean;\n  readonly #downstreamSender: DownstreamSender;\n  readonly #downstreamMsgTimer: NodeJS.Timeout | undefined;\n  readonly #webSocketErrors = getOrCreateCounter(\n    'sync',\n    'websocket.errors',\n    'Client WebSocket error events.',\n  );\n\n  #viewSyncerOutboundStream: Source<ViewSyncerDownstream> | undefined;\n  #pusherOutboundStream: Source<Downstream> | undefined;\n  #closed = false;\n  #loggedUnknownMessage = false;\n\n  constructor(\n    lc: LogContext,\n    connectParams: ConnectParams,\n    ws: WebSocket,\n    allowLegacyQueries: boolean,\n    messageHandler: MessageHandler,\n    onClose: () => void,\n  ) {\n    const {clientGroupID, clientID, wsID, protocolVersion} = connectParams;\n    this.#messageHandler = messageHandler;\n    this.#allowLegacyQueries = allowLegacyQueries;\n\n    this.#ws = ws;\n    this.#wsID = wsID;\n    this.#protocolVersion = protocolVersion;\n\n    this.#lc = lc\n      .withContext('connection')\n      .withContext('clientID', clientID)\n      .withContext('clientGroupID', clientGroupID)\n      .withContext('wsID', wsID);\n    this.#downstreamSender = new DownstreamSender(\n      this.#lc,\n      ws,\n      protocolVersion,\n    );\n    this.#lc.debug?.('new connection');\n    this.#onClose = onClose;\n\n    this.#ws.addEventListener('close', this.#handleClose);\n    this.#ws.addEventListener('error', this.#handleError);\n\n    this.#proxyInbound();\n    this.#downstreamMsgTimer = setInterval(\n      this.#maybeSendPong,\n      DOWNSTREAM_MSG_INTERVAL_MS / 2,\n    );\n  }\n\n  /**\n   * Checks the protocol version and errors for unsupported protocols,\n   * sending the initial `connected` response on success.\n   *\n   * This is early in the connection lifecycle because {@link #handleMessage}\n   * will only parse messages with schema(s) of supported protocol versions.\n   */\n  init(): boolean {\n    if (\n      this.#protocolVersion > PROTOCOL_VERSION ||\n      this.#protocolVersion < MIN_SERVER_SUPPORTED_SYNC_PROTOCOL\n    ) {\n      this.#closeWithError({\n        kind: ErrorKind.VersionNotSupported,\n        message: `server is at sync protocol v${PROTOCOL_VERSION} and does not support v${\n          this.#protocolVersion\n        }. The ${\n          this.#protocolVersion > PROTOCOL_VERSION ? 'server' : 'client'\n        } must be updated to a newer release.`,\n        origin: ErrorOrigin.ZeroCache,\n      });\n    } else {\n      const connectedMessage: ConnectedMessage = [\n        'connected',\n        {wsid: this.#wsID, timestamp: Date.now()},\n      ];\n      this.send(connectedMessage, 'ignore-backpressure');\n      return true;\n    }\n    return false;\n  }\n\n  close(reason: string, ...args: unknown[]) {\n    if (this.#closed) {\n      return;\n    }\n    this.#closed = true;\n    this.#lc.info?.(`closing connection: ${reason}`, ...args);\n    this.#ws.removeEventListener('close', this.#handleClose);\n    this.#ws.removeEventListener('error', this.#handleError);\n    this.#viewSyncerOutboundStream?.cancel();\n    this.#viewSyncerOutboundStream = undefined;\n    this.#pusherOutboundStream?.cancel();\n    this.#pusherOutboundStream = undefined;\n    this.#onClose();\n    if (this.#ws.readyState !== this.#ws.CLOSED) {\n      this.#ws.close();\n    }\n    clearTimeout(this.#downstreamMsgTimer);\n\n    // spin down services if we have\n    // no more client connections for the client group?\n  }\n\n  handleInitConnection(initConnectionMsg: string) {\n    return this.#handleMessage({data: initConnectionMsg});\n  }\n\n  #handleMessage = async (event: {data: Data}) => {\n    const data = event.data.toString();\n    if (this.#closed) {\n      this.#lc.debug?.('Ignoring message received after closed', data);\n      return;\n    }\n\n    let msg;\n    try {\n      const value = JSON.parse(data);\n      msg = parseUpstreamMessage(value, this.#allowLegacyQueries);\n    } catch (e) {\n      const errorBody = {\n        kind: ErrorKind.InvalidMessage,\n        message: String(e),\n        origin: ErrorOrigin.ZeroCache,\n      } as const;\n      this.#closeWithError(\n        errorBody,\n        new ProtocolErrorWithLevel(errorBody, 'warn'),\n      );\n      return;\n    }\n\n    if (!upstreamSchemaWithUnparsedAnalyzeQuery.isKnown(msg)) {\n      // A message kind from a newer client.\n      if (!this.#loggedUnknownMessage) {\n        this.#loggedUnknownMessage = true;\n        this.#lc.info?.(\n          `Ignoring messages of unknown type, starting with ${JSON.stringify(msg[0])}`,\n        );\n      }\n      return;\n    }\n\n    try {\n      const msgType = msg[0];\n      if (msgType === 'ping') {\n        this.send(['pong', {}], 'ignore-backpressure');\n        return;\n      }\n\n      const result = await this.#messageHandler.handleMessage(msg);\n      for (const r of result) {\n        this.#handleMessageResult(r);\n      }\n    } catch (e) {\n      this.#closeWithThrown(e);\n    }\n  };\n\n  #handleMessageResult(result: HandlerResult): void {\n    switch (result.type) {\n      case 'fatal':\n        this.#closeWithError(result.error);\n        break;\n      case 'ok':\n        break;\n      case 'stream': {\n        switch (result.source) {\n          case 'viewSyncer':\n            assert(\n              this.#viewSyncerOutboundStream === undefined,\n              'Outbound stream already set for this connection!',\n            );\n            this.#viewSyncerOutboundStream = result.stream;\n            break;\n          case 'pusher':\n            assert(\n              this.#pusherOutboundStream === undefined,\n              'Outbound stream already set for this connection!',\n            );\n            this.#pusherOutboundStream = result.stream;\n            break;\n        }\n        if (result.source === 'viewSyncer') {\n          this.#proxyOutbound(result.stream, (downstream, callback) =>\n            this.send(downstream.message, callback, downstream.serialized),\n          );\n        } else {\n          this.#proxyOutbound(result.stream, (downstream, callback) =>\n            this.send(downstream, callback),\n          );\n        }\n        break;\n      }\n      case 'transient': {\n        for (const error of result.errors) {\n          this.sendError(error);\n        }\n      }\n    }\n  }\n\n  #handleClose = (e: CloseEvent) => {\n    const {code, reason, wasClean} = e;\n    if (!wasClean) {\n      this.#recordWebSocketError('unclean_close');\n    }\n    this.close('WebSocket close event', {code, reason, wasClean});\n  };\n\n  #handleError = (e: ErrorEvent) => {\n    this.#recordWebSocketError('error_event');\n    this.#lc.warn?.('WebSocket error event', e.message, e.error);\n  };\n\n  #recordWebSocketError(eventType: string) {\n    this.#webSocketErrors.add(1, {\n      [PROTOCOL_VERSION_ATTRIBUTE]: this.#protocolVersion,\n      [EVENT_TYPE_ATTRIBUTE]: eventType,\n    });\n  }\n\n  #proxyInbound() {\n    pipeline(\n      createWebSocketStream(this.#ws),\n      new Writable({\n        write: (data, _encoding, callback) => {\n          this.#handleMessage({data}).then(() => callback(), callback);\n        },\n      }),\n      // The done callback is not used, as #handleClose and #handleError,\n      // configured on the underlying WebSocket, provide more complete\n      // information.\n      () => {},\n    );\n  }\n\n  #proxyOutbound<T>(\n    outboundStream: Source<T>,\n    sendMessage: (\n      downstream: T,\n      callback: (err?: Error | null) => void,\n    ) => void,\n  ) {\n    // Note: createWebSocketStream() is avoided here in order to control\n    //       exception handling with #closeWithThrown(). If the Writable\n    //       from createWebSocketStream() were instead used, exceptions\n    //       from the outboundStream result in the Writable closing the\n    //       the websocket before the error message can be sent.\n    pipeline(\n      Readable.from(outboundStream),\n      new Writable({\n        objectMode: true,\n        write: (downstream: T, _encoding, callback) =>\n          sendMessage(downstream, callback),\n      }),\n      e =>\n        e\n          ? this.#closeWithThrown(e)\n          : this.close(`downstream closed by ViewSyncer`),\n    );\n  }\n\n  #closeWithThrown(e: unknown) {\n    const errorBody =\n      findProtocolError(e)?.errorBody ?? wrapWithProtocolError(e).errorBody;\n\n    this.#closeWithError(errorBody, e);\n  }\n\n  #closeWithError(errorBody: ErrorBody, thrown?: unknown) {\n    this.sendError(errorBody, thrown);\n    this.close(\n      `${errorBody.kind} (${errorBody.origin}): ${errorBody.message}`,\n      errorBody,\n    );\n  }\n\n  #lastDownstreamMsgTime = Date.now();\n\n  #maybeSendPong = () => {\n    if (Date.now() - this.#lastDownstreamMsgTime > DOWNSTREAM_MSG_INTERVAL_MS) {\n      this.#lc.debug?.('manually sending pong');\n      this.send(['pong', {}], 'ignore-backpressure');\n    }\n  };\n\n  send(\n    data: Downstream,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n    serialized?: string | undefined,\n  ) {\n    this.#lastDownstreamMsgTime = Date.now();\n    this.#downstreamSender.send(data, callback, serialized);\n  }\n\n  sendError(errorBody: ErrorBody, thrown?: unknown) {\n    sendError(this.#lc, this.#ws, errorBody, thrown);\n  }\n}\n\nexport type WebSocketLike = Pick<WebSocket, 'readyState'> & {\n  send(data: string | PokeChunk, cb?: (err?: Error) => void): void;\n};\n\n// Exported for testing purposes.\nexport function send(\n  lc: LogContext,\n  ws: WebSocketLike,\n  data: Downstream,\n  callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  serialized?: string | undefined,\n) {\n  if (ws.readyState === WebSocket.OPEN) {\n    serialized ??= JSON.stringify(data);\n    if (callback === 'ignore-backpressure') {\n      ws.send(serialized);\n      return;\n    }\n\n    let completed = false;\n    const timer = setTimeout(() => {\n      complete(webSocketSendTimeoutError());\n    }, WEBSOCKET_SEND_TIMEOUT_MS);\n    timer.unref();\n\n    const complete = (error?: Error | null) => {\n      if (completed) {\n        return;\n      }\n      completed = true;\n      clearTimeout(timer);\n      callback(error);\n    };\n\n    ws.send(serialized, complete);\n  } else {\n    lc.debug?.(`Dropping outbound message on ws (state: ${ws.readyState})`, {\n      dropped: data,\n    });\n    if (callback !== 'ignore-backpressure') {\n      callback(\n        new ProtocolErrorWithLevel(\n          {\n            kind: ErrorKind.Internal,\n            message: 'WebSocket closed',\n            origin: ErrorOrigin.ZeroCache,\n          },\n          'info',\n        ),\n      );\n    }\n  }\n}\n\n/**\n * Sends downstream messages in the format supported by a connection's sync\n * protocol version. Keep version forks contained here so the view-syncer can\n * continue producing one canonical poke representation.\n *\n * Exported for compatibility testing.\n */\nexport class DownstreamSender {\n  readonly #lc: LogContext;\n  readonly #ws: WebSocketLike;\n  readonly #protocolVersion: number;\n  #pokeChunkEncoder: PokeChunkEncoder | undefined;\n\n  constructor(lc: LogContext, ws: WebSocketLike, protocolVersion: number) {\n    this.#lc = lc;\n    this.#ws = ws;\n    this.#protocolVersion = protocolVersion;\n  }\n\n  send(\n    data: Downstream,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n    serialized?: string | undefined,\n  ): void {\n    if (this.#protocolVersion >= POKE_CHUNK_PROTOCOL_VERSION) {\n      switch (data[0]) {\n        case 'pokeStart':\n          assert(\n            this.#pokeChunkEncoder === undefined,\n            'cannot start a poke while another poke is in progress',\n          );\n          this.#pokeChunkEncoder = new PokeChunkEncoder();\n          break;\n        case 'pokePart':\n          this.#sendPokePart(data, callback, serialized);\n          return;\n        case 'pokeEnd':\n          this.#sendPokeEnd(data, callback);\n          return;\n      }\n    }\n    send(this.#lc, this.#ws, data, callback, serialized);\n  }\n\n  #sendPokePart(\n    pokePart: PokePartMessage,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n    serialized?: string | undefined,\n  ): void {\n    const encoder = this.#pokeChunkEncoder;\n    assert(encoder, 'pokePart received without a pokeStart');\n\n    serialized ??= JSON.stringify(pokePart);\n    assert(\n      serialized.startsWith(POKE_PART_PREFIX) && serialized.endsWith(']'),\n      'invalid serialized pokePart',\n    );\n    void encoder\n      .addPatch(serialized.slice(POKE_PART_PREFIX.length, -1), chunk =>\n        sendBinary(this.#lc, this.#ws, chunk),\n      )\n      .then(\n        () => invokeCallback(callback),\n        error => invokeCallback(callback, toError(error)),\n      );\n  }\n\n  #sendPokeEnd(\n    pokeEnd: PokeEndMessage,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  ): void {\n    const encoder = this.#pokeChunkEncoder;\n    assert(encoder, 'pokeEnd received without a pokeStart');\n    this.#pokeChunkEncoder = undefined;\n\n    if (pokeEnd[1].cancel) {\n      encoder.cancel();\n      send(this.#lc, this.#ws, pokeEnd, callback);\n      return;\n    }\n\n    void encoder\n      .finish(chunk => sendBinary(this.#lc, this.#ws, chunk))\n      .then(() => sendAsync(this.#lc, this.#ws, pokeEnd))\n      .then(\n        () => invokeCallback(callback),\n        error => invokeCallback(callback, toError(error)),\n      );\n  }\n}\n\nconst POKE_PART_PREFIX = '[\"pokePart\",';\n\nfunction sendAsync(\n  lc: LogContext,\n  ws: WebSocketLike,\n  data: Downstream,\n): Promise<void> {\n  return new Promise((resolve, reject) => {\n    send(lc, ws, data, error => (error ? reject(error) : resolve()));\n  });\n}\n\nfunction sendBinary(\n  lc: LogContext,\n  ws: WebSocketLike,\n  chunk: PokeChunk,\n): Promise<void> {\n  return new Promise((resolve, reject) => {\n    if (ws.readyState !== WebSocket.OPEN) {\n      const error = webSocketClosedError();\n      lc.debug?.(\n        `Dropping outbound binary message on ws (state: ${ws.readyState})`,\n      );\n      reject(error);\n      return;\n    }\n    let completed = false;\n    const timer = setTimeout(\n      () => complete(webSocketSendTimeoutError()),\n      WEBSOCKET_SEND_TIMEOUT_MS,\n    );\n    timer.unref();\n\n    const complete = (error?: Error | null) => {\n      if (completed) {\n        return;\n      }\n      completed = true;\n      clearTimeout(timer);\n      error ? reject(error) : resolve();\n    };\n\n    ws.send(chunk, complete);\n  });\n}\n\nfunction invokeCallback(\n  callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  error?: Error,\n): void {\n  if (callback !== 'ignore-backpressure') {\n    callback(error);\n  }\n}\n\nfunction toError(error: unknown): Error {\n  return error instanceof Error ? error : new Error(String(error));\n}\n\nfunction webSocketClosedError(): ProtocolErrorWithLevel {\n  return new ProtocolErrorWithLevel(\n    {\n      kind: ErrorKind.Internal,\n      message: 'WebSocket closed',\n      origin: ErrorOrigin.ZeroCache,\n    },\n    'info',\n  );\n}\n\nfunction webSocketSendTimeoutError(): ProtocolErrorWithLevel {\n  return new ProtocolErrorWithLevel(\n    {\n      kind: ErrorKind.Internal,\n      message: `WebSocket send timed out after ${WEBSOCKET_SEND_TIMEOUT_MS} ms`,\n      origin: ErrorOrigin.ZeroCache,\n    },\n    'info',\n  );\n}\n\nexport function sendError(\n  lc: LogContext,\n  ws: WebSocket,\n  errorBody: ErrorBody,\n  thrown?: unknown,\n) {\n  lc = lc.withContext('errorKind', errorBody.kind);\n\n  let logLevel: LogLevel;\n\n  // If the thrown error is a ProtocolErrorWithLevel, its explicit logLevel takes precedence\n  if (thrown instanceof ProtocolErrorWithLevel) {\n    logLevel = thrown.logLevel;\n  }\n  // Errors with errno or transient socket codes are low-level, transient I/O issues\n  // (e.g., EPIPE, ECONNRESET) and should be warnings, not errors\n  else if (\n    hasErrno(thrown) ||\n    hasTransientSocketCode(thrown) ||\n    isTransientSocketMessage(errorBody.message)\n  ) {\n    logLevel = 'warn';\n  }\n  // Fallback: check errorBody.kind for errors that weren't thrown as ProtocolErrorWithLevel\n  else if (\n    errorBody.kind === ErrorKind.ClientNotFound ||\n    errorBody.kind === ErrorKind.TransformFailed\n  ) {\n    logLevel = 'warn';\n  } else {\n    logLevel = thrown ? getLogLevel(thrown) : 'info';\n  }\n\n  lc[logLevel]?.('Sending error on WebSocket', errorBody, thrown ?? '');\n  send(lc, ws, ['error', errorBody], 'ignore-backpressure');\n}\n\nexport function findProtocolError(error: unknown): ProtocolError | undefined {\n  if (isProtocolError(error)) {\n    return error;\n  }\n  if (error instanceof Error && error.cause) {\n    return findProtocolError(error.cause);\n  }\n  return undefined;\n}\n\nfunction hasErrno(error: unknown): boolean {\n  return Boolean(\n    error &&\n    typeof error === 'object' &&\n    'errno' in error &&\n    typeof (error as {errno: unknown}).errno !== 'undefined',\n  );\n}\n\n// System error codes that indicate transient socket conditions.\n// These are checked via the `code` property on errors.\nconst TRANSIENT_SOCKET_ERROR_CODES = new Set([\n  'EPIPE',\n  'ECONNRESET',\n  'ECANCELED',\n]);\n\n// Error messages that indicate transient socket conditions but don't have\n// standard error codes (e.g., WebSocket library errors).\nconst TRANSIENT_SOCKET_MESSAGE_PATTERNS = [\n  'socket was closed while data was being compressed',\n];\n\nfunction hasTransientSocketCode(error: unknown): boolean {\n  if (!error || typeof error !== 'object') {\n    return false;\n  }\n  const maybeCode =\n    'code' in error ? String((error as {code?: unknown}).code) : undefined;\n  return Boolean(\n    maybeCode && TRANSIENT_SOCKET_ERROR_CODES.has(maybeCode.toUpperCase()),\n  );\n}\n\nfunction isTransientSocketMessage(message: string | undefined): boolean {\n  if (!message) {\n    return false;\n  }\n  const lower = message.toLowerCase();\n  return TRANSIENT_SOCKET_MESSAGE_PATTERNS.some(pattern =>\n    lower.includes(pattern),\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;AAwEA,SAAS,OAAO,OAAgB,UAA2B;CACzD,OACE,OAAO,UAAU,YACjB,UAAU,QACV,OAAO,OAAO,OAAO,QAAQ;AAEjC;AAEA,SAAS,oBAAoB,SAA2B;CACtD,IAAI,CAAC,MAAM,QAAQ,OAAO,GACxB,OAAO;CAGT,MAAM,OAAO,QAAQ;CACrB,IAAI,OAAO,SAAS,YAAY,SAAS,MACvC,OAAO;CAGT,IACE,QAAQ,OAAO,oBACf,QAAQ,OAAO,wBAEf,OAAO;CAGT,MAAM,QAAQ,KAAK;CACnB,OACE,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAK,cAAa,OAAO,WAAW,KAAK,CAAC;AAE5E;AAEA,IAAM,kCACJ;AAeF,SAAgB,qBACd,OACA,oBAC6D;CAC7D,IAAI,CAAC,sBAAsB,oBAAoB,KAAK,GAClD,MAAM,IAAI,MAAM,+BAA+B;CAEjD,OAAOA,MAAa,OAAO,wCAAwC,OAAO;AAC5E;AAaA,IAAM,6BAA6B;AACnC,IAAa,4BAA4B;AACzC,IAAM,6BAA6B;AACnC,IAAM,uBAAuB;;;;;;;;;AAU7B,IAAa,aAAb,MAAwB;CACtB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA,mBAA4B,mBAC1B,QACA,oBACA,gCACF;CAEA;CACA;CACA,UAAU;CACV,wBAAwB;CAExB,YACE,IACA,eACA,IACA,oBACA,gBACA,SACA;EACA,MAAM,EAAC,eAAe,UAAU,MAAM,oBAAmB;EACzD,KAAK,kBAAkB;EACvB,KAAK,sBAAsB;EAE3B,KAAK,MAAM;EACX,KAAK,QAAQ;EACb,KAAK,mBAAmB;EAExB,KAAK,MAAM,GACR,YAAY,YAAY,CAAC,CACzB,YAAY,YAAY,QAAQ,CAAC,CACjC,YAAY,iBAAiB,aAAa,CAAC,CAC3C,YAAY,QAAQ,IAAI;EAC3B,KAAK,oBAAoB,IAAI,iBAC3B,KAAK,KACL,IACA,eACF;EACA,KAAK,IAAI,QAAQ,gBAAgB;EACjC,KAAK,WAAW;EAEhB,KAAK,IAAI,iBAAiB,SAAS,KAAK,YAAY;EACpD,KAAK,IAAI,iBAAiB,SAAS,KAAK,YAAY;EAEpD,KAAK,cAAc;EACnB,KAAK,sBAAsB,YACzB,KAAK,gBACL,6BAA6B,CAC/B;CACF;;;;;;;;CASA,OAAgB;EACd,IACE,KAAK,mBAAA,MACL,KAAK,mBAAA,IAEL,KAAK,gBAAgB;GACnB,MAAMC;GACN,SAAS,wDACP,KAAK,iBACN,QACC,KAAK,mBAAA,KAAsC,WAAW,SACvD;GACD,QAAQC;EACV,CAAC;OACI;GACL,MAAM,mBAAqC,CACzC,aACA;IAAC,MAAM,KAAK;IAAO,WAAW,KAAK,IAAI;GAAC,CAC1C;GACA,KAAK,KAAK,kBAAkB,qBAAqB;GACjD,OAAO;EACT;EACA,OAAO;CACT;CAEA,MAAM,QAAgB,GAAG,MAAiB;EACxC,IAAI,KAAK,SACP;EAEF,KAAK,UAAU;EACf,KAAK,IAAI,OAAO,uBAAuB,UAAU,GAAG,IAAI;EACxD,KAAK,IAAI,oBAAoB,SAAS,KAAK,YAAY;EACvD,KAAK,IAAI,oBAAoB,SAAS,KAAK,YAAY;EACvD,KAAK,2BAA2B,OAAO;EACvC,KAAK,4BAA4B,KAAA;EACjC,KAAK,uBAAuB,OAAO;EACnC,KAAK,wBAAwB,KAAA;EAC7B,KAAK,SAAS;EACd,IAAI,KAAK,IAAI,eAAe,KAAK,IAAI,QACnC,KAAK,IAAI,MAAM;EAEjB,aAAa,KAAK,mBAAmB;CAIvC;CAEA,qBAAqB,mBAA2B;EAC9C,OAAO,KAAK,eAAe,EAAC,MAAM,kBAAiB,CAAC;CACtD;CAEA,iBAAiB,OAAO,UAAwB;EAC9C,MAAM,OAAO,MAAM,KAAK,SAAS;EACjC,IAAI,KAAK,SAAS;GAChB,KAAK,IAAI,QAAQ,0CAA0C,IAAI;GAC/D;EACF;EAEA,IAAI;EACJ,IAAI;GAEF,MAAM,qBADQ,KAAK,MAAM,IACE,GAAO,KAAK,mBAAmB;EAC5D,SAAS,GAAG;GACV,MAAM,YAAY;IAChB,MAAMC;IACN,SAAS,OAAO,CAAC;IACjB,QAAQD;GACV;GACA,KAAK,gBACH,WACA,IAAI,uBAAuB,WAAW,MAAM,CAC9C;GACA;EACF;EAEA,IAAI,CAAC,uCAAuC,QAAQ,GAAG,GAAG;GAExD,IAAI,CAAC,KAAK,uBAAuB;IAC/B,KAAK,wBAAwB;IAC7B,KAAK,IAAI,OACP,oDAAoD,KAAK,UAAU,IAAI,EAAE,GAC3E;GACF;GACA;EACF;EAEA,IAAI;GAEF,IADgB,IAAI,OACJ,QAAQ;IACtB,KAAK,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,qBAAqB;IAC7C;GACF;GAEA,MAAM,SAAS,MAAM,KAAK,gBAAgB,cAAc,GAAG;GAC3D,KAAK,MAAM,KAAK,QACd,KAAK,qBAAqB,CAAC;EAE/B,SAAS,GAAG;GACV,KAAK,iBAAiB,CAAC;EACzB;CACF;CAEA,qBAAqB,QAA6B;EAChD,QAAQ,OAAO,MAAf;GACE,KAAK;IACH,KAAK,gBAAgB,OAAO,KAAK;IACjC;GACF,KAAK,MACH;GACF,KAAK;IACH,QAAQ,OAAO,QAAf;KACE,KAAK;MACH,OACE,KAAK,8BAA8B,KAAA,GACnC,kDACF;MACA,KAAK,4BAA4B,OAAO;MACxC;KACF,KAAK;MACH,OACE,KAAK,0BAA0B,KAAA,GAC/B,kDACF;MACA,KAAK,wBAAwB,OAAO;IAExC;IACA,IAAI,OAAO,WAAW,cACpB,KAAK,eAAe,OAAO,SAAS,YAAY,aAC9C,KAAK,KAAK,WAAW,SAAS,UAAU,WAAW,UAAU,CAC/D;SAEA,KAAK,eAAe,OAAO,SAAS,YAAY,aAC9C,KAAK,KAAK,YAAY,QAAQ,CAChC;IAEF;GAEF,KAAK,aACH,KAAK,MAAM,SAAS,OAAO,QACzB,KAAK,UAAU,KAAK;EAG1B;CACF;CAEA,gBAAgB,MAAkB;EAChC,MAAM,EAAC,MAAM,QAAQ,aAAY;EACjC,IAAI,CAAC,UACH,KAAK,sBAAsB,eAAe;EAE5C,KAAK,MAAM,yBAAyB;GAAC;GAAM;GAAQ;EAAQ,CAAC;CAC9D;CAEA,gBAAgB,MAAkB;EAChC,KAAK,sBAAsB,aAAa;EACxC,KAAK,IAAI,OAAO,yBAAyB,EAAE,SAAS,EAAE,KAAK;CAC7D;CAEA,sBAAsB,WAAmB;EACvC,KAAK,iBAAiB,IAAI,GAAG;IAC1B,6BAA6B,KAAK;IAClC,uBAAuB;EAC1B,CAAC;CACH;CAEA,gBAAgB;EACd,SACE,sBAAsB,KAAK,GAAG,GAC9B,IAAI,SAAS,EACX,QAAQ,MAAM,WAAW,aAAa;GACpC,KAAK,eAAe,EAAC,KAAI,CAAC,CAAC,CAAC,WAAW,SAAS,GAAG,QAAQ;EAC7D,EACF,CAAC,SAIK,CAAC,CACT;CACF;CAEA,eACE,gBACA,aAIA;EAMA,SACE,SAAS,KAAK,cAAc,GAC5B,IAAI,SAAS;GACX,YAAY;GACZ,QAAQ,YAAe,WAAW,aAChC,YAAY,YAAY,QAAQ;EACpC,CAAC,IACD,MACE,IACI,KAAK,iBAAiB,CAAC,IACvB,KAAK,MAAM,iCAAiC,CACpD;CACF;CAEA,iBAAiB,GAAY;EAC3B,MAAM,YACJ,kBAAkB,CAAC,CAAC,EAAE,aAAa,sBAAsB,CAAC,CAAC,CAAC;EAE9D,KAAK,gBAAgB,WAAW,CAAC;CACnC;CAEA,gBAAgB,WAAsB,QAAkB;EACtD,KAAK,UAAU,WAAW,MAAM;EAChC,KAAK,MACH,GAAG,UAAU,KAAK,IAAI,UAAU,OAAO,KAAK,UAAU,WACtD,SACF;CACF;CAEA,yBAAyB,KAAK,IAAI;CAElC,uBAAuB;EACrB,IAAI,KAAK,IAAI,IAAI,KAAK,yBAAyB,4BAA4B;GACzE,KAAK,IAAI,QAAQ,uBAAuB;GACxC,KAAK,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,qBAAqB;EAC/C;CACF;CAEA,KACE,MACA,UACA,YACA;EACA,KAAK,yBAAyB,KAAK,IAAI;EACvC,KAAK,kBAAkB,KAAK,MAAM,UAAU,UAAU;CACxD;CAEA,UAAU,WAAsB,QAAkB;EAChD,UAAU,KAAK,KAAK,KAAK,KAAK,WAAW,MAAM;CACjD;AACF;AAOA,SAAgB,KACd,IACA,IACA,MACA,UACA,YACA;CACA,IAAI,GAAG,eAAeE,YAAU,MAAM;EACpC,eAAe,KAAK,UAAU,IAAI;EAClC,IAAI,aAAa,uBAAuB;GACtC,GAAG,KAAK,UAAU;GAClB;EACF;EAEA,IAAI,YAAY;EAChB,MAAM,QAAQ,iBAAiB;GAC7B,SAAS,0BAA0B,CAAC;EACtC,GAAG,yBAAyB;EAC5B,MAAM,MAAM;EAEZ,MAAM,YAAY,UAAyB;GACzC,IAAI,WACF;GAEF,YAAY;GACZ,aAAa,KAAK;GAClB,SAAS,KAAK;EAChB;EAEA,GAAG,KAAK,YAAY,QAAQ;CAC9B,OAAO;EACL,GAAG,QAAQ,2CAA2C,GAAG,WAAW,IAAI,EACtE,SAAS,KACX,CAAC;EACD,IAAI,aAAa,uBACf,SACE,IAAI,uBACF;GACE,MAAMC;GACN,SAAS;GACT,QAAQH;EACV,GACA,MACF,CACF;CAEJ;AACF;;;;;;;;AASA,IAAa,mBAAb,MAA8B;CAC5B;CACA;CACA;CACA;CAEA,YAAY,IAAgB,IAAmB,iBAAyB;EACtE,KAAK,MAAM;EACX,KAAK,MAAM;EACX,KAAK,mBAAmB;CAC1B;CAEA,KACE,MACA,UACA,YACM;EACN,IAAI,KAAK,oBAAA,IACP,QAAQ,KAAK,IAAb;GACE,KAAK;IACH,OACE,KAAK,sBAAsB,KAAA,GAC3B,uDACF;IACA,KAAK,oBAAoB,IAAI,iBAAiB;IAC9C;GACF,KAAK;IACH,KAAK,cAAc,MAAM,UAAU,UAAU;IAC7C;GACF,KAAK;IACH,KAAK,aAAa,MAAM,QAAQ;IAChC;EACJ;EAEF,KAAK,KAAK,KAAK,KAAK,KAAK,MAAM,UAAU,UAAU;CACrD;CAEA,cACE,UACA,UACA,YACM;EACN,MAAM,UAAU,KAAK;EACrB,OAAO,SAAS,uCAAuC;EAEvD,eAAe,KAAK,UAAU,QAAQ;EACtC,OACE,WAAW,WAAW,gBAAgB,KAAK,WAAW,SAAS,GAAG,GAClE,6BACF;EACA,QACG,SAAS,WAAW,MAAM,iBAAiB,QAAQ,EAAE,IAAG,UACvD,WAAW,KAAK,KAAK,KAAK,KAAK,KAAK,CACtC,CAAC,CACA,WACO,eAAe,QAAQ,IAC7B,UAAS,eAAe,UAAU,QAAQ,KAAK,CAAC,CAClD;CACJ;CAEA,aACE,SACA,UACM;EACN,MAAM,UAAU,KAAK;EACrB,OAAO,SAAS,sCAAsC;EACtD,KAAK,oBAAoB,KAAA;EAEzB,IAAI,QAAQ,EAAE,CAAC,QAAQ;GACrB,QAAQ,OAAO;GACf,KAAK,KAAK,KAAK,KAAK,KAAK,SAAS,QAAQ;GAC1C;EACF;EAEA,QACG,QAAO,UAAS,WAAW,KAAK,KAAK,KAAK,KAAK,KAAK,CAAC,CAAC,CACtD,WAAW,UAAU,KAAK,KAAK,KAAK,KAAK,OAAO,CAAC,CAAC,CAClD,WACO,eAAe,QAAQ,IAC7B,UAAS,eAAe,UAAU,QAAQ,KAAK,CAAC,CAClD;CACJ;AACF;AAEA,IAAM,mBAAmB;AAEzB,SAAS,UACP,IACA,IACA,MACe;CACf,OAAO,IAAI,SAAS,SAAS,WAAW;EACtC,KAAK,IAAI,IAAI,OAAM,UAAU,QAAQ,OAAO,KAAK,IAAI,QAAQ,CAAE;CACjE,CAAC;AACH;AAEA,SAAS,WACP,IACA,IACA,OACe;CACf,OAAO,IAAI,SAAS,SAAS,WAAW;EACtC,IAAI,GAAG,eAAeE,YAAU,MAAM;GACpC,MAAM,QAAQ,qBAAqB;GACnC,GAAG,QACD,kDAAkD,GAAG,WAAW,EAClE;GACA,OAAO,KAAK;GACZ;EACF;EACA,IAAI,YAAY;EAChB,MAAM,QAAQ,iBACN,SAAS,0BAA0B,CAAC,GAC1C,yBACF;EACA,MAAM,MAAM;EAEZ,MAAM,YAAY,UAAyB;GACzC,IAAI,WACF;GAEF,YAAY;GACZ,aAAa,KAAK;GAClB,QAAQ,OAAO,KAAK,IAAI,QAAQ;EAClC;EAEA,GAAG,KAAK,OAAO,QAAQ;CACzB,CAAC;AACH;AAEA,SAAS,eACP,UACA,OACM;CACN,IAAI,aAAa,uBACf,SAAS,KAAK;AAElB;AAEA,SAAS,QAAQ,OAAuB;CACtC,OAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AACjE;AAEA,SAAS,uBAA+C;CACtD,OAAO,IAAI,uBACT;EACE,MAAMC;EACN,SAAS;EACT,QAAQH;CACV,GACA,MACF;AACF;AAEA,SAAS,4BAAoD;CAC3D,OAAO,IAAI,uBACT;EACE,MAAMG;EACN,SAAS,kCAAkC,0BAA0B;EACrE,QAAQH;CACV,GACA,MACF;AACF;AAEA,SAAgB,UACd,IACA,IACA,WACA,QACA;CACA,KAAK,GAAG,YAAY,aAAa,UAAU,IAAI;CAE/C,IAAI;CAGJ,IAAI,kBAAkB,wBACpB,WAAW,OAAO;MAIf,IACH,SAAS,MAAM,KACf,uBAAuB,MAAM,KAC7B,yBAAyB,UAAU,OAAO,GAE1C,WAAW;MAGR,IACH,UAAU,SAAS,oBACnB,UAAU,SAAS,mBAEnB,WAAW;MAEX,WAAW,SAAS,YAAY,MAAM,IAAI;CAG5C,GAAG,SAAS,GAAG,8BAA8B,WAAW,UAAU,EAAE;CACpE,KAAK,IAAI,IAAI,CAAC,SAAS,SAAS,GAAG,qBAAqB;AAC1D;AAEA,SAAgB,kBAAkB,OAA2C;CAC3E,IAAI,gBAAgB,KAAK,GACvB,OAAO;CAET,IAAI,iBAAiB,SAAS,MAAM,OAClC,OAAO,kBAAkB,MAAM,KAAK;AAGxC;AAEA,SAAS,SAAS,OAAyB;CACzC,OAAO,QACL,SACA,OAAO,UAAU,YACjB,WAAW,SACX,OAAQ,MAA2B,UAAU,WAC/C;AACF;AAIA,IAAM,+CAA+B,IAAI,IAAI;CAC3C;CACA;CACA;AACF,CAAC;AAID,IAAM,oCAAoC,CACxC,mDACF;AAEA,SAAS,uBAAuB,OAAyB;CACvD,IAAI,CAAC,SAAS,OAAO,UAAU,UAC7B,OAAO;CAET,MAAM,YACJ,UAAU,QAAQ,OAAQ,MAA2B,IAAI,IAAI,KAAA;CAC/D,OAAO,QACL,aAAa,6BAA6B,IAAI,UAAU,YAAY,CAAC,CACvE;AACF;AAEA,SAAS,yBAAyB,SAAsC;CACtE,IAAI,CAAC,SACH,OAAO;CAET,MAAM,QAAQ,QAAQ,YAAY;CAClC,OAAO,kCAAkC,MAAK,YAC5C,MAAM,SAAS,OAAO,CACxB;AACF"}
+{"version":3,"file":"connection.js","names":["valita.parse","ErrorKind.VersionNotSupported","ErrorOrigin.ZeroCache","ErrorKind.InvalidMessage","WebSocket","ErrorKind.Internal"],"sources":["../../../../../zero-cache/src/workers/connection.ts"],"sourcesContent":["import {pipeline, Readable, Writable} from 'node:stream';\nimport type {LogContext, LogLevel} from '@rocicorp/logger';\nimport type {CloseEvent, Data, ErrorEvent} from 'ws';\nimport WebSocket, {createWebSocketStream} from 'ws';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport * as valita from '../../../shared/src/valita.ts';\nimport type {ConnectedMessage} from '../../../zero-protocol/src/connect.ts';\nimport type {Downstream} from '../../../zero-protocol/src/down.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\nimport type {ErrorBody} from '../../../zero-protocol/src/error.ts';\nimport {\n  isProtocolError,\n  type ProtocolError,\n} from '../../../zero-protocol/src/error.ts';\nimport {\n  POKE_CHUNK_PROTOCOL_VERSION,\n  type PokeChunk,\n  type PokeEndMessage,\n  type PokePartMessage,\n} from '../../../zero-protocol/src/poke.ts';\nimport {\n  MIN_SERVER_SUPPORTED_SYNC_PROTOCOL,\n  PROTOCOL_VERSION,\n} from '../../../zero-protocol/src/protocol-version.ts';\nimport {\n  upstreamSchemaWithUnparsedAnalyzeQuery,\n  type UpstreamWithUnparsedAnalyzeQuery,\n} from '../../../zero-protocol/src/up.ts';\nimport {getOrCreateCounter} from '../observability/metrics.ts';\nimport type {ViewSyncerDownstream} from '../types/downstream.ts';\nimport {\n  ProtocolErrorWithLevel,\n  getLogLevel,\n  wrapWithProtocolError,\n} from '../types/error-with-level.ts';\nimport {PokeChunkEncoder} from '../types/poke-chunk.ts';\nimport type {Source} from '../types/streams.ts';\nimport type {ConnectParams} from './connect-params.ts';\n\nexport type HandlerResult =\n  | {\n      type: 'ok';\n    }\n  | {\n      type: 'fatal';\n      error: ErrorBody;\n    }\n  | {\n      type: 'transient';\n      errors: ErrorBody[];\n    }\n  | StreamResult;\n\nexport type StreamResult =\n  | {\n      type: 'stream';\n      source: 'viewSyncer';\n      stream: Source<ViewSyncerDownstream>;\n    }\n  | {\n      type: 'stream';\n      source: 'pusher';\n      stream: Source<Downstream>;\n    };\n\nexport interface MessageHandler {\n  handleMessage(\n    msg: UpstreamWithUnparsedAnalyzeQuery,\n  ): Promise<HandlerResult[]>;\n}\n\nfunction hasOwn(value: unknown, property: string): boolean {\n  return (\n    typeof value === 'object' &&\n    value !== null &&\n    Object.hasOwn(value, property)\n  );\n}\n\nfunction containsLegacyQuery(message: unknown): boolean {\n  if (!Array.isArray(message)) {\n    return false;\n  }\n\n  const body = message[1];\n  if (typeof body !== 'object' || body === null) {\n    return false;\n  }\n\n  if (\n    message[0] !== 'initConnection' &&\n    message[0] !== 'changeDesiredQueries'\n  ) {\n    return false;\n  }\n\n  const patch = body['desiredQueriesPatch'];\n  return (\n    Array.isArray(patch) && patch.some(operation => hasOwn(operation, 'ast'))\n  );\n}\n\nconst LEGACY_QUERIES_DISABLED_MESSAGE =\n  'Legacy queries are disabled by this Zero server. This client must use custom queries instead of sending query ASTs directly. To temporarily restore compatibility, set ZERO_ALLOW_LEGACY_QUERIES=true on zero-cache.';\n\n// Exported for testing purposes.\n//\n// Client messages are parsed leniently, so a newer client can add message\n// kinds and fields:\n// - A message of a kind this server doesn't know parses as an\n//   UnrecognizedTuple, for the connection to drop. Malformed messages still\n//   throw.\n// - Messages are parsed in strip mode, so fields this server doesn't know are\n//   dropped. Strip rather than passthrough: nothing this server doesn't know\n//   reaches the CVR or the API server, and valita's passthrough drops unknown\n//   fields anyway on any object with a transformed field below it (such as\n//   jsonSchema args). ASTs stay strict in the schema, because a server that\n//   ignored an AST field would run a different query.\nexport function parseUpstreamMessage(\n  value: unknown,\n  allowLegacyQueries: boolean,\n): UpstreamWithUnparsedAnalyzeQuery | valita.UnrecognizedTuple {\n  if (!allowLegacyQueries && containsLegacyQuery(value)) {\n    throw new Error(LEGACY_QUERIES_DISABLED_MESSAGE);\n  }\n  return valita.parse(value, upstreamSchemaWithUnparsedAnalyzeQuery, 'strip');\n}\n\n// Ensures that a downstream message is sent at least every interval, sending a\n// 'pong' if necessary. This is set to be slightly longer than the client-side\n// PING_INTERVAL of 5 seconds, so that in the common case, 'pong's are sent in\n// response to client-initiated 'ping's. However, if the inbound stream is\n// backed up because a command is taking a long time to process, the pings\n// will be stuck in the queue (i.e. back-pressured), in which case pongs will\n// be manually sent to notify the client of server liveness.\n//\n// This is equivalent to what is done for Postgres keepalives on the\n// replication stream (which can similarly be back-pressured):\n// https://github.com/rocicorp/mono/blob/f98cb369a2dbb15650328859c732db358f187ef0/packages/zero-cache/src/services/change-source/pg/logical-replication/stream.ts#L21\nconst DOWNSTREAM_MSG_INTERVAL_MS = 6_000;\nexport const WEBSOCKET_SEND_TIMEOUT_MS = 10_000;\nconst PROTOCOL_VERSION_ATTRIBUTE = 'protocol.version';\nconst EVENT_TYPE_ATTRIBUTE = 'event.type';\n\n/**\n * Represents a connection between the client and server.\n *\n * Handles incoming messages on the connection and dispatches\n * them to the correct service.\n *\n * Listens to the ViewSyncer and sends messages to the client.\n */\nexport class Connection {\n  readonly #ws: WebSocket;\n  readonly #wsID: string;\n  readonly #protocolVersion: number;\n  readonly #lc: LogContext;\n  readonly #onClose: () => void;\n  readonly #messageHandler: MessageHandler;\n  readonly #allowLegacyQueries: boolean;\n  readonly #downstreamSender: DownstreamSender;\n  readonly #downstreamMsgTimer: NodeJS.Timeout | undefined;\n  readonly #webSocketErrors = getOrCreateCounter(\n    'sync',\n    'websocket.errors',\n    'Client WebSocket error events.',\n  );\n\n  #viewSyncerOutboundStream: Source<ViewSyncerDownstream> | undefined;\n  #pusherOutboundStream: Source<Downstream> | undefined;\n  #closed = false;\n  #loggedUnknownMessage = false;\n\n  constructor(\n    lc: LogContext,\n    connectParams: ConnectParams,\n    ws: WebSocket,\n    allowLegacyQueries: boolean,\n    messageHandler: MessageHandler,\n    onClose: () => void,\n  ) {\n    const {clientGroupID, clientID, wsID, protocolVersion} = connectParams;\n    this.#messageHandler = messageHandler;\n    this.#allowLegacyQueries = allowLegacyQueries;\n\n    this.#ws = ws;\n    this.#wsID = wsID;\n    this.#protocolVersion = protocolVersion;\n\n    this.#lc = lc\n      .withContext('connection')\n      .withContext('clientID', clientID)\n      .withContext('clientGroupID', clientGroupID)\n      .withContext('wsID', wsID);\n    this.#downstreamSender = new DownstreamSender(\n      this.#lc,\n      ws,\n      protocolVersion,\n    );\n    this.#lc.debug?.('new connection');\n    this.#onClose = onClose;\n\n    this.#ws.addEventListener('close', this.#handleClose);\n    this.#ws.addEventListener('error', this.#handleError);\n\n    this.#proxyInbound();\n    this.#downstreamMsgTimer = setInterval(\n      this.#maybeSendPong,\n      DOWNSTREAM_MSG_INTERVAL_MS / 2,\n    );\n  }\n\n  /**\n   * Checks the protocol version and errors for unsupported protocols,\n   * sending the initial `connected` response on success.\n   *\n   * This is early in the connection lifecycle because {@link #handleMessage}\n   * will only parse messages with schema(s) of supported protocol versions.\n   */\n  init(): boolean {\n    if (\n      this.#protocolVersion > PROTOCOL_VERSION ||\n      this.#protocolVersion < MIN_SERVER_SUPPORTED_SYNC_PROTOCOL\n    ) {\n      this.#closeWithError({\n        kind: ErrorKind.VersionNotSupported,\n        message: `server is at sync protocol v${PROTOCOL_VERSION} and does not support v${\n          this.#protocolVersion\n        }. The ${\n          this.#protocolVersion > PROTOCOL_VERSION ? 'server' : 'client'\n        } must be updated to a newer release.`,\n        origin: ErrorOrigin.ZeroCache,\n      });\n    } else {\n      const connectedMessage: ConnectedMessage = [\n        'connected',\n        {wsid: this.#wsID, timestamp: Date.now()},\n      ];\n      this.send(connectedMessage, 'ignore-backpressure');\n      return true;\n    }\n    return false;\n  }\n\n  get wsID(): string {\n    return this.#wsID;\n  }\n\n  close(reason: string, ...args: unknown[]) {\n    if (this.#closed) {\n      return;\n    }\n    this.#closed = true;\n    this.#lc.info?.(`closing connection: ${reason}`, ...args);\n    this.#ws.removeEventListener('close', this.#handleClose);\n    this.#ws.removeEventListener('error', this.#handleError);\n    this.#viewSyncerOutboundStream?.cancel();\n    this.#viewSyncerOutboundStream = undefined;\n    this.#pusherOutboundStream?.cancel();\n    this.#pusherOutboundStream = undefined;\n    this.#onClose();\n    if (this.#ws.readyState !== this.#ws.CLOSED) {\n      this.#ws.close();\n    }\n    clearTimeout(this.#downstreamMsgTimer);\n\n    // spin down services if we have\n    // no more client connections for the client group?\n  }\n\n  handleInitConnection(initConnectionMsg: string) {\n    return this.#handleMessage({data: initConnectionMsg});\n  }\n\n  #handleMessage = async (event: {data: Data}) => {\n    const data = event.data.toString();\n    if (this.#closed) {\n      this.#lc.debug?.('Ignoring message received after closed', data);\n      return;\n    }\n\n    let msg;\n    try {\n      const value = JSON.parse(data);\n      msg = parseUpstreamMessage(value, this.#allowLegacyQueries);\n    } catch (e) {\n      const errorBody = {\n        kind: ErrorKind.InvalidMessage,\n        message: String(e),\n        origin: ErrorOrigin.ZeroCache,\n      } as const;\n      this.#closeWithError(\n        errorBody,\n        new ProtocolErrorWithLevel(errorBody, 'warn'),\n      );\n      return;\n    }\n\n    if (!upstreamSchemaWithUnparsedAnalyzeQuery.isKnown(msg)) {\n      // A message kind from a newer client.\n      if (!this.#loggedUnknownMessage) {\n        this.#loggedUnknownMessage = true;\n        this.#lc.info?.(\n          `Ignoring messages of unknown type, starting with ${JSON.stringify(msg[0])}`,\n        );\n      }\n      return;\n    }\n\n    try {\n      const msgType = msg[0];\n      if (msgType === 'ping') {\n        this.send(['pong', {}], 'ignore-backpressure');\n        return;\n      }\n\n      const result = await this.#messageHandler.handleMessage(msg);\n      for (const r of result) {\n        this.#handleMessageResult(r);\n      }\n    } catch (e) {\n      this.#closeWithThrown(e);\n    }\n  };\n\n  #handleMessageResult(result: HandlerResult): void {\n    switch (result.type) {\n      case 'fatal':\n        this.#closeWithError(result.error);\n        break;\n      case 'ok':\n        break;\n      case 'stream': {\n        switch (result.source) {\n          case 'viewSyncer':\n            assert(\n              this.#viewSyncerOutboundStream === undefined,\n              'Outbound stream already set for this connection!',\n            );\n            this.#viewSyncerOutboundStream = result.stream;\n            break;\n          case 'pusher':\n            assert(\n              this.#pusherOutboundStream === undefined,\n              'Outbound stream already set for this connection!',\n            );\n            this.#pusherOutboundStream = result.stream;\n            break;\n        }\n        if (result.source === 'viewSyncer') {\n          this.#proxyOutbound(result.stream, (downstream, callback) =>\n            this.send(downstream.message, callback, downstream.serialized),\n          );\n        } else {\n          this.#proxyOutbound(result.stream, (downstream, callback) =>\n            this.send(downstream, callback),\n          );\n        }\n        break;\n      }\n      case 'transient': {\n        for (const error of result.errors) {\n          this.sendError(error);\n        }\n      }\n    }\n  }\n\n  #handleClose = (e: CloseEvent) => {\n    const {code, reason, wasClean} = e;\n    if (!wasClean) {\n      this.#recordWebSocketError('unclean_close');\n    }\n    this.close('WebSocket close event', {code, reason, wasClean});\n  };\n\n  #handleError = (e: ErrorEvent) => {\n    this.#recordWebSocketError('error_event');\n    this.#lc.warn?.('WebSocket error event', e.message, e.error);\n  };\n\n  #recordWebSocketError(eventType: string) {\n    this.#webSocketErrors.add(1, {\n      [PROTOCOL_VERSION_ATTRIBUTE]: this.#protocolVersion,\n      [EVENT_TYPE_ATTRIBUTE]: eventType,\n    });\n  }\n\n  #proxyInbound() {\n    pipeline(\n      createWebSocketStream(this.#ws),\n      new Writable({\n        write: (data, _encoding, callback) => {\n          this.#handleMessage({data}).then(() => callback(), callback);\n        },\n      }),\n      // The done callback is not used, as #handleClose and #handleError,\n      // configured on the underlying WebSocket, provide more complete\n      // information.\n      () => {},\n    );\n  }\n\n  #proxyOutbound<T>(\n    outboundStream: Source<T>,\n    sendMessage: (\n      downstream: T,\n      callback: (err?: Error | null) => void,\n    ) => void,\n  ) {\n    // Note: createWebSocketStream() is avoided here in order to control\n    //       exception handling with #closeWithThrown(). If the Writable\n    //       from createWebSocketStream() were instead used, exceptions\n    //       from the outboundStream result in the Writable closing the\n    //       the websocket before the error message can be sent.\n    pipeline(\n      Readable.from(outboundStream),\n      new Writable({\n        objectMode: true,\n        write: (downstream: T, _encoding, callback) =>\n          sendMessage(downstream, callback),\n      }),\n      e =>\n        e\n          ? this.#closeWithThrown(e)\n          : this.close(`downstream closed by ViewSyncer`),\n    );\n  }\n\n  #closeWithThrown(e: unknown) {\n    const errorBody =\n      findProtocolError(e)?.errorBody ?? wrapWithProtocolError(e).errorBody;\n\n    this.#closeWithError(errorBody, e);\n  }\n\n  #closeWithError(errorBody: ErrorBody, thrown?: unknown) {\n    this.sendError(errorBody, thrown);\n    this.close(\n      `${errorBody.kind} (${errorBody.origin}): ${errorBody.message}`,\n      errorBody,\n    );\n  }\n\n  #lastDownstreamMsgTime = Date.now();\n\n  #maybeSendPong = () => {\n    if (Date.now() - this.#lastDownstreamMsgTime > DOWNSTREAM_MSG_INTERVAL_MS) {\n      this.#lc.debug?.('manually sending pong');\n      this.send(['pong', {}], 'ignore-backpressure');\n    }\n  };\n\n  send(\n    data: Downstream,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n    serialized?: string | undefined,\n  ) {\n    this.#lastDownstreamMsgTime = Date.now();\n    this.#downstreamSender.send(data, callback, serialized);\n  }\n\n  sendError(errorBody: ErrorBody, thrown?: unknown) {\n    sendError(this.#lc, this.#ws, errorBody, thrown);\n  }\n}\n\nexport type WebSocketLike = Pick<WebSocket, 'readyState'> & {\n  send(data: string | PokeChunk, cb?: (err?: Error) => void): void;\n};\n\n// Exported for testing purposes.\nexport function send(\n  lc: LogContext,\n  ws: WebSocketLike,\n  data: Downstream,\n  callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  serialized?: string | undefined,\n) {\n  if (ws.readyState === WebSocket.OPEN) {\n    serialized ??= JSON.stringify(data);\n    if (callback === 'ignore-backpressure') {\n      ws.send(serialized);\n      return;\n    }\n\n    let completed = false;\n    const timer = setTimeout(() => {\n      complete(webSocketSendTimeoutError());\n    }, WEBSOCKET_SEND_TIMEOUT_MS);\n    timer.unref();\n\n    const complete = (error?: Error | null) => {\n      if (completed) {\n        return;\n      }\n      completed = true;\n      clearTimeout(timer);\n      callback(error);\n    };\n\n    ws.send(serialized, complete);\n  } else {\n    lc.debug?.(`Dropping outbound message on ws (state: ${ws.readyState})`, {\n      dropped: data,\n    });\n    if (callback !== 'ignore-backpressure') {\n      callback(\n        new ProtocolErrorWithLevel(\n          {\n            kind: ErrorKind.Internal,\n            message: 'WebSocket closed',\n            origin: ErrorOrigin.ZeroCache,\n          },\n          'info',\n        ),\n      );\n    }\n  }\n}\n\n/**\n * Sends downstream messages in the format supported by a connection's sync\n * protocol version. Keep version forks contained here so the view-syncer can\n * continue producing one canonical poke representation.\n *\n * Exported for compatibility testing.\n */\nexport class DownstreamSender {\n  readonly #lc: LogContext;\n  readonly #ws: WebSocketLike;\n  readonly #protocolVersion: number;\n  #pokeChunkEncoder: PokeChunkEncoder | undefined;\n\n  constructor(lc: LogContext, ws: WebSocketLike, protocolVersion: number) {\n    this.#lc = lc;\n    this.#ws = ws;\n    this.#protocolVersion = protocolVersion;\n  }\n\n  send(\n    data: Downstream,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n    serialized?: string | undefined,\n  ): void {\n    if (this.#protocolVersion >= POKE_CHUNK_PROTOCOL_VERSION) {\n      switch (data[0]) {\n        case 'pokeStart':\n          assert(\n            this.#pokeChunkEncoder === undefined,\n            'cannot start a poke while another poke is in progress',\n          );\n          this.#pokeChunkEncoder = new PokeChunkEncoder();\n          break;\n        case 'pokePart':\n          this.#sendPokePart(data, callback, serialized);\n          return;\n        case 'pokeEnd':\n          this.#sendPokeEnd(data, callback);\n          return;\n      }\n    }\n    send(this.#lc, this.#ws, data, callback, serialized);\n  }\n\n  #sendPokePart(\n    pokePart: PokePartMessage,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n    serialized?: string | undefined,\n  ): void {\n    const encoder = this.#pokeChunkEncoder;\n    assert(encoder, 'pokePart received without a pokeStart');\n\n    serialized ??= JSON.stringify(pokePart);\n    assert(\n      serialized.startsWith(POKE_PART_PREFIX) && serialized.endsWith(']'),\n      'invalid serialized pokePart',\n    );\n    void encoder\n      .addPatch(serialized.slice(POKE_PART_PREFIX.length, -1), chunk =>\n        sendBinary(this.#lc, this.#ws, chunk),\n      )\n      .then(\n        () => invokeCallback(callback),\n        error => invokeCallback(callback, toError(error)),\n      );\n  }\n\n  #sendPokeEnd(\n    pokeEnd: PokeEndMessage,\n    callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  ): void {\n    const encoder = this.#pokeChunkEncoder;\n    assert(encoder, 'pokeEnd received without a pokeStart');\n    this.#pokeChunkEncoder = undefined;\n\n    if (pokeEnd[1].cancel) {\n      encoder.cancel();\n      send(this.#lc, this.#ws, pokeEnd, callback);\n      return;\n    }\n\n    void encoder\n      .finish(chunk => sendBinary(this.#lc, this.#ws, chunk))\n      .then(() => sendAsync(this.#lc, this.#ws, pokeEnd))\n      .then(\n        () => invokeCallback(callback),\n        error => invokeCallback(callback, toError(error)),\n      );\n  }\n}\n\nconst POKE_PART_PREFIX = '[\"pokePart\",';\n\nfunction sendAsync(\n  lc: LogContext,\n  ws: WebSocketLike,\n  data: Downstream,\n): Promise<void> {\n  return new Promise((resolve, reject) => {\n    send(lc, ws, data, error => (error ? reject(error) : resolve()));\n  });\n}\n\nfunction sendBinary(\n  lc: LogContext,\n  ws: WebSocketLike,\n  chunk: PokeChunk,\n): Promise<void> {\n  return new Promise((resolve, reject) => {\n    if (ws.readyState !== WebSocket.OPEN) {\n      const error = webSocketClosedError();\n      lc.debug?.(\n        `Dropping outbound binary message on ws (state: ${ws.readyState})`,\n      );\n      reject(error);\n      return;\n    }\n    let completed = false;\n    const timer = setTimeout(\n      () => complete(webSocketSendTimeoutError()),\n      WEBSOCKET_SEND_TIMEOUT_MS,\n    );\n    timer.unref();\n\n    const complete = (error?: Error | null) => {\n      if (completed) {\n        return;\n      }\n      completed = true;\n      clearTimeout(timer);\n      error ? reject(error) : resolve();\n    };\n\n    ws.send(chunk, complete);\n  });\n}\n\nfunction invokeCallback(\n  callback: ((err?: Error | null) => void) | 'ignore-backpressure',\n  error?: Error,\n): void {\n  if (callback !== 'ignore-backpressure') {\n    callback(error);\n  }\n}\n\nfunction toError(error: unknown): Error {\n  return error instanceof Error ? error : new Error(String(error));\n}\n\nfunction webSocketClosedError(): ProtocolErrorWithLevel {\n  return new ProtocolErrorWithLevel(\n    {\n      kind: ErrorKind.Internal,\n      message: 'WebSocket closed',\n      origin: ErrorOrigin.ZeroCache,\n    },\n    'info',\n  );\n}\n\nfunction webSocketSendTimeoutError(): ProtocolErrorWithLevel {\n  return new ProtocolErrorWithLevel(\n    {\n      kind: ErrorKind.Internal,\n      message: `WebSocket send timed out after ${WEBSOCKET_SEND_TIMEOUT_MS} ms`,\n      origin: ErrorOrigin.ZeroCache,\n    },\n    'info',\n  );\n}\n\nexport function sendError(\n  lc: LogContext,\n  ws: WebSocket,\n  errorBody: ErrorBody,\n  thrown?: unknown,\n) {\n  lc = lc.withContext('errorKind', errorBody.kind);\n\n  let logLevel: LogLevel;\n\n  // If the thrown error is a ProtocolErrorWithLevel, its explicit logLevel takes precedence\n  if (thrown instanceof ProtocolErrorWithLevel) {\n    logLevel = thrown.logLevel;\n  }\n  // Errors with errno or transient socket codes are low-level, transient I/O issues\n  // (e.g., EPIPE, ECONNRESET) and should be warnings, not errors\n  else if (\n    hasErrno(thrown) ||\n    hasTransientSocketCode(thrown) ||\n    isTransientSocketMessage(errorBody.message)\n  ) {\n    logLevel = 'warn';\n  }\n  // Fallback: check errorBody.kind for errors that weren't thrown as ProtocolErrorWithLevel\n  else if (\n    errorBody.kind === ErrorKind.ClientNotFound ||\n    errorBody.kind === ErrorKind.TransformFailed\n  ) {\n    logLevel = 'warn';\n  } else {\n    logLevel = thrown ? getLogLevel(thrown) : 'info';\n  }\n\n  lc[logLevel]?.('Sending error on WebSocket', errorBody, thrown ?? '');\n  send(lc, ws, ['error', errorBody], 'ignore-backpressure');\n}\n\nfunction findProtocolError(error: unknown): ProtocolError | undefined {\n  if (isProtocolError(error)) {\n    return error;\n  }\n  if (error instanceof Error && error.cause) {\n    return findProtocolError(error.cause);\n  }\n  return undefined;\n}\n\nfunction hasErrno(error: unknown): boolean {\n  return Boolean(\n    error &&\n    typeof error === 'object' &&\n    'errno' in error &&\n    typeof (error as {errno: unknown}).errno !== 'undefined',\n  );\n}\n\n// System error codes that indicate transient socket conditions.\n// These are checked via the `code` property on errors.\nconst TRANSIENT_SOCKET_ERROR_CODES = new Set([\n  'EPIPE',\n  'ECONNRESET',\n  'ECANCELED',\n]);\n\n// Error messages that indicate transient socket conditions but don't have\n// standard error codes (e.g., WebSocket library errors).\nconst TRANSIENT_SOCKET_MESSAGE_PATTERNS = [\n  'socket was closed while data was being compressed',\n];\n\nfunction hasTransientSocketCode(error: unknown): boolean {\n  if (!error || typeof error !== 'object') {\n    return false;\n  }\n  const maybeCode =\n    'code' in error ? String((error as {code?: unknown}).code) : undefined;\n  return Boolean(\n    maybeCode && TRANSIENT_SOCKET_ERROR_CODES.has(maybeCode.toUpperCase()),\n  );\n}\n\nfunction isTransientSocketMessage(message: string | undefined): boolean {\n  if (!message) {\n    return false;\n  }\n  const lower = message.toLowerCase();\n  return TRANSIENT_SOCKET_MESSAGE_PATTERNS.some(pattern =>\n    lower.includes(pattern),\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;AAwEA,SAAS,OAAO,OAAgB,UAA2B;CACzD,OACE,OAAO,UAAU,YACjB,UAAU,QACV,OAAO,OAAO,OAAO,QAAQ;AAEjC;AAEA,SAAS,oBAAoB,SAA2B;CACtD,IAAI,CAAC,MAAM,QAAQ,OAAO,GACxB,OAAO;CAGT,MAAM,OAAO,QAAQ;CACrB,IAAI,OAAO,SAAS,YAAY,SAAS,MACvC,OAAO;CAGT,IACE,QAAQ,OAAO,oBACf,QAAQ,OAAO,wBAEf,OAAO;CAGT,MAAM,QAAQ,KAAK;CACnB,OACE,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAK,cAAa,OAAO,WAAW,KAAK,CAAC;AAE5E;AAEA,IAAM,kCACJ;AAeF,SAAgB,qBACd,OACA,oBAC6D;CAC7D,IAAI,CAAC,sBAAsB,oBAAoB,KAAK,GAClD,MAAM,IAAI,MAAM,+BAA+B;CAEjD,OAAOA,MAAa,OAAO,wCAAwC,OAAO;AAC5E;AAaA,IAAM,6BAA6B;AACnC,IAAa,4BAA4B;AACzC,IAAM,6BAA6B;AACnC,IAAM,uBAAuB;;;;;;;;;AAU7B,IAAa,aAAb,MAAwB;CACtB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA,mBAA4B,mBAC1B,QACA,oBACA,gCACF;CAEA;CACA;CACA,UAAU;CACV,wBAAwB;CAExB,YACE,IACA,eACA,IACA,oBACA,gBACA,SACA;EACA,MAAM,EAAC,eAAe,UAAU,MAAM,oBAAmB;EACzD,KAAK,kBAAkB;EACvB,KAAK,sBAAsB;EAE3B,KAAK,MAAM;EACX,KAAK,QAAQ;EACb,KAAK,mBAAmB;EAExB,KAAK,MAAM,GACR,YAAY,YAAY,CAAC,CACzB,YAAY,YAAY,QAAQ,CAAC,CACjC,YAAY,iBAAiB,aAAa,CAAC,CAC3C,YAAY,QAAQ,IAAI;EAC3B,KAAK,oBAAoB,IAAI,iBAC3B,KAAK,KACL,IACA,eACF;EACA,KAAK,IAAI,QAAQ,gBAAgB;EACjC,KAAK,WAAW;EAEhB,KAAK,IAAI,iBAAiB,SAAS,KAAK,YAAY;EACpD,KAAK,IAAI,iBAAiB,SAAS,KAAK,YAAY;EAEpD,KAAK,cAAc;EACnB,KAAK,sBAAsB,YACzB,KAAK,gBACL,6BAA6B,CAC/B;CACF;;;;;;;;CASA,OAAgB;EACd,IACE,KAAK,mBAAA,MACL,KAAK,mBAAA,IAEL,KAAK,gBAAgB;GACnB,MAAMC;GACN,SAAS,wDACP,KAAK,iBACN,QACC,KAAK,mBAAA,KAAsC,WAAW,SACvD;GACD,QAAQC;EACV,CAAC;OACI;GACL,MAAM,mBAAqC,CACzC,aACA;IAAC,MAAM,KAAK;IAAO,WAAW,KAAK,IAAI;GAAC,CAC1C;GACA,KAAK,KAAK,kBAAkB,qBAAqB;GACjD,OAAO;EACT;EACA,OAAO;CACT;CAEA,IAAI,OAAe;EACjB,OAAO,KAAK;CACd;CAEA,MAAM,QAAgB,GAAG,MAAiB;EACxC,IAAI,KAAK,SACP;EAEF,KAAK,UAAU;EACf,KAAK,IAAI,OAAO,uBAAuB,UAAU,GAAG,IAAI;EACxD,KAAK,IAAI,oBAAoB,SAAS,KAAK,YAAY;EACvD,KAAK,IAAI,oBAAoB,SAAS,KAAK,YAAY;EACvD,KAAK,2BAA2B,OAAO;EACvC,KAAK,4BAA4B,KAAA;EACjC,KAAK,uBAAuB,OAAO;EACnC,KAAK,wBAAwB,KAAA;EAC7B,KAAK,SAAS;EACd,IAAI,KAAK,IAAI,eAAe,KAAK,IAAI,QACnC,KAAK,IAAI,MAAM;EAEjB,aAAa,KAAK,mBAAmB;CAIvC;CAEA,qBAAqB,mBAA2B;EAC9C,OAAO,KAAK,eAAe,EAAC,MAAM,kBAAiB,CAAC;CACtD;CAEA,iBAAiB,OAAO,UAAwB;EAC9C,MAAM,OAAO,MAAM,KAAK,SAAS;EACjC,IAAI,KAAK,SAAS;GAChB,KAAK,IAAI,QAAQ,0CAA0C,IAAI;GAC/D;EACF;EAEA,IAAI;EACJ,IAAI;GAEF,MAAM,qBADQ,KAAK,MAAM,IACE,GAAO,KAAK,mBAAmB;EAC5D,SAAS,GAAG;GACV,MAAM,YAAY;IAChB,MAAMC;IACN,SAAS,OAAO,CAAC;IACjB,QAAQD;GACV;GACA,KAAK,gBACH,WACA,IAAI,uBAAuB,WAAW,MAAM,CAC9C;GACA;EACF;EAEA,IAAI,CAAC,uCAAuC,QAAQ,GAAG,GAAG;GAExD,IAAI,CAAC,KAAK,uBAAuB;IAC/B,KAAK,wBAAwB;IAC7B,KAAK,IAAI,OACP,oDAAoD,KAAK,UAAU,IAAI,EAAE,GAC3E;GACF;GACA;EACF;EAEA,IAAI;GAEF,IADgB,IAAI,OACJ,QAAQ;IACtB,KAAK,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,qBAAqB;IAC7C;GACF;GAEA,MAAM,SAAS,MAAM,KAAK,gBAAgB,cAAc,GAAG;GAC3D,KAAK,MAAM,KAAK,QACd,KAAK,qBAAqB,CAAC;EAE/B,SAAS,GAAG;GACV,KAAK,iBAAiB,CAAC;EACzB;CACF;CAEA,qBAAqB,QAA6B;EAChD,QAAQ,OAAO,MAAf;GACE,KAAK;IACH,KAAK,gBAAgB,OAAO,KAAK;IACjC;GACF,KAAK,MACH;GACF,KAAK;IACH,QAAQ,OAAO,QAAf;KACE,KAAK;MACH,OACE,KAAK,8BAA8B,KAAA,GACnC,kDACF;MACA,KAAK,4BAA4B,OAAO;MACxC;KACF,KAAK;MACH,OACE,KAAK,0BAA0B,KAAA,GAC/B,kDACF;MACA,KAAK,wBAAwB,OAAO;IAExC;IACA,IAAI,OAAO,WAAW,cACpB,KAAK,eAAe,OAAO,SAAS,YAAY,aAC9C,KAAK,KAAK,WAAW,SAAS,UAAU,WAAW,UAAU,CAC/D;SAEA,KAAK,eAAe,OAAO,SAAS,YAAY,aAC9C,KAAK,KAAK,YAAY,QAAQ,CAChC;IAEF;GAEF,KAAK,aACH,KAAK,MAAM,SAAS,OAAO,QACzB,KAAK,UAAU,KAAK;EAG1B;CACF;CAEA,gBAAgB,MAAkB;EAChC,MAAM,EAAC,MAAM,QAAQ,aAAY;EACjC,IAAI,CAAC,UACH,KAAK,sBAAsB,eAAe;EAE5C,KAAK,MAAM,yBAAyB;GAAC;GAAM;GAAQ;EAAQ,CAAC;CAC9D;CAEA,gBAAgB,MAAkB;EAChC,KAAK,sBAAsB,aAAa;EACxC,KAAK,IAAI,OAAO,yBAAyB,EAAE,SAAS,EAAE,KAAK;CAC7D;CAEA,sBAAsB,WAAmB;EACvC,KAAK,iBAAiB,IAAI,GAAG;IAC1B,6BAA6B,KAAK;IAClC,uBAAuB;EAC1B,CAAC;CACH;CAEA,gBAAgB;EACd,SACE,sBAAsB,KAAK,GAAG,GAC9B,IAAI,SAAS,EACX,QAAQ,MAAM,WAAW,aAAa;GACpC,KAAK,eAAe,EAAC,KAAI,CAAC,CAAC,CAAC,WAAW,SAAS,GAAG,QAAQ;EAC7D,EACF,CAAC,SAIK,CAAC,CACT;CACF;CAEA,eACE,gBACA,aAIA;EAMA,SACE,SAAS,KAAK,cAAc,GAC5B,IAAI,SAAS;GACX,YAAY;GACZ,QAAQ,YAAe,WAAW,aAChC,YAAY,YAAY,QAAQ;EACpC,CAAC,IACD,MACE,IACI,KAAK,iBAAiB,CAAC,IACvB,KAAK,MAAM,iCAAiC,CACpD;CACF;CAEA,iBAAiB,GAAY;EAC3B,MAAM,YACJ,kBAAkB,CAAC,CAAC,EAAE,aAAa,sBAAsB,CAAC,CAAC,CAAC;EAE9D,KAAK,gBAAgB,WAAW,CAAC;CACnC;CAEA,gBAAgB,WAAsB,QAAkB;EACtD,KAAK,UAAU,WAAW,MAAM;EAChC,KAAK,MACH,GAAG,UAAU,KAAK,IAAI,UAAU,OAAO,KAAK,UAAU,WACtD,SACF;CACF;CAEA,yBAAyB,KAAK,IAAI;CAElC,uBAAuB;EACrB,IAAI,KAAK,IAAI,IAAI,KAAK,yBAAyB,4BAA4B;GACzE,KAAK,IAAI,QAAQ,uBAAuB;GACxC,KAAK,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,qBAAqB;EAC/C;CACF;CAEA,KACE,MACA,UACA,YACA;EACA,KAAK,yBAAyB,KAAK,IAAI;EACvC,KAAK,kBAAkB,KAAK,MAAM,UAAU,UAAU;CACxD;CAEA,UAAU,WAAsB,QAAkB;EAChD,UAAU,KAAK,KAAK,KAAK,KAAK,WAAW,MAAM;CACjD;AACF;AAOA,SAAgB,KACd,IACA,IACA,MACA,UACA,YACA;CACA,IAAI,GAAG,eAAeE,YAAU,MAAM;EACpC,eAAe,KAAK,UAAU,IAAI;EAClC,IAAI,aAAa,uBAAuB;GACtC,GAAG,KAAK,UAAU;GAClB;EACF;EAEA,IAAI,YAAY;EAChB,MAAM,QAAQ,iBAAiB;GAC7B,SAAS,0BAA0B,CAAC;EACtC,GAAG,yBAAyB;EAC5B,MAAM,MAAM;EAEZ,MAAM,YAAY,UAAyB;GACzC,IAAI,WACF;GAEF,YAAY;GACZ,aAAa,KAAK;GAClB,SAAS,KAAK;EAChB;EAEA,GAAG,KAAK,YAAY,QAAQ;CAC9B,OAAO;EACL,GAAG,QAAQ,2CAA2C,GAAG,WAAW,IAAI,EACtE,SAAS,KACX,CAAC;EACD,IAAI,aAAa,uBACf,SACE,IAAI,uBACF;GACE,MAAMC;GACN,SAAS;GACT,QAAQH;EACV,GACA,MACF,CACF;CAEJ;AACF;;;;;;;;AASA,IAAa,mBAAb,MAA8B;CAC5B;CACA;CACA;CACA;CAEA,YAAY,IAAgB,IAAmB,iBAAyB;EACtE,KAAK,MAAM;EACX,KAAK,MAAM;EACX,KAAK,mBAAmB;CAC1B;CAEA,KACE,MACA,UACA,YACM;EACN,IAAI,KAAK,oBAAA,IACP,QAAQ,KAAK,IAAb;GACE,KAAK;IACH,OACE,KAAK,sBAAsB,KAAA,GAC3B,uDACF;IACA,KAAK,oBAAoB,IAAI,iBAAiB;IAC9C;GACF,KAAK;IACH,KAAK,cAAc,MAAM,UAAU,UAAU;IAC7C;GACF,KAAK;IACH,KAAK,aAAa,MAAM,QAAQ;IAChC;EACJ;EAEF,KAAK,KAAK,KAAK,KAAK,KAAK,MAAM,UAAU,UAAU;CACrD;CAEA,cACE,UACA,UACA,YACM;EACN,MAAM,UAAU,KAAK;EACrB,OAAO,SAAS,uCAAuC;EAEvD,eAAe,KAAK,UAAU,QAAQ;EACtC,OACE,WAAW,WAAW,gBAAgB,KAAK,WAAW,SAAS,GAAG,GAClE,6BACF;EACA,QACG,SAAS,WAAW,MAAM,iBAAiB,QAAQ,EAAE,IAAG,UACvD,WAAW,KAAK,KAAK,KAAK,KAAK,KAAK,CACtC,CAAC,CACA,WACO,eAAe,QAAQ,IAC7B,UAAS,eAAe,UAAU,QAAQ,KAAK,CAAC,CAClD;CACJ;CAEA,aACE,SACA,UACM;EACN,MAAM,UAAU,KAAK;EACrB,OAAO,SAAS,sCAAsC;EACtD,KAAK,oBAAoB,KAAA;EAEzB,IAAI,QAAQ,EAAE,CAAC,QAAQ;GACrB,QAAQ,OAAO;GACf,KAAK,KAAK,KAAK,KAAK,KAAK,SAAS,QAAQ;GAC1C;EACF;EAEA,QACG,QAAO,UAAS,WAAW,KAAK,KAAK,KAAK,KAAK,KAAK,CAAC,CAAC,CACtD,WAAW,UAAU,KAAK,KAAK,KAAK,KAAK,OAAO,CAAC,CAAC,CAClD,WACO,eAAe,QAAQ,IAC7B,UAAS,eAAe,UAAU,QAAQ,KAAK,CAAC,CAClD;CACJ;AACF;AAEA,IAAM,mBAAmB;AAEzB,SAAS,UACP,IACA,IACA,MACe;CACf,OAAO,IAAI,SAAS,SAAS,WAAW;EACtC,KAAK,IAAI,IAAI,OAAM,UAAU,QAAQ,OAAO,KAAK,IAAI,QAAQ,CAAE;CACjE,CAAC;AACH;AAEA,SAAS,WACP,IACA,IACA,OACe;CACf,OAAO,IAAI,SAAS,SAAS,WAAW;EACtC,IAAI,GAAG,eAAeE,YAAU,MAAM;GACpC,MAAM,QAAQ,qBAAqB;GACnC,GAAG,QACD,kDAAkD,GAAG,WAAW,EAClE;GACA,OAAO,KAAK;GACZ;EACF;EACA,IAAI,YAAY;EAChB,MAAM,QAAQ,iBACN,SAAS,0BAA0B,CAAC,GAC1C,yBACF;EACA,MAAM,MAAM;EAEZ,MAAM,YAAY,UAAyB;GACzC,IAAI,WACF;GAEF,YAAY;GACZ,aAAa,KAAK;GAClB,QAAQ,OAAO,KAAK,IAAI,QAAQ;EAClC;EAEA,GAAG,KAAK,OAAO,QAAQ;CACzB,CAAC;AACH;AAEA,SAAS,eACP,UACA,OACM;CACN,IAAI,aAAa,uBACf,SAAS,KAAK;AAElB;AAEA,SAAS,QAAQ,OAAuB;CACtC,OAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AACjE;AAEA,SAAS,uBAA+C;CACtD,OAAO,IAAI,uBACT;EACE,MAAMC;EACN,SAAS;EACT,QAAQH;CACV,GACA,MACF;AACF;AAEA,SAAS,4BAAoD;CAC3D,OAAO,IAAI,uBACT;EACE,MAAMG;EACN,SAAS,kCAAkC,0BAA0B;EACrE,QAAQH;CACV,GACA,MACF;AACF;AAEA,SAAgB,UACd,IACA,IACA,WACA,QACA;CACA,KAAK,GAAG,YAAY,aAAa,UAAU,IAAI;CAE/C,IAAI;CAGJ,IAAI,kBAAkB,wBACpB,WAAW,OAAO;MAIf,IACH,SAAS,MAAM,KACf,uBAAuB,MAAM,KAC7B,yBAAyB,UAAU,OAAO,GAE1C,WAAW;MAGR,IACH,UAAU,SAAS,oBACnB,UAAU,SAAS,mBAEnB,WAAW;MAEX,WAAW,SAAS,YAAY,MAAM,IAAI;CAG5C,GAAG,SAAS,GAAG,8BAA8B,WAAW,UAAU,EAAE;CACpE,KAAK,IAAI,IAAI,CAAC,SAAS,SAAS,GAAG,qBAAqB;AAC1D;AAEA,SAAS,kBAAkB,OAA2C;CACpE,IAAI,gBAAgB,KAAK,GACvB,OAAO;CAET,IAAI,iBAAiB,SAAS,MAAM,OAClC,OAAO,kBAAkB,MAAM,KAAK;AAGxC;AAEA,SAAS,SAAS,OAAyB;CACzC,OAAO,QACL,SACA,OAAO,UAAU,YACjB,WAAW,SACX,OAAQ,MAA2B,UAAU,WAC/C;AACF;AAIA,IAAM,+CAA+B,IAAI,IAAI;CAC3C;CACA;CACA;AACF,CAAC;AAID,IAAM,oCAAoC,CACxC,mDACF;AAEA,SAAS,uBAAuB,OAAyB;CACvD,IAAI,CAAC,SAAS,OAAO,UAAU,UAC7B,OAAO;CAET,MAAM,YACJ,UAAU,QAAQ,OAAQ,MAA2B,IAAI,IAAI,KAAA;CAC/D,OAAO,QACL,aAAa,6BAA6B,IAAI,UAAU,YAAY,CAAC,CACvE;AACF;AAEA,SAAS,yBAAyB,SAAsC;CACtE,IAAI,CAAC,SACH,OAAO;CAET,MAAM,QAAQ,QAAQ,YAAY;CAClC,OAAO,kCAAkC,MAAK,YAC5C,MAAM,SAAS,OAAO,CACxB;AACF"}

package/out/zero-cache/src/workers/mutator.js

@@ -1,4 +1,4 @@
-import { resolver } from "@rocicorp/resolver";
+import { resolver } from "../../../shared/src/resolver.js";
 import { pid } from "node:process";
 //#region ../zero-cache/src/workers/mutator.ts
 var Mutator = class {

package/out/zero-cache/src/workers/mutator.js.map

@@ -1 +1 @@
-{"version":3,"file":"mutator.js","names":[],"sources":["../../../../../zero-cache/src/workers/mutator.ts"],"sourcesContent":["import {pid} from 'node:process';\nimport {resolver} from '@rocicorp/resolver';\nimport type {SingletonService} from '../services/service.ts';\n\n// TODO:\n// - install websocket receiver\n// - spin up pusher services for each unique client group that connects\nexport class Mutator implements SingletonService {\n  readonly id = `mutator-${pid}`;\n  readonly #stopped;\n\n  constructor() {\n    this.#stopped = resolver();\n  }\n\n  run(): Promise<void> {\n    return this.#stopped.promise;\n  }\n\n  stop(): Promise<void> {\n    this.#stopped.resolve();\n    return this.#stopped.promise;\n  }\n\n  drain(): Promise<void> {\n    this.#stopped.resolve();\n    return this.#stopped.promise;\n  }\n}\n"],"mappings":";;;AAOA,IAAa,UAAb,MAAiD;CAC/C,KAAc,WAAW;CACzB;CAEA,cAAc;EACZ,KAAK,WAAW,SAAS;CAC3B;CAEA,MAAqB;EACnB,OAAO,KAAK,SAAS;CACvB;CAEA,OAAsB;EACpB,KAAK,SAAS,QAAQ;EACtB,OAAO,KAAK,SAAS;CACvB;CAEA,QAAuB;EACrB,KAAK,SAAS,QAAQ;EACtB,OAAO,KAAK,SAAS;CACvB;AACF"}
+{"version":3,"file":"mutator.js","names":[],"sources":["../../../../../zero-cache/src/workers/mutator.ts"],"sourcesContent":["import {pid} from 'node:process';\nimport {resolver} from '../../../shared/src/resolver.ts';\nimport type {SingletonService} from '../services/service.ts';\n\n// TODO:\n// - install websocket receiver\n// - spin up pusher services for each unique client group that connects\nexport class Mutator implements SingletonService {\n  readonly id = `mutator-${pid}`;\n  readonly #stopped;\n\n  constructor() {\n    this.#stopped = resolver();\n  }\n\n  run(): Promise<void> {\n    return this.#stopped.promise;\n  }\n\n  stop(): Promise<void> {\n    this.#stopped.resolve();\n    return this.#stopped.promise;\n  }\n\n  drain(): Promise<void> {\n    this.#stopped.resolve();\n    return this.#stopped.promise;\n  }\n}\n"],"mappings":";;;AAOA,IAAa,UAAb,MAAiD;CAC/C,KAAc,WAAW;CACzB;CAEA,cAAc;EACZ,KAAK,WAAW,SAAS;CAC3B;CAEA,MAAqB;EACnB,OAAO,KAAK,SAAS;CACvB;CAEA,OAAsB;EACpB,KAAK,SAAS,QAAQ;EACtB,OAAO,KAAK,SAAS;CACvB;CAEA,QAAuB;EACrB,KAAK,SAAS,QAAQ;EACtB,OAAO,KAAK,SAAS;CACvB;AACF"}

package/out/zero-cache/src/workers/syncer.js

@@ -1,5 +1,6 @@
 import { promiseVoid } from "../../../shared/src/resolved-promises.js";
-import { Unauthorized } from "../../../zero-protocol/src/error-kind-enum.js";
+import { resolver } from "../../../shared/src/resolver.js";
+import { Rehome, Unauthorized } from "../../../zero-protocol/src/error-kind-enum.js";
 import { ZeroCache } from "../../../zero-protocol/src/error-origin-enum.js";
 import { ProtocolError, isProtocolError } from "../../../zero-protocol/src/error.js";
 import "../config/zero-config.js";
@@ -8,12 +9,12 @@ import { installWebSocketReceiver } from "../types/websocket-handoff.js";
 import { createNotifierFrom, subscribeTo } from "./replicator.js";
 import { recordConnectionAttempted, recordConnectionSuccess, setActiveClientGroupsGetter } from "../server/anonymous-otel-start.js";
 import { tokenConfigOptions } from "../auth/jwt.js";
+import { ProtocolErrorWithLevel } from "../types/error-with-level.js";
 import { resolveAuth } from "../auth/auth.js";
 import { ServiceRunner } from "../services/runner.js";
 import { DrainCoordinator, TARGET_UTILIZATION } from "../services/view-syncer/drain-coordinator.js";
 import { Connection, sendError } from "./connection.js";
 import { SyncerWsMessageHandler } from "./syncer-ws-message-handler.js";
-import { resolver } from "@rocicorp/resolver";
 import { WebSocket, WebSocketServer } from "ws";
 import { pid } from "node:process";
 //#region ../zero-cache/src/workers/syncer.ts
@@ -158,6 +159,7 @@ var Syncer = class {
 	#drainCoordinator = new DrainCoordinator();
 	#parent;
 	#wss;
+	#draining = false;
 	#stopped = resolver();
 	#config;
 	#validateLegacyJWT;
@@ -195,6 +197,14 @@ var Syncer = class {
 		if (mutagenFactory) this.#mutagens = new ServiceRunner(lc, mutagenFactory, (m) => m.hasRefs());
 		if (pusherFactory) this.#pushers = new ServiceRunner(lc, (id) => pusherFactory(id, this.#viewSyncers.getService(id).connContextManager), (p) => p.hasRefs());
 		this.#wss = new WebSocketServer(getWebSocketServerOptions(config));
+		const cookie = config.affinityCookie;
+		if (cookie?.name) {
+			const { name, path, domain } = cookie;
+			const expiredCookie = `${name}=; Max-Age=0; Path=${path}${domain ? `; Domain=${domain}` : ""}; Secure; HttpOnly; SameSite=None`;
+			this.#wss.on("headers", (headers) => {
+				if (this.#draining) headers.push(`Set-Cookie: ${expiredCookie}`);
+			});
+		}
 		installWebSocketReceiver(lc, this.#wss, this.#receiveConnection, this.#parent, (params) => {
 			this.#releasePending(params.clientGroupID, params.generation);
 		}, (params) => {
@@ -378,6 +388,11 @@ var Syncer = class {
 			recordConnectionFailure("closed_during_auth");
 			return;
 		}
+		if (this.#draining) {
+			recordConnectionFailure("draining");
+			this.#rehome(ws);
+			return;
+		}
 		const viewSyncer = this.#viewSyncers.getService(clientGroupID);
 		const connContextManager = viewSyncer.connContextManager;
 		const group = connContextManager.getGroupState();
@@ -394,6 +409,7 @@ var Syncer = class {
 		}
 		const existing = this.#connections.get(clientID);
 		if (existing) {
+			if (existing.wsID === params.wsID) this.#lc.warn?.(`duplicate connection for client ${clientID} with the same wsID ${params.wsID}, closing existing connection`, { clientGroupID });
 			this.#lc.debug?.(`client ${clientID} already connected, closing existing connection`);
 			existing.close(`replaced by ${params.wsID}`);
 		}
@@ -438,11 +454,28 @@ var Syncer = class {
 	};
 	#receiveConnection = async (ws, params) => {
 		try {
+			if (this.#draining) {
+				this.#rehome(ws);
+				return;
+			}
 			await this.#createConnection(ws, params);
 		} finally {
 			this.#releasePending(params.clientGroupID, params.generation);
 		}
 	};
+	#rehome(ws) {
+		ws.on("error", (error) => {
+			this.#lc.warn?.("error while rehoming draining connection", error);
+			ws.terminate();
+		});
+		const error = new ProtocolErrorWithLevel({
+			kind: Rehome,
+			message: "Server is draining",
+			origin: ZeroCache
+		}, "info");
+		sendError(this.#lc, ws, error.errorBody, error);
+		ws.close(1001, "Server is draining");
+	}
 	#releasePending(clientGroupID, generation) {
 		const remaining = (this.#pendingConnections.get(clientGroupID) ?? 1) - 1;
 		if (remaining <= 0) {
@@ -468,6 +501,7 @@ var Syncer = class {
 	* when the clients reconnect.
 	*/
 	async drain() {
+		this.#draining = true;
 		const start = Date.now();
 		this.#lc.info?.(`draining ${this.#viewSyncers.size} view-syncers`);
 		this.#drainCoordinator.drainNextIn(0);