Function makeWebSocketExecutor

packages/http-recorder/src/websocket.ts:64–173

(
  options: WebSocketRecordReplayOptions<E>,
)

  })

export const makeWebSocketExecutor = <E>(
  options: WebSocketRecordReplayOptions<E>,
): Effect.Effect<WebSocketExecutor<E>, never, Scope.Scope> =>
  Effect.gen(function* () {
    const mode = options.mode ?? (yield* resolveAutoMode(options.cassette, options.name))
    const redactor = options.redactor ?? make()
    const openSnapshot = (request: WebSocketRequest) => {
      const snapshot = redactor.request({
        method: "GET",
        url: request.url,
        headers: headersRecord(request.headers),
        body: "",
      })
      return { url: snapshot.url, headers: snapshot.headers }
    }
    const redactEvent = (event: WebSocketEvent) => {
      if (event.kind === "binary") return event
      const body =
        event.direction === "client"
          ? redactor.request({ method: "WEBSOCKET", url: "", headers: {}, body: event.body }).body
          : redactor.response({ status: 101, headers: {}, body: event.body }).body
      return { ...event, body }
    }

    if (mode === "passthrough") return options.live

    if (mode === "record") {
      return {
        open: (request) =>
          Effect.gen(function* () {
            const events: WebSocketEvent[] = []
            const connection = yield* options.live.open(request)
            const closed = yield* Ref.make(false)
            const closeLock = yield* Semaphore.make(1)
            return {
              sendText: (message) =>
                Effect.sync(() => events.push(redactEvent(textEvent("client", message)))).pipe(
                  Effect.andThen(connection.sendText(message)),
                ),
              messages: connection.messages.pipe(
                Stream.tap((message) =>
                  Effect.sync(() =>
                    events.push(
                      typeof message === "string"
                        ? redactEvent(textEvent("server", message))
                        : {
                            direction: "server",
                            kind: "binary",
                            body: Buffer.from(message).toString("base64"),
                            bodyEncoding: "base64",
                          },
                    ),
                  ),
                ),
              ),
              close: closeLock.withPermit(
                Effect.gen(function* () {
                  if (yield* Ref.get(closed)) return