fix(stream): try to forward headers

adonisjs · Oct 20, 2023 · 415a888 · 415a888
1 parent 4615efb
commit 415a888
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/stream.ts b/src/stream.ts
@@ -52,10 +52,15 @@ export class Stream extends Transform {
     return this.#uid
   }
 
-  pipe<T extends HeaderStream>(destination: T, options?: { end?: boolean }): T {
+  pipe<T extends HeaderStream>(
+    destination: T,
+    options?: { end?: boolean },
+    headers?: Record<string, any>
+  ): T {
     if (destination.writeHead) {
       // @see https://github.com/dunglas/mercure/blob/9e080c8dc9a141d4294412d14efdecfb15bf7f43/subscribe.go#L219
       destination.writeHead(200, {
+        ...headers,
         'Cache-Control': 'private, no-cache, no-store, must-revalidate, max-age=0, no-transform',
         'Connection': 'keep-alive',
         'Content-Type': 'text/event-stream',

diff --git a/src/transmit.ts b/src/transmit.ts
@@ -71,7 +71,7 @@ export class Transmit extends Emittery<TransmitLifecycleHooks> {
    */
   $createStream(request: Request, response: Response): void {
     const stream = new Stream(request.input('uid'), request.request)
-    stream.pipe(response.response)
+    stream.pipe(response.response, undefined, response.getHeaders())
 
     void this.emit('connect', { uid: stream.getUid() })