Function webSocketFrameToAcpStream

packages/ai-acp/src/transport/websocket.ts:67–108

(ws: WebSocket)

 * One JSON-RPC object per WebSocket text frame (e.g. `grok agent serve`).
 */
export function webSocketFrameToAcpStream(ws: WebSocket): AcpJsonRpcStream {
  const decoder = new TextDecoder()

  const readable = new ReadableStream<unknown>({
    start(rawController) {
      const controller = idempotentController(rawController)
      ws.addEventListener('message', (event) => {
        const text =
          typeof event.data === 'string'
            ? event.data
            : decoder.decode(event.data as ArrayBuffer)
        const trimmed = text.trim()
        if (trimmed === '') return
        try {
          controller.enqueue(JSON.parse(trimmed))
        } catch (error) {
          controller.error(
            error instanceof Error ? error : new Error(String(error)),
          )
        }
      })
      ws.addEventListener('close', () => controller.close())
      ws.addEventListener('error', () =>
        controller.error(new Error('WebSocket connection error')),
      )
    },
    cancel() {
      ws.close()
    },
  })

  const writable = new WritableStream({
    write(message) {
      ws.send(JSON.stringify(message))
    },
    close() {
      ws.close()
    },
  })

  return { readable, writable } as AcpJsonRpcStream
}

function webSocketNdjsonToAcpStream(ws: WebSocket): AcpJsonRpcStream {
  const encoder = new TextEncoder()