fix download

Nurutomo · Oct 9, 2022 · 91c2ee9 · 91c2ee9
1 parent 24b3aff
commit 91c2ee9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git 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
@@ -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
 }