Function createSessionReadyHandler

services/cloud-agent-next/wrapper/src/server.ts:969–1004

(deps: ServerDependencies)

  return { code: 1011, reason: event.reason || 'PTY upstream closed' };
}

type BunUpgradeServer = {
  upgrade: (req: Request, options: { data: WrapperWebSocketData }) => boolean;
};

function createWebSocketHandlers(config: ServerConfig, deps: ServerDependencies) {
  const ptyUpstreams = new WeakMap<object, WebSocket>();

  return {
    open(ws: Bun.ServerWebSocket<WrapperWebSocketData>) {
      const ptyId = ws.data.ptyId;
      if (!ptyId) {
        ws.close(1011, 'Missing PTY');
        return;
      }

      const upstream = new WebSocket(buildPtyUpstreamUrl(config, deps, ptyId));
      upstream.binaryType = 'arraybuffer';
      ptyUpstreams.set(ws, upstream);

      upstream.onmessage = event => {
        try {
          ws.send(event.data instanceof ArrayBuffer ? event.data : String(event.data));
        } catch {
          // Client disconnected.
        }
      };
      upstream.onclose = event => {
        const close = resolvePtyClientClose({ code: event.code, reason: event.reason });
        try {
          ws.close(close.code, close.reason);
        } catch {
          // Already closed.
        }
      };
      upstream.onerror = () => {
        try {
          ws.close(1011, 'PTY upstream error');