8752464816
This PR brings back @Schniz awesome contribution to bring in bodies to middleware. It was reverted to leave it out of the stable release and to have some time to test it out in canary before officially releasing it. This PR is simply a `cherry-pick` of his original work. Closes: #30953 Closes: https://github.com/vercel/next.js/pull/34490 Co-authored-by: Gal Schlezinger <2054772+Schniz@users.noreply.github.com>
87 lines
2.2 KiB
TypeScript
87 lines
2.2 KiB
TypeScript
import type { IncomingMessage } from 'http'
|
|
import { Readable } from 'stream'
|
|
import { TransformStream } from 'next/dist/compiled/web-streams-polyfill'
|
|
|
|
type BodyStream = ReadableStream<Uint8Array>
|
|
|
|
/**
|
|
* Creates a ReadableStream from a Node.js HTTP request
|
|
*/
|
|
function requestToBodyStream(request: IncomingMessage): BodyStream {
|
|
const transform = new TransformStream<Uint8Array, Uint8Array>({
|
|
start(controller) {
|
|
request.on('data', (chunk) => controller.enqueue(chunk))
|
|
request.on('end', () => controller.terminate())
|
|
request.on('error', (err) => controller.error(err))
|
|
},
|
|
})
|
|
|
|
return transform.readable as unknown as ReadableStream<Uint8Array>
|
|
}
|
|
|
|
function bodyStreamToNodeStream(bodyStream: BodyStream): Readable {
|
|
const reader = bodyStream.getReader()
|
|
return Readable.from(
|
|
(async function* () {
|
|
while (true) {
|
|
const { done, value } = await reader.read()
|
|
if (done) {
|
|
return
|
|
}
|
|
yield value
|
|
}
|
|
})()
|
|
)
|
|
}
|
|
|
|
function replaceRequestBody<T extends IncomingMessage>(
|
|
base: T,
|
|
stream: Readable
|
|
): T {
|
|
for (const key in stream) {
|
|
let v = stream[key as keyof Readable] as any
|
|
if (typeof v === 'function') {
|
|
v = v.bind(stream)
|
|
}
|
|
base[key as keyof T] = v
|
|
}
|
|
|
|
return base
|
|
}
|
|
|
|
/**
|
|
* An interface that encapsulates body stream cloning
|
|
* of an incoming request.
|
|
*/
|
|
export function clonableBodyForRequest<T extends IncomingMessage>(
|
|
incomingMessage: T
|
|
) {
|
|
let bufferedBodyStream: BodyStream | null = null
|
|
|
|
return {
|
|
/**
|
|
* Replaces the original request body if necessary.
|
|
* This is done because once we read the body from the original request,
|
|
* we can't read it again.
|
|
*/
|
|
finalize(): void {
|
|
if (bufferedBodyStream) {
|
|
replaceRequestBody(
|
|
incomingMessage,
|
|
bodyStreamToNodeStream(bufferedBodyStream)
|
|
)
|
|
}
|
|
},
|
|
/**
|
|
* Clones the body stream
|
|
* to pass into a middleware
|
|
*/
|
|
cloneBodyStream(): BodyStream {
|
|
const originalStream =
|
|
bufferedBodyStream ?? requestToBodyStream(incomingMessage)
|
|
const [stream1, stream2] = originalStream.tee()
|
|
bufferedBodyStream = stream1
|
|
return stream2
|
|
},
|
|
}
|
|
}
|