File size: 1,510 Bytes
b91e262 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | import { NEXT_REQUEST_ID_HEADER } from '../components/app-router-headers'
import { InvariantError } from '../../shared/lib/invariant-error'
export interface DebugChannelReadableWriterPair {
readonly readable: ReadableStream<Uint8Array>
readonly writer: WritableStreamDefaultWriter<Uint8Array>
}
const pairs = new Map<string, DebugChannelReadableWriterPair>()
export function getOrCreateDebugChannelReadableWriterPair(
requestId: string
): DebugChannelReadableWriterPair {
let pair = pairs.get(requestId)
if (!pair) {
const { readable, writable } = new TransformStream<Uint8Array, Uint8Array>()
pair = { readable, writer: writable.getWriter() }
pairs.set(requestId, pair)
pair.writer.closed.finally(() => pairs.delete(requestId))
}
return pair
}
export function createDebugChannel(
requestHeaders: Record<string, string> | undefined
): {
writable?: WritableStream
readable?: ReadableStream
} {
let requestId: string | undefined
if (requestHeaders) {
requestId = requestHeaders[NEXT_REQUEST_ID_HEADER] ?? undefined
if (!requestId) {
throw new InvariantError(
`Expected a ${JSON.stringify(NEXT_REQUEST_ID_HEADER)} request header.`
)
}
} else {
requestId = self.__next_r
if (!requestId) {
throw new InvariantError(
`Expected a request ID to be defined for the document via self.__next_r.`
)
}
}
const { readable } = getOrCreateDebugChannelReadableWriterPair(requestId)
return { readable }
}
|