@copilotkit/aimock 1.42.1 → 1.44.0

package/dist/ws-responses.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ws-responses.cjs","names":["requestWantsEncryptedReasoning","requestIncludesEncryptedReasoning","responsesToCompletionRequest","DEFAULT_TEST_ID","matchFixtureDiagnostic","resolveStrictMode","strictNoMatchMessage","strictNoMatchLogLine","flattenHeaders","strictOverrideField","resolveResponse","isErrorResponse","isContentWithToolCallsResponse","buildContentWithToolCallsStreamEvents","resolveReasoningForModel","extractOverrides","createInterruptionSignal","isTextResponse","buildTextStreamEvents","isToolCallResponse","buildToolCallStreamEvents","calculateDelay","delay"],"sources":["../src/ws-responses.ts"],"sourcesContent":["/**\n * WebSocket handler for OpenAI Responses API.\n *\n * Accepts `{ type: \"response.create\", model: \"...\", input: [...] }` messages over\n * WebSocket and sends back the same Responses API SSE events as the HTTP\n * handler, but as individual WebSocket text frames.\n */\n\nimport type { ChatCompletionRequest, Fixture } from \"./types.js\";\nimport { matchFixtureDiagnostic } from \"./router.js\";\nimport {\n  responsesToCompletionRequest,\n  requestIncludesEncryptedReasoning,\n  requestWantsEncryptedReasoning,\n  buildTextStreamEvents,\n  buildToolCallStreamEvents,\n  buildContentWithToolCallsStreamEvents,\n  type ResponsesSSEEvent,\n} from \"./responses.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  extractOverrides,\n  resolveResponse,\n  resolveStrictMode,\n  resolveReasoningForModel,\n  strictOverrideField,\n  flattenHeaders,\n  strictNoMatchMessage,\n  strictNoMatchLogLine,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\ninterface ResponseCreateMessage {\n  type: \"response.create\";\n  model?: string;\n  input?: unknown[];\n  instructions?: string;\n  tools?: unknown[];\n  tool_choice?: string | object;\n  stream?: boolean;\n  temperature?: number;\n  max_output_tokens?: number;\n  [key: string]: unknown;\n}\n\nfunction isResponseCreateMessage(msg: unknown): msg is ResponseCreateMessage {\n  return (\n    typeof msg === \"object\" &&\n    msg !== null &&\n    (msg as ResponseCreateMessage).type === \"response.create\"\n  );\n}\n\nfunction buildErrorEvent(\n  message: string,\n  type = \"invalid_request_error\",\n  code?: string,\n): ResponsesSSEEvent {\n  return {\n    type: \"error\",\n    error: { message, type, code },\n  };\n}\n\nexport function handleWebSocketResponses(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  // Serialize message processing to prevent event interleaving\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket responses error: ${msg}`);\n        try {\n          ws.send(JSON.stringify(buildErrorEvent(msg, \"server_error\")));\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): Promise<void> {\n  let parsed: unknown;\n  try {\n    parsed = JSON.parse(raw);\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(`Malformed JSON: ${detail}`, \"invalid_request_error\", \"invalid_json\"),\n      ),\n    );\n    return;\n  }\n\n  if (!isResponseCreateMessage(parsed)) {\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\n          'Expected message type \"response.create\"',\n          \"invalid_request_error\",\n          \"invalid_message_type\",\n        ),\n      ),\n    );\n    return;\n  }\n\n  const responsesReq = {\n    model: parsed.model ?? defaults.model,\n    input: (parsed.input ?? []) as {\n      role?: string;\n      type?: string;\n      content?: string | { type: string; text?: string }[];\n      call_id?: string;\n      name?: string;\n      arguments?: string;\n      output?: string;\n      id?: string;\n    }[],\n    instructions: parsed.instructions,\n    tools: parsed.tools as\n      | {\n          type: \"function\";\n          name: string;\n          description?: string;\n          parameters?: object;\n          strict?: boolean;\n        }[]\n      | undefined,\n    tool_choice: parsed.tool_choice,\n    stream: parsed.stream,\n    temperature: parsed.temperature,\n    max_output_tokens: parsed.max_output_tokens,\n    include: (parsed as { include?: string[] }).include,\n    store: (parsed as { store?: boolean }).store,\n  };\n\n  // Gate encrypted-reasoning emission identically to the HTTP transport so the\n  // agent-framework#7233 stateless-replay path works over WebSocket too.\n  const emitEncryptedReasoning = requestWantsEncryptedReasoning(responsesReq);\n  // Synthesizing a reasoning item the fixture never declared takes the NARROWER\n  // explicit-`include` gate, transport-identically to HTTP — a `store: false`\n  // request with a summary-less fixture still gets no reasoning item.\n  const synthesizeSummarylessReasoning = requestIncludesEncryptedReasoning(responsesReq);\n\n  const completionReq = responsesToCompletionRequest(responsesReq);\n  completionReq._endpointType = \"chat\";\n  const contextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  completionReq._context =\n    typeof contextHeader === \"string\"\n      ? contextHeader\n      : Array.isArray(contextHeader) && contextHeader.length > 0\n        ? contextHeader[0]\n        : undefined;\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);\n      defaults.logger.error(strictNoMatchLogLine(\"WS\", \"/v1/responses\", skippedBySequenceOrTurn));\n      journal.add({\n        method: \"WS\",\n        path: \"/v1/responses\",\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, strictMessage);\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\"No fixture matched\", \"invalid_request_error\", \"no_fixture_match\"),\n      ),\n    );\n    return;\n  }\n\n  const response = await resolveResponse(fixture, completionReq);\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // The WS path has no per-request `req.headers`; strict is resolved from the\n  // connection's upgrade headers (see the `!fixture` branch above). Used below\n  // to gate the synthesized reasoning channel on the requested model's capability.\n  const effectiveStrict = resolveStrictMode(defaults.strict, defaults.upgradeHeaders);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(response.error.message, response.error.type, response.error.code),\n      ),\n    );\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isContentWithToolCallsResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const events = buildContentWithToolCallsStreamEvents(\n      response.content ?? \"\",\n      response.toolCalls ?? [],\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      response.blocks,\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const events = buildTextStreamEvents(\n      response.content,\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    const events = buildToolCallStreamEvents(\n      response.toolCalls,\n      completionReq.model,\n      chunkSize,\n      // Gate the synthesized reasoning channel on the requested model's\n      // capability, matching the WS text / content+tool branches and the HTTP\n      // tool-only path so reasoning emission is transport-independent.\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path: \"/v1/responses\",\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify(\n      buildErrorEvent(\"Fixture response did not match any known type\", \"server_error\"),\n    ),\n  );\n}\n\nasync function sendEvents(\n  ws: WebSocketConnection,\n  events: ResponsesSSEEvent[],\n  latency: number,\n  signal?: AbortSignal,\n  onChunkSent?: () => void,\n  recordedTimings?: import(\"./types.js\").RecordedTimings,\n  replaySpeed?: number,\n): Promise<boolean> {\n  let eventIndex = 0;\n  for (const event of events) {\n    if (ws.isClosed) return false;\n    const chunkDelay = calculateDelay(eventIndex, undefined, latency, recordedTimings, replaySpeed);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (ws.isClosed) return false;\n    ws.send(JSON.stringify(event));\n    eventIndex++;\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;AAoDA,SAAS,wBAAwB,KAA4C;AAC3E,QACE,OAAO,QAAQ,YACf,QAAQ,QACP,IAA8B,SAAS;;AAI5C,SAAS,gBACP,SACA,OAAO,yBACP,MACmB;AACnB,QAAO;EACL,MAAM;EACN,OAAO;GAAE;GAAS;GAAM;GAAM;EAC/B;;AAGH,SAAgB,yBACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CAEnB,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,SAAS,CAAC,OAAO,QAAiB;GAC3E,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,8BAA8B,MAAM;AACjD,OAAI;AACF,OAAG,KAAK,KAAK,UAAU,gBAAgB,KAAK,eAAe,CAAC,CAAC;YACtD,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UACH,gBAAgB,mBAAmB,UAAU,yBAAyB,eAAe,CACtF,CACF;AACD;;AAGF,KAAI,CAAC,wBAAwB,OAAO,EAAE;AACpC,KAAG,KACD,KAAK,UACH,gBACE,6CACA,yBACA,uBACD,CACF,CACF;AACD;;CAGF,MAAM,eAAe;EACnB,OAAO,OAAO,SAAS,SAAS;EAChC,OAAQ,OAAO,SAAS,EAAE;EAU1B,cAAc,OAAO;EACrB,OAAO,OAAO;EASd,aAAa,OAAO;EACpB,QAAQ,OAAO;EACf,aAAa,OAAO;EACpB,mBAAmB,OAAO;EAC1B,SAAU,OAAkC;EAC5C,OAAQ,OAA+B;EACxC;CAID,MAAM,yBAAyBA,iDAA+B,aAAa;CAI3E,MAAM,iCAAiCC,oDAAkC,aAAa;CAEtF,MAAM,gBAAgBC,+CAA6B,aAAa;AAChE,eAAc,gBAAgB;CAC9B,MAAM,gBAAgB,SAAS,iBAAiB;AAChD,eAAc,WACZ,OAAO,kBAAkB,WACrB,gBACA,MAAM,QAAQ,cAAc,IAAI,cAAc,SAAS,IACrD,cAAc,KACd;CACR,MAAM,SAAS,SAAS,UAAUC;CAClC,MAAM,EAAE,SAAS,4BAA4BC,sCAC3C,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAIC,kCAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;GAC/D,MAAM,gBAAgBC,qCAAqB,wBAAwB;AACnE,YAAS,OAAO,MAAMC,qCAAqB,MAAM,iBAAiB,wBAAwB,CAAC;AAC3F,WAAQ,IAAI;IACV,QAAQ;IACR,MAAM;IACN,SAASC,+BAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,cAAc;AAC7B;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAASD,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,sBAAsB,yBAAyB,mBAAmB,CACnF,CACF;AACD;;CAGF,MAAM,WAAW,MAAMC,gCAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;CAKtE,MAAM,kBAAkBL,kCAAkB,SAAS,QAAQ,SAAS,eAAe;AAGnF,KAAIM,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAASH,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,SAAS,MAAM,SAAS,SAAS,MAAM,MAAM,SAAS,MAAM,KAAK,CAClF,CACF;AACD;;AAIF,KAAII,+CAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASJ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,SAASK,wDACb,SAAS,WAAW,IACpB,SAAS,aAAa,EAAE,EACxB,cAAc,OACd,WACAC,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,SAAS,QACT,wBACA,+BACD;EAED,MAAM,eAAeC,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAIC,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAST,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,SAASU,wCACb,SAAS,SACT,cAAc,OACd,WACAJ,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAeC,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAIG,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASX,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EACF,MAAM,SAASY,4CACb,SAAS,WACT,cAAc,OACd,WAIAN,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAeC,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR,MAAM;EACN,SAASR,+BAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UACH,gBAAgB,iDAAiD,eAAe,CACjF,CACF;;AAGH,eAAe,WACb,IACA,QACA,SACA,QACA,aACA,iBACA,aACkB;CAClB,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,GAAG,SAAU,QAAO;EACxB,MAAM,aAAaa,kCAAe,YAAY,QAAW,SAAS,iBAAiB,YAAY;AAC/F,MAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,GAAG,SAAU,QAAO;AACxB,KAAG,KAAK,KAAK,UAAU,MAAM,CAAC;AAC9B;AACA,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;;AAE9B,QAAO"}
+{"version":3,"file":"ws-responses.cjs","names":["requestWantsEncryptedReasoning","requestIncludesEncryptedReasoning","responsesToCompletionRequest","DEFAULT_TEST_ID","matchFixtureDiagnostic","resolveStrictMode","strictNoMatchMessage","strictNoMatchLogLine","flattenHeaders","strictOverrideField","resolveResponse","isErrorResponse","isContentWithToolCallsResponse","buildContentWithToolCallsStreamEvents","resolveReasoningForModel","extractOverrides","createInterruptionSignal","isTextResponse","buildTextStreamEvents","isToolCallResponse","buildToolCallStreamEvents","calculateDelay","delay"],"sources":["../src/ws-responses.ts"],"sourcesContent":["/**\n * WebSocket handler for OpenAI Responses API.\n *\n * Accepts `{ type: \"response.create\", model: \"...\", input: [...] }` messages over\n * WebSocket and sends back the same Responses API SSE events as the HTTP\n * handler, but as individual WebSocket text frames.\n */\n\nimport type { ChatCompletionRequest, Fixture } from \"./types.js\";\nimport { matchFixtureDiagnostic } from \"./router.js\";\nimport {\n  responsesToCompletionRequest,\n  requestIncludesEncryptedReasoning,\n  requestWantsEncryptedReasoning,\n  buildTextStreamEvents,\n  buildToolCallStreamEvents,\n  buildContentWithToolCallsStreamEvents,\n  type ResponsesSSEEvent,\n} from \"./responses.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  extractOverrides,\n  resolveResponse,\n  resolveStrictMode,\n  resolveReasoningForModel,\n  strictOverrideField,\n  flattenHeaders,\n  strictNoMatchMessage,\n  strictNoMatchLogLine,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\ninterface ResponseCreateMessage {\n  type: \"response.create\";\n  model?: string;\n  input?: unknown[];\n  instructions?: string;\n  tools?: unknown[];\n  tool_choice?: string | object;\n  stream?: boolean;\n  temperature?: number;\n  max_output_tokens?: number;\n  [key: string]: unknown;\n}\n\nfunction isResponseCreateMessage(msg: unknown): msg is ResponseCreateMessage {\n  return (\n    typeof msg === \"object\" &&\n    msg !== null &&\n    (msg as ResponseCreateMessage).type === \"response.create\"\n  );\n}\n\nfunction buildErrorEvent(\n  message: string,\n  type = \"invalid_request_error\",\n  code?: string,\n): ResponsesSSEEvent {\n  return {\n    type: \"error\",\n    error: { message, type, code },\n  };\n}\n\nexport function handleWebSocketResponses(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  // Serialize message processing to prevent event interleaving\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket responses error: ${msg}`);\n        try {\n          ws.send(JSON.stringify(buildErrorEvent(msg, \"server_error\")));\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): Promise<void> {\n  let parsed: unknown;\n  try {\n    parsed = JSON.parse(raw);\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(`Malformed JSON: ${detail}`, \"invalid_request_error\", \"invalid_json\"),\n      ),\n    );\n    return;\n  }\n\n  if (!isResponseCreateMessage(parsed)) {\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\n          'Expected message type \"response.create\"',\n          \"invalid_request_error\",\n          \"invalid_message_type\",\n        ),\n      ),\n    );\n    return;\n  }\n\n  const responsesReq = {\n    model: parsed.model ?? defaults.model,\n    input: (parsed.input ?? []) as {\n      role?: string;\n      type?: string;\n      content?: string | { type: string; text?: string }[];\n      call_id?: string;\n      name?: string;\n      arguments?: string;\n      output?: string;\n      id?: string;\n    }[],\n    instructions: parsed.instructions,\n    tools: parsed.tools as\n      | {\n          type: \"function\";\n          name: string;\n          description?: string;\n          parameters?: object;\n          strict?: boolean;\n        }[]\n      | undefined,\n    tool_choice: parsed.tool_choice,\n    stream: parsed.stream,\n    temperature: parsed.temperature,\n    max_output_tokens: parsed.max_output_tokens,\n    include: (parsed as { include?: string[] }).include,\n    store: (parsed as { store?: boolean }).store,\n  };\n\n  // Gate encrypted-reasoning emission identically to the HTTP transport so the\n  // agent-framework#7233 stateless-replay path works over WebSocket too.\n  const emitEncryptedReasoning = requestWantsEncryptedReasoning(responsesReq);\n  // Synthesizing a reasoning item the fixture never declared takes the NARROWER\n  // explicit-`include` gate, transport-identically to HTTP — a `store: false`\n  // request with a summary-less fixture still gets no reasoning item.\n  const synthesizeSummarylessReasoning = requestIncludesEncryptedReasoning(responsesReq);\n\n  const completionReq = responsesToCompletionRequest(responsesReq);\n  completionReq._endpointType = \"chat\";\n  const contextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  completionReq._context =\n    typeof contextHeader === \"string\"\n      ? contextHeader\n      : Array.isArray(contextHeader) && contextHeader.length > 0\n        ? contextHeader[0]\n        : undefined;\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);\n      defaults.logger.error(strictNoMatchLogLine(\"WS\", \"/v1/responses\", skippedBySequenceOrTurn));\n      journal.add({\n        method: \"WS\",\n        path: \"/v1/responses\",\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, strictMessage);\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\"No fixture matched\", \"invalid_request_error\", \"no_fixture_match\"),\n      ),\n    );\n    return;\n  }\n\n  const response = await resolveResponse(fixture, completionReq);\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // The WS path has no per-request `req.headers`; strict is resolved from the\n  // connection's upgrade headers (see the `!fixture` branch above). Used below\n  // to gate the synthesized reasoning channel on the requested model's capability.\n  const effectiveStrict = resolveStrictMode(defaults.strict, defaults.upgradeHeaders);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(response.error.message, response.error.type, response.error.code),\n      ),\n    );\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isContentWithToolCallsResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    let events: ResponsesSSEEvent[];\n    try {\n      events = buildContentWithToolCallsStreamEvents(\n        response.content ?? \"\",\n        response.toolCalls ?? [],\n        completionReq.model,\n        chunkSize,\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        response.blocks,\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      throw error;\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const events = buildTextStreamEvents(\n      response.content,\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    let events: ResponsesSSEEvent[];\n    try {\n      events = buildToolCallStreamEvents(\n        response.toolCalls,\n        completionReq.model,\n        chunkSize,\n        // Gate the synthesized reasoning channel on the requested model's\n        // capability, matching the WS text / content+tool branches and the HTTP\n        // tool-only path so reasoning emission is transport-independent.\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      throw error;\n    }\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path: \"/v1/responses\",\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify(\n      buildErrorEvent(\"Fixture response did not match any known type\", \"server_error\"),\n    ),\n  );\n}\n\nasync function sendEvents(\n  ws: WebSocketConnection,\n  events: ResponsesSSEEvent[],\n  latency: number,\n  signal?: AbortSignal,\n  onChunkSent?: () => void,\n  recordedTimings?: import(\"./types.js\").RecordedTimings,\n  replaySpeed?: number,\n): Promise<boolean> {\n  let eventIndex = 0;\n  for (const event of events) {\n    if (ws.isClosed) return false;\n    const chunkDelay = calculateDelay(eventIndex, undefined, latency, recordedTimings, replaySpeed);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (ws.isClosed) return false;\n    ws.send(JSON.stringify(event));\n    eventIndex++;\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;AAoDA,SAAS,wBAAwB,KAA4C;AAC3E,QACE,OAAO,QAAQ,YACf,QAAQ,QACP,IAA8B,SAAS;;AAI5C,SAAS,gBACP,SACA,OAAO,yBACP,MACmB;AACnB,QAAO;EACL,MAAM;EACN,OAAO;GAAE;GAAS;GAAM;GAAM;EAC/B;;AAGH,SAAgB,yBACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CAEnB,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,SAAS,CAAC,OAAO,QAAiB;GAC3E,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,8BAA8B,MAAM;AACjD,OAAI;AACF,OAAG,KAAK,KAAK,UAAU,gBAAgB,KAAK,eAAe,CAAC,CAAC;YACtD,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UACH,gBAAgB,mBAAmB,UAAU,yBAAyB,eAAe,CACtF,CACF;AACD;;AAGF,KAAI,CAAC,wBAAwB,OAAO,EAAE;AACpC,KAAG,KACD,KAAK,UACH,gBACE,6CACA,yBACA,uBACD,CACF,CACF;AACD;;CAGF,MAAM,eAAe;EACnB,OAAO,OAAO,SAAS,SAAS;EAChC,OAAQ,OAAO,SAAS,EAAE;EAU1B,cAAc,OAAO;EACrB,OAAO,OAAO;EASd,aAAa,OAAO;EACpB,QAAQ,OAAO;EACf,aAAa,OAAO;EACpB,mBAAmB,OAAO;EAC1B,SAAU,OAAkC;EAC5C,OAAQ,OAA+B;EACxC;CAID,MAAM,yBAAyBA,iDAA+B,aAAa;CAI3E,MAAM,iCAAiCC,oDAAkC,aAAa;CAEtF,MAAM,gBAAgBC,+CAA6B,aAAa;AAChE,eAAc,gBAAgB;CAC9B,MAAM,gBAAgB,SAAS,iBAAiB;AAChD,eAAc,WACZ,OAAO,kBAAkB,WACrB,gBACA,MAAM,QAAQ,cAAc,IAAI,cAAc,SAAS,IACrD,cAAc,KACd;CACR,MAAM,SAAS,SAAS,UAAUC;CAClC,MAAM,EAAE,SAAS,4BAA4BC,sCAC3C,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAIC,kCAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;GAC/D,MAAM,gBAAgBC,qCAAqB,wBAAwB;AACnE,YAAS,OAAO,MAAMC,qCAAqB,MAAM,iBAAiB,wBAAwB,CAAC;AAC3F,WAAQ,IAAI;IACV,QAAQ;IACR,MAAM;IACN,SAASC,+BAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,cAAc;AAC7B;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAASD,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,sBAAsB,yBAAyB,mBAAmB,CACnF,CACF;AACD;;CAGF,MAAM,WAAW,MAAMC,gCAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;CAKtE,MAAM,kBAAkBL,kCAAkB,SAAS,QAAQ,SAAS,eAAe;AAGnF,KAAIM,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAASH,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,SAAS,MAAM,SAAS,SAAS,MAAM,MAAM,SAAS,MAAM,KAAK,CAClF,CACF;AACD;;AAIF,KAAII,+CAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASJ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,IAAI;AACJ,MAAI;AACF,YAASK,wDACP,SAAS,WAAW,IACpB,SAAS,aAAa,EAAE,EACxB,cAAc,OACd,WACAC,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,SAAS,QACT,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,SAAM;;EAGR,MAAM,eAAeC,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAIC,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAST,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,SAASU,wCACb,SAAS,SACT,cAAc,OACd,WACAJ,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAeC,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAIG,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASX,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EACF,IAAI;AACJ,MAAI;AACF,YAASY,4CACP,SAAS,WACT,cAAc,OACd,WAIAN,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,SAAM;;EAER,MAAM,eAAeC,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR,MAAM;EACN,SAASR,+BAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UACH,gBAAgB,iDAAiD,eAAe,CACjF,CACF;;AAGH,eAAe,WACb,IACA,QACA,SACA,QACA,aACA,iBACA,aACkB;CAClB,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,GAAG,SAAU,QAAO;EACxB,MAAM,aAAaa,kCAAe,YAAY,QAAW,SAAS,iBAAiB,YAAY;AAC/F,MAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,GAAG,SAAU,QAAO;AACxB,KAAG,KAAK,KAAK,UAAU,MAAM,CAAC;AAC9B;AACA,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;;AAE9B,QAAO"}

package/dist/ws-responses.js

@@ -1,8 +1,8 @@
 import { DEFAULT_TEST_ID } from "./constants.js";
 import { extractOverrides, flattenHeaders, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, resolveReasoningForModel, resolveResponse, resolveStrictMode, strictNoMatchLogLine, strictNoMatchMessage, strictOverrideField } from "./helpers.js";
-import "./journal.js";
-import { matchFixtureDiagnostic } from "./router.js";
 import { calculateDelay, delay } from "./sse-writer.js";
+import { matchFixtureDiagnostic } from "./router.js";
+import "./journal.js";
 import { createInterruptionSignal } from "./interruption.js";
 import { buildContentWithToolCallsStreamEvents, buildTextStreamEvents, buildToolCallStreamEvents, requestIncludesEncryptedReasoning, requestWantsEncryptedReasoning, responsesToCompletionRequest } from "./responses.js";
 
@@ -131,7 +131,13 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 				fixture
 			}
 		});
-		const events = buildContentWithToolCallsStreamEvents(response.content ?? "", response.toolCalls ?? [], completionReq.model, chunkSize, resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, extractOverrides(response), response.blocks, emitEncryptedReasoning, synthesizeSummarylessReasoning);
+		let events;
+		try {
+			events = buildContentWithToolCallsStreamEvents(response.content ?? "", response.toolCalls ?? [], completionReq.model, chunkSize, resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, extractOverrides(response), response.blocks, emitEncryptedReasoning, synthesizeSummarylessReasoning);
+		} catch (error) {
+			journalEntry.response.status = 500;
+			throw error;
+		}
 		const interruption = createInterruptionSignal(fixture);
 		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
 			ws.destroy();
@@ -173,7 +179,13 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 				fixture
 			}
 		});
-		const events = buildToolCallStreamEvents(response.toolCalls, completionReq.model, chunkSize, resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, extractOverrides(response), emitEncryptedReasoning, synthesizeSummarylessReasoning);
+		let events;
+		try {
+			events = buildToolCallStreamEvents(response.toolCalls, completionReq.model, chunkSize, resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, extractOverrides(response), emitEncryptedReasoning, synthesizeSummarylessReasoning);
+		} catch (error) {
+			journalEntry.response.status = 500;
+			throw error;
+		}
 		const interruption = createInterruptionSignal(fixture);
 		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
 			ws.destroy();

package/dist/ws-responses.js.map

@@ -1 +1 @@
-{"version":3,"file":"ws-responses.js","names":[],"sources":["../src/ws-responses.ts"],"sourcesContent":["/**\n * WebSocket handler for OpenAI Responses API.\n *\n * Accepts `{ type: \"response.create\", model: \"...\", input: [...] }` messages over\n * WebSocket and sends back the same Responses API SSE events as the HTTP\n * handler, but as individual WebSocket text frames.\n */\n\nimport type { ChatCompletionRequest, Fixture } from \"./types.js\";\nimport { matchFixtureDiagnostic } from \"./router.js\";\nimport {\n  responsesToCompletionRequest,\n  requestIncludesEncryptedReasoning,\n  requestWantsEncryptedReasoning,\n  buildTextStreamEvents,\n  buildToolCallStreamEvents,\n  buildContentWithToolCallsStreamEvents,\n  type ResponsesSSEEvent,\n} from \"./responses.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  extractOverrides,\n  resolveResponse,\n  resolveStrictMode,\n  resolveReasoningForModel,\n  strictOverrideField,\n  flattenHeaders,\n  strictNoMatchMessage,\n  strictNoMatchLogLine,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\ninterface ResponseCreateMessage {\n  type: \"response.create\";\n  model?: string;\n  input?: unknown[];\n  instructions?: string;\n  tools?: unknown[];\n  tool_choice?: string | object;\n  stream?: boolean;\n  temperature?: number;\n  max_output_tokens?: number;\n  [key: string]: unknown;\n}\n\nfunction isResponseCreateMessage(msg: unknown): msg is ResponseCreateMessage {\n  return (\n    typeof msg === \"object\" &&\n    msg !== null &&\n    (msg as ResponseCreateMessage).type === \"response.create\"\n  );\n}\n\nfunction buildErrorEvent(\n  message: string,\n  type = \"invalid_request_error\",\n  code?: string,\n): ResponsesSSEEvent {\n  return {\n    type: \"error\",\n    error: { message, type, code },\n  };\n}\n\nexport function handleWebSocketResponses(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  // Serialize message processing to prevent event interleaving\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket responses error: ${msg}`);\n        try {\n          ws.send(JSON.stringify(buildErrorEvent(msg, \"server_error\")));\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): Promise<void> {\n  let parsed: unknown;\n  try {\n    parsed = JSON.parse(raw);\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(`Malformed JSON: ${detail}`, \"invalid_request_error\", \"invalid_json\"),\n      ),\n    );\n    return;\n  }\n\n  if (!isResponseCreateMessage(parsed)) {\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\n          'Expected message type \"response.create\"',\n          \"invalid_request_error\",\n          \"invalid_message_type\",\n        ),\n      ),\n    );\n    return;\n  }\n\n  const responsesReq = {\n    model: parsed.model ?? defaults.model,\n    input: (parsed.input ?? []) as {\n      role?: string;\n      type?: string;\n      content?: string | { type: string; text?: string }[];\n      call_id?: string;\n      name?: string;\n      arguments?: string;\n      output?: string;\n      id?: string;\n    }[],\n    instructions: parsed.instructions,\n    tools: parsed.tools as\n      | {\n          type: \"function\";\n          name: string;\n          description?: string;\n          parameters?: object;\n          strict?: boolean;\n        }[]\n      | undefined,\n    tool_choice: parsed.tool_choice,\n    stream: parsed.stream,\n    temperature: parsed.temperature,\n    max_output_tokens: parsed.max_output_tokens,\n    include: (parsed as { include?: string[] }).include,\n    store: (parsed as { store?: boolean }).store,\n  };\n\n  // Gate encrypted-reasoning emission identically to the HTTP transport so the\n  // agent-framework#7233 stateless-replay path works over WebSocket too.\n  const emitEncryptedReasoning = requestWantsEncryptedReasoning(responsesReq);\n  // Synthesizing a reasoning item the fixture never declared takes the NARROWER\n  // explicit-`include` gate, transport-identically to HTTP — a `store: false`\n  // request with a summary-less fixture still gets no reasoning item.\n  const synthesizeSummarylessReasoning = requestIncludesEncryptedReasoning(responsesReq);\n\n  const completionReq = responsesToCompletionRequest(responsesReq);\n  completionReq._endpointType = \"chat\";\n  const contextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  completionReq._context =\n    typeof contextHeader === \"string\"\n      ? contextHeader\n      : Array.isArray(contextHeader) && contextHeader.length > 0\n        ? contextHeader[0]\n        : undefined;\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);\n      defaults.logger.error(strictNoMatchLogLine(\"WS\", \"/v1/responses\", skippedBySequenceOrTurn));\n      journal.add({\n        method: \"WS\",\n        path: \"/v1/responses\",\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, strictMessage);\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\"No fixture matched\", \"invalid_request_error\", \"no_fixture_match\"),\n      ),\n    );\n    return;\n  }\n\n  const response = await resolveResponse(fixture, completionReq);\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // The WS path has no per-request `req.headers`; strict is resolved from the\n  // connection's upgrade headers (see the `!fixture` branch above). Used below\n  // to gate the synthesized reasoning channel on the requested model's capability.\n  const effectiveStrict = resolveStrictMode(defaults.strict, defaults.upgradeHeaders);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(response.error.message, response.error.type, response.error.code),\n      ),\n    );\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isContentWithToolCallsResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const events = buildContentWithToolCallsStreamEvents(\n      response.content ?? \"\",\n      response.toolCalls ?? [],\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      response.blocks,\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const events = buildTextStreamEvents(\n      response.content,\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    const events = buildToolCallStreamEvents(\n      response.toolCalls,\n      completionReq.model,\n      chunkSize,\n      // Gate the synthesized reasoning channel on the requested model's\n      // capability, matching the WS text / content+tool branches and the HTTP\n      // tool-only path so reasoning emission is transport-independent.\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path: \"/v1/responses\",\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify(\n      buildErrorEvent(\"Fixture response did not match any known type\", \"server_error\"),\n    ),\n  );\n}\n\nasync function sendEvents(\n  ws: WebSocketConnection,\n  events: ResponsesSSEEvent[],\n  latency: number,\n  signal?: AbortSignal,\n  onChunkSent?: () => void,\n  recordedTimings?: import(\"./types.js\").RecordedTimings,\n  replaySpeed?: number,\n): Promise<boolean> {\n  let eventIndex = 0;\n  for (const event of events) {\n    if (ws.isClosed) return false;\n    const chunkDelay = calculateDelay(eventIndex, undefined, latency, recordedTimings, replaySpeed);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (ws.isClosed) return false;\n    ws.send(JSON.stringify(event));\n    eventIndex++;\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;AAoDA,SAAS,wBAAwB,KAA4C;AAC3E,QACE,OAAO,QAAQ,YACf,QAAQ,QACP,IAA8B,SAAS;;AAI5C,SAAS,gBACP,SACA,OAAO,yBACP,MACmB;AACnB,QAAO;EACL,MAAM;EACN,OAAO;GAAE;GAAS;GAAM;GAAM;EAC/B;;AAGH,SAAgB,yBACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CAEnB,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,SAAS,CAAC,OAAO,QAAiB;GAC3E,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,8BAA8B,MAAM;AACjD,OAAI;AACF,OAAG,KAAK,KAAK,UAAU,gBAAgB,KAAK,eAAe,CAAC,CAAC;YACtD,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UACH,gBAAgB,mBAAmB,UAAU,yBAAyB,eAAe,CACtF,CACF;AACD;;AAGF,KAAI,CAAC,wBAAwB,OAAO,EAAE;AACpC,KAAG,KACD,KAAK,UACH,gBACE,6CACA,yBACA,uBACD,CACF,CACF;AACD;;CAGF,MAAM,eAAe;EACnB,OAAO,OAAO,SAAS,SAAS;EAChC,OAAQ,OAAO,SAAS,EAAE;EAU1B,cAAc,OAAO;EACrB,OAAO,OAAO;EASd,aAAa,OAAO;EACpB,QAAQ,OAAO;EACf,aAAa,OAAO;EACpB,mBAAmB,OAAO;EAC1B,SAAU,OAAkC;EAC5C,OAAQ,OAA+B;EACxC;CAID,MAAM,yBAAyB,+BAA+B,aAAa;CAI3E,MAAM,iCAAiC,kCAAkC,aAAa;CAEtF,MAAM,gBAAgB,6BAA6B,aAAa;AAChE,eAAc,gBAAgB;CAC9B,MAAM,gBAAgB,SAAS,iBAAiB;AAChD,eAAc,WACZ,OAAO,kBAAkB,WACrB,gBACA,MAAM,QAAQ,cAAc,IAAI,cAAc,SAAS,IACrD,cAAc,KACd;CACR,MAAM,SAAS,SAAS,UAAU;CAClC,MAAM,EAAE,SAAS,4BAA4B,uBAC3C,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAI,kBAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;GAC/D,MAAM,gBAAgB,qBAAqB,wBAAwB;AACnE,YAAS,OAAO,MAAM,qBAAqB,MAAM,iBAAiB,wBAAwB,CAAC;AAC3F,WAAQ,IAAI;IACV,QAAQ;IACR,MAAM;IACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,cAAc;AAC7B;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,sBAAsB,yBAAyB,mBAAmB,CACnF,CACF;AACD;;CAGF,MAAM,WAAW,MAAM,gBAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;CAKtE,MAAM,kBAAkB,kBAAkB,SAAS,QAAQ,SAAS,eAAe;AAGnF,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,SAAS,MAAM,SAAS,SAAS,MAAM,MAAM,SAAS,MAAM,KAAK,CAClF,CACF;AACD;;AAIF,KAAI,+BAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,SAAS,sCACb,SAAS,WAAW,IACpB,SAAS,aAAa,EAAE,EACxB,cAAc,OACd,WACA,yBACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACT,iBAAiB,SAAS,EAC1B,SAAS,QACT,wBACA,+BACD;EAED,MAAM,eAAe,yBAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAI,eAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,SAAS,sBACb,SAAS,SACT,cAAc,OACd,WACA,yBACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACT,iBAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAe,yBAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAI,mBAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EACF,MAAM,SAAS,0BACb,SAAS,WACT,cAAc,OACd,WAIA,yBACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACT,iBAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAe,yBAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR,MAAM;EACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UACH,gBAAgB,iDAAiD,eAAe,CACjF,CACF;;AAGH,eAAe,WACb,IACA,QACA,SACA,QACA,aACA,iBACA,aACkB;CAClB,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,GAAG,SAAU,QAAO;EACxB,MAAM,aAAa,eAAe,YAAY,QAAW,SAAS,iBAAiB,YAAY;AAC/F,MAAI,aAAa,EAAG,OAAM,MAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,GAAG,SAAU,QAAO;AACxB,KAAG,KAAK,KAAK,UAAU,MAAM,CAAC;AAC9B;AACA,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;;AAE9B,QAAO"}
+{"version":3,"file":"ws-responses.js","names":[],"sources":["../src/ws-responses.ts"],"sourcesContent":["/**\n * WebSocket handler for OpenAI Responses API.\n *\n * Accepts `{ type: \"response.create\", model: \"...\", input: [...] }` messages over\n * WebSocket and sends back the same Responses API SSE events as the HTTP\n * handler, but as individual WebSocket text frames.\n */\n\nimport type { ChatCompletionRequest, Fixture } from \"./types.js\";\nimport { matchFixtureDiagnostic } from \"./router.js\";\nimport {\n  responsesToCompletionRequest,\n  requestIncludesEncryptedReasoning,\n  requestWantsEncryptedReasoning,\n  buildTextStreamEvents,\n  buildToolCallStreamEvents,\n  buildContentWithToolCallsStreamEvents,\n  type ResponsesSSEEvent,\n} from \"./responses.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  extractOverrides,\n  resolveResponse,\n  resolveStrictMode,\n  resolveReasoningForModel,\n  strictOverrideField,\n  flattenHeaders,\n  strictNoMatchMessage,\n  strictNoMatchLogLine,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\ninterface ResponseCreateMessage {\n  type: \"response.create\";\n  model?: string;\n  input?: unknown[];\n  instructions?: string;\n  tools?: unknown[];\n  tool_choice?: string | object;\n  stream?: boolean;\n  temperature?: number;\n  max_output_tokens?: number;\n  [key: string]: unknown;\n}\n\nfunction isResponseCreateMessage(msg: unknown): msg is ResponseCreateMessage {\n  return (\n    typeof msg === \"object\" &&\n    msg !== null &&\n    (msg as ResponseCreateMessage).type === \"response.create\"\n  );\n}\n\nfunction buildErrorEvent(\n  message: string,\n  type = \"invalid_request_error\",\n  code?: string,\n): ResponsesSSEEvent {\n  return {\n    type: \"error\",\n    error: { message, type, code },\n  };\n}\n\nexport function handleWebSocketResponses(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  // Serialize message processing to prevent event interleaving\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket responses error: ${msg}`);\n        try {\n          ws.send(JSON.stringify(buildErrorEvent(msg, \"server_error\")));\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): Promise<void> {\n  let parsed: unknown;\n  try {\n    parsed = JSON.parse(raw);\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(`Malformed JSON: ${detail}`, \"invalid_request_error\", \"invalid_json\"),\n      ),\n    );\n    return;\n  }\n\n  if (!isResponseCreateMessage(parsed)) {\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\n          'Expected message type \"response.create\"',\n          \"invalid_request_error\",\n          \"invalid_message_type\",\n        ),\n      ),\n    );\n    return;\n  }\n\n  const responsesReq = {\n    model: parsed.model ?? defaults.model,\n    input: (parsed.input ?? []) as {\n      role?: string;\n      type?: string;\n      content?: string | { type: string; text?: string }[];\n      call_id?: string;\n      name?: string;\n      arguments?: string;\n      output?: string;\n      id?: string;\n    }[],\n    instructions: parsed.instructions,\n    tools: parsed.tools as\n      | {\n          type: \"function\";\n          name: string;\n          description?: string;\n          parameters?: object;\n          strict?: boolean;\n        }[]\n      | undefined,\n    tool_choice: parsed.tool_choice,\n    stream: parsed.stream,\n    temperature: parsed.temperature,\n    max_output_tokens: parsed.max_output_tokens,\n    include: (parsed as { include?: string[] }).include,\n    store: (parsed as { store?: boolean }).store,\n  };\n\n  // Gate encrypted-reasoning emission identically to the HTTP transport so the\n  // agent-framework#7233 stateless-replay path works over WebSocket too.\n  const emitEncryptedReasoning = requestWantsEncryptedReasoning(responsesReq);\n  // Synthesizing a reasoning item the fixture never declared takes the NARROWER\n  // explicit-`include` gate, transport-identically to HTTP — a `store: false`\n  // request with a summary-less fixture still gets no reasoning item.\n  const synthesizeSummarylessReasoning = requestIncludesEncryptedReasoning(responsesReq);\n\n  const completionReq = responsesToCompletionRequest(responsesReq);\n  completionReq._endpointType = \"chat\";\n  const contextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  completionReq._context =\n    typeof contextHeader === \"string\"\n      ? contextHeader\n      : Array.isArray(contextHeader) && contextHeader.length > 0\n        ? contextHeader[0]\n        : undefined;\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);\n      defaults.logger.error(strictNoMatchLogLine(\"WS\", \"/v1/responses\", skippedBySequenceOrTurn));\n      journal.add({\n        method: \"WS\",\n        path: \"/v1/responses\",\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, strictMessage);\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\"No fixture matched\", \"invalid_request_error\", \"no_fixture_match\"),\n      ),\n    );\n    return;\n  }\n\n  const response = await resolveResponse(fixture, completionReq);\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // The WS path has no per-request `req.headers`; strict is resolved from the\n  // connection's upgrade headers (see the `!fixture` branch above). Used below\n  // to gate the synthesized reasoning channel on the requested model's capability.\n  const effectiveStrict = resolveStrictMode(defaults.strict, defaults.upgradeHeaders);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(response.error.message, response.error.type, response.error.code),\n      ),\n    );\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isContentWithToolCallsResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    let events: ResponsesSSEEvent[];\n    try {\n      events = buildContentWithToolCallsStreamEvents(\n        response.content ?? \"\",\n        response.toolCalls ?? [],\n        completionReq.model,\n        chunkSize,\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        response.blocks,\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      throw error;\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const events = buildTextStreamEvents(\n      response.content,\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    let events: ResponsesSSEEvent[];\n    try {\n      events = buildToolCallStreamEvents(\n        response.toolCalls,\n        completionReq.model,\n        chunkSize,\n        // Gate the synthesized reasoning channel on the requested model's\n        // capability, matching the WS text / content+tool branches and the HTTP\n        // tool-only path so reasoning emission is transport-independent.\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      throw error;\n    }\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path: \"/v1/responses\",\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify(\n      buildErrorEvent(\"Fixture response did not match any known type\", \"server_error\"),\n    ),\n  );\n}\n\nasync function sendEvents(\n  ws: WebSocketConnection,\n  events: ResponsesSSEEvent[],\n  latency: number,\n  signal?: AbortSignal,\n  onChunkSent?: () => void,\n  recordedTimings?: import(\"./types.js\").RecordedTimings,\n  replaySpeed?: number,\n): Promise<boolean> {\n  let eventIndex = 0;\n  for (const event of events) {\n    if (ws.isClosed) return false;\n    const chunkDelay = calculateDelay(eventIndex, undefined, latency, recordedTimings, replaySpeed);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (ws.isClosed) return false;\n    ws.send(JSON.stringify(event));\n    eventIndex++;\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;AAoDA,SAAS,wBAAwB,KAA4C;AAC3E,QACE,OAAO,QAAQ,YACf,QAAQ,QACP,IAA8B,SAAS;;AAI5C,SAAS,gBACP,SACA,OAAO,yBACP,MACmB;AACnB,QAAO;EACL,MAAM;EACN,OAAO;GAAE;GAAS;GAAM;GAAM;EAC/B;;AAGH,SAAgB,yBACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CAEnB,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,SAAS,CAAC,OAAO,QAAiB;GAC3E,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,8BAA8B,MAAM;AACjD,OAAI;AACF,OAAG,KAAK,KAAK,UAAU,gBAAgB,KAAK,eAAe,CAAC,CAAC;YACtD,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UACH,gBAAgB,mBAAmB,UAAU,yBAAyB,eAAe,CACtF,CACF;AACD;;AAGF,KAAI,CAAC,wBAAwB,OAAO,EAAE;AACpC,KAAG,KACD,KAAK,UACH,gBACE,6CACA,yBACA,uBACD,CACF,CACF;AACD;;CAGF,MAAM,eAAe;EACnB,OAAO,OAAO,SAAS,SAAS;EAChC,OAAQ,OAAO,SAAS,EAAE;EAU1B,cAAc,OAAO;EACrB,OAAO,OAAO;EASd,aAAa,OAAO;EACpB,QAAQ,OAAO;EACf,aAAa,OAAO;EACpB,mBAAmB,OAAO;EAC1B,SAAU,OAAkC;EAC5C,OAAQ,OAA+B;EACxC;CAID,MAAM,yBAAyB,+BAA+B,aAAa;CAI3E,MAAM,iCAAiC,kCAAkC,aAAa;CAEtF,MAAM,gBAAgB,6BAA6B,aAAa;AAChE,eAAc,gBAAgB;CAC9B,MAAM,gBAAgB,SAAS,iBAAiB;AAChD,eAAc,WACZ,OAAO,kBAAkB,WACrB,gBACA,MAAM,QAAQ,cAAc,IAAI,cAAc,SAAS,IACrD,cAAc,KACd;CACR,MAAM,SAAS,SAAS,UAAU;CAClC,MAAM,EAAE,SAAS,4BAA4B,uBAC3C,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAI,kBAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;GAC/D,MAAM,gBAAgB,qBAAqB,wBAAwB;AACnE,YAAS,OAAO,MAAM,qBAAqB,MAAM,iBAAiB,wBAAwB,CAAC;AAC3F,WAAQ,IAAI;IACV,QAAQ;IACR,MAAM;IACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,cAAc;AAC7B;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,sBAAsB,yBAAyB,mBAAmB,CACnF,CACF;AACD;;CAGF,MAAM,WAAW,MAAM,gBAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;CAKtE,MAAM,kBAAkB,kBAAkB,SAAS,QAAQ,SAAS,eAAe;AAGnF,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,SAAS,MAAM,SAAS,SAAS,MAAM,MAAM,SAAS,MAAM,KAAK,CAClF,CACF;AACD;;AAIF,KAAI,+BAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,IAAI;AACJ,MAAI;AACF,YAAS,sCACP,SAAS,WAAW,IACpB,SAAS,aAAa,EAAE,EACxB,cAAc,OACd,WACA,yBACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACT,iBAAiB,SAAS,EAC1B,SAAS,QACT,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,SAAM;;EAGR,MAAM,eAAe,yBAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAI,eAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,SAAS,sBACb,SAAS,SACT,cAAc,OACd,WACA,yBACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACT,iBAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAe,yBAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAI,mBAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EACF,IAAI;AACJ,MAAI;AACF,YAAS,0BACP,SAAS,WACT,cAAc,OACd,WAIA,yBACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACT,iBAAiB,SAAS,EAC1B,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,SAAM;;EAER,MAAM,eAAe,yBAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR,MAAM;EACN,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UACH,gBAAgB,iDAAiD,eAAe,CACjF,CACF;;AAGH,eAAe,WACb,IACA,QACA,SACA,QACA,aACA,iBACA,aACkB;CAClB,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,GAAG,SAAU,QAAO;EACxB,MAAM,aAAa,eAAe,YAAY,QAAW,SAAS,iBAAiB,YAAY;AAC/F,MAAI,aAAa,EAAG,OAAM,MAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,GAAG,SAAU,QAAO;AACxB,KAAG,KAAK,KAAK,UAAU,MAAM,CAAC;AAC9B;AACA,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;;AAE9B,QAAO"}

package/dist/ws-upstream.cjs

@@ -0,0 +1,118 @@
+const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
+const require_ws_framing = require('./ws-framing.cjs');
+let node_http = require("node:http");
+node_http = require_runtime.__toESM(node_http);
+let node_crypto = require("node:crypto");
+let node_https = require("node:https");
+node_https = require_runtime.__toESM(node_https);
+
+//#region src/ws-upstream.ts
+const REFUSAL_BODY_LIMIT = 64 * 1024;
+const UPGRADE_TIMEOUT_MS = 3e4;
+const MAX_TIMER_DELAY_MS = 2147483647;
+const PROVIDER_HEADERS = new Set(["authorization"]);
+/** Connect to a provider using verified TLS, or an explicitly configured loopback test peer.
+* The caller's signal owns the session lifetime; the upgrade budget defaults to thirty seconds.
+*/
+async function connectUpstreamWebSocket(url, headers, limits, signal, upstreamTimeoutMs = UPGRADE_TIMEOUT_MS) {
+	if (!Number.isSafeInteger(upstreamTimeoutMs) || upstreamTimeoutMs <= 0) throw new RangeError("Upstream WebSocket timeout must be a positive safe integer");
+	const target = new URL(url.href);
+	const secure = target.protocol === "https:" || target.protocol === "wss:";
+	const loopback = target.hostname === "localhost" || target.hostname === "[::1]" || /^127\.(?:\d{1,3}\.){2}\d{1,3}$/.test(target.hostname);
+	if (target.username || target.password || target.hash || !secure && (!(target.protocol === "http:" || target.protocol === "ws:") || !loopback)) throw new Error("Invalid upstream WebSocket URL");
+	for (const value of [
+		limits.maxMessageBytes,
+		limits.maxBufferedBytes,
+		limits.maxWriteBytes
+	]) if (value !== void 0 && (!Number.isSafeInteger(value) || value <= 0)) throw new RangeError("WebSocket limits must be positive safe integers");
+	if (signal.aborted) throw new Error("Upstream WebSocket aborted");
+	target.protocol = secure ? "https:" : "http:";
+	const key = (0, node_crypto.randomBytes)(16).toString("base64");
+	const outgoing = {};
+	for (const [name, value] of Object.entries(headers)) if (PROVIDER_HEADERS.has(name.toLowerCase())) outgoing[name.toLowerCase()] = value;
+	Object.assign(outgoing, {
+		Upgrade: "websocket",
+		Connection: "Upgrade",
+		"Sec-WebSocket-Key": key,
+		"Sec-WebSocket-Version": "13"
+	});
+	return new Promise((resolve, reject) => {
+		let settled = false;
+		let connection;
+		const request = (secure ? node_https : node_http).request(target, {
+			method: "GET",
+			headers: outgoing,
+			agent: false,
+			...secure ? { rejectUnauthorized: true } : {}
+		});
+		const started = performance.now();
+		const onTimeout = () => {
+			const remaining = upstreamTimeoutMs - (performance.now() - started);
+			if (remaining > 0) timer = setTimeout(onTimeout, Math.min(remaining, MAX_TIMER_DELAY_MS));
+			else fail("Upstream WebSocket upgrade timed out");
+		};
+		let timer = setTimeout(onTimeout, Math.min(upstreamTimeoutMs, MAX_TIMER_DELAY_MS));
+		const cleanup = () => {
+			clearTimeout(timer);
+			signal.removeEventListener("abort", abort);
+		};
+		const fail = (message) => {
+			if (settled) return;
+			settled = true;
+			cleanup();
+			request.destroy();
+			reject(new Error(message));
+		};
+		const abort = () => {
+			if (connection) connection.destroy();
+			else fail("Upstream WebSocket aborted");
+		};
+		signal.addEventListener("abort", abort, { once: true });
+		request.on("error", () => fail("Upstream WebSocket connection failed"));
+		request.once("response", (response) => {
+			let bytes = 0;
+			response.on("data", (chunk) => {
+				bytes += chunk.length;
+				if (bytes > REFUSAL_BODY_LIMIT) {
+					fail("Upstream WebSocket refusal body limit exceeded");
+					response.destroy();
+				}
+			});
+			response.once("end", () => fail(`Upstream WebSocket refused (HTTP ${response.statusCode ?? 0})`));
+			response.once("error", () => fail("Upstream WebSocket response failed"));
+			response.once("aborted", () => fail("Upstream WebSocket response failed"));
+		});
+		request.once("upgrade", (response, socket, head) => {
+			if (settled) {
+				socket.destroy();
+				return;
+			}
+			const tokens = (value) => value?.split(",").map((token) => token.trim().toLowerCase()) ?? [];
+			if (response.statusCode !== 101 || !tokens(response.headers.upgrade).includes("websocket") || !tokens(response.headers.connection).includes("upgrade") || response.headers["sec-websocket-accept"] !== require_ws_framing.computeAcceptKey(key) || response.headers["sec-websocket-extensions"] !== void 0 || response.headers["sec-websocket-protocol"] !== void 0) {
+				socket.destroy();
+				fail("Invalid upstream WebSocket upgrade");
+				return;
+			}
+			socket.pause();
+			connection = new require_ws_framing.WebSocketConnection(socket, {
+				maxMessageBytes: limits.maxMessageBytes ?? 1024 * 1024,
+				maxBufferedBytes: limits.maxBufferedBytes ?? 2 * 1024 * 1024,
+				maxWriteBytes: limits.maxWriteBytes ?? 1024 * 1024,
+				maskOutgoing: true
+			});
+			connection.once("close", cleanup);
+			settled = true;
+			clearTimeout(timer);
+			if (head.length) socket.unshift(head);
+			resolve(connection);
+			setImmediate(() => {
+				if (!socket.destroyed) socket.resume();
+			});
+		});
+		request.end();
+	});
+}
+
+//#endregion
+exports.connectUpstreamWebSocket = connectUpstreamWebSocket;
+//# sourceMappingURL=ws-upstream.cjs.map

package/dist/ws-upstream.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-upstream.cjs","names":["https","http","computeAcceptKey","WebSocketConnection"],"sources":["../src/ws-upstream.ts"],"sourcesContent":["import * as http from \"node:http\";\nimport * as https from \"node:https\";\nimport { randomBytes } from \"node:crypto\";\nimport { WebSocketConnection, computeAcceptKey, type WebSocketLimits } from \"./ws-framing.js\";\n\nconst REFUSAL_BODY_LIMIT = 64 * 1024;\nconst UPGRADE_TIMEOUT_MS = 30_000;\nconst MAX_TIMER_DELAY_MS = 2_147_483_647;\n// Reviewed Live startup uses Authorization plus connector-owned RFC 6455 headers.\nconst PROVIDER_HEADERS = new Set([\"authorization\"]);\n\n/** Connect to a provider using verified TLS, or an explicitly configured loopback test peer.\n * The caller's signal owns the session lifetime; the upgrade budget defaults to thirty seconds.\n */\nexport async function connectUpstreamWebSocket(\n  url: URL,\n  headers: Record<string, string>,\n  limits: WebSocketLimits,\n  signal: AbortSignal,\n  upstreamTimeoutMs = UPGRADE_TIMEOUT_MS,\n): Promise<WebSocketConnection> {\n  if (!Number.isSafeInteger(upstreamTimeoutMs) || upstreamTimeoutMs <= 0) {\n    throw new RangeError(\"Upstream WebSocket timeout must be a positive safe integer\");\n  }\n  const target = new URL(url.href);\n  const secure = target.protocol === \"https:\" || target.protocol === \"wss:\";\n  const loopback =\n    target.hostname === \"localhost\" ||\n    target.hostname === \"[::1]\" ||\n    /^127\\.(?:\\d{1,3}\\.){2}\\d{1,3}$/.test(target.hostname);\n  if (\n    target.username ||\n    target.password ||\n    target.hash ||\n    (!secure && (!(target.protocol === \"http:\" || target.protocol === \"ws:\") || !loopback))\n  ) {\n    throw new Error(\"Invalid upstream WebSocket URL\");\n  }\n  for (const value of [limits.maxMessageBytes, limits.maxBufferedBytes, limits.maxWriteBytes]) {\n    if (value !== undefined && (!Number.isSafeInteger(value) || value <= 0))\n      throw new RangeError(\"WebSocket limits must be positive safe integers\");\n  }\n  if (signal.aborted) throw new Error(\"Upstream WebSocket aborted\");\n  target.protocol = secure ? \"https:\" : \"http:\";\n  const key = randomBytes(16).toString(\"base64\");\n  const outgoing: Record<string, string> = {};\n  for (const [name, value] of Object.entries(headers)) {\n    if (PROVIDER_HEADERS.has(name.toLowerCase())) outgoing[name.toLowerCase()] = value;\n  }\n  Object.assign(outgoing, {\n    Upgrade: \"websocket\",\n    Connection: \"Upgrade\",\n    \"Sec-WebSocket-Key\": key,\n    \"Sec-WebSocket-Version\": \"13\",\n  });\n  return new Promise<WebSocketConnection>((resolve, reject) => {\n    let settled = false;\n    let connection: WebSocketConnection | undefined;\n    const request = (secure ? https : http).request(target, {\n      method: \"GET\",\n      headers: outgoing,\n      agent: false,\n      ...(secure ? { rejectUnauthorized: true } : {}),\n    });\n    const started = performance.now();\n    const onTimeout = () => {\n      const remaining = upstreamTimeoutMs - (performance.now() - started);\n      if (remaining > 0) timer = setTimeout(onTimeout, Math.min(remaining, MAX_TIMER_DELAY_MS));\n      else fail(\"Upstream WebSocket upgrade timed out\");\n    };\n    // Node clamps overflowing setTimeout delays to one millisecond. Keep the\n    // full validated budget by scheduling bounded segments against elapsed time.\n    let timer = setTimeout(onTimeout, Math.min(upstreamTimeoutMs, MAX_TIMER_DELAY_MS));\n    const cleanup = () => {\n      clearTimeout(timer);\n      signal.removeEventListener(\"abort\", abort);\n    };\n    const fail = (message: string) => {\n      if (settled) return;\n      settled = true;\n      cleanup();\n      request.destroy();\n      reject(new Error(message));\n    };\n    const abort = () => {\n      if (connection) connection.destroy();\n      else fail(\"Upstream WebSocket aborted\");\n    };\n    signal.addEventListener(\"abort\", abort, { once: true });\n    request.on(\"error\", () => fail(\"Upstream WebSocket connection failed\"));\n    request.once(\"response\", (response) => {\n      let bytes = 0;\n      response.on(\"data\", (chunk: Buffer) => {\n        bytes += chunk.length;\n        if (bytes > REFUSAL_BODY_LIMIT) {\n          fail(\"Upstream WebSocket refusal body limit exceeded\");\n          response.destroy();\n        }\n      });\n      response.once(\"end\", () =>\n        fail(`Upstream WebSocket refused (HTTP ${response.statusCode ?? 0})`),\n      );\n      response.once(\"error\", () => fail(\"Upstream WebSocket response failed\"));\n      response.once(\"aborted\", () => fail(\"Upstream WebSocket response failed\"));\n    });\n    request.once(\"upgrade\", (response, socket, head) => {\n      if (settled) {\n        socket.destroy();\n        return;\n      }\n      const tokens = (value: string | undefined) =>\n        value?.split(\",\").map((token) => token.trim().toLowerCase()) ?? [];\n      if (\n        response.statusCode !== 101 ||\n        !tokens(response.headers.upgrade).includes(\"websocket\") ||\n        !tokens(response.headers.connection).includes(\"upgrade\") ||\n        response.headers[\"sec-websocket-accept\"] !== computeAcceptKey(key) ||\n        response.headers[\"sec-websocket-extensions\"] !== undefined ||\n        response.headers[\"sec-websocket-protocol\"] !== undefined\n      ) {\n        socket.destroy();\n        fail(\"Invalid upstream WebSocket upgrade\");\n        return;\n      }\n      socket.pause();\n      connection = new WebSocketConnection(socket, {\n        maxMessageBytes: limits.maxMessageBytes ?? 1024 * 1024,\n        maxBufferedBytes: limits.maxBufferedBytes ?? 2 * 1024 * 1024,\n        maxWriteBytes: limits.maxWriteBytes ?? 1024 * 1024,\n        maskOutgoing: true,\n      });\n      connection.once(\"close\", cleanup);\n      settled = true;\n      clearTimeout(timer);\n      // Pause until the promise consumer can register its message/error listeners.\n      // unshift preserves ordering between upgrade head and subsequent socket bytes.\n      if (head.length) socket.unshift(head);\n      resolve(connection);\n      setImmediate(() => {\n        if (!socket.destroyed) socket.resume();\n      });\n    });\n    request.end();\n  });\n}\n"],"mappings":";;;;;;;;;AAKA,MAAM,qBAAqB,KAAK;AAChC,MAAM,qBAAqB;AAC3B,MAAM,qBAAqB;AAE3B,MAAM,mBAAmB,IAAI,IAAI,CAAC,gBAAgB,CAAC;;;;AAKnD,eAAsB,yBACpB,KACA,SACA,QACA,QACA,oBAAoB,oBACU;AAC9B,KAAI,CAAC,OAAO,cAAc,kBAAkB,IAAI,qBAAqB,EACnE,OAAM,IAAI,WAAW,6DAA6D;CAEpF,MAAM,SAAS,IAAI,IAAI,IAAI,KAAK;CAChC,MAAM,SAAS,OAAO,aAAa,YAAY,OAAO,aAAa;CACnE,MAAM,WACJ,OAAO,aAAa,eACpB,OAAO,aAAa,WACpB,iCAAiC,KAAK,OAAO,SAAS;AACxD,KACE,OAAO,YACP,OAAO,YACP,OAAO,QACN,CAAC,WAAW,EAAE,OAAO,aAAa,WAAW,OAAO,aAAa,UAAU,CAAC,UAE7E,OAAM,IAAI,MAAM,iCAAiC;AAEnD,MAAK,MAAM,SAAS;EAAC,OAAO;EAAiB,OAAO;EAAkB,OAAO;EAAc,CACzF,KAAI,UAAU,WAAc,CAAC,OAAO,cAAc,MAAM,IAAI,SAAS,GACnE,OAAM,IAAI,WAAW,kDAAkD;AAE3E,KAAI,OAAO,QAAS,OAAM,IAAI,MAAM,6BAA6B;AACjE,QAAO,WAAW,SAAS,WAAW;CACtC,MAAM,mCAAkB,GAAG,CAAC,SAAS,SAAS;CAC9C,MAAM,WAAmC,EAAE;AAC3C,MAAK,MAAM,CAAC,MAAM,UAAU,OAAO,QAAQ,QAAQ,CACjD,KAAI,iBAAiB,IAAI,KAAK,aAAa,CAAC,CAAE,UAAS,KAAK,aAAa,IAAI;AAE/E,QAAO,OAAO,UAAU;EACtB,SAAS;EACT,YAAY;EACZ,qBAAqB;EACrB,yBAAyB;EAC1B,CAAC;AACF,QAAO,IAAI,SAA8B,SAAS,WAAW;EAC3D,IAAI,UAAU;EACd,IAAI;EACJ,MAAM,WAAW,SAASA,aAAQC,WAAM,QAAQ,QAAQ;GACtD,QAAQ;GACR,SAAS;GACT,OAAO;GACP,GAAI,SAAS,EAAE,oBAAoB,MAAM,GAAG,EAAE;GAC/C,CAAC;EACF,MAAM,UAAU,YAAY,KAAK;EACjC,MAAM,kBAAkB;GACtB,MAAM,YAAY,qBAAqB,YAAY,KAAK,GAAG;AAC3D,OAAI,YAAY,EAAG,SAAQ,WAAW,WAAW,KAAK,IAAI,WAAW,mBAAmB,CAAC;OACpF,MAAK,uCAAuC;;EAInD,IAAI,QAAQ,WAAW,WAAW,KAAK,IAAI,mBAAmB,mBAAmB,CAAC;EAClF,MAAM,gBAAgB;AACpB,gBAAa,MAAM;AACnB,UAAO,oBAAoB,SAAS,MAAM;;EAE5C,MAAM,QAAQ,YAAoB;AAChC,OAAI,QAAS;AACb,aAAU;AACV,YAAS;AACT,WAAQ,SAAS;AACjB,UAAO,IAAI,MAAM,QAAQ,CAAC;;EAE5B,MAAM,cAAc;AAClB,OAAI,WAAY,YAAW,SAAS;OAC/B,MAAK,6BAA6B;;AAEzC,SAAO,iBAAiB,SAAS,OAAO,EAAE,MAAM,MAAM,CAAC;AACvD,UAAQ,GAAG,eAAe,KAAK,uCAAuC,CAAC;AACvE,UAAQ,KAAK,aAAa,aAAa;GACrC,IAAI,QAAQ;AACZ,YAAS,GAAG,SAAS,UAAkB;AACrC,aAAS,MAAM;AACf,QAAI,QAAQ,oBAAoB;AAC9B,UAAK,iDAAiD;AACtD,cAAS,SAAS;;KAEpB;AACF,YAAS,KAAK,aACZ,KAAK,oCAAoC,SAAS,cAAc,EAAE,GAAG,CACtE;AACD,YAAS,KAAK,eAAe,KAAK,qCAAqC,CAAC;AACxE,YAAS,KAAK,iBAAiB,KAAK,qCAAqC,CAAC;IAC1E;AACF,UAAQ,KAAK,YAAY,UAAU,QAAQ,SAAS;AAClD,OAAI,SAAS;AACX,WAAO,SAAS;AAChB;;GAEF,MAAM,UAAU,UACd,OAAO,MAAM,IAAI,CAAC,KAAK,UAAU,MAAM,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE;AACpE,OACE,SAAS,eAAe,OACxB,CAAC,OAAO,SAAS,QAAQ,QAAQ,CAAC,SAAS,YAAY,IACvD,CAAC,OAAO,SAAS,QAAQ,WAAW,CAAC,SAAS,UAAU,IACxD,SAAS,QAAQ,4BAA4BC,oCAAiB,IAAI,IAClE,SAAS,QAAQ,gCAAgC,UACjD,SAAS,QAAQ,8BAA8B,QAC/C;AACA,WAAO,SAAS;AAChB,SAAK,qCAAqC;AAC1C;;AAEF,UAAO,OAAO;AACd,gBAAa,IAAIC,uCAAoB,QAAQ;IAC3C,iBAAiB,OAAO,mBAAmB,OAAO;IAClD,kBAAkB,OAAO,oBAAoB,IAAI,OAAO;IACxD,eAAe,OAAO,iBAAiB,OAAO;IAC9C,cAAc;IACf,CAAC;AACF,cAAW,KAAK,SAAS,QAAQ;AACjC,aAAU;AACV,gBAAa,MAAM;AAGnB,OAAI,KAAK,OAAQ,QAAO,QAAQ,KAAK;AACrC,WAAQ,WAAW;AACnB,sBAAmB;AACjB,QAAI,CAAC,OAAO,UAAW,QAAO,QAAQ;KACtC;IACF;AACF,UAAQ,KAAK;GACb"}

package/dist/ws-upstream.js

@@ -0,0 +1,115 @@
+import { WebSocketConnection, computeAcceptKey } from "./ws-framing.js";
+import * as http$1 from "node:http";
+import { randomBytes } from "node:crypto";
+import * as https from "node:https";
+
+//#region src/ws-upstream.ts
+const REFUSAL_BODY_LIMIT = 64 * 1024;
+const UPGRADE_TIMEOUT_MS = 3e4;
+const MAX_TIMER_DELAY_MS = 2147483647;
+const PROVIDER_HEADERS = new Set(["authorization"]);
+/** Connect to a provider using verified TLS, or an explicitly configured loopback test peer.
+* The caller's signal owns the session lifetime; the upgrade budget defaults to thirty seconds.
+*/
+async function connectUpstreamWebSocket(url, headers, limits, signal, upstreamTimeoutMs = UPGRADE_TIMEOUT_MS) {
+	if (!Number.isSafeInteger(upstreamTimeoutMs) || upstreamTimeoutMs <= 0) throw new RangeError("Upstream WebSocket timeout must be a positive safe integer");
+	const target = new URL(url.href);
+	const secure = target.protocol === "https:" || target.protocol === "wss:";
+	const loopback = target.hostname === "localhost" || target.hostname === "[::1]" || /^127\.(?:\d{1,3}\.){2}\d{1,3}$/.test(target.hostname);
+	if (target.username || target.password || target.hash || !secure && (!(target.protocol === "http:" || target.protocol === "ws:") || !loopback)) throw new Error("Invalid upstream WebSocket URL");
+	for (const value of [
+		limits.maxMessageBytes,
+		limits.maxBufferedBytes,
+		limits.maxWriteBytes
+	]) if (value !== void 0 && (!Number.isSafeInteger(value) || value <= 0)) throw new RangeError("WebSocket limits must be positive safe integers");
+	if (signal.aborted) throw new Error("Upstream WebSocket aborted");
+	target.protocol = secure ? "https:" : "http:";
+	const key = randomBytes(16).toString("base64");
+	const outgoing = {};
+	for (const [name, value] of Object.entries(headers)) if (PROVIDER_HEADERS.has(name.toLowerCase())) outgoing[name.toLowerCase()] = value;
+	Object.assign(outgoing, {
+		Upgrade: "websocket",
+		Connection: "Upgrade",
+		"Sec-WebSocket-Key": key,
+		"Sec-WebSocket-Version": "13"
+	});
+	return new Promise((resolve, reject) => {
+		let settled = false;
+		let connection;
+		const request = (secure ? https : http$1).request(target, {
+			method: "GET",
+			headers: outgoing,
+			agent: false,
+			...secure ? { rejectUnauthorized: true } : {}
+		});
+		const started = performance.now();
+		const onTimeout = () => {
+			const remaining = upstreamTimeoutMs - (performance.now() - started);
+			if (remaining > 0) timer = setTimeout(onTimeout, Math.min(remaining, MAX_TIMER_DELAY_MS));
+			else fail("Upstream WebSocket upgrade timed out");
+		};
+		let timer = setTimeout(onTimeout, Math.min(upstreamTimeoutMs, MAX_TIMER_DELAY_MS));
+		const cleanup = () => {
+			clearTimeout(timer);
+			signal.removeEventListener("abort", abort);
+		};
+		const fail = (message) => {
+			if (settled) return;
+			settled = true;
+			cleanup();
+			request.destroy();
+			reject(new Error(message));
+		};
+		const abort = () => {
+			if (connection) connection.destroy();
+			else fail("Upstream WebSocket aborted");
+		};
+		signal.addEventListener("abort", abort, { once: true });
+		request.on("error", () => fail("Upstream WebSocket connection failed"));
+		request.once("response", (response) => {
+			let bytes = 0;
+			response.on("data", (chunk) => {
+				bytes += chunk.length;
+				if (bytes > REFUSAL_BODY_LIMIT) {
+					fail("Upstream WebSocket refusal body limit exceeded");
+					response.destroy();
+				}
+			});
+			response.once("end", () => fail(`Upstream WebSocket refused (HTTP ${response.statusCode ?? 0})`));
+			response.once("error", () => fail("Upstream WebSocket response failed"));
+			response.once("aborted", () => fail("Upstream WebSocket response failed"));
+		});
+		request.once("upgrade", (response, socket, head) => {
+			if (settled) {
+				socket.destroy();
+				return;
+			}
+			const tokens = (value) => value?.split(",").map((token) => token.trim().toLowerCase()) ?? [];
+			if (response.statusCode !== 101 || !tokens(response.headers.upgrade).includes("websocket") || !tokens(response.headers.connection).includes("upgrade") || response.headers["sec-websocket-accept"] !== computeAcceptKey(key) || response.headers["sec-websocket-extensions"] !== void 0 || response.headers["sec-websocket-protocol"] !== void 0) {
+				socket.destroy();
+				fail("Invalid upstream WebSocket upgrade");
+				return;
+			}
+			socket.pause();
+			connection = new WebSocketConnection(socket, {
+				maxMessageBytes: limits.maxMessageBytes ?? 1024 * 1024,
+				maxBufferedBytes: limits.maxBufferedBytes ?? 2 * 1024 * 1024,
+				maxWriteBytes: limits.maxWriteBytes ?? 1024 * 1024,
+				maskOutgoing: true
+			});
+			connection.once("close", cleanup);
+			settled = true;
+			clearTimeout(timer);
+			if (head.length) socket.unshift(head);
+			resolve(connection);
+			setImmediate(() => {
+				if (!socket.destroyed) socket.resume();
+			});
+		});
+		request.end();
+	});
+}
+
+//#endregion
+export { connectUpstreamWebSocket };
+//# sourceMappingURL=ws-upstream.js.map

package/dist/ws-upstream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-upstream.js","names":["http"],"sources":["../src/ws-upstream.ts"],"sourcesContent":["import * as http from \"node:http\";\nimport * as https from \"node:https\";\nimport { randomBytes } from \"node:crypto\";\nimport { WebSocketConnection, computeAcceptKey, type WebSocketLimits } from \"./ws-framing.js\";\n\nconst REFUSAL_BODY_LIMIT = 64 * 1024;\nconst UPGRADE_TIMEOUT_MS = 30_000;\nconst MAX_TIMER_DELAY_MS = 2_147_483_647;\n// Reviewed Live startup uses Authorization plus connector-owned RFC 6455 headers.\nconst PROVIDER_HEADERS = new Set([\"authorization\"]);\n\n/** Connect to a provider using verified TLS, or an explicitly configured loopback test peer.\n * The caller's signal owns the session lifetime; the upgrade budget defaults to thirty seconds.\n */\nexport async function connectUpstreamWebSocket(\n  url: URL,\n  headers: Record<string, string>,\n  limits: WebSocketLimits,\n  signal: AbortSignal,\n  upstreamTimeoutMs = UPGRADE_TIMEOUT_MS,\n): Promise<WebSocketConnection> {\n  if (!Number.isSafeInteger(upstreamTimeoutMs) || upstreamTimeoutMs <= 0) {\n    throw new RangeError(\"Upstream WebSocket timeout must be a positive safe integer\");\n  }\n  const target = new URL(url.href);\n  const secure = target.protocol === \"https:\" || target.protocol === \"wss:\";\n  const loopback =\n    target.hostname === \"localhost\" ||\n    target.hostname === \"[::1]\" ||\n    /^127\\.(?:\\d{1,3}\\.){2}\\d{1,3}$/.test(target.hostname);\n  if (\n    target.username ||\n    target.password ||\n    target.hash ||\n    (!secure && (!(target.protocol === \"http:\" || target.protocol === \"ws:\") || !loopback))\n  ) {\n    throw new Error(\"Invalid upstream WebSocket URL\");\n  }\n  for (const value of [limits.maxMessageBytes, limits.maxBufferedBytes, limits.maxWriteBytes]) {\n    if (value !== undefined && (!Number.isSafeInteger(value) || value <= 0))\n      throw new RangeError(\"WebSocket limits must be positive safe integers\");\n  }\n  if (signal.aborted) throw new Error(\"Upstream WebSocket aborted\");\n  target.protocol = secure ? \"https:\" : \"http:\";\n  const key = randomBytes(16).toString(\"base64\");\n  const outgoing: Record<string, string> = {};\n  for (const [name, value] of Object.entries(headers)) {\n    if (PROVIDER_HEADERS.has(name.toLowerCase())) outgoing[name.toLowerCase()] = value;\n  }\n  Object.assign(outgoing, {\n    Upgrade: \"websocket\",\n    Connection: \"Upgrade\",\n    \"Sec-WebSocket-Key\": key,\n    \"Sec-WebSocket-Version\": \"13\",\n  });\n  return new Promise<WebSocketConnection>((resolve, reject) => {\n    let settled = false;\n    let connection: WebSocketConnection | undefined;\n    const request = (secure ? https : http).request(target, {\n      method: \"GET\",\n      headers: outgoing,\n      agent: false,\n      ...(secure ? { rejectUnauthorized: true } : {}),\n    });\n    const started = performance.now();\n    const onTimeout = () => {\n      const remaining = upstreamTimeoutMs - (performance.now() - started);\n      if (remaining > 0) timer = setTimeout(onTimeout, Math.min(remaining, MAX_TIMER_DELAY_MS));\n      else fail(\"Upstream WebSocket upgrade timed out\");\n    };\n    // Node clamps overflowing setTimeout delays to one millisecond. Keep the\n    // full validated budget by scheduling bounded segments against elapsed time.\n    let timer = setTimeout(onTimeout, Math.min(upstreamTimeoutMs, MAX_TIMER_DELAY_MS));\n    const cleanup = () => {\n      clearTimeout(timer);\n      signal.removeEventListener(\"abort\", abort);\n    };\n    const fail = (message: string) => {\n      if (settled) return;\n      settled = true;\n      cleanup();\n      request.destroy();\n      reject(new Error(message));\n    };\n    const abort = () => {\n      if (connection) connection.destroy();\n      else fail(\"Upstream WebSocket aborted\");\n    };\n    signal.addEventListener(\"abort\", abort, { once: true });\n    request.on(\"error\", () => fail(\"Upstream WebSocket connection failed\"));\n    request.once(\"response\", (response) => {\n      let bytes = 0;\n      response.on(\"data\", (chunk: Buffer) => {\n        bytes += chunk.length;\n        if (bytes > REFUSAL_BODY_LIMIT) {\n          fail(\"Upstream WebSocket refusal body limit exceeded\");\n          response.destroy();\n        }\n      });\n      response.once(\"end\", () =>\n        fail(`Upstream WebSocket refused (HTTP ${response.statusCode ?? 0})`),\n      );\n      response.once(\"error\", () => fail(\"Upstream WebSocket response failed\"));\n      response.once(\"aborted\", () => fail(\"Upstream WebSocket response failed\"));\n    });\n    request.once(\"upgrade\", (response, socket, head) => {\n      if (settled) {\n        socket.destroy();\n        return;\n      }\n      const tokens = (value: string | undefined) =>\n        value?.split(\",\").map((token) => token.trim().toLowerCase()) ?? [];\n      if (\n        response.statusCode !== 101 ||\n        !tokens(response.headers.upgrade).includes(\"websocket\") ||\n        !tokens(response.headers.connection).includes(\"upgrade\") ||\n        response.headers[\"sec-websocket-accept\"] !== computeAcceptKey(key) ||\n        response.headers[\"sec-websocket-extensions\"] !== undefined ||\n        response.headers[\"sec-websocket-protocol\"] !== undefined\n      ) {\n        socket.destroy();\n        fail(\"Invalid upstream WebSocket upgrade\");\n        return;\n      }\n      socket.pause();\n      connection = new WebSocketConnection(socket, {\n        maxMessageBytes: limits.maxMessageBytes ?? 1024 * 1024,\n        maxBufferedBytes: limits.maxBufferedBytes ?? 2 * 1024 * 1024,\n        maxWriteBytes: limits.maxWriteBytes ?? 1024 * 1024,\n        maskOutgoing: true,\n      });\n      connection.once(\"close\", cleanup);\n      settled = true;\n      clearTimeout(timer);\n      // Pause until the promise consumer can register its message/error listeners.\n      // unshift preserves ordering between upgrade head and subsequent socket bytes.\n      if (head.length) socket.unshift(head);\n      resolve(connection);\n      setImmediate(() => {\n        if (!socket.destroyed) socket.resume();\n      });\n    });\n    request.end();\n  });\n}\n"],"mappings":";;;;;;AAKA,MAAM,qBAAqB,KAAK;AAChC,MAAM,qBAAqB;AAC3B,MAAM,qBAAqB;AAE3B,MAAM,mBAAmB,IAAI,IAAI,CAAC,gBAAgB,CAAC;;;;AAKnD,eAAsB,yBACpB,KACA,SACA,QACA,QACA,oBAAoB,oBACU;AAC9B,KAAI,CAAC,OAAO,cAAc,kBAAkB,IAAI,qBAAqB,EACnE,OAAM,IAAI,WAAW,6DAA6D;CAEpF,MAAM,SAAS,IAAI,IAAI,IAAI,KAAK;CAChC,MAAM,SAAS,OAAO,aAAa,YAAY,OAAO,aAAa;CACnE,MAAM,WACJ,OAAO,aAAa,eACpB,OAAO,aAAa,WACpB,iCAAiC,KAAK,OAAO,SAAS;AACxD,KACE,OAAO,YACP,OAAO,YACP,OAAO,QACN,CAAC,WAAW,EAAE,OAAO,aAAa,WAAW,OAAO,aAAa,UAAU,CAAC,UAE7E,OAAM,IAAI,MAAM,iCAAiC;AAEnD,MAAK,MAAM,SAAS;EAAC,OAAO;EAAiB,OAAO;EAAkB,OAAO;EAAc,CACzF,KAAI,UAAU,WAAc,CAAC,OAAO,cAAc,MAAM,IAAI,SAAS,GACnE,OAAM,IAAI,WAAW,kDAAkD;AAE3E,KAAI,OAAO,QAAS,OAAM,IAAI,MAAM,6BAA6B;AACjE,QAAO,WAAW,SAAS,WAAW;CACtC,MAAM,MAAM,YAAY,GAAG,CAAC,SAAS,SAAS;CAC9C,MAAM,WAAmC,EAAE;AAC3C,MAAK,MAAM,CAAC,MAAM,UAAU,OAAO,QAAQ,QAAQ,CACjD,KAAI,iBAAiB,IAAI,KAAK,aAAa,CAAC,CAAE,UAAS,KAAK,aAAa,IAAI;AAE/E,QAAO,OAAO,UAAU;EACtB,SAAS;EACT,YAAY;EACZ,qBAAqB;EACrB,yBAAyB;EAC1B,CAAC;AACF,QAAO,IAAI,SAA8B,SAAS,WAAW;EAC3D,IAAI,UAAU;EACd,IAAI;EACJ,MAAM,WAAW,SAAS,QAAQA,QAAM,QAAQ,QAAQ;GACtD,QAAQ;GACR,SAAS;GACT,OAAO;GACP,GAAI,SAAS,EAAE,oBAAoB,MAAM,GAAG,EAAE;GAC/C,CAAC;EACF,MAAM,UAAU,YAAY,KAAK;EACjC,MAAM,kBAAkB;GACtB,MAAM,YAAY,qBAAqB,YAAY,KAAK,GAAG;AAC3D,OAAI,YAAY,EAAG,SAAQ,WAAW,WAAW,KAAK,IAAI,WAAW,mBAAmB,CAAC;OACpF,MAAK,uCAAuC;;EAInD,IAAI,QAAQ,WAAW,WAAW,KAAK,IAAI,mBAAmB,mBAAmB,CAAC;EAClF,MAAM,gBAAgB;AACpB,gBAAa,MAAM;AACnB,UAAO,oBAAoB,SAAS,MAAM;;EAE5C,MAAM,QAAQ,YAAoB;AAChC,OAAI,QAAS;AACb,aAAU;AACV,YAAS;AACT,WAAQ,SAAS;AACjB,UAAO,IAAI,MAAM,QAAQ,CAAC;;EAE5B,MAAM,cAAc;AAClB,OAAI,WAAY,YAAW,SAAS;OAC/B,MAAK,6BAA6B;;AAEzC,SAAO,iBAAiB,SAAS,OAAO,EAAE,MAAM,MAAM,CAAC;AACvD,UAAQ,GAAG,eAAe,KAAK,uCAAuC,CAAC;AACvE,UAAQ,KAAK,aAAa,aAAa;GACrC,IAAI,QAAQ;AACZ,YAAS,GAAG,SAAS,UAAkB;AACrC,aAAS,MAAM;AACf,QAAI,QAAQ,oBAAoB;AAC9B,UAAK,iDAAiD;AACtD,cAAS,SAAS;;KAEpB;AACF,YAAS,KAAK,aACZ,KAAK,oCAAoC,SAAS,cAAc,EAAE,GAAG,CACtE;AACD,YAAS,KAAK,eAAe,KAAK,qCAAqC,CAAC;AACxE,YAAS,KAAK,iBAAiB,KAAK,qCAAqC,CAAC;IAC1E;AACF,UAAQ,KAAK,YAAY,UAAU,QAAQ,SAAS;AAClD,OAAI,SAAS;AACX,WAAO,SAAS;AAChB;;GAEF,MAAM,UAAU,UACd,OAAO,MAAM,IAAI,CAAC,KAAK,UAAU,MAAM,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE;AACpE,OACE,SAAS,eAAe,OACxB,CAAC,OAAO,SAAS,QAAQ,QAAQ,CAAC,SAAS,YAAY,IACvD,CAAC,OAAO,SAAS,QAAQ,WAAW,CAAC,SAAS,UAAU,IACxD,SAAS,QAAQ,4BAA4B,iBAAiB,IAAI,IAClE,SAAS,QAAQ,gCAAgC,UACjD,SAAS,QAAQ,8BAA8B,QAC/C;AACA,WAAO,SAAS;AAChB,SAAK,qCAAqC;AAC1C;;AAEF,UAAO,OAAO;AACd,gBAAa,IAAI,oBAAoB,QAAQ;IAC3C,iBAAiB,OAAO,mBAAmB,OAAO;IAClD,kBAAkB,OAAO,oBAAoB,IAAI,OAAO;IACxD,eAAe,OAAO,iBAAiB,OAAO;IAC9C,cAAc;IACf,CAAC;AACF,cAAW,KAAK,SAAS,QAAQ;AACjC,aAAU;AACV,gBAAa,MAAM;AAGnB,OAAI,KAAK,OAAQ,QAAO,QAAQ,KAAK;AACrC,WAAQ,WAAW;AACnB,sBAAmB;AACjB,QAAI,CAAC,OAAO,UAAW,QAAO,QAAQ;KACtC;IACF;AACF,UAAQ,KAAK;GACb"}