@hansaka02/baileys 7.3.2 → 7.3.4

package/lib/Utils/message-retry-manager.js

@@ -2,7 +2,7 @@
 
 Object.defineProperty(exports, "__esModule", { value: true })
 
-const LRUCache_1 = require("lru-cache")
+const { LRUCache } = require("lru-cache")
 
 /** Number of sent messages to cache in memory for handling retry receipts */
 const RECENT_MESSAGES_SIZE = 512
@@ -15,14 +15,14 @@ const PHONE_REQUEST_DELAY = 3000
 class MessageRetryManager {
     constructor(logger, maxMsgRetryCount) {
         this.logger = logger
-        this.recentMessagesMap = new LRUCache_1.LRUCache({
+        this.recentMessagesMap = new LRUCache({
             max: RECENT_MESSAGES_SIZE
         })
-        this.sessionRecreateHistory = new LRUCache_1.LRUCache({
+        this.sessionRecreateHistory = new LRUCache({
             ttl: RECREATE_SESSION_TIMEOUT * 2,
             ttlAutopurge: true
         })
-        this.retryCounters = new LRUCache_1.LRUCache({
+        this.retryCounters = new LRUCache({
             ttl: 15 * 60 * 1000,
             ttlAutopurge: true,
             updateAgeOnGet: true

package/lib/Utils/messages-media.js

@@ -1,51 +1,46 @@
 "use strict"
 
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k
-    var desc = Object.getOwnPropertyDescriptor(m, k)
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k] } }
-    }
-    Object.defineProperty(o, k2, desc)
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k
-    o[k2] = m[k]
-}))
-
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v })
-}) : function(o, v) {
-    o["default"] = v
-})
-
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod
-    var result = {}
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k)
-    __setModuleDefault(result, mod)
-    return result
-}
-
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod }
-}
-
 Object.defineProperty(exports, "__esModule", { value: true })
 
-const boom_1 = require("@hapi/boom")
-const axios_1 = __importDefault(require("axios"))
-const child_process_1 = require("child_process")
-const Crypto = __importStar(require("crypto"))
-const events_1 = require("events")
-const fs_1 = require("fs")
-const os_1 = require("os")
-const path_1 = require("path")
-const stream_1 = require("stream")
-const WAProto_1 = require("../../WAProto")
-const Defaults_1 = require("../Defaults")
-const WABinary_1 = require("../WABinary")
-const crypto_1 = require("./crypto")
-const generics_1 = require("./generics")
+const { Boom } = require("@hapi/boom")
+const { spawn } = require("child_process")
+const { once } = require("events")
+const { 
+  createHash, 
+  randomBytes, 
+  createHmac, 
+  createCipheriv, 
+  createDecipheriv
+} = require("crypto")
+const {
+  promises, 
+  createReadStream, 
+  createWriteStream
+} = require("fs")
+const { tmpdir } = require("os")
+const { join } = require("path")
+const {
+  Readable, 
+  Transform
+} = require("stream")
+const axios_1 =  require("axios")
+const { proto } = require("../../WAProto")
+const {
+  MEDIA_PATH_MAP, 
+  MEDIA_HKDF_KEY_MAPPING
+} = require("../Defaults/media")
+const { DEFAULT_ORIGIN } = require("../Defaults/constants")
+const { 
+  getBinaryNodeChild,
+  getBinaryNodeChildBuffer, 
+  jidNormalizedUser 
+} = require("../WABinary")
+const {
+  aesDecryptGCM, 
+  aesEncryptGCM,
+  hkdf 
+} = require("./crypto")
+const { generateMessageID } = require("./generics")
 
 const getImageProcessingLibrary = async () => {
     const [_jimp, sharp] = await Promise.all([
@@ -65,11 +60,11 @@ const getImageProcessingLibrary = async () => {
     if (jimp) {
         return { jimp }
     }
-    throw new boom_1.Boom('No image processing library available')
+    throw new Boom('No image processing library available')
 }
 
 const hkdfInfoKey = (type) => {
-    const hkdfInfo = Defaults_1.MEDIA_HKDF_KEY_MAPPING[type]
+    const hkdfInfo = MEDIA_HKDF_KEY_MAPPING[type]
     return `WhatsApp ${hkdfInfo} Keys`
 }
 
@@ -78,9 +73,9 @@ const getRawMediaUploadData = async (media, mediaType, logger) => {
     
     logger?.debug('got stream for raw upload')
     
-    const hasher = Crypto.createHash('sha256')
-    const filePath = path_1.join(os_1.tmpdir(), mediaType + generics_1.generateMessageID())
-    const fileWriteStream = fs_1.createWriteStream(filePath)
+    const hasher = createHash('sha256')
+    const filePath = join(tmpdir(), mediaType + generateMessageID())
+    const fileWriteStream = createWriteStream(filePath)
     
     let fileLength = 0
     
@@ -90,12 +85,12 @@ const getRawMediaUploadData = async (media, mediaType, logger) => {
             hasher.update(data)
             
             if (!fileWriteStream.write(data)) {
-                await events_1.once(fileWriteStream, 'drain')
+                await once(fileWriteStream, 'drain')
             }
         }
         
         fileWriteStream.end()
-        await events_1.once(fileWriteStream, 'finish')
+        await once(fileWriteStream, 'finish')
         stream.destroy()
         
         const fileSha256 = hasher.digest()
@@ -113,7 +108,7 @@ const getRawMediaUploadData = async (media, mediaType, logger) => {
         stream.destroy()
         
         try {
-            await fs_1.promises.unlink(filePath)
+            await promises.unlink(filePath)
         }
         catch {
             //
@@ -125,13 +120,13 @@ const getRawMediaUploadData = async (media, mediaType, logger) => {
 /** generates all the keys required to encrypt/decrypt & sign a media message */
 async function getMediaKeys(buffer, mediaType) {
     if (!buffer) {
-        throw new boom_1.Boom('Cannot derive from empty media key')
+        throw new Boom('Cannot derive from empty media key')
     }
     if (typeof buffer === 'string') {
         buffer = Buffer.from(buffer.replace('data:base64,', ''), 'base64')
     }
     // expand using HKDF to 112 bytes, also pass in the relevant app info
-    const expandedMediaKey = await crypto_1.hkdf(buffer, 112, { info: hkdfInfoKey(mediaType) })
+    const expandedMediaKey = await hkdf(buffer, 112, { info: hkdfInfoKey(mediaType) })
     return {
         iv: expandedMediaKey.slice(0, 16),
         cipherKey: expandedMediaKey.slice(16, 48),
@@ -153,7 +148,7 @@ const extractVideoThumb = (videoPath, time = '00:00:00', size = { width: 320 })
             'pipe:1'
         ]
 
-        const ffmpeg = child_process_1.spawn('ffmpeg', args)
+        const ffmpeg = spawn('ffmpeg', args)
         const chunks = []
         let errorOutput = ''
 
@@ -174,7 +169,7 @@ const extractImageThumb = async (bufferOrFilePath, width = 32, quality = 50) =>
         const response = await axios_1.default.get(bufferOrFilePath, { responseType: "arraybuffer" })
         bufferOrFilePath = Buffer.from(response.data)
     }
-    if (bufferOrFilePath instanceof stream_1.Readable) {
+    if (bufferOrFilePath instanceof Readable) {
         bufferOrFilePath = await toBuffer(bufferOrFilePath)
     }
     const lib = await getImageProcessingLibrary()
@@ -215,7 +210,7 @@ const extractImageThumb = async (bufferOrFilePath, width = 32, quality = 50) =>
         }
     }
     else {
-        throw new boom_1.Boom('No image processing library available')
+        throw new Boom('No image processing library available')
     }
 }
 
@@ -254,7 +249,7 @@ const generateProfilePicture = async (mediaUpload) => {
         img = await cropped.scaleToFit(720, 720).getBufferAsync(MIME_JPEG)
     }
     else {
-        throw new boom_1.Boom('No image processing library available')
+        throw new Boom('No image processing library available')
     }
     return {
         img: await img,
@@ -299,7 +294,7 @@ async function getAudioWaveform(buffer, logger) {
             audioData = buffer
         }
         else if (typeof buffer === 'string') {
-            const rStream = fs_1.createReadStream(buffer)
+            const rStream = createReadStream(buffer)
             audioData = await toBuffer(rStream)
         }
         else {
@@ -331,7 +326,7 @@ async function getAudioWaveform(buffer, logger) {
 }
 
 const toReadable = (buffer) => {
-    const readable = new stream_1.Readable({ read: () => { } })
+    const readable = new Readable({ read: () => { } })
     readable.push(buffer)
     readable.push(null)
     return readable
@@ -366,7 +361,7 @@ const getStream = async (item, opts) => {
         return { stream: await getHttpStream(item.url, opts), type: 'remote' }
     }
     
-    return { stream: fs_1.createReadStream(item.url), type: 'file' }
+    return { stream: createReadStream(item.url), type: 'file' }
 }
 
 /** generates a thumbnail for a given media, if required */
@@ -407,8 +402,8 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
     const { stream, type } = await getStream(media, opts)
     logger?.debug('fetched media stream')
 
-    const encFilePath = path_1.join(os_1.tmpdir(), mediaType + generics_1.generateMessageID() + '-plain')
-    const encFileWriteStream = fs_1.createWriteStream(encFilePath)
+    const encFilePath = join(tmpdir(), mediaType + generateMessageID() + '-plain')
+    const encFileWriteStream = createWriteStream(encFilePath)
 
     let originalFilePath
     let originalFileStream
@@ -416,12 +411,12 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
     if (type === 'file') {
         originalFilePath = media.url.toString()
     } else if (saveOriginalFileIfRequired) {
-        originalFilePath = path_1.join(os_1.tmpdir(), mediaType + generics_1.generateMessageID() + '-original')
-        originalFileStream = fs_1.createWriteStream(originalFilePath)
+        originalFilePath = join(tmpdir(), mediaType + generateMessageID() + '-original')
+        originalFileStream = createWriteStream(originalFilePath)
     }
 
     let fileLength = 0
-    const sha256 = Crypto.createHash('sha256')
+    const sha256 = createHash('sha256')
 
     try {
         for await (const data of stream) {
@@ -430,7 +425,7 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
             if (type === 'remote'
                 && opts?.maxContentLength
                 && fileLength + data.length > opts.maxContentLength) {
-                throw new boom_1.Boom(`content length exceeded when preparing "${type}"`, {
+                throw new Boom(`content length exceeded when preparing "${type}"`, {
                     data: { media, type }
                 })
             }
@@ -439,7 +434,7 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
             encFileWriteStream.write(data)
 
             if (originalFileStream && !originalFileStream.write(data)) {
-                await events_1.once(originalFileStream, 'drain')
+                await once(originalFileStream, 'drain')
             }
         }
 
@@ -466,9 +461,9 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
         sha256.destroy()
         stream.destroy()
         try {
-            await fs_1.promises.unlink(encFilePath)
+            await promises.unlink(encFilePath)
             if (originalFilePath && didSaveToTmpPath) {
-                await fs_1.promises.unlink(originalFilePath)
+                await promises.unlink(originalFilePath)
             }
         } catch (err) {
             logger?.error({ err }, 'failed deleting tmp files')
@@ -480,21 +475,21 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
 const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts } = {}) => {
     const { stream, type } = await getStream(media, opts)
     logger?.debug('fetched media stream')
-    const mediaKey = Crypto.randomBytes(32)
+    const mediaKey = randomBytes(32)
     const { cipherKey, iv, macKey } = await getMediaKeys(mediaKey, mediaType)
-    const encFilePath = path_1.join(os_1.tmpdir(), mediaType + generics_1.generateMessageID() + '-enc')
-    const encFileWriteStream = fs_1.createWriteStream(encFilePath)
+    const encFilePath = join(tmpdir(), mediaType + generateMessageID() + '-enc')
+    const encFileWriteStream = createWriteStream(encFilePath)
     let originalFileStream
     let originalFilePath
     if (saveOriginalFileIfRequired) {
-        originalFilePath = path_1.join(os_1.tmpdir(), mediaType + generics_1.generateMessageID() + '-original') 
-        originalFileStream = fs_1.createWriteStream(originalFilePath)
+        originalFilePath = join(tmpdir(), mediaType + generateMessageID() + '-original') 
+        originalFileStream = createWriteStream(originalFilePath)
     }
     let fileLength = 0
-    const aes = Crypto.createCipheriv('aes-256-cbc', cipherKey, iv)
-    const hmac = Crypto.createHmac('sha256', macKey).update(iv)
-    const sha256Plain = Crypto.createHash('sha256')
-    const sha256Enc = Crypto.createHash('sha256')
+    const aes = createCipheriv('aes-256-cbc', cipherKey, iv)
+    const hmac = createHmac('sha256', macKey).update(iv)
+    const sha256Plain = createHash('sha256')
+    const sha256Enc = createHash('sha256')
     const onChunk = (buff) => {
         sha256Enc.update(buff)
         hmac.update(buff)
@@ -506,13 +501,13 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
             if (type === 'remote'
                 && opts?.maxContentLength
                 && fileLength + data.length > opts.maxContentLength) {
-                throw new boom_1.Boom(`content length exceeded when encrypting "${type}"`, {
+                throw new Boom(`content length exceeded when encrypting "${type}"`, {
                     data: { media, type }
                 })
             }
             if (originalFileStream) {
                 if (!originalFileStream.write(data)) {
-                    await events_1.once(originalFileStream, 'drain')
+                    await once(originalFileStream, 'drain')
                 }
             }
             sha256Plain.update(data)
@@ -548,9 +543,9 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
         sha256Enc.destroy()
         stream.destroy()
         try {
-            await fs_1.promises.unlink(encFilePath)
+            await promises.unlink(encFilePath)
             if (originalFilePath) {
-                await fs_1.promises.unlink(originalFilePath)
+                await promises.unlink(originalFilePath)
             }
         }
         catch (err) {
@@ -574,7 +569,7 @@ const downloadContentFromMessage = async ({ mediaKey, directPath, url }, type, o
     const downloadUrl = isValidMediaUrl ? url : getUrlFromDirectPath(directPath)
     
     if (!downloadUrl) {
-    	throw new boom_1.Boom('No valid media URL or directPath present in message', { statusCode: 400 }) 
+    	throw new Boom('No valid media URL or directPath present in message', { statusCode: 400 }) 
     }
     
     const keys = await getMediaKeys(mediaKey, type)
@@ -601,7 +596,7 @@ const downloadEncryptedContent = async (downloadUrl, { cipherKey, iv }, { startB
     const endChunk = endByte ? toSmallestChunkSize(endByte || 0) + AES_CHUNK_SIZE : undefined
     const headers = {
         ...(options?.headers) || {},
-        Origin: Defaults_1.DEFAULT_ORIGIN,
+        Origin: DEFAULT_ORIGIN
     }
     if (startChunk || endChunk) {
         headers.Range = `bytes=${startChunk}-`
@@ -629,7 +624,7 @@ const downloadEncryptedContent = async (downloadUrl, { cipherKey, iv }, { startB
             push(bytes)
         }
     }
-    const output = new stream_1.Transform({
+    const output = new Transform({
         transform(chunk, _, callback) {
             let data = Buffer.concat([remainingBytes, chunk])
             const decryptLength = toSmallestChunkSize(data.length)
@@ -641,7 +636,7 @@ const downloadEncryptedContent = async (downloadUrl, { cipherKey, iv }, { startB
                     ivValue = data.slice(0, AES_CHUNK_SIZE)
                     data = data.slice(AES_CHUNK_SIZE)
                 }
-                aes = Crypto.createDecipheriv('aes-256-cbc', cipherKey, ivValue)
+                aes = createDecipheriv('aes-256-cbc', cipherKey, ivValue)
                 // if an end byte that is not EOF is specified
                 // stop auto padding (PKCS7) -- otherwise throws an error for decryption
                 if (endByte) {
@@ -690,7 +685,7 @@ const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options },
         // send a query JSON to obtain the url & auth token to upload our media
         let uploadInfo = await refreshMediaConn(false)
         let urls
-        let media = Defaults_1.MEDIA_PATH_MAP[mediaType]
+        let media = MEDIA_PATH_MAP[mediaType]
         const hosts = [...customUploadHosts, ...uploadInfo.hosts]
         fileEncSha256B64 = encodeBase64EncodedStringForUpload(fileEncSha256B64)
         if (newsletter) {
@@ -703,13 +698,13 @@ const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options },
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             let result
             try {
-                const body = await axios_1.default.post(url, fs_1.createReadStream(filePath), {
+                const body = await axios_1.default.post(url, createReadStream(filePath), {
                     ...options,
                     maxRedirects: 0,
                     headers: {
                         ...options.headers || {},
                         'Content-Type': 'application/octet-stream',
-                        'Origin': Defaults_1.DEFAULT_ORIGIN
+                        'Origin': DEFAULT_ORIGIN
                     },
                     httpsAgent: fetchAgent,
                     timeout: timeoutMs,
@@ -742,29 +737,29 @@ const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options },
             }
         }
         if (!urls) {
-            throw new boom_1.Boom('Media upload failed on all hosts', { statusCode: 500 })
+            throw new Boom('Media upload failed on all hosts', { statusCode: 500 })
         }
         return urls
     }
 }
 
 const getMediaRetryKey = (mediaKey) => {
-    return crypto_1.hkdf(mediaKey, 32, { info: 'WhatsApp Media Retry Notification' })
+    return hkdf(mediaKey, 32, { info: 'WhatsApp Media Retry Notification' })
 }
 /**
  * Generate a binary node that will request the phone to re-upload the media & return the newly uploaded URL
  */
 const encryptMediaRetryRequest = async (key, mediaKey, meId) => {
     const recp = { stanzaId: key.id }
-    const recpBuffer = WAProto_1.proto.ServerErrorReceipt.encode(recp).finish()
-    const iv = Crypto.randomBytes(12)
+    const recpBuffer = proto.ServerErrorReceipt.encode(recp).finish()
+    const iv = randomBytes(12)
     const retryKey = await getMediaRetryKey(mediaKey)
-    const ciphertext = crypto_1.aesEncryptGCM(recpBuffer, retryKey, iv, Buffer.from(key.id))
+    const ciphertext = aesEncryptGCM(recpBuffer, retryKey, iv, Buffer.from(key.id))
     const req = {
         tag: 'receipt',
         attrs: {
             id: key.id,
-            to: WABinary_1.jidNormalizedUser(meId),
+            to: jidNormalizedUser(meId),
             type: 'server-error'
         },
         content: [
@@ -794,7 +789,7 @@ const encryptMediaRetryRequest = async (key, mediaKey, meId) => {
 }
 
 const decodeMediaRetryNode = (node) => {
-    const rmrNode = WABinary_1.getBinaryNodeChild(node, 'rmr')
+    const rmrNode = getBinaryNodeChild(node, 'rmr')
     const event = {
         key: {
             id: node.attrs.id,
@@ -803,20 +798,20 @@ const decodeMediaRetryNode = (node) => {
             participant: rmrNode.attrs.participant
         }
     }
-    const errorNode = WABinary_1.getBinaryNodeChild(node, 'error')
+    const errorNode = getBinaryNodeChild(node, 'error')
     if (errorNode) {
         const errorCode = +errorNode.attrs.code
-        event.error = new boom_1.Boom(`Failed to re-upload media (${errorCode})`, { data: errorNode.attrs, statusCode: getStatusCodeForMediaRetry(errorCode) })
+        event.error = new Boom(`Failed to re-upload media (${errorCode})`, { data: errorNode.attrs, statusCode: getStatusCodeForMediaRetry(errorCode) })
     }
     else {
-        const encryptedInfoNode = WABinary_1.getBinaryNodeChild(node, 'encrypt')
-        const ciphertext = WABinary_1.getBinaryNodeChildBuffer(encryptedInfoNode, 'enc_p')
-        const iv = WABinary_1.getBinaryNodeChildBuffer(encryptedInfoNode, 'enc_iv')
+        const encryptedInfoNode = getBinaryNodeChild(node, 'encrypt')
+        const ciphertext = getBinaryNodeChildBuffer(encryptedInfoNode, 'enc_p')
+        const iv = getBinaryNodeChildBuffer(encryptedInfoNode, 'enc_iv')
         if (ciphertext && iv) {
             event.media = { ciphertext, iv }
         }
         else {
-            event.error = new boom_1.Boom('Failed to re-upload media (missing ciphertext)', { statusCode: 404 })
+            event.error = new Boom('Failed to re-upload media (missing ciphertext)', { statusCode: 404 })
         }
     }
     return event
@@ -824,17 +819,17 @@ const decodeMediaRetryNode = (node) => {
 
 const decryptMediaRetryData = async ({ ciphertext, iv }, mediaKey, msgId) => {
     const retryKey = await getMediaRetryKey(mediaKey)
-    const plaintext = crypto_1.aesDecryptGCM(ciphertext, retryKey, iv, Buffer.from(msgId))
-    return WAProto_1.proto.MediaRetryNotification.decode(plaintext)
+    const plaintext = aesDecryptGCM(ciphertext, retryKey, iv, Buffer.from(msgId))
+    return proto.MediaRetryNotification.decode(plaintext)
 }
 
 const getStatusCodeForMediaRetry = (code) => MEDIA_RETRY_STATUS_MAP[code]
 
 const MEDIA_RETRY_STATUS_MAP = {
-    [WAProto_1.proto.MediaRetryNotification.ResultType.SUCCESS]: 200,
-    [WAProto_1.proto.MediaRetryNotification.ResultType.DECRYPTION_ERROR]: 412,
-    [WAProto_1.proto.MediaRetryNotification.ResultType.NOT_FOUND]: 404,
-    [WAProto_1.proto.MediaRetryNotification.ResultType.GENERAL_ERROR]: 418,
+    [proto.MediaRetryNotification.ResultType.SUCCESS]: 200,
+    [proto.MediaRetryNotification.ResultType.DECRYPTION_ERROR]: 412,
+    [proto.MediaRetryNotification.ResultType.NOT_FOUND]: 404,
+    [proto.MediaRetryNotification.ResultType.GENERAL_ERROR]: 418,
 }
 
 module.exports = {