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: map all WA media types

Browse Source
This commit is contained in:
Adhiraj Singh
2022-09-15 15:08:49 +05:30
parent b05ce25dfa
commit 3a891e071e
6 changed files with 32 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Utils/business.ts
View File

@@ -239,7 +239,7 @@ export const uploadingNecessaryImages = async(
const { directPath } = await waUploadToServer(
toReadable(Buffer.concat(contentBlocks)),
{
mediaType: 'product-image',
mediaType: 'product-catalog-image',
fileEncSha256B64: sha,
timeoutMs
}

2
src/Utils/history.ts
View File

@@ -10,7 +10,7 @@ import { downloadContentFromMessage } from './messages-media'
const inflatePromise = promisify(inflate)
export const downloadHistory = async(msg: proto.Message.IHistorySyncNotification) => {
const stream = await downloadContentFromMessage(msg, 'history')
const stream = await downloadContentFromMessage(msg, 'md-msg-hist')
const bufferArray: Buffer[] = []
for await (const chunk of stream) {
bufferArray.push(chunk)

13
src/Utils/messages-media.ts
View File

@@ -11,7 +11,7 @@ import type { Logger } from 'pino'
import { Readable, Transform } from 'stream'
import { URL } from 'url'
import { proto } from '../../WAProto'
import { DEFAULT_ORIGIN, MEDIA_PATH_MAP } from '../Defaults'
import { DEFAULT_ORIGIN, MEDIA_HKDF_KEY_MAPPING, MEDIA_PATH_MAP } from '../Defaults'
import { BaileysEventMap, CommonSocketConfig, DownloadableMessage, MediaConnInfo, MediaDecryptionKeyInfo, MediaType, MessageType, WAGenericMediaMessage, WAMediaUpload, WAMediaUploadFunction, WAMessageContent } from '../Types'
import { BinaryNode, getBinaryNodeChild, getBinaryNodeChildBuffer, jidNormalizedUser } from '../WABinary'
import { aesDecryptGCM, aesEncryptGCM, hkdf } from './crypto'
@@ -50,16 +50,7 @@ const getImageProcessingLibrary = async() => {
}
export const hkdfInfoKey = (type: MediaType) => {
let str: string = type
if(type === 'sticker') {
str = 'image'
}
if(type === 'md-app-state') {
str = 'App State'
}
const hkdfInfo = str[0].toUpperCase() + str.slice(1)
const hkdfInfo = MEDIA_HKDF_KEY_MAPPING[type]
return `WhatsApp ${hkdfInfo} Keys`
}

6
src/Utils/messages.ts
View File

@@ -37,15 +37,13 @@ type MediaUploadData = {
mimetype?: string
}
const MIMETYPE_MAP: { [T in MediaType]: string } = {
const MIMETYPE_MAP: { [T in MediaType]?: string } = {
image: 'image/jpeg',
video: 'video/mp4',
document: 'application/pdf',
audio: 'audio/ogg; codecs=opus',
sticker: 'image/webp',
history: 'application/x-protobuf',
'product-image': 'image/jpeg',
'md-app-state': 'application/x-protobuf',
'product-catalog-image': 'image/jpeg',
}
const MessageTypeProto = {