@ubedsockets/baileys 1.6.8 → 1.7.0

package/lib/Utils/messages-media.js

@@ -36,26 +36,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 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.extractVideoThumb =
-exports.hkdfInfoKey = void 0;
+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.extractVideoThumb = exports.hkdfInfoKey = void 0;
 exports.getMediaKeys = getMediaKeys;
 exports.uploadFile = uploadFile;
 exports.vid2jpg = vid2jpg;
@@ -63,7 +44,6 @@ exports.getAudioDuration = getAudioDuration;
 exports.getAudioWaveform = getAudioWaveform;
 exports.generateThumbnail = generateThumbnail;
 exports.extensionForMediaMessage = extensionForMediaMessage;
-// 🆅🆈🆉🅴🅽 🅱🅰🅸🅻
 const boom_1 = require("@hapi/boom");
 const axios_1 = __importDefault(require("axios"));
 const form_data_1 = __importDefault(require("form-data"));
@@ -369,173 +349,119 @@ const mediaMessageSHA256B64 = (message) => {
 };
 exports.mediaMessageSHA256B64 = mediaMessageSHA256B64;
 async function getAudioDuration(buffer) {
-    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;
+    const musicMetadata = await eval('import(\'music-metadata\')');
+    let metadata;
+    const options = {
+        duration: true
+    };
+    if (Buffer.isBuffer(buffer)) {
+        metadata = await musicMetadata.parseBuffer(buffer, options);
+    }
+    else if (typeof buffer === 'string') {
+        metadata = await musicMetadata.parseFile(buffer, options);
+    }
+    else {
+        metadata = await musicMetadata.parseStream(buffer, options);
     }
+    return metadata.format.duration;
 }
-exports.getAudioDuration = getAudioDuration;
+const generateFallbackWaveform = (audioData) => {
+    const samples = 64;
+    const waveform = new Uint8Array(samples);
+    let seed = audioData.length;
+    for (let i = 0; i < Math.min(audioData.length, 100); i++) {
+        seed = (seed * 31 + audioData[i]) >>> 0;
+    }
+    const random = () => {
+        seed = (seed * 1664525 + 1013904223) >>> 0;
+        return (seed >>> 16) / 65536;
+    };
+    for (let i = 0; i < samples; i++) {
+        const position = i / samples;
+        const envelope = Math.sin(position * Math.PI);
+        const baseAmplitude = 30 + random() * 40;
+        waveform[i] = Math.min(100, Math.max(0, Math.floor(baseAmplitude * envelope)));
+    }
+    return waveform;
+};
 async function getAudioWaveform(buffer, logger) {
+    let audioData;
+    if (Buffer.isBuffer(buffer)) {
+        audioData = 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);
+    }
     try {
-        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);
+        const { default: decoder } = await eval('import(\'audio-decode\')');
+        const audioBuffer = await decoder(audioData);
+        const rawData = audioBuffer.getChannelData(0);
+        const samples = 64;
+        const blockSize = Math.floor(rawData.length / samples);
+        const filteredData = [];
+        for (let i = 0; i < samples; i++) {
+            const blockStart = blockSize * i;
+            let sum = 0;
+            for (let j = 0; j < blockSize; j++) {
+                sum = sum + Math.abs(rawData[blockStart + j]);
+            }
+            filteredData.push(sum / blockSize);
         }
-
-        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);
+        const multiplier = Math.pow(Math.max(...filteredData), -1);
+        const normalizedData = filteredData.map((n) => n * multiplier);
+        const waveform = new Uint8Array(normalizedData.map((n) => Math.floor(100 * n)));
+        return waveform;
     }
-}
-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;
+    catch (e) {
+        logger === null || logger === void 0 ? void 0 : logger.debug('Failed to generate waveform using audio-decode, using fallback: ' + e);
+        return generateFallbackWaveform(audioData);
     }
 }
-exports.convertToOpusBuffer = convertToOpusBuffer;
-async function convertToMp4Buffer(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)
-                .videoCodec('libx264')
-                .audioCodec('aac')
-                .format('mp4')
-                .outputOptions([
-                    '-preset', 'veryfast',
-                    '-crf', '23',
-                    '-movflags', 'faststart',
-                    '-map_metadata', '-1'
-                ])
-                .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;
+async function transcodeAudio(audio, options = {}) {
+    const { bitrate = '64k' } = options;
+    let audioData;
+    if (Buffer.isBuffer(audio)) {
+        audioData = audio;
+    }
+    else if (typeof audio === 'string') {
+        const rStream = (0, fs_1.createReadStream)(audio);
+        audioData = await (0, exports.toBuffer)(rStream);
     }
+    else {
+        audioData = await (0, exports.toBuffer)(audio);
+    }
+    return await new Promise((resolve, reject) => {
+        const ffmpeg = (0, child_process_1.spawn)('ffmpeg', [
+            '-i', 'pipe:0',
+            '-vn',
+            '-ac', '1',
+            '-c:a', 'libopus',
+            '-b:a', bitrate,
+            '-f', 'ogg',
+            'pipe:1'
+        ]);
+        const chunks = [];
+        let errorOutput = '';
+        ffmpeg.stdout.on('data', chunk => chunks.push(chunk));
+        ffmpeg.stderr.on('data', data => {
+            errorOutput += data.toString();
+        });
+        ffmpeg.on('error', reject);
+        ffmpeg.on('close', code => {
+            if (code === 0) {
+                resolve(Buffer.concat(chunks));
+                return;
+            }
+            reject(new Error(`ffmpeg exited with code ${code}\n${errorOutput}`));
+        });
+        ffmpeg.stdin.end(audioData);
+    });
 }
-
-exports.convertToMp4Buffer = convertToMp4Buffer;
+exports.transcodeAudio = transcodeAudio;
 const toReadable = (buffer) => {
     const readable = new stream_1.Readable({ read: () => { } });
     readable.push(buffer);
@@ -661,90 +587,56 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
     }
 };
 exports.prepareStream = prepareStream;
-const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts, isPtt, forceOpus, convertVideo } = {}) => {
+const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts } = {}) => {
     const { stream, type } = await (0, exports.getStream)(media, opts);
-    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);
-            logger?.debug('converted audio to Opus');
-        } catch (error) {
-            logger?.error('failed to convert audio to Opus, fallback to original stream');
-            const { stream: newStream } = await (0, exports.getStream)(media, opts);
-            finalStream = newStream;
-        }
-    }
-
-    if (mediaType === 'video' && convertVideo === true) {
-        try {
-            const buffer = await (0, exports.toBuffer)(stream);
-            const mp4Buffer = await exports.convertToMp4Buffer(buffer, logger);
-            finalStream = (0, exports.toReadable)(mp4Buffer);
-            logger?.debug('converted video to mp4');
-        } catch (error) {
-            logger?.error('failed to convert video to mp4, fallback to original stream');
-            const { stream: newStream } = await (0, exports.getStream)(media, opts);
-            finalStream = newStream;
-        }
-    }
-
+    logger === null || logger === void 0 ? void 0 : logger.debug('fetched media stream');
     const mediaKey = Crypto.randomBytes(32);
     const { cipherKey, iv, macKey } = await getMediaKeys(mediaKey, mediaType);
-    const encWriteStream = new stream_1.Readable({ read: () => {} });
+    const encWriteStream = new stream_1.Readable({ read: () => { } });
     let bodyPath;
     let writeStream;
     let didSaveToTmpPath = false;
-
     if (type === 'file') {
         bodyPath = media.url;
-    } else if (saveOriginalFileIfRequired) {
+    }
+    else if (saveOriginalFileIfRequired) {
         bodyPath = (0, path_1.join)(getTmpFilesDirectory(), mediaType + (0, generics_1.generateMessageIDV2)());
         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 finalStream) {
+        for await (const data of stream) {
             fileLength += data.length;
-            if (type === 'remote' && (opts?.maxContentLength) && fileLength + data.length > opts.maxContentLength) {
-                throw new boom_1.Boom(`content length exceeded when encrypting "${type}"`, { data: { media, type } });
+            if (type === 'remote'
+                && (opts === null || opts === void 0 ? void 0 : opts.maxContentLength)
+                && fileLength + data.length > opts.maxContentLength) {
+                throw new boom_1.Boom(`content length exceeded when encrypting "${type}"`, {
+                    data: { media, type }
+                });
             }
-
             sha256Plain = sha256Plain.update(data);
-
             if (writeStream) {
                 if (!writeStream.write(data)) {
                     await (0, events_1.once)(writeStream, 'drain');
                 }
             }
-
             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?.end();
-        finalStream.destroy();
-
-        logger?.debug('encrypted data successfully');
-
+        writeStream === null || writeStream === void 0 ? void 0 : writeStream.end();
+        stream.destroy();
+        logger === null || logger === void 0 ? void 0 : logger.debug('encrypted data successfully');
         return {
             mediaKey,
             encWriteStream,
@@ -755,32 +647,31 @@ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfReq
             fileLength,
             didSaveToTmpPath
         };
-    } catch (error) {
+    }
+    catch (error) {
         encWriteStream.destroy();
-        writeStream?.destroy();
+        writeStream === null || writeStream === void 0 ? void 0 : writeStream.destroy();
         aes.destroy();
         hmac.destroy();
         sha256Plain.destroy();
         sha256Enc.destroy();
-        finalStream.destroy();
-
+        stream.destroy();
         if (didSaveToTmpPath) {
             try {
                 await fs_1.promises.unlink(bodyPath);
-            } catch (err) {
-                logger?.error({ err }, 'failed to save to tmp path');
+            }
+            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);
         encWriteStream.push(buff);
     }
 };
-
 exports.encryptedStream = encryptedStream;
 const DEF_HOST = 'mmg.whatsapp.net';
 const AES_CHUNK_SIZE = 16;
@@ -1046,4 +937,4 @@ const MEDIA_RETRY_STATUS_MAP = {
 };
 function __importStar(arg0) {
     throw new Error('Function not implemented.');
-}
+}