franpcode/Baileys
1
0
Fork 0
mirror of https://github.com/FranP-code/Baileys.git synced 2025-10-13 00:32:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: allow downloading extendedTextMessage thumbs

Browse Source
This commit is contained in:
Adhiraj Singh
2022-09-15 15:13:54 +05:30
parent 3a891e071e
commit 561163c88d
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/Utils/messages.ts
View File

@@ -7,6 +7,7 @@ import { MEDIA_KEYS, URL_EXCLUDE_REGEX, URL_REGEX, WA_DEFAULT_EPHEMERAL } from '
import { import {
AnyMediaMessageContent, AnyMediaMessageContent,
AnyMessageContent, AnyMessageContent,
DownloadableMessage,
MediaGenerationOptions, MediaGenerationOptions,
MediaType, MediaType,
MessageContentGenerationOptions, MessageContentGenerationOptions,
@@ -687,13 +688,25 @@ export const downloadMediaMessage = async(
} }
const contentType = getContentType(mContent) const contentType = getContentType(mContent)
const mediaType = contentType?.replace('Message', '') as MediaType let mediaType = contentType?.replace('Message', '') as MediaType
const media = mContent[contentType!] const media = mContent[contentType!]
if(!media || typeof media !== 'object' || !('url' in media)) {
if(!media || typeof media !== 'object' || (!('url' in media) && !('thumbnailDirectPath' in media))) {
throw new Boom(`"${contentType}" message is not a media message`) throw new Boom(`"${contentType}" message is not a media message`)
} }
const stream = await downloadContentFromMessage(media, mediaType, options) let download: DownloadableMessage
if('thumbnailDirectPath' in media && !('url' in media)) {
download = {
directPath: media.thumbnailDirectPath,
mediaKey: media.mediaKey
}
mediaType = 'thumbnail-link'
} else {
download = media
}
const stream = await downloadContentFromMessage(download, mediaType, options)
if(type === 'buffer') { if(type === 'buffer') {
const bufferArray: Buffer[] = [] const bufferArray: Buffer[] = []
for await (const chunk of stream) { for await (const chunk of stream) {