fix: no Brotli decompression on an empty HEAD response (#150)

Closes #149
apify · Nov 20, 2024 · 8d9808a · 8d9808a
1 parent eea94e5
commit 8d9808a
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "got-scraping",
-    "version": "4.0.7",
+    "version": "4.0.8",
     "description": "HTTP client made for scraping based on got.",
     "engines": {
         "node": ">=16"

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);
     }