From db941f147278f64b473653b2691242b257eba93e Mon Sep 17 00:00:00 2001
From: Adhiraj Singh <adhirajsingh1001@gmail.com>
Date: Sun, 19 Dec 2021 22:25:17 +0530
Subject: [PATCH] fix: download on axios

---
 src/Utils/messages-media.ts | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/src/Utils/messages-media.ts b/src/Utils/messages-media.ts
index 03d7bbb..eed5158 100644
--- a/src/Utils/messages-media.ts
+++ b/src/Utils/messages-media.ts
@@ -324,19 +324,18 @@ export const downloadContentFromMessage = async(
             firstBlockIsIV = true
         }
     }
-    const endChunk = endByte ? toSmallestChunkSize(endByte || 0)+AES_CHUNK_SIZE : undefined
-    let rangeHeader: string | undefined = undefined
-    if(startChunk || endChunk) {
-        rangeHeader = `bytes=${startChunk}-`
-        if(endChunk) rangeHeader += endChunk
+    const endChunk = endByte ? toSmallestChunkSize(endByte || 0)+AES_CHUNK_SIZE : undefined    
+
+    const headers: { [_: string]: string } = {
+        Origin: DEFAULT_ORIGIN,
     }
+    if(startChunk || endChunk) {
+        headers.Range = `bytes=${startChunk}-`
+        if(endChunk) headers.Range += endChunk
+    }
+
     // download the message
-    const fetched = await getHttpStream(downloadUrl, {
-        headers: { 
-            Origin: DEFAULT_ORIGIN,
-            Range: rangeHeader
-        }
-    })
+    const fetched = await getHttpStream(downloadUrl, { headers })
 
     let remainingBytes = Buffer.from([])
     const { cipherKey, iv } = getMediaKeys(mediaKey, type)