From 91c2ee944afdfe6d2cc051bc5983d53f14743a50 Mon Sep 17 00:00:00 2001
From: Nurutomo <tixbig@gmail.com>
Date: Sun, 9 Oct 2022 09:26:47 +0700
Subject: [PATCH] fix download

---
 src/util/Connection.ts    |  1 +
 src/util/MessageParser.ts | 11 +++++++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/util/Connection.ts b/src/util/Connection.ts
index 4d019ef..5751f99 100644
--- a/src/util/Connection.ts
+++ b/src/util/Connection.ts
@@ -12,6 +12,7 @@ import PermissionHandler from './PermissionHandler'
 const Logger = pino({ transport: { target: 'pino-pretty' }, prettyPrint: { levelFirst: true, ignore: 'hostname', translateTime: true } })
 export default class Connection {
     static isModule: boolean = true
+    static Logger = Logger
     
     sock: ReturnType<typeof Helper>
     store: ReturnType<typeof makeInMemoryStore>
diff --git a/src/util/MessageParser.ts b/src/util/MessageParser.ts
index bc93245..21ca681 100644
--- a/src/util/MessageParser.ts
+++ b/src/util/MessageParser.ts
@@ -1,5 +1,6 @@
 import { downloadMediaMessage, extractMessageContent, getContentType, getDevice, WAMessage, WAProto, areJidsSameUser, isJidGroup } from '@adiwajshing/baileys'
 import { ParsedMessage, ParserOptions, AnyWASocket } from '../types'
+import Connection from './Connection'
 
 function MessageParser(conn: AnyWASocket, m: WAMessage, options: ParserOptions = {}): ParsedMessage {
     const {
@@ -74,7 +75,10 @@ function MessageParser(conn: AnyWASocket, m: WAMessage, options: ParserOptions =
                             })
                         }
                         parsed.quoted.download = (type = 'buffer', options) => {
-                            return downloadMediaMessage(parsed.quoted.fakeObj, type, options)
+                            return downloadMediaMessage(parsed.quoted.fakeObj, type, options, {
+                                logger: Connection.Logger.child({ class: 'Download' }),
+                                reuploadRequest: conn.updateMediaMessage
+                            })
                         }
                     }
                 }
@@ -95,7 +99,10 @@ function MessageParser(conn: AnyWASocket, m: WAMessage, options: ParserOptions =
         })
     }
     parsed.download = (type = 'buffer', options) => {
-        return downloadMediaMessage(m, type, options)
+        return downloadMediaMessage(m, type, options, {
+            logger: Connection.Logger.child({ class: 'Download' }),
+            reuploadRequest: conn.updateMediaMessage
+        })
     }
     return parsed
 }