fix: no Brotli decompression on empty HEAD response

apify · Nov 20, 2024 · 1c4772d · 1c4772d
1 parent eea94e5
commit 1c4772d
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/hooks/fix-decompress.ts b/src/hooks/fix-decompress.ts
@@ -56,7 +56,22 @@ const onResponse = (response: IncomingMessage, propagate: (fixedResponse: Incomi
             }
         });
     } else if (encoding === 'br') {
-        useDecompressor(zlib.createBrotliDecompress());
+        let read = false;
+
+        response.once('data', (chunk: Buffer) => {
+            read = true;
+
+            response.unshift(chunk);
+
+            const decompressor = zlib.createBrotliDecompress();
+            useDecompressor(decompressor);
+        });
+
+        response.once('end', () => {
+            if (!read) {
+                propagate(response);
+            }
+        });
     } else {
         propagate(response);
     }