@aseppxyzz12/baileys 1.1.8

package/lib/Utils/messages-media.js

@@ -0,0 +1,806 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const { Boom: Boom } = require("@hapi/boom");
+const { exec: exec } = require("child_process");
+const { once: once } = require("events");
+const {
+    createHash: createHash,
+    randomBytes: randomBytes,
+    createHmac: createHmac,
+    createCipheriv: createCipheriv,
+    createDecipheriv: createDecipheriv,
+} = require("crypto");
+const {
+    promises: promises,
+    createReadStream: createReadStream,
+    createWriteStream: createWriteStream,
+} = require("fs");
+const {
+    parseBuffer: parseBuffer,
+    parseFile: parseFile,
+    parseStream: parseStream,
+} = require("music-metadata");
+const { tmpdir: tmpdir } = require("os");
+const { join: join } = require("path");
+const { Readable: Readable, Transform: Transform } = require("stream");
+const { proto: proto } = require("../../WAProto");
+const {
+    MEDIA_PATH_MAP: MEDIA_PATH_MAP,
+    MEDIA_HKDF_KEY_MAPPING: MEDIA_HKDF_KEY_MAPPING,
+} = require("../Defaults/media");
+const { DEFAULT_ORIGIN: DEFAULT_ORIGIN } = require("../Defaults/constants");
+const {
+    getBinaryNodeChild: getBinaryNodeChild,
+    getBinaryNodeChildBuffer: getBinaryNodeChildBuffer,
+    jidNormalizedUser: jidNormalizedUser,
+} = require("../WABinary");
+const {
+    aesDecryptGCM: aesDecryptGCM,
+    aesEncryptGCM: aesEncryptGCM,
+    hkdf: hkdf,
+} = require("./crypto");
+const { generateMessageID: generateMessageID } = require("./generics");
+const getTmpFilesDirectory = () => tmpdir();
+const getImageProcessingLibrary = () => {
+    let sharp, jimp;
+    try {
+        sharp = require("sharp");
+    } catch {}
+    if (sharp) {
+        return { sharp: sharp };
+    }
+    try {
+        jimp = require("jimp");
+    } catch {}
+    if (jimp) {
+        return { jimp: jimp };
+    }
+    throw new Boom("No image processing library available");
+};
+const hkdfInfoKey = (type) => {
+    const hkdfInfo = MEDIA_HKDF_KEY_MAPPING[type];
+    return `WhatsApp ${hkdfInfo} Keys`;
+};
+const getRawMediaUploadData = async (media, mediaType, logger) => {
+    const { stream: stream } = await getStream(media);
+    logger?.debug("got stream for raw upload");
+    const hasher = createHash("sha256");
+    const filePath = join(getTmpFilesDirectory(), mediaType + generateMessageID());
+    const fileWriteStream = createWriteStream(filePath);
+    let fileLength = 0;
+    try {
+        for await (const data of stream) {
+            fileLength += data.length;
+            hasher.update(data);
+            if (!fileWriteStream.write(data)) {
+                await once(fileWriteStream, "drain");
+            }
+        }
+        fileWriteStream.end();
+        await once(fileWriteStream, "finish");
+        stream.destroy();
+        const fileSha256 = hasher.digest();
+        logger?.debug("hashed data for raw upload");
+        return {
+            filePath: filePath,
+            fileSha256: fileSha256,
+            fileLength: fileLength,
+        };
+    } catch (error) {
+        fileWriteStream.destroy();
+        stream.destroy();
+        try {
+            await promises.unlink(filePath);
+        } catch {}
+        throw error;
+    }
+};
+async function getMediaKeys(buffer, mediaType) {
+    if (!buffer) {
+        throw new Boom("Cannot derive from empty media key");
+    }
+    if (typeof buffer === "string") {
+        buffer = Buffer.from(buffer.replace("data:base64,", ""), "base64");
+    }
+    const expandedMediaKey = await hkdf(buffer, 112, {
+        info: hkdfInfoKey(mediaType),
+    });
+    return {
+        iv: expandedMediaKey.slice(0, 16),
+        cipherKey: expandedMediaKey.slice(16, 48),
+        macKey: expandedMediaKey.slice(48, 80),
+    };
+}
+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}`;
+        exec(cmd, (err) => {
+            if (err) {
+                reject(err);
+            } else {
+                resolve();
+            }
+        });
+    });
+const extractImageThumb = async (bufferOrFilePath, width = 32, quality = 50) => {
+    if (bufferOrFilePath instanceof Readable) {
+        bufferOrFilePath = await toBuffer(bufferOrFilePath);
+    }
+    const lib = await getImageProcessingLibrary();
+    if ("sharp" in lib && typeof lib.sharp === "function") {
+        const img = lib.sharp(bufferOrFilePath);
+        const dimensions = await img.metadata();
+        const buffer = await img.resize(width).jpeg({ quality: 50 }).toBuffer();
+        return {
+            buffer: buffer,
+            original: { width: dimensions.width, height: dimensions.height },
+        };
+    } else if ("jimp" in lib && typeof lib.jimp.read === "function") {
+        const {
+            read: read,
+            MIME_JPEG: MIME_JPEG,
+            RESIZE_BEZIER: RESIZE_BEZIER,
+            AUTO: AUTO,
+        } = lib.jimp;
+        const jimp = await read(bufferOrFilePath);
+        const dimensions = { width: jimp.getWidth(), height: jimp.getHeight() };
+        const buffer = await jimp
+            .quality(quality)
+            .resize(width, AUTO, RESIZE_BEZIER)
+            .getBufferAsync(MIME_JPEG);
+        return { buffer: buffer, original: dimensions };
+    } else {
+        throw new Boom("No image processing library available");
+    }
+};
+const encodeBase64EncodedStringForUpload = (b64) =>
+    encodeURIComponent(b64.replace(/\+/g, "-").replace(/\//g, "_").replace(/\=+$/, ""));
+const generateProfilePicture = async (mediaUpload, dimensions) => {
+    let buffer;
+    const { width: w = 640, height: h = 640 } = dimensions || {};
+    if (Buffer.isBuffer(mediaUpload)) {
+        buffer = mediaUpload;
+    } else {
+        const { stream: stream } = await getStream(mediaUpload);
+        buffer = await toBuffer(stream);
+    }
+    const lib = await getImageProcessingLibrary();
+    let img;
+    if ("sharp" in lib && typeof lib.sharp?.default === "function") {
+        img = lib.sharp.default(buffer).resize(w, h).jpeg({ quality: 50 }).toBuffer();
+    } else if ("jimp" in lib && typeof lib.jimp?.read === "function") {
+        const jimp = await lib.jimp.read(buffer);
+        const min = Math.min(jimp.width, jimp.height);
+        const cropped = jimp.crop({ x: 0, y: 0, w: min, h: min });
+        img = cropped
+            .resize({ w: w, h: h, mode: lib.jimp.ResizeStrategy.BILINEAR })
+            .getBuffer("image/jpeg", { quality: 50 });
+    } else {
+        throw new Boom("No image processing library available");
+    }
+    return { img: await img };
+};
+const mediaMessageSHA256B64 = (message) => {
+    const media = Object.values(message)[0];
+    return media?.fileSha256 && Buffer.from(media.fileSha256).toString("base64");
+};
+async function getAudioDuration(buffer) {
+    const options = { duration: true };
+    let metadata;
+    if (Buffer.isBuffer(buffer)) {
+        metadata = await parseBuffer(buffer, undefined, options);
+    } else if (typeof buffer === "string") {
+        metadata = await parseFile(buffer, options);
+    } else {
+        metadata = await parseStream(buffer, undefined, options);
+    }
+    return metadata.format?.duration;
+}
+async function getAudioWaveform(buffer, logger) {
+    try {
+        const { default: decoder } = await eval("import('audio-decode')");
+        let audioData;
+        if (Buffer.isBuffer(buffer)) {
+            audioData = buffer;
+        } else if (typeof buffer === "string") {
+            const rStream = createReadStream(buffer);
+            audioData = await toBuffer(rStream);
+        } else {
+            audioData = await toBuffer(buffer);
+        }
+        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);
+        }
+        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;
+    } catch (e) {
+        logger?.debug("Failed to generate waveform: " + e);
+    }
+}
+const toReadable = (buffer) => {
+    const readable = new Readable({ read: () => {} });
+    readable.push(buffer);
+    readable.push(null);
+    return readable;
+};
+const toBuffer = async (stream) => {
+    const chunks = [];
+    for await (const chunk of stream) {
+        chunks.push(chunk);
+    }
+    stream.destroy();
+    return Buffer.concat(chunks);
+};
+const getStream = async (item, opts) => {
+    if (Buffer.isBuffer(item)) {
+        return { stream: toReadable(item), type: "buffer" };
+    }
+    if ("stream" in item) {
+        return { stream: item.stream, type: "readable" };
+    }
+    const urlStr = item.url.toString();
+    if (urlStr.startsWith("data:")) {
+        const buffer = Buffer.from(urlStr.split(",")[1], "base64");
+        return { stream: await toReadable(buffer), type: "buffer" };
+    }
+    if (urlStr.startsWith("http://") || urlStr.startsWith("https://")) {
+        return { stream: await getHttpStream(item.url, opts), type: "remote" };
+    }
+    return { stream: createReadStream(item.url), type: "file" };
+};
+async function generateThumbnail(file, mediaType, options) {
+    let thumbnail;
+    let originalImageDimensions;
+    if (mediaType === "image") {
+        const { buffer: buffer, original: original } = await extractImageThumb(file);
+        thumbnail = buffer.toString("base64");
+        if (original.width && original.height) {
+            originalImageDimensions = {
+                width: original.width,
+                height: original.height,
+            };
+        }
+    } else if (mediaType === "video") {
+        const imgFilename = join(getTmpFilesDirectory(), generateMessageID() + ".jpg");
+        try {
+            await extractVideoThumb(file, imgFilename, "00:00:00", {
+                width: 32,
+                height: 32,
+            });
+            const buff = await promises.readFile(imgFilename);
+            thumbnail = buff.toString("base64");
+            await promises.unlink(imgFilename);
+        } catch (err) {
+            options.logger?.debug("could not generate video thumb: " + err);
+        }
+    }
+    return {
+        thumbnail: thumbnail,
+        originalImageDimensions: originalImageDimensions,
+    };
+}
+const getHttpStream = async (url, options = {}) => {
+    const response = await fetch(url.toString(), {
+        dispatcher: options.dispatcher,
+        method: "GET",
+        headers: options.headers,
+    });
+    if (!response.ok) {
+        throw new Boom(`Failed to fetch stream from ${url}`, {
+            statusCode: response.status,
+            data: { url: url },
+        });
+    }
+    return response.body instanceof Readable ? response.body : Readable.fromWeb(response.body);
+};
+const encryptedStream = async (
+    media,
+    mediaType,
+    { logger: logger, saveOriginalFileIfRequired: saveOriginalFileIfRequired, opts: opts } = {}
+) => {
+    const { stream: stream, type: type } = await getStream(media, opts);
+    logger?.debug("fetched media stream");
+    const mediaKey = randomBytes(32);
+    const {
+        cipherKey: cipherKey,
+        iv: iv,
+        macKey: macKey,
+    } = await getMediaKeys(mediaKey, mediaType);
+    const encFilePath = join(getTmpFilesDirectory(), mediaType + generateMessageID() + "-enc");
+    const encFileWriteStream = createWriteStream(encFilePath);
+    let originalFileStream;
+    let originalFilePath;
+    if (saveOriginalFileIfRequired) {
+        originalFilePath = join(
+            getTmpFilesDirectory(),
+            mediaType + generateMessageID() + "-original"
+        );
+        originalFileStream = createWriteStream(originalFilePath);
+    }
+    let fileLength = 0;
+    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 = async (buff) => {
+        sha256Enc.update(buff);
+        hmac.update(buff);
+        if (!encFileWriteStream.write(buff)) {
+            await once(encFileWriteStream, "drain");
+        }
+    };
+    try {
+        for await (const data of stream) {
+            fileLength += data.length;
+            if (
+                type === "remote" &&
+                opts?.maxContentLength &&
+                fileLength + data.length > opts.maxContentLength
+            ) {
+                throw new Boom(`content length exceeded when encrypting "${type}"`, {
+                    data: { media: media, type: type },
+                });
+            }
+            if (originalFileStream) {
+                if (!originalFileStream.write(data)) {
+                    await once(originalFileStream, "drain");
+                }
+            }
+            sha256Plain.update(data);
+            await onChunk(aes.update(data));
+        }
+        await onChunk(aes.final());
+        const mac = hmac.digest().slice(0, 10);
+        sha256Enc.update(mac);
+        const fileSha256 = sha256Plain.digest();
+        const fileEncSha256 = sha256Enc.digest();
+        encFileWriteStream.write(mac);
+        const encFinishPromise = once(encFileWriteStream, "finish");
+        const originalFinishPromise = originalFileStream
+            ? once(originalFileStream, "finish")
+            : Promise.resolve();
+        encFileWriteStream.end();
+        originalFileStream?.end?.();
+        stream.destroy();
+        await encFinishPromise;
+        await originalFinishPromise;
+        logger?.debug("encrypted data successfully");
+        return {
+            mediaKey: mediaKey,
+            originalFilePath: originalFilePath,
+            encFilePath: encFilePath,
+            mac: mac,
+            fileEncSha256: fileEncSha256,
+            fileSha256: fileSha256,
+            fileLength: fileLength,
+        };
+    } catch (error) {
+        encFileWriteStream.destroy();
+        originalFileStream?.destroy?.();
+        aes.destroy();
+        hmac.destroy();
+        sha256Plain.destroy();
+        sha256Enc.destroy();
+        stream.destroy();
+        try {
+            await promises.unlink(encFilePath);
+            if (originalFilePath) {
+                await promises.unlink(originalFilePath);
+            }
+        } catch (err) {
+            logger?.error({ err: err }, "failed deleting tmp files");
+        }
+        throw error;
+    }
+};
+const DEF_HOST = "mmg.whatsapp.net";
+const AES_CHUNK_SIZE = 16;
+const toSmallestChunkSize = (num) => Math.floor(num / AES_CHUNK_SIZE) * AES_CHUNK_SIZE;
+const getUrlFromDirectPath = (directPath) => `https://${DEF_HOST}${directPath}`;
+const downloadContentFromMessage = async (
+    { mediaKey: mediaKey, directPath: directPath, url: url },
+    type,
+    opts = {}
+) => {
+    const isValidMediaUrl = url?.startsWith("https://mmg.whatsapp.net/");
+    const downloadUrl = isValidMediaUrl ? url : getUrlFromDirectPath(directPath);
+    if (!downloadUrl) {
+        throw new Boom("No valid media URL or directPath present in message", {
+            statusCode: 400,
+        });
+    }
+    const keys = await getMediaKeys(mediaKey, type);
+    return downloadEncryptedContent(downloadUrl, keys, opts);
+};
+const downloadEncryptedContent = async (
+    downloadUrl,
+    { cipherKey: cipherKey, iv: iv },
+    { startByte: startByte, endByte: endByte, options: options } = {}
+) => {
+    let bytesFetched = 0;
+    let startChunk = 0;
+    let firstBlockIsIV = false;
+    if (startByte) {
+        const chunk = toSmallestChunkSize(startByte || 0);
+        if (chunk) {
+            startChunk = chunk - AES_CHUNK_SIZE;
+            bytesFetched = chunk;
+            firstBlockIsIV = true;
+        }
+    }
+    const endChunk = endByte ? toSmallestChunkSize(endByte || 0) + AES_CHUNK_SIZE : undefined;
+    const headersInit = options?.headers ? options.headers : undefined;
+    const headers = {
+        ...(headersInit
+            ? Array.isArray(headersInit)
+                ? Object.fromEntries(headersInit)
+                : headersInit
+            : {}),
+        Origin: DEFAULT_ORIGIN,
+    };
+    if (startChunk || endChunk) {
+        headers.Range = `bytes=${startChunk}-`;
+        if (endChunk) {
+            headers.Range += endChunk;
+        }
+    }
+    const fetched = await getHttpStream(downloadUrl, {
+        ...(options || {}),
+        headers: headers,
+    });
+    let remainingBytes = Buffer.from([]);
+    let aes;
+    const pushBytes = (bytes, push) => {
+        if (startByte || endByte) {
+            const start =
+                bytesFetched >= startByte ? undefined : Math.max(startByte - bytesFetched, 0);
+            const end =
+                bytesFetched + bytes.length < endByte
+                    ? undefined
+                    : Math.max(endByte - bytesFetched, 0);
+            push(bytes.slice(start, end));
+            bytesFetched += bytes.length;
+        } else {
+            push(bytes);
+        }
+    };
+    const output = new Transform({
+        transform(chunk, _, callback) {
+            let data = Buffer.concat([remainingBytes, chunk]);
+            const decryptLength = toSmallestChunkSize(data.length);
+            remainingBytes = data.slice(decryptLength);
+            data = data.slice(0, decryptLength);
+            if (!aes) {
+                let ivValue = iv;
+                if (firstBlockIsIV) {
+                    ivValue = data.slice(0, AES_CHUNK_SIZE);
+                    data = data.slice(AES_CHUNK_SIZE);
+                }
+                aes = createDecipheriv("aes-256-cbc", cipherKey, ivValue);
+                if (endByte) {
+                    aes.setAutoPadding(false);
+                }
+            }
+            try {
+                pushBytes(aes.update(data), (b) => this.push(b));
+                callback();
+            } catch (error) {
+                callback(error);
+            }
+        },
+        final(callback) {
+            try {
+                pushBytes(aes.final(), (b) => this.push(b));
+                callback();
+            } catch (error) {
+                callback(error);
+            }
+        },
+    });
+    return fetched.pipe(output, { end: true });
+};
+function extensionForMediaMessage(message) {
+    const getExtension = (mimetype) => mimetype.split("")[0].split("/")[1];
+    const type = Object.keys(message)[0];
+    let extension;
+    if (type === "locationMessage" || type === "liveLocationMessage" || type === "productMessage") {
+        extension = ".jpeg";
+    } else {
+        const messageContent = message[type];
+        extension = getExtension(messageContent.mimetype);
+    }
+    return extension;
+}
+const isNodeRuntime = () =>
+    typeof process !== "undefined" &&
+    process.versions?.node !== null &&
+    typeof process.versions.bun === "undefined" &&
+    typeof globalThis.Deno === "undefined";
+const uploadWithNodeHttp = async (
+    { url: url, filePath: filePath, headers: headers, timeoutMs: timeoutMs, agent: agent },
+    redirectCount = 0
+) => {
+    if (redirectCount > 5) {
+        throw new Error("Too many redirects");
+    }
+    const parsedUrl = new URL(url);
+    const httpModule = parsedUrl.protocol === "https:" ? require("https") : require("http");
+    const fileStats = await promises.stat(filePath);
+    const fileSize = fileStats.size;
+    return new Promise((resolve, reject) => {
+        const req = httpModule.request(
+            {
+                hostname: parsedUrl.hostname,
+                port: parsedUrl.port || (parsedUrl.protocol === "https:" ? 443 : 80),
+                path: parsedUrl.pathname + parsedUrl.search,
+                method: "POST",
+                headers: { ...headers, "Content-Length": fileSize },
+                agent: agent,
+                timeout: timeoutMs,
+            },
+            (res) => {
+                if (
+                    res.statusCode &&
+                    res.statusCode >= 300 &&
+                    res.statusCode < 400 &&
+                    res.headers.location
+                ) {
+                    res.resume();
+                    const newUrl = new URL(res.headers.location, url).toString();
+                    resolve(
+                        uploadWithNodeHttp(
+                            {
+                                url: newUrl,
+                                filePath: filePath,
+                                headers: headers,
+                                timeoutMs: timeoutMs,
+                                agent: agent,
+                            },
+                            redirectCount + 1
+                        )
+                    );
+                    return;
+                }
+                let body = "";
+                res.on("data", (chunk) => (body += chunk));
+                res.on("end", () => {
+                    try {
+                        resolve(JSON.parse(body));
+                    } catch {
+                        resolve(undefined);
+                    }
+                });
+            }
+        );
+        req.on("error", reject);
+        req.on("timeout", () => {
+            req.destroy();
+            reject(new Error("Upload timeout"));
+        });
+        const stream = createReadStream(filePath);
+        stream.pipe(req);
+        stream.on("error", (err) => {
+            req.destroy();
+            reject(err);
+        });
+    });
+};
+const uploadWithFetch = async ({
+    url: url,
+    filePath: filePath,
+    headers: headers,
+    timeoutMs: timeoutMs,
+    agent: agent,
+}) => {
+    const nodeStream = createReadStream(filePath);
+    const webStream = Readable.toWeb(nodeStream);
+    const response = await fetch(url, {
+        dispatcher: agent,
+        method: "POST",
+        body: webStream,
+        headers: headers,
+        duplex: "half",
+        signal: timeoutMs ? AbortSignal.timeout(timeoutMs) : undefined,
+    });
+    try {
+        return await response.json();
+    } catch {
+        return undefined;
+    }
+};
+const uploadMedia = async (params, logger) => {
+    if (isNodeRuntime()) {
+        logger?.debug("Using Node.js https module for upload (avoids undici buffering bug)");
+        return uploadWithNodeHttp(params);
+    } else {
+        logger?.debug("Using web-standard Fetch API for upload");
+        return uploadWithFetch(params);
+    }
+};
+const getWAUploadToServer =
+    (
+        {
+            customUploadHosts: customUploadHosts,
+            fetchAgent: fetchAgent,
+            logger: logger,
+            options: options,
+        },
+        refreshMediaConn
+    ) =>
+    async (
+        filePath,
+        { mediaType: mediaType, fileEncSha256B64: fileEncSha256B64, timeoutMs: timeoutMs }
+    ) => {
+        let uploadInfo = await refreshMediaConn(false);
+        let urls;
+        const hosts = [...customUploadHosts, ...uploadInfo.hosts];
+        fileEncSha256B64 = encodeBase64EncodedStringForUpload(fileEncSha256B64);
+        const customHeaders = (() => {
+            const hdrs = options?.headers;
+            if (!hdrs) return {};
+            return Array.isArray(hdrs) ? Object.fromEntries(hdrs) : hdrs;
+        })();
+        const headers = {
+            ...customHeaders,
+            "Content-Type": "application/octet-stream",
+            Origin: DEFAULT_ORIGIN,
+        };
+        for (const { hostname: hostname } of hosts) {
+            logger.debug(`uploading to "${hostname}"`);
+            const auth = encodeURIComponent(uploadInfo.auth);
+            const url = `https://${hostname}${MEDIA_PATH_MAP[mediaType]}/${fileEncSha256B64}?auth=${auth}&token=${fileEncSha256B64}`;
+            let result;
+            try {
+                result = await uploadMedia(
+                    {
+                        url: url,
+                        filePath: filePath,
+                        headers: headers,
+                        timeoutMs: timeoutMs,
+                        agent: fetchAgent,
+                    },
+                    logger
+                );
+                if (result?.url || result?.direct_path) {
+                    urls = {
+                        mediaUrl: result.url,
+                        directPath: result.direct_path,
+                        meta_hmac: result.meta_hmac,
+                        fbid: result.fbid,
+                        ts: result.ts,
+                    };
+                    break;
+                } else {
+                    uploadInfo = await refreshMediaConn(true);
+                    throw new Error(`upload failed, reason: ${JSON.stringify(result)}`);
+                }
+            } catch (error) {
+                const isLast = hostname === hosts[uploadInfo.hosts.length - 1]?.hostname;
+                logger.warn(
+                    { trace: error?.stack, uploadResult: result },
+                    `Error in uploading to ${hostname} ${isLast ? "" : ", retrying..."}`
+                );
+            }
+        }
+        if (!urls) {
+            throw new Boom("Media upload failed on all hosts", { statusCode: 500 });
+        }
+        return urls;
+    };
+const getMediaRetryKey = (mediaKey) =>
+    hkdf(mediaKey, 32, { info: "WhatsApp Media Retry Notification" });
+const encryptMediaRetryRequest = async (key, mediaKey, meId) => {
+    const recp = { stanzaId: key.id };
+    const recpBuffer = proto.ServerErrorReceipt.encode(recp).finish();
+    const iv = randomBytes(12);
+    const retryKey = await getMediaRetryKey(mediaKey);
+    const ciphertext = aesEncryptGCM(recpBuffer, retryKey, iv, Buffer.from(key.id));
+    const req = {
+        tag: "receipt",
+        attrs: { id: key.id, to: jidNormalizedUser(meId), type: "server-error" },
+        content: [
+            {
+                tag: "encrypt",
+                attrs: {},
+                content: [
+                    { tag: "enc_p", attrs: {}, content: ciphertext },
+                    { tag: "enc_iv", attrs: {}, content: iv },
+                ],
+            },
+            {
+                tag: "rmr",
+                attrs: {
+                    jid: key.remoteJid,
+                    from_me: (!!key.fromMe).toString(),
+                    participant: key.participant || undefined,
+                },
+            },
+        ],
+    };
+    return req;
+};
+const decodeMediaRetryNode = (node) => {
+    const rmrNode = getBinaryNodeChild(node, "rmr");
+    const event = {
+        key: {
+            id: node.attrs.id,
+            remoteJid: rmrNode.attrs.jid,
+            fromMe: rmrNode.attrs.from_me === "true",
+            participant: rmrNode.attrs.participant,
+        },
+    };
+    const errorNode = getBinaryNodeChild(node, "error");
+    if (errorNode) {
+        const errorCode = +errorNode.attrs.code;
+        event.error = new Boom(`Failed to re-upload media (${errorCode})`, {
+            data: errorNode.attrs,
+            statusCode: getStatusCodeForMediaRetry(errorCode),
+        });
+    } else {
+        const encryptedInfoNode = getBinaryNodeChild(node, "encrypt");
+        const ciphertext = getBinaryNodeChildBuffer(encryptedInfoNode, "enc_p");
+        const iv = getBinaryNodeChildBuffer(encryptedInfoNode, "enc_iv");
+        if (ciphertext && iv) {
+            event.media = { ciphertext: ciphertext, iv: iv };
+        } else {
+            event.error = new Boom("Failed to re-upload media (missing ciphertext)", {
+                statusCode: 404,
+            });
+        }
+    }
+    return event;
+};
+const decryptMediaRetryData = async ({ ciphertext: ciphertext, iv: iv }, mediaKey, msgId) => {
+    const retryKey = await getMediaRetryKey(mediaKey);
+    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 = {
+    [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 = {
+    hkdfInfoKey: hkdfInfoKey,
+    getMediaKeys: getMediaKeys,
+    extractVideoThumb: extractVideoThumb,
+    extractImageThumb: extractImageThumb,
+    encodeBase64EncodedStringForUpload: encodeBase64EncodedStringForUpload,
+    generateProfilePicture: generateProfilePicture,
+    mediaMessageSHA256B64: mediaMessageSHA256B64,
+    getAudioDuration: getAudioDuration,
+    getAudioWaveform: getAudioWaveform,
+    toReadable: toReadable,
+    toBuffer: toBuffer,
+    getStream: getStream,
+    generateThumbnail: generateThumbnail,
+    getHttpStream: getHttpStream,
+    encryptedStream: encryptedStream,
+    getUrlFromDirectPath: getUrlFromDirectPath,
+    downloadContentFromMessage: downloadContentFromMessage,
+    downloadEncryptedContent: downloadEncryptedContent,
+    extensionForMediaMessage: extensionForMediaMessage,
+    uploadWithNodeHttp: uploadWithNodeHttp,
+    getRawMediaUploadData: getRawMediaUploadData,
+    getWAUploadToServer: getWAUploadToServer,
+    getMediaRetryKey: getMediaRetryKey,
+    encryptMediaRetryRequest: encryptMediaRetryRequest,
+    decodeMediaRetryNode: decodeMediaRetryNode,
+    decryptMediaRetryData: decryptMediaRetryData,
+    getStatusCodeForMediaRetry: getStatusCodeForMediaRetry,
+    MEDIA_RETRY_STATUS_MAP: MEDIA_RETRY_STATUS_MAP,
+};