From 3a2926225616f4e1fead3fa98d5b5d99cad530e0 Mon Sep 17 00:00:00 2001
From: nsankbeil <nsankbeil@gmail.com>
Date: Mon, 5 Jun 2023 21:43:49 -0400
Subject: [PATCH] feat: support compressed responses

---
 src/utils/proxy/http.js | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/utils/proxy/http.js b/src/utils/proxy/http.js
index e022fb462..f869cd0bd 100644
--- a/src/utils/proxy/http.js
+++ b/src/utils/proxy/http.js
@@ -1,5 +1,7 @@
 /* eslint-disable prefer-promise-reject-errors */
 /* eslint-disable no-param-reassign */
+import { createUnzip } from "node:zlib";
+
 import { http, https } from "follow-redirects";
 
 import { addCookieToJar, setCookieHeader } from "./cookie-jar";
@@ -28,12 +30,19 @@ function handleRequest(requestor, url, params) {
 
     const request = requestor.request(url, params, (response) => {
       const data = [];
+      const contentEncoding = response.headers['content-encoding']?.trim().toLowerCase();
+
+      let responseContent = response;
+      if (contentEncoding === 'gzip' || contentEncoding === 'deflate') {
+        responseContent = createUnzip();
+        response.pipe(responseContent);
+      }
 
-      response.on("data", (chunk) => {
+      responseContent.on("data", (chunk) => {
         data.push(chunk);
       });
 
-      response.on("end", () => {
+      responseContent.on("end", () => {
         addCookieToJar(url, response.headers);
         resolve([response.statusCode, response.headers["content-type"], Buffer.concat(data), response.headers]);
       });