From da53d521e198e7db832f003ac14ffcb91b53baf5 Mon Sep 17 00:00:00 2001
From: Adhiraj <adhirajsingh1001@gmail.com>
Date: Sat, 25 Jul 2020 17:22:04 +0530
Subject: [PATCH] Download media bug fix + switch to ES2020

---
 src/WAClient/Messages.ts | 9 ++++++---
 tsconfig.json            | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/WAClient/Messages.ts b/src/WAClient/Messages.ts
index 38ceba6..95e0a21 100644
--- a/src/WAClient/Messages.ts
+++ b/src/WAClient/Messages.ts
@@ -17,7 +17,7 @@ import {
     WAUrlInfo,
 } from './Constants'
 import { generateMessageID, sha256, hmacSign, aesEncrypWithIV, randomBytes } from '../WAConnection/Utils'
-import { WAMessageContent, WAMetric, WAFlag, WANode, WAMessage, WAMessageProto, BaileysError } from '../WAConnection/Constants'
+import { WAMessageContent, WAMetric, WAFlag, WANode, WAMessage, WAMessageProto, BaileysError, MessageLogLevel } from '../WAConnection/Constants'
 import { validateJIDForSending, generateThumbnail, getMediaKeys, decodeMediaMessageBuffer, extensionForMediaMessage } from './Utils'
 import { proto } from '../../WAMessage/WAMessage'
 
@@ -342,11 +342,14 @@ export default class WhatsAppWebMessages extends WhatsAppWebGroups {
      */
     async downloadMediaMessage (message: WAMessage) {
         try {
-            return decodeMediaMessageBuffer (message.message)
+            const buff = await decodeMediaMessageBuffer (message.message)
+            return buff
         } catch (error) {
             if (error instanceof BaileysError && error.status === 404) { // media needs to be updated
+                this.log (`updating media of message: ${message.key.id}`, MessageLogLevel.info)
                 await this.updateMediaMessage (message)
-                return decodeMediaMessageBuffer (message.message)
+                const buff = await decodeMediaMessageBuffer (message.message)
+                return buff
             }
             throw error
         }
diff --git a/tsconfig.json b/tsconfig.json
index 1c05a6c..af8c916 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -1,6 +1,6 @@
 {
   "compilerOptions": {
-    "target": "es5",
+    "target": "es2020",
     "module": "commonjs",
     "allowJs": true,
     "checkJs": false,