@heavstaltech/baileys 2.3.4 → 3.2.4

package/lib/Utils/messages-media.d.ts

@@ -1,7 +1,10 @@
+/// <reference types="node" />
+/// <reference types="node" />
+/// <reference types="node" />
 import { Boom } from '@hapi/boom';
 import { AxiosRequestConfig } from 'axios';
 import type { Logger } from 'pino';
-import { Readable } from 'stream';
+import { Readable, Transform } from 'stream';
 import { URL } from 'url';
 import { proto } from '../../WAProto';
 import { DownloadableMessage, MediaConnInfo, MediaDecryptionKeyInfo, MediaType, SocketConfig, WAMediaUpload, WAMediaUploadFunction, WAMessageContent } from '../Types';
@@ -10,10 +13,10 @@ export declare const hkdfInfoKey: (type: MediaType) => string;
 /** generates all the keys required to encrypt/decrypt & sign a media message */
 export declare function getMediaKeys(buffer: Uint8Array | string | null | undefined, mediaType: MediaType): MediaDecryptionKeyInfo;
 export declare const extractImageThumb: (bufferOrFilePath: Readable | Buffer | string, width?: number) => Promise<{
-    buffer: any;
+    buffer: Buffer;
     original: {
-        width: any;
-        height: any;
+        width: number | undefined;
+        height: number | undefined;
     };
 }>;
 export declare const encodeBase64EncodedStringForUpload: (b64: string) => string;
@@ -21,25 +24,25 @@ export declare const generateProfilePicture: (mediaUpload: WAMediaUpload) => Pro
     img: Buffer;
 }>;
 /** gets the SHA256 of the given media message */
-export declare const mediaMessageSHA256B64: (message: WAMessageContent) => any;
-export declare function getAudioDuration(buffer: Buffer | string | Readable): Promise<any>;
+export declare const mediaMessageSHA256B64: (message: WAMessageContent) => string | null | undefined;
+export declare function getAudioDuration(buffer: Buffer | string | Readable): Promise<number | undefined>;
 /**
   referenced from and modifying https://github.com/wppconnect-team/wa-js/blob/main/src/chat/functions/prepareAudioWaveform.ts
  */
-export declare function getAudioWaveform(buffer: Buffer | string | Readable, logger?: Logger): Promise<Uint8Array<ArrayBuffer> | undefined>;
-export declare const toReadable: (buffer: Buffer) => any;
-export declare const toBuffer: (stream: Readable) => Promise<any>;
+export declare function getAudioWaveform(buffer: Buffer | string | Readable, logger?: Logger): Promise<Uint8Array | undefined>;
+export declare const toReadable: (buffer: Buffer) => Readable;
+export declare const toBuffer: (stream: Readable) => Promise<Buffer>;
 export declare const getStream: (item: WAMediaUpload, opts?: AxiosRequestConfig) => Promise<{
-    readonly stream: any;
+    readonly stream: Readable;
     readonly type: "buffer";
 } | {
-    readonly stream: any;
+    readonly stream: Readable;
     readonly type: "readable";
 } | {
     readonly stream: Readable;
     readonly type: "remote";
 } | {
-    readonly stream: any;
+    readonly stream: import("fs").ReadStream;
     readonly type: "file";
 }>;
 /** generates a thumbnail for a given media, if required */
@@ -62,20 +65,20 @@ type EncryptedStreamOptions = {
 };
 export declare const prepareStream: (media: WAMediaUpload, mediaType: MediaType, { logger, saveOriginalFileIfRequired, opts }?: EncryptedStreamOptions) => Promise<{
     mediaKey: undefined;
-    encWriteStream: any;
-    fileLength: any;
-    fileSha256: any;
+    encWriteStream: Buffer;
+    fileLength: number;
+    fileSha256: Buffer;
     fileEncSha256: undefined;
     bodyPath: string | undefined;
     didSaveToTmpPath: boolean;
 }>;
 export declare const encryptedStream: (media: WAMediaUpload, mediaType: MediaType, { logger, saveOriginalFileIfRequired, opts }?: EncryptedStreamOptions) => Promise<{
-    mediaKey: any;
-    encWriteStream: any;
+    mediaKey: Buffer;
+    encWriteStream: Readable;
     bodyPath: string | undefined;
-    mac: any;
-    fileEncSha256: any;
-    fileSha256: any;
+    mac: Buffer;
+    fileEncSha256: Buffer;
+    fileSha256: Buffer;
     fileLength: number;
     didSaveToTmpPath: boolean;
 }>;
@@ -85,12 +88,12 @@ export type MediaDownloadOptions = {
     options?: AxiosRequestConfig<any>;
 };
 export declare const getUrlFromDirectPath: (directPath: string) => string;
-export declare const downloadContentFromMessage: ({ mediaKey, directPath, url }: DownloadableMessage, type: MediaType, opts?: MediaDownloadOptions) => Promise<any>;
+export declare const downloadContentFromMessage: ({ mediaKey, directPath, url }: DownloadableMessage, type: MediaType, opts?: MediaDownloadOptions) => Promise<Transform>;
 /**
  * Decrypts and downloads an AES256-CBC encrypted file given the keys.
  * Assumes the SHA256 of the plaintext is appended to the end of the ciphertext
  * */
-export declare const downloadEncryptedContent: (downloadUrl: string, { cipherKey, iv }: MediaDecryptionKeyInfo, { startByte, endByte, options }?: MediaDownloadOptions) => Promise<any>;
+export declare const downloadEncryptedContent: (downloadUrl: string, { cipherKey, iv }: MediaDecryptionKeyInfo, { startByte, endByte, options }?: MediaDownloadOptions) => Promise<Transform>;
 export declare function extensionForMediaMessage(message: WAMessageContent): string;
 export declare const getWAUploadToServer: ({ customUploadHosts, fetchAgent, logger, options }: SocketConfig, refreshMediaConn: (force: boolean) => Promise<MediaConnInfo>) => WAMediaUploadFunction;
 /**
@@ -98,12 +101,12 @@ export declare const getWAUploadToServer: ({ customUploadHosts, fetchAgent, logg
  */
 export declare const encryptMediaRetryRequest: (key: proto.IMessageKey, mediaKey: Buffer | Uint8Array, meId: string) => BinaryNode;
 export declare const decodeMediaRetryNode: (node: BinaryNode) => {
-    key: import("../Types").WAMessageKey;
+    key: proto.IMessageKey;
     media?: {
         ciphertext: Uint8Array;
         iv: Uint8Array;
-    };
-    error?: Boom;
+    } | undefined;
+    error?: Boom<any> | undefined;
 };
 export declare const decryptMediaRetryData: ({ ciphertext, iv }: {
     ciphertext: Uint8Array;

package/lib/Utils/messages-media.js

@@ -15,30 +15,15 @@ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (
 }) : function(o, v) {
     o["default"] = v;
 });
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
+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;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getStatusCodeForMediaRetry = exports.decryptMediaRetryData = exports.decodeMediaRetryNode = exports.encryptMediaRetryRequest = exports.getWAUploadToServer = exports.downloadEncryptedContent = exports.downloadContentFromMessage = exports.getUrlFromDirectPath = exports.encryptedStream = exports.prepareStream = exports.getHttpStream = exports.getStream = exports.toBuffer = exports.toReadable = exports.mediaMessageSHA256B64 = exports.generateProfilePicture = exports.encodeBase64EncodedStringForUpload = exports.extractImageThumb = exports.hkdfInfoKey = void 0;
-exports.getMediaKeys = getMediaKeys;
-exports.getAudioDuration = getAudioDuration;
-exports.getAudioWaveform = getAudioWaveform;
-exports.generateThumbnail = generateThumbnail;
-exports.extensionForMediaMessage = extensionForMediaMessage;
+exports.getStatusCodeForMediaRetry = exports.decryptMediaRetryData = exports.decodeMediaRetryNode = exports.encryptMediaRetryRequest = exports.getWAUploadToServer = exports.extensionForMediaMessage = exports.downloadEncryptedContent = exports.downloadContentFromMessage = exports.getUrlFromDirectPath = exports.encryptedStream = exports.prepareStream = exports.getHttpStream = exports.generateThumbnail = exports.getStream = exports.toBuffer = exports.toReadable = exports.getAudioWaveform = exports.getAudioDuration = exports.mediaMessageSHA256B64 = exports.generateProfilePicture = exports.encodeBase64EncodedStringForUpload = exports.extractImageThumb = exports.getMediaKeys = exports.hkdfInfoKey = void 0;
 const boom_1 = require("@hapi/boom");
 const child_process_1 = require("child_process");
 const Crypto = __importStar(require("crypto"));
@@ -81,7 +66,7 @@ const hkdfInfoKey = (type) => {
 };
 exports.hkdfInfoKey = hkdfInfoKey;
 /** generates all the keys required to encrypt/decrypt & sign a media message */
-async function getMediaKeys(buffer, mediaType) {
+function getMediaKeys(buffer, mediaType) {
     if (!buffer) {
         throw new boom_1.Boom('Cannot derive from empty media key');
     }
@@ -89,13 +74,14 @@ async function getMediaKeys(buffer, mediaType) {
         buffer = Buffer.from(buffer.replace('data:;base64,', ''), 'base64');
     }
     // expand using HKDF to 112 bytes, also pass in the relevant app info
-    const expandedMediaKey = await (0, crypto_1.hkdf)(buffer, 112, { info: (0, exports.hkdfInfoKey)(mediaType) });
+    const expandedMediaKey = (0, crypto_1.hkdf)(buffer, 112, { info: (0, exports.hkdfInfoKey)(mediaType) });
     return {
         iv: expandedMediaKey.slice(0, 16),
         cipherKey: expandedMediaKey.slice(16, 48),
         macKey: expandedMediaKey.slice(48, 80),
     };
 }
+exports.getMediaKeys = getMediaKeys;
 /** Extracts video thumb using FFMPEG */
 const extractVideoThumb = async (path, destPath, time, size) => new Promise((resolve, reject) => {
     const cmd = `ffmpeg -ss ${time} -i ${path} -y -vf scale=${size.width}:-1 -vframes 1 -f image2 ${destPath}`;
@@ -156,23 +142,42 @@ const encodeBase64EncodedStringForUpload = (b64) => (encodeURIComponent(b64
     .replace(/\=+$/, '')));
 exports.encodeBase64EncodedStringForUpload = encodeBase64EncodedStringForUpload;
 const generateProfilePicture = async (mediaUpload) => {
-    const bufferOrFilePath = Buffer.isBuffer(mediaUpload)
-        ? mediaUpload
-        : typeof mediaUpload === 'object' && 'url' in mediaUpload
-            ? mediaUpload.url.toString()
-            : await (0, exports.toBuffer)(mediaUpload.stream);
+    var _a, _b;
+    let bufferOrFilePath;
+    if (Buffer.isBuffer(mediaUpload)) {
+        bufferOrFilePath = mediaUpload;
+    }
+    else if ('url' in mediaUpload) {
+        bufferOrFilePath = mediaUpload.url.toString();
+    }
+    else {
+        bufferOrFilePath = await (0, exports.toBuffer)(mediaUpload.stream);
+    }
+    const lib = await getImageProcessingLibrary();
     let img;
-    const { read, MIME_JPEG, AUTO } = require('jimp');
-    const jimp = await read(bufferOrFilePath);
-    const min = jimp.getWidth();
-    const max = jimp.getHeight();
-    const cropped = jimp.crop(0, 0, min, max);
-    img = cropped
-        .quality(100)
-        .scaleToFit(720, 720, AUTO)
-        .getBufferAsync(MIME_JPEG);
+    if ('sharp' in lib && typeof ((_a = lib.sharp) === null || _a === void 0 ? void 0 : _a.default) === 'function') {
+        img = lib.sharp.default(bufferOrFilePath)
+            .resize(640, 640)
+            .jpeg({
+            quality: 50,
+        })
+            .toBuffer();
+    }
+    else if ('jimp' in lib && typeof ((_b = lib.jimp) === null || _b === void 0 ? void 0 : _b.read) === 'function') {
+        const { read, MIME_JPEG, RESIZE_BILINEAR } = lib.jimp;
+        const jimp = await read(bufferOrFilePath);
+        const min = Math.min(jimp.getWidth(), jimp.getHeight());
+        const cropped = jimp.crop(0, 0, min, min);
+        img = cropped
+            .quality(50)
+            .resize(640, 640, RESIZE_BILINEAR)
+            .getBufferAsync(MIME_JPEG);
+    }
+    else {
+        throw new boom_1.Boom('No image processing library available');
+    }
     return {
-        img: await img
+        img: await img,
     };
 };
 exports.generateProfilePicture = generateProfilePicture;
@@ -183,66 +188,138 @@ const mediaMessageSHA256B64 = (message) => {
 };
 exports.mediaMessageSHA256B64 = mediaMessageSHA256B64;
 async function getAudioDuration(buffer) {
-    const musicMetadata = await import('music-metadata');
-    let metadata;
-    if (Buffer.isBuffer(buffer)) {
-        metadata = await musicMetadata.parseBuffer(buffer, undefined, { duration: true });
-    }
-    else if (typeof buffer === 'string') {
-        const rStream = (0, fs_1.createReadStream)(buffer);
-        try {
-            metadata = await musicMetadata.parseStream(rStream, undefined, { duration: true });
-        }
-        finally {
-            rStream.destroy();
+    try {
+        const { PassThrough } = require('stream');
+        const ff = require('fluent-ffmpeg');
+        
+        return await new Promise((resolve, reject) => {
+            const inputStream = new PassThrough();
+            inputStream.end(buffer);
+            
+            ff(inputStream)
+                .ffprobe((err, data) => {
+                    if (err) reject(err);
+                    else resolve(data.format.duration);
+                });
+        });
+    } catch (error) {
+        const musicMetadata = await import('music-metadata');
+        let metadata;
+        if (Buffer.isBuffer(buffer)) {
+            metadata = await musicMetadata.parseBuffer(buffer, undefined, { 
+                duration: true 
+            });
+        } else if (typeof buffer === 'string') {
+            const rStream = (0, fs_1.createReadStream)(buffer);
+            try {
+                metadata = await musicMetadata.parseStream(rStream, undefined, { 
+                    duration: true 
+                });
+            } finally {
+                rStream.destroy();
+            }
+        } else {
+            metadata = await musicMetadata.parseStream(buffer, undefined, { 
+                duration: true 
+            });
         }
+        return metadata.format.duration;
     }
-    else {
-        metadata = await musicMetadata.parseStream(buffer, undefined, { duration: true });
-    }
-    return metadata.format.duration;
 }
-/**
-  referenced from and modifying https://github.com/wppconnect-team/wa-js/blob/main/src/chat/functions/prepareAudioWaveform.ts
- */
+exports.getAudioDuration = getAudioDuration;
 async function getAudioWaveform(buffer, logger) {
     try {
-        const audioDecode = (buffer) => import('audio-decode').then(({ default: audioDecode }) => audioDecode(buffer));
+        const { PassThrough } = require('stream');
+        const ff = require('fluent-ffmpeg');
+        
         let audioData;
         if (Buffer.isBuffer(buffer)) {
             audioData = buffer;
+        } else if (typeof buffer === 'string') {
+            const rStream = require('fs').createReadStream(buffer);
+            audioData = await exports.toBuffer(rStream);
+        } else {
+            audioData = await exports.toBuffer(buffer);
         }
-        else if (typeof buffer === 'string') {
-            const rStream = (0, fs_1.createReadStream)(buffer);
-            audioData = await (0, exports.toBuffer)(rStream);
-        }
-        else {
-            audioData = await (0, exports.toBuffer)(buffer);
-        }
-        const audioBuffer = await audioDecode(audioData);
-        const rawData = audioBuffer.getChannelData(0); // We only need to work with one channel of data
-        const samples = 64; // Number of samples we want to have in our final data set
-        const blockSize = Math.floor(rawData.length / samples); // the number of samples in each subdivision
-        const filteredData = [];
-        for (let i = 0; i < samples; i++) {
-            const blockStart = blockSize * i; // the location of the first sample in the block
-            let sum = 0;
-            for (let j = 0; j < blockSize; j++) {
-                sum = sum + Math.abs(rawData[blockStart + j]); // find the sum of all the samples in the block
-            }
-            filteredData.push(sum / blockSize); // divide the sum by the block size to get the average
-        }
-        // This guarantees that the largest data point will be set to 1, and the rest of the data will scale proportionally.
-        const multiplier = Math.pow(Math.max(...filteredData), -1);
-        const normalizedData = filteredData.map((n) => n * multiplier);
-        // Generate waveform like WhatsApp
-        const waveform = new Uint8Array(normalizedData.map((n) => Math.floor(100 * n)));
-        return waveform;
+
+        return await new Promise((resolve, reject) => {
+            const inputStream = new PassThrough();
+            inputStream.end(audioData);
+            const chunks = [];
+            const bars = 64;
+            
+            ff(inputStream)
+                .audioChannels(1)
+                .audioFrequency(16000)
+                .format('s16le')
+                .on('error', reject)
+                .on('end', () => {
+                    const rawData = Buffer.concat(chunks);
+                    const samples = rawData.length / 2;
+                    const amplitudes = [];
+                    
+                    for (let i = 0; i < samples; i++) {
+                        amplitudes.push(Math.abs(rawData.readInt16LE(i * 2)) / 32768);
+                    }
+                    
+                    const blockSize = Math.floor(amplitudes.length / bars);
+                    const avg = [];
+                    for (let i = 0; i < bars; i++) {
+                        const block = amplitudes.slice(i * blockSize, (i + 1) * blockSize);
+                        avg.push(block.reduce((a, b) => a + b, 0) / block.length);
+                    }
+                    
+                    const max = Math.max(...avg);
+                    const normalized = avg.map(v => Math.floor((v / max) * 100));
+                    resolve(new Uint8Array(normalized));
+                })
+                .pipe()
+                .on('data', chunk => chunks.push(chunk));
+        });
+    } catch (e) {
+        logger?.debug(e);
     }
-    catch (e) {
-        logger === null || logger === void 0 ? void 0 : logger.debug('Failed to generate waveform: ' + e);
+}
+exports.getAudioWaveform = getAudioWaveform;
+async function convertToOpusBuffer(buffer, logger) {
+    try {
+        const { PassThrough } = require('stream');
+        const ff = require('fluent-ffmpeg');
+        
+        return await new Promise((resolve, reject) => {
+            const inStream = new PassThrough();
+            const outStream = new PassThrough();
+            const chunks = [];
+            inStream.end(buffer);
+            
+            ff(inStream)
+                .noVideo()
+                .audioCodec('libopus')
+                .format('ogg')
+                .audioBitrate('48k')
+                .audioChannels(1)
+                .audioFrequency(48000)
+                .outputOptions([
+                    '-vn',
+                    '-b:a 64k',
+                    '-ac 2',
+                    '-ar 48000',
+                    '-map_metadata', '-1',
+                    '-application', 'voip'
+                ])
+                .on('error', reject)
+                .on('end', () => resolve(Buffer.concat(chunks)))
+                .pipe(outStream, {
+                end: true 
+            });       
+            outStream.on('data', c => chunks.push(c));
+        });
+    } catch (e) {
+        logger?.debug(e);
+        throw e;
     }
 }
+exports.convertToOpusBuffer = convertToOpusBuffer;
 const toReadable = (buffer) => {
     const readable = new stream_1.Readable({ read: () => { } });
     readable.push(buffer);
@@ -304,6 +381,7 @@ async function generateThumbnail(file, mediaType, options) {
         originalImageDimensions
     };
 }
+exports.generateThumbnail = generateThumbnail;
 const getHttpStream = async (url, options = {}) => {
     const { default: axios } = await import('axios');
     const fetched = await axios.get(url.toString(), { ...options, responseType: 'stream' });
@@ -354,15 +432,28 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
     }
 };
 exports.prepareStream = prepareStream;
-const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts } = {}) => {
+const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts, isPtt, forceOpus } = {}) => {
     const { stream, type } = await (0, exports.getStream)(media, opts);
-    logger === null || logger === void 0 ? void 0 : logger.debug('fetched media stream');
+    
+    let finalStream = stream;  
+    if (mediaType === 'audio' && (isPtt === true || forceOpus === true)) {
+        try {
+            const buffer = await (0, exports.toBuffer)(stream);
+            const opusBuffer = await exports.convertToOpusBuffer(buffer, logger);
+            finalStream = (0, exports.toReadable)(opusBuffer);
+        } catch (error) {
+            const { stream: newStream } = await (0, exports.getStream)(media, opts);
+            finalStream = newStream;
+        }
+    }
+
     const mediaKey = Crypto.randomBytes(32);
-    const { cipherKey, iv, macKey } = await getMediaKeys(mediaKey, mediaType);
+    const { cipherKey, iv, macKey } = getMediaKeys(mediaKey, mediaType);
     const encWriteStream = new stream_1.Readable({ read: () => { } });
     let bodyPath;
     let writeStream;
     let didSaveToTmpPath = false;
+    
     if (type === 'file') {
         bodyPath = media.url;
     }
@@ -371,13 +462,15 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
         writeStream = (0, fs_1.createWriteStream)(bodyPath);
         didSaveToTmpPath = true;
     }
+    
     let fileLength = 0;
     const aes = Crypto.createCipheriv('aes-256-cbc', cipherKey, iv);
     let hmac = Crypto.createHmac('sha256', macKey).update(iv);
     let sha256Plain = Crypto.createHash('sha256');
     let sha256Enc = Crypto.createHash('sha256');
+    
     try {
-        for await (const data of stream) {
+        for await (const data of finalStream) {
             fileLength += data.length;
             if (type === 'remote'
                 && (opts === null || opts === void 0 ? void 0 : opts.maxContentLength)
@@ -386,6 +479,7 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
                     data: { media, type }
                 });
             }
+            
             sha256Plain = sha256Plain.update(data);
             if (writeStream) {
                 if (!writeStream.write(data)) {
@@ -394,16 +488,18 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
             }
             onChunk(aes.update(data));
         }
+        
         onChunk(aes.final());
         const mac = hmac.digest().slice(0, 10);
         sha256Enc = sha256Enc.update(mac);
         const fileSha256 = sha256Plain.digest();
         const fileEncSha256 = sha256Enc.digest();
+        
         encWriteStream.push(mac);
         encWriteStream.push(null);
         writeStream === null || writeStream === void 0 ? void 0 : writeStream.end();
-        stream.destroy();
-        logger === null || logger === void 0 ? void 0 : logger.debug('encrypted data successfully');
+        finalStream.destroy();
+        
         return {
             mediaKey,
             encWriteStream,
@@ -416,24 +512,24 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
         };
     }
     catch (error) {
-        // destroy all streams with error
         encWriteStream.destroy();
         writeStream === null || writeStream === void 0 ? void 0 : writeStream.destroy();
         aes.destroy();
         hmac.destroy();
         sha256Plain.destroy();
         sha256Enc.destroy();
-        stream.destroy();
+        finalStream.destroy();
+        
         if (didSaveToTmpPath) {
             try {
                 await fs_1.promises.unlink(bodyPath);
             }
             catch (err) {
-                logger === null || logger === void 0 ? void 0 : logger.error({ err }, 'failed to save to tmp path');
             }
         }
         throw error;
     }
+    
     function onChunk(buff) {
         sha256Enc = sha256Enc.update(buff);
         hmac = hmac.update(buff);
@@ -448,9 +544,9 @@ const toSmallestChunkSize = (num) => {
 };
 const getUrlFromDirectPath = (directPath) => `https://${DEF_HOST}${directPath}`;
 exports.getUrlFromDirectPath = getUrlFromDirectPath;
-const downloadContentFromMessage = async ({ mediaKey, directPath, url }, type, opts = {}) => {
+const downloadContentFromMessage = ({ mediaKey, directPath, url }, type, opts = {}) => {
     const downloadUrl = url || (0, exports.getUrlFromDirectPath)(directPath);
-    const keys = await getMediaKeys(mediaKey, type);
+    const keys = getMediaKeys(mediaKey, type);
     return (0, exports.downloadEncryptedContent)(downloadUrl, keys, opts);
 };
 exports.downloadContentFromMessage = downloadContentFromMessage;
@@ -557,6 +653,7 @@ function extensionForMediaMessage(message) {
     }
     return extension;
 }
+exports.extensionForMediaMessage = extensionForMediaMessage;
 const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options }, refreshMediaConn) => {
     return async (stream, { mediaType, fileEncSha256B64, newsletter, timeoutMs }) => {
         var _a, _b;
@@ -634,11 +731,11 @@ const getMediaRetryKey = (mediaKey) => {
 /**
  * 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 encryptMediaRetryRequest = (key, mediaKey, meId) => {
     const recp = { stanzaId: key.id };
     const recpBuffer = WAProto_1.proto.ServerErrorReceipt.encode(recp).finish();
     const iv = Crypto.randomBytes(12);
-    const retryKey = await getMediaRetryKey(mediaKey);
+    const retryKey = getMediaRetryKey(mediaKey);
     const ciphertext = (0, crypto_1.aesEncryptGCM)(recpBuffer, retryKey, iv, Buffer.from(key.id));
     const req = {
         tag: 'receipt',
@@ -702,8 +799,8 @@ const decodeMediaRetryNode = (node) => {
     return event;
 };
 exports.decodeMediaRetryNode = decodeMediaRetryNode;
-const decryptMediaRetryData = async ({ ciphertext, iv }, mediaKey, msgId) => {
-    const retryKey = await getMediaRetryKey(mediaKey);
+const decryptMediaRetryData = ({ ciphertext, iv }, mediaKey, msgId) => {
+    const retryKey = getMediaRetryKey(mediaKey);
     const plaintext = (0, crypto_1.aesDecryptGCM)(ciphertext, retryKey, iv, Buffer.from(msgId));
     return WAProto_1.proto.MediaRetryNotification.decode(plaintext);
 };
@@ -716,6 +813,7 @@ const MEDIA_RETRY_STATUS_MAP = {
     [WAProto_1.proto.MediaRetryNotification.ResultType.NOT_FOUND]: 404,
     [WAProto_1.proto.MediaRetryNotification.ResultType.GENERAL_ERROR]: 418,
 };
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
 function __importStar(arg0) {
     throw new Error('Function not implemented.');
 }