@badzz88/baileys 8.5.2 → 8.5.4

package/lib/Utils/make-mutex.js

@@ -25,7 +25,7 @@ const makeKeyedMutex = () => {
 				return await entry.mutex.runExclusive(task)
 			} finally {
 				entry.refCount--
-				// only delete it if this is still the current entry
+				
 				if (entry.refCount === 0 && map.get(key) === entry) {
 					map.delete(key)
 				}

package/lib/Utils/message-inspect.js

@@ -0,0 +1,393 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+
+const boom_1 = require('@hapi/boom')
+const { proto } = require('../../WAProto/index.js')
+const WABinary_1 = require('../WABinary')
+const crypto_2 = require('./crypto')
+const generics_1 = require('./generics')
+const messages_media_1 = require('./messages-media')
+
+
+const getContentType = content => {
+	if (content) {
+		const keys = Object.keys(content)
+		const key = keys.find(k => (k === 'conversation' || k.includes('Message')) && k !== 'senderKeyDistributionMessage')
+		return key
+	}
+}
+exports.getContentType = getContentType
+
+
+const normalizeMessageContent = content => {
+	if (!content) {
+		return undefined
+	}
+	
+	for (let i = 0; i < 5; i++) {
+		const inner = getFutureProofMessage(content)
+		if (!inner) {
+			break
+		}
+		content = inner.message
+	}
+	return content
+	function getFutureProofMessage(message) {
+		return (
+			message.ephemeralMessage ||
+			message.viewOnceMessage ||
+			message.documentWithCaptionMessage ||
+			message.viewOnceMessageV2 ||
+			message.viewOnceMessageV2Extension ||
+			message.editedMessage ||
+			message.groupMentionedMessage ||
+			message.botInvokeMessage ||
+			message.lottieStickerMessage ||
+			message.eventCoverImage ||
+			message.statusMentionMessage ||
+			message.pollCreationOptionImageMessage ||
+			message.associatedChildMessage ||
+			message.groupStatusMentionMessage ||
+			message.pollCreationMessageV4 ||
+			message.pollCreationMessageV5 ||
+			message.statusAddYours ||
+			message.groupStatusMessage ||
+			message.limitSharingMessage ||
+			message.botTaskMessage ||
+			message.questionMessage ||
+			message.groupStatusMessageV2 ||
+			message.botForwardedMessage ||
+			message.questionReplyMessage ||
+			message.newsletterAdminProfileMessage ||
+			message.newsletterAdminProfileMessageV2 ||
+			message.newsletterAdminProfileStatusMessage ||
+			message.spoilerMessage ||
+			message.groupStatusV3Message ||
+			message.pollCreationMessageV6
+		)
+	}
+}
+exports.normalizeMessageContent = normalizeMessageContent
+
+
+const extractMessageContent = content => {
+	const extractFromTemplateMessage = msg => {
+		if (msg.imageMessage) {
+			return { imageMessage: msg.imageMessage }
+		} else if (msg.documentMessage) {
+			return { documentMessage: msg.documentMessage }
+		} else if (msg.videoMessage) {
+			return { videoMessage: msg.videoMessage }
+		} else if (msg.locationMessage) {
+			return { locationMessage: msg.locationMessage }
+		} else {
+			return {
+				conversation:
+					'contentText' in msg ? msg.contentText : 'hydratedContentText' in msg ? msg.hydratedContentText : ''
+			}
+		}
+	}
+	content = normalizeMessageContent(content)
+	if (content?.buttonsMessage) {
+		return extractFromTemplateMessage(content.buttonsMessage)
+	}
+	if (content?.templateMessage?.hydratedFourRowTemplate) {
+		return extractFromTemplateMessage(content?.templateMessage?.hydratedFourRowTemplate)
+	}
+	if (content?.templateMessage?.hydratedTemplate) {
+		return extractFromTemplateMessage(content?.templateMessage?.hydratedTemplate)
+	}
+	if (content?.templateMessage?.fourRowTemplate) {
+		return extractFromTemplateMessage(content?.templateMessage?.fourRowTemplate)
+	}
+	return content
+}
+exports.extractMessageContent = extractMessageContent
+
+
+const getDevice = id =>
+	/^3A.{18}$/.test(id)
+		? 'ios'
+		: /^3E.{20}$/.test(id)
+			? 'web'
+			: /^(.{21}|.{32})$/.test(id)
+				? 'android'
+				: /^(3F|.{18}$)/.test(id)
+					? 'desktop'
+					: 'api/baileys'
+exports.getDevice = getDevice
+
+
+const updateMessageWithReceipt = (msg, receipt) => {
+	msg.userReceipt = msg.userReceipt || []
+	const recp = msg.userReceipt.find(m => m.userJid === receipt.userJid)
+	if (recp) {
+		Object.assign(recp, receipt)
+	} else {
+		msg.userReceipt.push(receipt)
+	}
+}
+exports.updateMessageWithReceipt = updateMessageWithReceipt
+
+
+const updateMessageWithReaction = (msg, reaction) => {
+	const authorID = (0, generics_1.getKeyAuthor)(reaction.key)
+	const reactions = (msg.reactions || []).filter(r => (0, generics_1.getKeyAuthor)(r.key) !== authorID)
+	reaction.text = reaction.text || ''
+	reactions.push(reaction)
+	msg.reactions = reactions
+}
+exports.updateMessageWithReaction = updateMessageWithReaction
+
+
+const updateMessageWithPollUpdate = (msg, update) => {
+	const authorID = (0, generics_1.getKeyAuthor)(update.pollUpdateMessageKey)
+	const reactions = (msg.pollUpdates || []).filter(
+		r => (0, generics_1.getKeyAuthor)(r.pollUpdateMessageKey) !== authorID
+	)
+	if (update.vote?.selectedOptions?.length) {
+		reactions.push(update)
+	}
+	msg.pollUpdates = reactions
+}
+exports.updateMessageWithPollUpdate = updateMessageWithPollUpdate
+
+
+const updateMessageWithEventResponse = (msg, update) => {
+	const authorID = (0, generics_1.getKeyAuthor)(update.eventResponseMessageKey)
+	const responses = (msg.eventResponses || []).filter(
+		r => (0, generics_1.getKeyAuthor)(r.eventResponseMessageKey) !== authorID
+	)
+	responses.push(update)
+	msg.eventResponses = responses
+}
+exports.updateMessageWithEventResponse = updateMessageWithEventResponse
+
+
+function getAggregateVotesInPollMessage({ message, pollUpdates }, meId) {
+	const opts =
+		message?.pollCreationMessage?.options ||
+		message?.pollCreationMessageV2?.options ||
+		message?.pollCreationMessageV3?.options ||
+		message?.pollCreationMessageV5?.options ||
+		message?.pollCreationMessageV6?.options ||
+		message?.pollCreationMessageV4?.message?.pollCreationMessage?.options ||
+		message?.pollCreationMessageV4?.message?.pollCreationMessageV3?.options ||
+		[]
+	const voteHashMap = opts.reduce((acc, opt) => {
+		const hash = (0, crypto_2.sha256)(Buffer.from(opt.optionName || '')).toString()
+		acc[hash] = {
+			name: opt.optionName || '',
+			voters: []
+		}
+		return acc
+	}, {})
+	for (const update of pollUpdates || []) {
+		const { vote } = update
+		if (!vote) {
+			continue
+		}
+		for (const option of vote.selectedOptions || []) {
+			const hash = option.toString()
+			let data = voteHashMap[hash]
+			if (!data) {
+				voteHashMap[hash] = {
+					name: 'Unknown',
+					voters: []
+				}
+				data = voteHashMap[hash]
+			}
+			voteHashMap[hash].voters.push((0, generics_1.getKeyAuthor)(update.pollUpdateMessageKey, meId))
+		}
+	}
+	return Object.values(voteHashMap)
+}
+exports.getAggregateVotesInPollMessage = getAggregateVotesInPollMessage
+
+
+const aggregateMessageKeysNotFromMe = keys => {
+	const keyMap = {}
+	for (const { remoteJid, id, participant, fromMe, sts } of keys) {
+		if (!fromMe) {
+			const uqKey = `${remoteJid}:${participant || ''}`
+			if (!keyMap[uqKey]) {
+				keyMap[uqKey] = {
+					jid: remoteJid,
+					participant: participant,
+					messageIds: [],
+					
+					sts: sts
+				}
+			} else if (keyMap[uqKey].sts !== sts) {
+				
+				keyMap[uqKey].sts = undefined
+			}
+			keyMap[uqKey].messageIds.push(id)
+		}
+	}
+	return Object.values(keyMap)
+}
+exports.aggregateMessageKeysNotFromMe = aggregateMessageKeysNotFromMe
+
+
+const isScheduledMessage = msg => !!msg?.scheduledMessageMetadata?.scheduledTime
+exports.isScheduledMessage = isScheduledMessage
+
+
+const getScheduledMessageTime = msg => {
+	const t = msg?.scheduledMessageMetadata?.scheduledTime
+	if (!t) return null
+	return new Date(Number(t) * 1000)
+}
+exports.getScheduledMessageTime = getScheduledMessageTime
+
+
+const getMessagePaymentInfo = msg => msg?.paymentInfo || msg?.quotedPaymentInfo || null
+exports.getMessagePaymentInfo = getMessagePaymentInfo
+
+
+const getMessageCommentMetadata = msg => msg?.commentMetadata || null
+exports.getMessageCommentMetadata = getMessageCommentMetadata
+
+
+const getMessageAddOns = msg => msg?.messageAddOns || []
+exports.getMessageAddOns = getMessageAddOns
+
+
+const getPollCorrectAnswer = pollMsg => {
+	const poll =
+		pollMsg?.pollCreationMessage ||
+		pollMsg?.pollCreationMessageV2 ||
+		pollMsg?.pollCreationMessageV3 ||
+		pollMsg?.pollCreationMessageV5 ||
+		pollMsg?.pollCreationMessageV6
+	if (!poll) return null
+	const isQuiz = poll.pollType === proto.Message.PollType?.QUIZ || poll.pollType === 1
+	return isQuiz ? poll.correctAnswer?.optionName || null : null
+}
+exports.getPollCorrectAnswer = getPollCorrectAnswer
+
+
+function getAggregateResponsesInEventMessage({ eventResponses }, meLid) {
+	const responseTypes = ['GOING', 'NOT_GOING', 'MAYBE']
+	const responseMap = {}
+
+	for (const type of responseTypes) {
+		responseMap[type] = {
+			response: type,
+			responders: []
+		}
+	}
+	for (const update of eventResponses) {
+		const { response } = update.response || {}
+		const responseType = proto.Message.EventResponseMessage.EventResponseType[response]
+		if (responseType !== 'UNKNOWN' && responseMap[responseType]) {
+			responseMap[responseType].responders.push(generics_1.getKeyAuthor(update.eventResponseMessageKey, meLid))
+		}
+	}
+
+	return Object.values(responseMap)
+}
+exports.getAggregateResponsesInEventMessage = getAggregateResponsesInEventMessage
+
+const REUPLOAD_REQUIRED_STATUS = [410, 404]
+
+
+const downloadMediaMessage = async (message, type, options, ctx) => {
+	const result = await downloadMsg().catch(async error => {
+		
+		
+		
+		const errorStatus =
+			typeof error?.status === 'number'
+				? error.status
+				: typeof error?.output?.statusCode === 'number'
+					? error.output.statusCode
+					: undefined
+		if (ctx && typeof errorStatus === 'number' && REUPLOAD_REQUIRED_STATUS.includes(errorStatus)) {
+			ctx.logger.info({ key: message.key }, 'sending reupload media request...')
+			message = await ctx.reuploadRequest(message)
+			const result = await downloadMsg()
+			return result
+		}
+		throw error
+	})
+	return result
+	async function downloadMsg() {
+		const mContent = extractMessageContent(message.message)
+		if (!mContent) {
+			throw new boom_1.Boom('No message present', { statusCode: 400, data: message })
+		}
+		const contentType = getContentType(mContent)
+		let mediaType = 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 Buffer.concat(bufferArray)
+		}
+		return stream
+	}
+}
+exports.downloadMediaMessage = downloadMediaMessage
+
+
+const assertMediaContent = content => {
+	content = extractMessageContent(content)
+	const mediaContent =
+		content?.documentMessage ||
+		content?.imageMessage ||
+		content?.videoMessage ||
+		content?.audioMessage ||
+		content?.stickerMessage
+	if (!mediaContent) {
+		throw new boom_1.Boom('given message is not a media message', { statusCode: 400, data: content })
+	}
+	return mediaContent
+}
+exports.assertMediaContent = assertMediaContent
+
+
+const toJid = id => {
+	if (!id) return ''
+	if (id.includes('@')) return id
+	return `${id}@s.whatsapp.net`
+}
+exports.toJid = toJid
+
+
+const getSenderLid = message => {
+	const k = message.key
+	if (!k) {
+		return { jid: '', lid: '' }
+	}
+	const jid = k.participant || k.remoteJid || ''
+	if (jid.endsWith('@lid') || jid.endsWith('@hosted.lid')) {
+		return { jid, lid: jid }
+	}
+	if (k.lid && typeof k.lid === 'string') {
+		const lid = k.lid.includes('@') ? k.lid : (0, WABinary_1.jidEncode)(k.lid, 'lid')
+		return { jid, lid }
+	}
+	if (k.participantLid && (0, WABinary_1.isLidUser)(k.participantLid)) {
+		return { jid, lid: k.participantLid }
+	}
+	return { jid, lid: '' }
+}
+exports.getSenderLid = getSenderLid

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

@@ -2,24 +2,24 @@
 Object.defineProperty(exports, '__esModule', { value: true })
 exports.MessageRetryManager = exports.RetryReason = void 0
 const lru_cache_1 = require('lru-cache')
-/** Number of sent messages to cache in memory for handling retry receipts */
+
 const RECENT_MESSAGES_SIZE = 512
 const MESSAGE_KEY_SEPARATOR = '\u0000'
-/** Timeout for session recreation - 1 hour */
-const RECREATE_SESSION_TIMEOUT = 60 * 60 * 1000 // 1 hour in milliseconds
+
+const RECREATE_SESSION_TIMEOUT = 60 * 60 * 1000 
 const PHONE_REQUEST_DELAY = 3000
-// Retry reason codes matching WhatsApp Web's Signal error codes.
+
 var RetryReason
 ;(function (RetryReason) {
 	RetryReason[(RetryReason['UnknownError'] = 0)] = 'UnknownError'
 	RetryReason[(RetryReason['SignalErrorNoSession'] = 1)] = 'SignalErrorNoSession'
 	RetryReason[(RetryReason['SignalErrorInvalidKey'] = 2)] = 'SignalErrorInvalidKey'
 	RetryReason[(RetryReason['SignalErrorInvalidKeyId'] = 3)] = 'SignalErrorInvalidKeyId'
-	/** MAC verification failed - most common cause of decryption failures */
+	
 	RetryReason[(RetryReason['SignalErrorInvalidMessage'] = 4)] = 'SignalErrorInvalidMessage'
 	RetryReason[(RetryReason['SignalErrorInvalidSignature'] = 5)] = 'SignalErrorInvalidSignature'
 	RetryReason[(RetryReason['SignalErrorFutureMessage'] = 6)] = 'SignalErrorFutureMessage'
-	/** Explicit MAC failure - session is definitely out of sync */
+	
 	RetryReason[(RetryReason['SignalErrorBadMac'] = 7)] = 'SignalErrorBadMac'
 	RetryReason[(RetryReason['SignalErrorInvalidSession'] = 8)] = 'SignalErrorInvalidSession'
 	RetryReason[(RetryReason['SignalErrorInvalidMsgKey'] = 9)] = 'SignalErrorInvalidMsgKey'
@@ -28,7 +28,7 @@ var RetryReason
 	RetryReason[(RetryReason['AdvFailure'] = 12)] = 'AdvFailure'
 	RetryReason[(RetryReason['StatusRevokeDelay'] = 13)] = 'StatusRevokeDelay'
 })(RetryReason || (exports.RetryReason = RetryReason = {}))
-/** Error codes that indicate a MAC failure and require immediate session recreation */
+
 const MAC_ERROR_CODES = new Set([RetryReason.SignalErrorInvalidMessage, RetryReason.SignalErrorBadMac])
 class MessageRetryManager {
 	constructor(logger, maxMsgRetryCount) {
@@ -54,12 +54,7 @@ class MessageRetryManager {
 			ttl: 15 * 60 * 1000,
 			ttlAutopurge: true,
 			updateAgeOnGet: true
-		}) // 15 minutes TTL
-		this.baseKeys = new lru_cache_1.LRUCache({
-			max: 1024,
-			ttl: 15 * 60 * 1000,
-			ttlAutopurge: true
-		})
+		}) 
 		this.pendingPhoneRequests = {}
 		this.maxMsgRetryCount = 5
 		this.statistics = {
@@ -72,13 +67,11 @@ class MessageRetryManager {
 		}
 		this.maxMsgRetryCount = maxMsgRetryCount
 	}
-	/**
-	 * Add a recent message to the cache for retry handling
-	 */
+	
 	addRecentMessage(to, id, message) {
 		const key = { to, id }
 		const keyStr = this.keyToString(key)
-		// Add new message
+		
 		this.recentMessagesMap.set(keyStr, {
 			message,
 			timestamp: Date.now()
@@ -86,20 +79,15 @@ class MessageRetryManager {
 		this.messageKeyIndex.set(id, keyStr)
 		this.logger.debug(`Added message to retry cache: ${to}/${id}`)
 	}
-	/**
-	 * Get a recent message from the cache
-	 */
+	
 	getRecentMessage(to, id) {
 		const key = { to, id }
 		const keyStr = this.keyToString(key)
 		return this.recentMessagesMap.get(keyStr)
 	}
-	/**
-	 * Check if a session should be recreated based on retry count, history, and error code.
-	 * MAC errors (codes 4 and 7) trigger immediate session recreation regardless of timeout.
-	 */
+	
 	shouldRecreateSession(jid, hasSession, errorCode) {
-		// If we don't have a session, always recreate
+		
 		if (!hasSession) {
 			this.sessionRecreateHistory.set(jid, Date.now())
 			this.statistics.sessionRecreations++
@@ -108,7 +96,7 @@ class MessageRetryManager {
 				recreate: true
 			}
 		}
-		// IMMEDIATE recreation for MAC errors - session is definitely out of sync
+		
 		if (errorCode !== undefined && MAC_ERROR_CODES.has(errorCode)) {
 			this.sessionRecreateHistory.set(jid, Date.now())
 			this.statistics.sessionRecreations++
@@ -123,7 +111,7 @@ class MessageRetryManager {
 		}
 		const now = Date.now()
 		const prevTime = this.sessionRecreateHistory.get(jid)
-		// If no previous recreation or it's been more than an hour
+		
 		if (!prevTime || now - prevTime > RECREATE_SESSION_TIMEOUT) {
 			this.sessionRecreateHistory.set(jid, now)
 			this.statistics.sessionRecreations++
@@ -134,10 +122,7 @@ class MessageRetryManager {
 		}
 		return { reason: '', recreate: false }
 	}
-	/**
-	 * Parse error code from retry receipt's retry node.
-	 * Returns undefined if no error code is present.
-	 */
+	
 	parseRetryErrorCode(errorAttr) {
 		if (errorAttr === undefined || errorAttr === '') {
 			return undefined
@@ -146,62 +131,48 @@ class MessageRetryManager {
 		if (Number.isNaN(code)) {
 			return undefined
 		}
-		// Validate it's a known RetryReason
+		
 		if (code >= RetryReason.UnknownError && code <= RetryReason.StatusRevokeDelay) {
 			return code
 		}
 		return RetryReason.UnknownError
 	}
-	/**
-	 * Check if an error code indicates a MAC failure
-	 */
+	
 	isMacError(errorCode) {
 		return errorCode !== undefined && MAC_ERROR_CODES.has(errorCode)
 	}
-	/**
-	 * Increment retry counter for a message
-	 */
+	
 	incrementRetryCount(messageId) {
 		this.retryCounters.set(messageId, (this.retryCounters.get(messageId) || 0) + 1)
 		this.statistics.totalRetries++
 		return this.retryCounters.get(messageId)
 	}
-	/**
-	 * Get retry count for a message
-	 */
+	
 	getRetryCount(messageId) {
 		return this.retryCounters.get(messageId) || 0
 	}
-	/**
-	 * Check if message has exceeded maximum retry attempts
-	 */
+	
 	hasExceededMaxRetries(messageId) {
 		return this.getRetryCount(messageId) >= this.maxMsgRetryCount
 	}
-	/**
-	 * Mark retry as successful
-	 */
+	
 	markRetrySuccess(messageId) {
 		this.statistics.successfulRetries++
-		// Clean up retry counter for successful message
+		
 		this.retryCounters.delete(messageId)
 		this.cancelPendingPhoneRequest(messageId)
 		this.removeRecentMessage(messageId)
 	}
-	/**
-	 * Mark retry as failed
-	 */
+	
 	markRetryFailed(messageId) {
 		this.statistics.failedRetries++
 		this.retryCounters.delete(messageId)
 		this.cancelPendingPhoneRequest(messageId)
 		this.removeRecentMessage(messageId)
 	}
-	/**
-	 * Schedule a phone request with delay
-	 */
+	
 	schedulePhoneRequest(messageId, callback, delay = PHONE_REQUEST_DELAY) {
-		// Cancel any existing request for this message
+		
 		this.cancelPendingPhoneRequest(messageId)
 		this.pendingPhoneRequests[messageId] = setTimeout(() => {
 			delete this.pendingPhoneRequests[messageId]
@@ -210,9 +181,7 @@ class MessageRetryManager {
 		}, delay)
 		this.logger.debug(`Scheduled phone request for message ${messageId} with ${delay}ms delay`)
 	}
-	/**
-	 * Cancel pending phone request
-	 */
+	
 	cancelPendingPhoneRequest(messageId) {
 		const timeout = this.pendingPhoneRequests[messageId]
 		if (timeout) {
@@ -221,40 +190,6 @@ class MessageRetryManager {
 			this.logger.debug(`Cancelled pending phone request for message ${messageId}`)
 		}
 	}
-	clear() {
-		this.recentMessagesMap.clear()
-		this.messageKeyIndex.clear()
-		this.sessionRecreateHistory.clear()
-		this.retryCounters.clear()
-		this.baseKeys.clear()
-		for (const messageId of Object.keys(this.pendingPhoneRequests)) {
-			this.cancelPendingPhoneRequest(messageId)
-		}
-		this.statistics = {
-			totalRetries: 0,
-			successfulRetries: 0,
-			failedRetries: 0,
-			mediaRetries: 0,
-			sessionRecreations: 0,
-			phoneRequests: 0
-		}
-	}
-	saveBaseKey(addr, msgId, baseKey) {
-		this.baseKeys.set(`${addr}:${msgId}`, baseKey)
-	}
-	hasSameBaseKey(addr, msgId, baseKey) {
-		const stored = this.baseKeys.get(`${addr}:${msgId}`)
-		if (!stored || stored.length !== baseKey.length) {
-			return false
-		}
-		for (let i = 0; i < stored.length; i++) {
-			if (stored[i] !== baseKey[i]) return false
-		}
-		return true
-	}
-	deleteBaseKey(addr, msgId) {
-		this.baseKeys.delete(`${addr}:${msgId}`)
-	}
 	keyToString(key) {
 		return `${key.to}${MESSAGE_KEY_SEPARATOR}${key.id}`
 	}