gifted-baileys 1.5.0 → 1.5.5

package/{lib → src}/Utils/messages.js

@@ -1,9 +1,26 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.assertMediaContent = exports.downloadMediaMessage = exports.aggregateMessageKeysNotFromMe = exports.getAggregateVotesInPollMessage = exports.updateMessageWithPollUpdate = exports.updateMessageWithReaction = exports.updateMessageWithReceipt = exports.getDevice = exports.extractMessageContent = exports.normalizeMessageContent = exports.getContentType = exports.generateWAMessage = exports.generateWAMessageFromContent = exports.generateWAMessageContent = exports.generateForwardMessageContent = exports.prepareDisappearingMessageSettingContent = exports.prepareWAMessageMedia = exports.generateLinkPreviewIfRequired = exports.extractUrlFromText = void 0;
+exports.assertMediaContent = exports.downloadMediaMessage = exports.aggregateMessageKeysNotFromMe = exports.updateMessageWithPollUpdate = exports.updateMessageWithReaction = exports.updateMessageWithReceipt = exports.getDevice = exports.extractMessageContent = exports.normalizeMessageContent = exports.getContentType = exports.generateWAMessage = exports.generateWAMessageFromContent = exports.generateWAMessageContent = exports.generateForwardMessageContent = exports.prepareDisappearingMessageSettingContent = exports.prepareWAMessageMedia = exports.generateLinkPreviewIfRequired = exports.extractUrlFromText = void 0;
+exports.getAggregateVotesInPollMessage = getAggregateVotesInPollMessage;
 const boom_1 = require("@hapi/boom");
 const axios_1 = __importDefault(require("axios"));
 const crypto_1 = require("crypto");
@@ -36,25 +53,22 @@ const ButtonType = WAProto_1.proto.Message.ButtonsMessage.HeaderType;
  * @param text eg. hello https://google.com
  * @returns the URL, eg. https://google.com
  */
-const extractUrlFromText = (text) => {
-    var _a;
-    return (!Defaults_1.URL_EXCLUDE_REGEX.test(text) ? (_a = text.match(Defaults_1.URL_REGEX)) === null || _a === void 0 ? void 0 : _a[0] : undefined);
-};
+const extractUrlFromText = (text) => { var _a; return (_a = text.match(Defaults_1.URL_REGEX)) === null || _a === void 0 ? void 0 : _a[0]; };
 exports.extractUrlFromText = extractUrlFromText;
-const generateLinkPreviewIfRequired = async (text, getUrlInfo, logger) => {
+const generateLinkPreviewIfRequired = (text, getUrlInfo, logger) => __awaiter(void 0, void 0, void 0, function* () {
     const url = (0, exports.extractUrlFromText)(text);
     if (!!getUrlInfo && url) {
         try {
-            const urlInfo = await getUrlInfo(url);
+            const urlInfo = yield getUrlInfo(url);
             return urlInfo;
         }
         catch (error) { // ignore if fails
             logger === null || logger === void 0 ? void 0 : logger.warn({ trace: error.stack }, 'url generation failed');
         }
     }
-};
+});
 exports.generateLinkPreviewIfRequired = generateLinkPreviewIfRequired;
-const assertColor = async (color) => {
+const assertColor = (color) => __awaiter(void 0, void 0, void 0, function* () {
     let assertedColor;
     if (typeof color === 'number') {
         assertedColor = color > 0 ? color : 0xffffffff + Number(color) + 1;
@@ -67,8 +81,8 @@ const assertColor = async (color) => {
         assertedColor = parseInt(hex, 16);
         return assertedColor;
     }
-};
-const prepareWAMessageMedia = async (message, options) => {
+});
+const prepareWAMessageMedia = (message, options) => __awaiter(void 0, void 0, void 0, function* () {
     const logger = options.logger;
     let mediaType;
     for (const key of Defaults_1.MEDIA_KEYS) {
@@ -79,10 +93,7 @@ const prepareWAMessageMedia = async (message, options) => {
     if (!mediaType) {
         throw new boom_1.Boom('Invalid media type', { statusCode: 400 });
     }
-    const uploadData = {
-        ...message,
-        media: message[mediaType]
-    };
+    const uploadData = Object.assign(Object.assign({}, message), { media: message[mediaType] });
     delete uploadData[mediaType];
     // check if cacheable + generate cache key
     const cacheableKey = typeof uploadData.media === 'object' &&
@@ -104,7 +115,7 @@ const prepareWAMessageMedia = async (message, options) => {
             logger === null || logger === void 0 ? void 0 : logger.debug({ cacheableKey }, 'got media cache hit');
             const obj = Types_1.WAProto.Message.decode(mediaBuff);
             const key = `${mediaType}Message`;
-            Object.assign(obj[key], { ...uploadData, media: undefined });
+            Object.assign(obj[key], Object.assign(Object.assign({}, uploadData), { media: undefined }));
             return obj;
         }
     }
@@ -114,23 +125,23 @@ const prepareWAMessageMedia = async (message, options) => {
     const requiresWaveformProcessing = mediaType === 'audio' && uploadData.ptt === true;
     const requiresAudioBackground = options.backgroundColor && mediaType === 'audio' && uploadData.ptt === true;
     const requiresOriginalForSomeProcessing = requiresDurationComputation || requiresThumbnailComputation;
-    const { mediaKey, encWriteStream, bodyPath, fileEncSha256, fileSha256, fileLength, didSaveToTmpPath } = await (0, messages_media_1.encryptedStream)(uploadData.media, options.mediaTypeOverride || mediaType, {
+    const { mediaKey, encWriteStream, bodyPath, fileEncSha256, fileSha256, fileLength, didSaveToTmpPath, } = yield (options.newsletter ? messages_media_1.prepareStream : messages_media_1.encryptedStream)(uploadData.media, options.mediaTypeOverride || mediaType, {
         logger,
         saveOriginalFileIfRequired: requiresOriginalForSomeProcessing,
         opts: options.options
     });
     // url safe Base64 encode the SHA256 hash of the body
-    const fileEncSha256B64 = fileEncSha256.toString('base64');
-    const [{ mediaUrl, directPath }] = await Promise.all([
-        (async () => {
-            const result = await options.upload(encWriteStream, { fileEncSha256B64, mediaType, timeoutMs: options.mediaUploadTimeoutMs });
+    const fileEncSha256B64 = (options.newsletter ? fileSha256 : fileEncSha256 !== null && fileEncSha256 !== void 0 ? fileEncSha256 : fileSha256).toString('base64');
+    const [{ mediaUrl, directPath, handle }] = yield Promise.all([
+        (() => __awaiter(void 0, void 0, void 0, function* () {
+            const result = yield options.upload(encWriteStream, { fileEncSha256B64, mediaType, timeoutMs: options.mediaUploadTimeoutMs });
             logger === null || logger === void 0 ? void 0 : logger.debug({ mediaType, cacheableKey }, 'uploaded media');
             return result;
-        })(),
-        (async () => {
+        }))(),
+        (() => __awaiter(void 0, void 0, void 0, function* () {
             try {
                 if (requiresThumbnailComputation) {
-                    const { thumbnail, originalImageDimensions } = await (0, messages_media_1.generateThumbnail)(bodyPath, mediaType, options);
+                    const { thumbnail, originalImageDimensions } = yield (0, messages_media_1.generateThumbnail)(bodyPath, mediaType, options);
                     uploadData.jpegThumbnail = thumbnail;
                     if (!uploadData.width && originalImageDimensions) {
                         uploadData.width = originalImageDimensions.width;
@@ -140,54 +151,51 @@ const prepareWAMessageMedia = async (message, options) => {
                     logger === null || logger === void 0 ? void 0 : logger.debug('generated thumbnail');
                 }
                 if (requiresDurationComputation) {
-                    uploadData.seconds = await (0, messages_media_1.getAudioDuration)(bodyPath);
+                    uploadData.seconds = yield (0, messages_media_1.getAudioDuration)(bodyPath);
                     logger === null || logger === void 0 ? void 0 : logger.debug('computed audio duration');
                 }
                 if (requiresWaveformProcessing) {
-                    uploadData.waveform = await (0, messages_media_1.getAudioWaveform)(bodyPath, logger);
+                    uploadData.waveform = yield (0, messages_media_1.getAudioWaveform)(bodyPath, logger);
                     logger === null || logger === void 0 ? void 0 : logger.debug('processed waveform');
                 }
                 if (requiresWaveformProcessing) {
-                    uploadData.waveform = await (0, messages_media_1.getAudioWaveform)(bodyPath, logger);
+                    uploadData.waveform = yield (0, messages_media_1.getAudioWaveform)(bodyPath, logger);
                     logger === null || logger === void 0 ? void 0 : logger.debug('processed waveform');
                 }
                 if (requiresAudioBackground) {
-                    uploadData.backgroundArgb = await assertColor(options.backgroundColor);
+                    uploadData.backgroundArgb = yield assertColor(options.backgroundColor);
                     logger === null || logger === void 0 ? void 0 : logger.debug('computed backgroundColor audio status');
                 }
             }
             catch (error) {
                 logger === null || logger === void 0 ? void 0 : logger.warn({ trace: error.stack }, 'failed to obtain extra info');
             }
-        })(),
+        }))(),
     ])
-        .finally(async () => {
-        encWriteStream.destroy();
+        .finally(() => __awaiter(void 0, void 0, void 0, function* () {
+        if (!Buffer.isBuffer(encWriteStream)) {
+            encWriteStream.destroy();
+        }
         // remove tmp files
         if (didSaveToTmpPath && bodyPath) {
-            await fs_1.promises.unlink(bodyPath);
+            yield fs_1.promises.unlink(bodyPath);
             logger === null || logger === void 0 ? void 0 : logger.debug('removed tmp files');
         }
-    });
+    }));
     const obj = Types_1.WAProto.Message.fromObject({
-        [`${mediaType}Message`]: MessageTypeProto[mediaType].fromObject({
-            url: mediaUrl,
-            directPath,
-            mediaKey,
-            fileEncSha256,
-            fileSha256,
-            fileLength,
-            mediaKeyTimestamp: (0, generics_1.unixTimestampSeconds)(),
-            ...uploadData,
-            media: undefined
-        })
+        [`${mediaType}Message`]: MessageTypeProto[mediaType].fromObject(Object.assign(Object.assign({ url: handle ? undefined : mediaUrl, directPath, mediaKey: mediaKey, fileEncSha256: fileEncSha256, fileSha256,
+            fileLength, mediaKeyTimestamp: handle ? undefined : (0, generics_1.unixTimestampSeconds)() }, uploadData), { media: undefined }))
     });
+    if (uploadData.ptv) {
+        obj.ptvMessage = obj.videoMessage;
+        delete obj.videoMessage;
+    }
     if (cacheableKey) {
         logger === null || logger === void 0 ? void 0 : logger.debug({ cacheableKey }, 'set cache');
         options.mediaCache.set(cacheableKey, Types_1.WAProto.Message.encode(obj).finish());
     }
     return obj;
-};
+});
 exports.prepareWAMessageMedia = prepareWAMessageMedia;
 const prepareDisappearingMessageSettingContent = (ephemeralExpiration) => {
     ephemeralExpiration = ephemeralExpiration || 0;
@@ -235,7 +243,7 @@ const generateForwardMessageContent = (message, forceForward) => {
     return content;
 };
 exports.generateForwardMessageContent = generateForwardMessageContent;
-const generateWAMessageContent = async (message, options) => {
+const generateWAMessageContent = (message, options) => __awaiter(void 0, void 0, void 0, function* () {
     var _a;
     var _b;
     let m = {};
@@ -243,7 +251,7 @@ const generateWAMessageContent = async (message, options) => {
         const extContent = { text: message.text };
         let urlInfo = message.linkPreview;
         if (typeof urlInfo === 'undefined') {
-            urlInfo = await (0, exports.generateLinkPreviewIfRequired)(message.text, options.getUrlInfo, options.logger);
+            urlInfo = yield (0, exports.generateLinkPreviewIfRequired)(message.text, options.getUrlInfo, options.logger);
         }
         if (urlInfo) {
             extContent.canonicalUrl = urlInfo['canonical-url'];
@@ -264,7 +272,7 @@ const generateWAMessageContent = async (message, options) => {
             }
         }
         if (options.backgroundColor) {
-            extContent.backgroundArgb = await assertColor(options.backgroundColor);
+            extContent.backgroundArgb = yield assertColor(options.backgroundColor);
         }
         if (options.font) {
             extContent.font = options.font;
@@ -326,17 +334,11 @@ const generateWAMessageContent = async (message, options) => {
         }
     }
     else if ('product' in message) {
-        const { imageMessage } = await (0, exports.prepareWAMessageMedia)({ image: message.product.productImage }, options);
-        m.productMessage = Types_1.WAProto.Message.ProductMessage.fromObject({
-            ...message,
-            product: {
-                ...message.product,
-                productImage: imageMessage,
-            }
-        });
+        const { imageMessage } = yield (0, exports.prepareWAMessageMedia)({ image: message.product.productImage }, options);
+        m.productMessage = Types_1.WAProto.Message.ProductMessage.fromObject(Object.assign(Object.assign({}, message), { product: Object.assign(Object.assign({}, message.product), { productImage: imageMessage }) }));
     }
     else if ('listReply' in message) {
-        m.listResponseMessage = { ...message.listReply };
+        m.listResponseMessage = Object.assign({}, message.listReply);
     }
     else if ('poll' in message) {
         (_b = message.poll).selectableCount || (_b.selectableCount = 0);
@@ -366,11 +368,11 @@ const generateWAMessageContent = async (message, options) => {
         m.requestPhoneNumberMessage = {};
     }
     else {
-        m = await (0, exports.prepareWAMessageMedia)(message, options);
+        m = yield (0, exports.prepareWAMessageMedia)(message, options);
     }
     if ('buttons' in message && !!message.buttons) {
         const buttonsMessage = {
-            buttons: message.buttons.map(b => ({ ...b, type: WAProto_1.proto.Message.ButtonsMessage.Button.Type.RESPONSE }))
+            buttons: message.buttons.map(b => (Object.assign(Object.assign({}, b), { type: WAProto_1.proto.Message.ButtonsMessage.Button.Type.RESPONSE })))
         };
         if ('text' in message) {
             buttonsMessage.contentText = message.text;
@@ -447,7 +449,7 @@ const generateWAMessageContent = async (message, options) => {
         m[messageType].contextInfo = message.contextInfo;
     }
     return Types_1.WAProto.Message.fromObject(m);
-};
+});
 exports.generateWAMessageContent = generateWAMessageContent;
 const generateWAMessageFromContent = (jid, message, options) => {
     // set timestamp to now
@@ -455,7 +457,8 @@ const generateWAMessageFromContent = (jid, message, options) => {
     if (!options.timestamp) {
         options.timestamp = new Date();
     }
-    const key = Object.keys(message)[0];
+    const innerMessage = (0, exports.normalizeMessageContent)(message);
+    const key = (0, exports.getContentType)(innerMessage);
     const timestamp = (0, generics_1.unixTimestampSeconds)(options.timestamp);
     const { quoted, userJid } = options;
     if (quoted) {
@@ -468,7 +471,7 @@ const generateWAMessageFromContent = (jid, message, options) => {
         if (typeof quotedContent === 'object' && quotedContent && 'contextInfo' in quotedContent) {
             delete quotedContent.contextInfo;
         }
-        const contextInfo = message[key].contextInfo || {};
+        const contextInfo = innerMessage[key].contextInfo || {};
         contextInfo.participant = (0, WABinary_1.jidNormalizedUser)(participant);
         contextInfo.stanzaId = quoted.key.id;
         contextInfo.quotedMessage = quotedMsg;
@@ -477,7 +480,7 @@ const generateWAMessageFromContent = (jid, message, options) => {
         if (jid !== quoted.key.remoteJid) {
             contextInfo.remoteJid = quoted.key.remoteJid;
         }
-        message[key].contextInfo = contextInfo;
+        innerMessage[key].contextInfo = contextInfo;
     }
     if (
     // if we want to send a disappearing message
@@ -486,11 +489,7 @@ const generateWAMessageFromContent = (jid, message, options) => {
         key !== 'protocolMessage' &&
         // already not converted to disappearing message
         key !== 'ephemeralMessage') {
-        message[key].contextInfo = {
-            ...(message[key].contextInfo || {}),
-            expiration: options.ephemeralExpiration || Defaults_1.WA_DEFAULT_EPHEMERAL,
-            //ephemeralSettingTimestamp: options.ephemeralOptions.eph_setting_ts?.toString()
-        };
+        innerMessage[key].contextInfo = Object.assign(Object.assign({}, (innerMessage[key].contextInfo || {})), { expiration: options.ephemeralExpiration || Defaults_1.WA_DEFAULT_EPHEMERAL });
     }
     message = Types_1.WAProto.Message.fromObject(message);
     const messageJSON = {
@@ -508,12 +507,12 @@ const generateWAMessageFromContent = (jid, message, options) => {
     return Types_1.WAProto.WebMessageInfo.fromObject(messageJSON);
 };
 exports.generateWAMessageFromContent = generateWAMessageFromContent;
-const generateWAMessage = async (jid, content, options) => {
+const generateWAMessage = (jid, content, options) => __awaiter(void 0, void 0, void 0, function* () {
     var _a;
     // ensure msg ID is with every log
     options.logger = (_a = options === null || options === void 0 ? void 0 : options.logger) === null || _a === void 0 ? void 0 : _a.child({ msgId: options.messageId });
-    return (0, exports.generateWAMessageFromContent)(jid, await (0, exports.generateWAMessageContent)(content, options), options);
-};
+    return (0, exports.generateWAMessageFromContent)(jid, yield (0, exports.generateWAMessageContent)(content, Object.assign({ newsletter: (0, WABinary_1.isJidNewsLetter)(jid) }, options)), options);
+});
 exports.generateWAMessage = generateWAMessage;
 /** Get the key to access the true type of content */
 const getContentType = (content) => {
@@ -673,7 +672,6 @@ function getAggregateVotesInPollMessage({ message, pollUpdates }, meId) {
     }
     return Object.values(voteHashMap);
 }
-exports.getAggregateVotesInPollMessage = getAggregateVotesInPollMessage;
 /** Given a list of message keys, aggregates them by chat & sender. Useful for sending read receipts in bulk */
 const aggregateMessageKeysNotFromMe = (keys) => {
     const keyMap = {};
@@ -697,60 +695,73 @@ const REUPLOAD_REQUIRED_STATUS = [410, 404];
 /**
  * Downloads the given message. Throws an error if it's not a media message
  */
-const downloadMediaMessage = async (message, type, options, ctx) => {
-    var _a;
-    try {
-        const result = await downloadMsg();
-        return result;
-    }
-    catch (error) {
+const downloadMediaMessage = (message, type, options, ctx) => __awaiter(void 0, void 0, void 0, function* () {
+    const result = yield downloadMsg()
+        .catch((error) => __awaiter(void 0, void 0, void 0, function* () {
+        var _a;
         if (ctx) {
             if (axios_1.default.isAxiosError(error)) {
                 // check if the message requires a reupload
                 if (REUPLOAD_REQUIRED_STATUS.includes((_a = error.response) === null || _a === void 0 ? void 0 : _a.status)) {
                     ctx.logger.info({ key: message.key }, 'sending reupload media request...');
                     // request reupload
-                    message = await ctx.reuploadRequest(message);
-                    const result = await downloadMsg();
+                    message = yield ctx.reuploadRequest(message);
+                    const result = yield downloadMsg();
                     return result;
                 }
             }
         }
         throw error;
-    }
-    async function downloadMsg() {
-        const mContent = (0, exports.extractMessageContent)(message.message);
-        if (!mContent) {
-            throw new boom_1.Boom('No message present', { statusCode: 400, data: message });
-        }
-        const contentType = (0, exports.getContentType)(mContent);
-        let mediaType = contentType === null || contentType === void 0 ? void 0 : contentType.replace('Message', '');
-        const media = mContent[contentType];
-        if (!media || typeof media !== 'object' || (!('url' in media) && !('thumbnailDirectPath' in media))) {
-            throw new boom_1.Boom(`"${contentType}" message is not a media message`);
-        }
-        let download;
-        if ('thumbnailDirectPath' in media && !('url' in media)) {
-            download = {
-                directPath: media.thumbnailDirectPath,
-                mediaKey: media.mediaKey
-            };
-            mediaType = 'thumbnail-link';
-        }
-        else {
-            download = media;
-        }
-        const stream = await (0, messages_media_1.downloadContentFromMessage)(download, mediaType, options);
-        if (type === 'buffer') {
-            const bufferArray = [];
-            for await (const chunk of stream) {
-                bufferArray.push(chunk);
+    }));
+    return result;
+    function downloadMsg() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, e_1, _b, _c;
+            const mContent = (0, exports.extractMessageContent)(message.message);
+            if (!mContent) {
+                throw new boom_1.Boom('No message present', { statusCode: 400, data: message });
             }
-            return Buffer.concat(bufferArray);
-        }
-        return stream;
+            const contentType = (0, exports.getContentType)(mContent);
+            let mediaType = contentType === null || contentType === void 0 ? void 0 : contentType.replace('Message', '');
+            const media = mContent[contentType];
+            if (!media || typeof media !== 'object' || (!('url' in media) && !('thumbnailDirectPath' in media))) {
+                throw new boom_1.Boom(`"${contentType}" message is not a media message`);
+            }
+            let download;
+            if ('thumbnailDirectPath' in media && !('url' in media)) {
+                download = {
+                    directPath: media.thumbnailDirectPath,
+                    mediaKey: media.mediaKey
+                };
+                mediaType = 'thumbnail-link';
+            }
+            else {
+                download = media;
+            }
+            const stream = yield (0, messages_media_1.downloadContentFromMessage)(download, mediaType, options);
+            if (type === 'buffer') {
+                const bufferArray = [];
+                try {
+                    for (var _d = true, stream_1 = __asyncValues(stream), stream_1_1; stream_1_1 = yield stream_1.next(), _a = stream_1_1.done, !_a; _d = true) {
+                        _c = stream_1_1.value;
+                        _d = false;
+                        const chunk = _c;
+                        bufferArray.push(chunk);
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (!_d && !_a && (_b = stream_1.return)) yield _b.call(stream_1);
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+                return Buffer.concat(bufferArray);
+            }
+            return stream;
+        });
     }
-};
+});
 exports.downloadMediaMessage = downloadMediaMessage;
 /** Checks whether the given message is a media message; if it is returns the inner content */
 const assertMediaContent = (content) => {