@nanazx/baileys 0.0.0-stage → 1.0.0

package/lib/Socket/interactive-handler.js

@@ -0,0 +1,577 @@
+import crypto from 'crypto'
+import {  proto  } from '../../WAProto/index.js'
+import { 
+	delay,
+	generateMessageID,
+	generateWAMessage,
+	generateWAMessageContent,
+	generateWAMessageFromContent,
+	getUrlFromDirectPath,
+	normalizeMessageContent,
+	prepareWAMessageMedia
+ } from '../Utils/index.js'
+import {  isJidGroup, isPnUser, jidNormalizedUser, STORIES_JID  } from '../WABinary/index.js'
+
+class Baron {
+	constructor(waUploadToServer, relayMessageFn, config, sock) {
+		this.relayMessage = relayMessageFn
+		this.waUploadToServer = waUploadToServer
+		this.config = config
+		this.sock = sock
+	}
+	detectType(content) {
+		if (content.requestPaymentMessage) return 'PAYMENT'
+		if (content.productMessage) return 'PRODUCT'
+		if (content.interactiveButtons) return 'INTERACTIVE_BUTTONS'
+		if (content.interactiveMessage) return 'INTERACTIVE'
+		if (content.albumMessage || content.album) return 'ALBUM'
+		if (content.eventMessage) return 'EVENT'
+		if (content.pollResultMessage) return 'POLL_RESULT'
+		if (content.groupStatusMessage) return 'GROUP_STORY'
+		return null
+	}
+	async handlePayment(content, quoted) {
+		const data = content.requestPaymentMessage
+		let notes = {}
+		if (data.sticker?.stickerMessage) {
+			notes = {
+				stickerMessage: {
+					...data.sticker.stickerMessage,
+					contextInfo: {
+						stanzaId: quoted?.key?.id,
+						participant: quoted?.key?.participant || content.sender,
+						quotedMessage: quoted?.message
+					}
+				}
+			}
+		} else if (data.note) {
+			notes = {
+				extendedTextMessage: {
+					text: data.note,
+					contextInfo: {
+						stanzaId: quoted?.key?.id,
+						participant: quoted?.key?.participant || content.sender,
+						quotedMessage: quoted?.message
+					}
+				}
+			}
+		}
+		return {
+			requestPaymentMessage: proto.Message.RequestPaymentMessage.fromObject({
+				expiryTimestamp: data.expiry || 0,
+				amount1000: data.amount || 0,
+				currencyCodeIso4217: data.currency || 'IDR',
+				requestFrom: data.from || '0@s.whatsapp.net',
+				noteMessage: notes,
+				background: data.background ?? {
+					id: 'DEFAULT',
+					placeholderArgb: 0xfff0f0f0
+				}
+			})
+		}
+	}
+	async handleProduct(content, _jid, _quoted) {
+		const {
+			title,
+			description,
+			thumbnail,
+			productId,
+			retailerId,
+			url,
+			body = '',
+			footer = '',
+			buttons = [],
+			priceAmount1000 = null,
+			currencyCode = 'IDR'
+		} = content.productMessage
+		let productImage
+		if (Buffer.isBuffer(thumbnail)) {
+			const { imageMessage } = await generateWAMessageContent({ image: thumbnail }, { upload: this.waUploadToServer })
+			productImage = imageMessage
+		} else if (typeof thumbnail === 'object' && thumbnail.url) {
+			const { imageMessage } = await generateWAMessageContent(
+				{ image: { url: thumbnail.url } },
+				{ upload: this.waUploadToServer }
+			)
+			productImage = imageMessage
+		}
+		return {
+			viewOnceMessage: {
+				message: {
+					interactiveMessage: {
+						body: { text: body },
+						footer: { text: footer },
+						header: {
+							title,
+							hasMediaAttachment: true,
+							productMessage: {
+								product: {
+									productImage,
+									productId,
+									title,
+									description,
+									currencyCode,
+									priceAmount1000,
+									retailerId,
+									url,
+									productImageCount: 1
+								},
+								businessOwnerJid: '0@s.whatsapp.net'
+							}
+						},
+						nativeFlowMessage: { buttons }
+					}
+				}
+			}
+		}
+	}
+	async handleInteractive(content, _jid, _quoted) {
+		const {
+			title,
+			footer,
+			thumbnail,
+			image,
+			video,
+			document,
+			mimetype,
+			fileName,
+			jpegThumbnail,
+			contextInfo,
+			externalAdReply,
+			buttons = [],
+			nativeFlowMessage,
+			header
+		} = content.interactiveMessage
+		let media = null
+		let _mediaType = null
+		if (thumbnail) {
+			media = await prepareWAMessageMedia({ image: { url: thumbnail } }, { upload: this.waUploadToServer })
+			_mediaType = 'image'
+		} else if (image) {
+			const src = typeof image === 'object' && image.url ? { image: { url: image.url } } : { image }
+			media = await prepareWAMessageMedia(src, { upload: this.waUploadToServer })
+			_mediaType = 'image'
+		} else if (video) {
+			const src = typeof video === 'object' && video.url ? { video: { url: video.url } } : { video }
+			media = await prepareWAMessageMedia(src, { upload: this.waUploadToServer })
+			_mediaType = 'video'
+		} else if (document) {
+			const docPayload = { document }
+			if (jpegThumbnail) {
+				docPayload.jpegThumbnail =
+					typeof jpegThumbnail === 'object' && jpegThumbnail.url ? { url: jpegThumbnail.url } : jpegThumbnail
+			}
+			media = await prepareWAMessageMedia(docPayload, { upload: this.waUploadToServer })
+			if (fileName) media.documentMessage.fileName = fileName
+			if (mimetype) media.documentMessage.mimetype = mimetype
+			_mediaType = 'document'
+		}
+		const interactiveMessage = {
+			body: { text: title || '' },
+			footer: { text: footer || '' }
+		}
+		if (buttons && buttons.length > 0) {
+			interactiveMessage.nativeFlowMessage = { buttons }
+			if (nativeFlowMessage) {
+				interactiveMessage.nativeFlowMessage = {
+					...interactiveMessage.nativeFlowMessage,
+					...nativeFlowMessage
+				}
+			}
+		} else if (nativeFlowMessage) {
+			interactiveMessage.nativeFlowMessage = nativeFlowMessage
+		}
+		if (media) {
+			interactiveMessage.header = {
+				title: header || '',
+				hasMediaAttachment: true,
+				...media
+			}
+		} else {
+			interactiveMessage.header = {
+				title: header || '',
+				hasMediaAttachment: false
+			}
+		}
+		const finalContextInfo = {}
+		if (contextInfo) {
+			Object.assign(finalContextInfo, {
+				mentionedJid: contextInfo.mentionedJid || [],
+				forwardingScore: contextInfo.forwardingScore || 0,
+				isForwarded: contextInfo.isForwarded || false,
+				...contextInfo
+			})
+		}
+		if (externalAdReply) {
+			finalContextInfo.externalAdReply = {
+				title: externalAdReply.title || '',
+				body: externalAdReply.body || '',
+				mediaType: externalAdReply.mediaType || 1,
+				thumbnailUrl: externalAdReply.thumbnailUrl || '',
+				mediaUrl: externalAdReply.mediaUrl || '',
+				sourceUrl: externalAdReply.sourceUrl || '',
+				showAdAttribution: externalAdReply.showAdAttribution || false,
+				renderLargerThumbnail: externalAdReply.renderLargerThumbnail || false,
+				...externalAdReply
+			}
+		}
+		if (Object.keys(finalContextInfo).length > 0) {
+			interactiveMessage.contextInfo = finalContextInfo
+		}
+		return { interactiveMessage }
+	}
+	async handleInteractiveButtons(content, _jid, _quoted) {
+		const {
+			text,
+			caption,
+			title,
+			subtitle,
+			footer,
+			interactiveButtons,
+			hasMediaAttachment,
+			image,
+			video,
+			document,
+			mimetype,
+			jpegThumbnail,
+			location,
+			product,
+			businessOwnerJid
+		} = content
+		const bodyText = text || caption || ''
+		const buttons = interactiveButtons.map(btn => ({
+			name: btn.name,
+			buttonParamsJson:
+				typeof btn.buttonParamsJson === 'string' ? btn.buttonParamsJson : JSON.stringify(btn.buttonParamsJson)
+		}))
+		let headerContent = {}
+		let mediaAttached = typeof hasMediaAttachment === 'boolean' ? hasMediaAttachment : false
+		if (image) {
+			const src = typeof image === 'object' && image.url ? { image: { url: image.url } } : { image }
+			const uploaded = await prepareWAMessageMedia(src, { upload: this.waUploadToServer })
+			headerContent = { ...uploaded }
+			mediaAttached = typeof hasMediaAttachment === 'boolean' ? hasMediaAttachment : true
+		} else if (video) {
+			const src = typeof video === 'object' && video.url ? { video: { url: video.url } } : { video }
+			const uploaded = await prepareWAMessageMedia(src, { upload: this.waUploadToServer })
+			headerContent = { ...uploaded }
+			mediaAttached = typeof hasMediaAttachment === 'boolean' ? hasMediaAttachment : true
+		} else if (document) {
+			const docPayload =
+				typeof document === 'object' && document.url ? { document: { url: document.url } } : { document }
+			if (mimetype) docPayload.mimetype = mimetype
+			const uploaded = await prepareWAMessageMedia(docPayload, { upload: this.waUploadToServer })
+			if (jpegThumbnail) {
+				uploaded.documentMessage.jpegThumbnail =
+					typeof jpegThumbnail === 'string' ? Buffer.from(jpegThumbnail, 'base64') : jpegThumbnail
+			}
+			headerContent = { ...uploaded }
+			mediaAttached = typeof hasMediaAttachment === 'boolean' ? hasMediaAttachment : true
+		} else if (location) {
+			headerContent = {
+				locationMessage: {
+					degreesLatitude: location.degressLatitude || location.degreesLatitude || 0,
+					degreesLongitude: location.degressLongitude || location.degreesLongitude || 0,
+					name: location.name || ''
+				}
+			}
+			mediaAttached = typeof hasMediaAttachment === 'boolean' ? hasMediaAttachment : true
+		} else if (product) {
+			let productImage
+			if (product.productImage) {
+				const imgSrc =
+					typeof product.productImage === 'object' && product.productImage.url
+						? { image: { url: product.productImage.url } }
+						: { image: product.productImage }
+				const uploaded = await prepareWAMessageMedia(imgSrc, { upload: this.waUploadToServer })
+				productImage = uploaded.imageMessage
+			}
+			headerContent = {
+				productMessage: {
+					product: {
+						productImage,
+						productId: product.productId,
+						title: product.title,
+						description: product.description,
+						currencyCode: product.currencyCode || 'IDR',
+						priceAmount1000: product.priceAmount1000,
+						retailerId: product.retailerId,
+						url: product.url,
+						productImageCount: product.productImageCount || 1
+					},
+					businessOwnerJid: businessOwnerJid || '0@s.whatsapp.net'
+				}
+			}
+			mediaAttached = typeof hasMediaAttachment === 'boolean' ? hasMediaAttachment : true
+		}
+		const interactiveMessage = {
+			body: { text: bodyText },
+			footer: { text: footer || '' },
+			header: {
+				title: title || '',
+				subtitle: subtitle || '',
+				hasMediaAttachment: mediaAttached,
+				...headerContent
+			},
+			nativeFlowMessage: { buttons }
+		}
+		return {
+			viewOnceMessage: {
+				message: {
+					messageContextInfo: {
+						deviceListMetadata: {},
+						deviceListMetadataVersion: 2,
+						messageSecret: crypto.randomBytes(32)
+					},
+					interactiveMessage
+				}
+			}
+		}
+	}
+	async handleAlbum(content, jid, quoted) {
+		const array = content.albumMessage || content.album
+		const ctxInfo = content.contextInfo || {}
+		const album = await generateWAMessageFromContent(
+			jid,
+			{
+				messageContextInfo: {
+					messageSecret: crypto.randomBytes(32)
+				},
+				albumMessage: {
+					expectedImageCount: array.filter(a => 'image' in a).length,
+					expectedVideoCount: array.filter(a => 'video' in a).length
+				}
+			},
+			{ userJid: jidNormalizedUser(this.sock.authState?.creds?.me?.id || '') }
+		)
+		await this.relayMessage(jid, album.message, {
+			messageId: album.key.id
+		})
+		for (let item of array) {
+			if (ctxInfo && Object.keys(ctxInfo).length > 0 && !item.contextInfo) {
+				item = { ...item, contextInfo: ctxInfo }
+			}
+			const img = await generateWAMessage(jid, item, {
+				upload: this.waUploadToServer,
+				userJid: jidNormalizedUser(this.sock.authState?.creds?.me?.id || '')
+			})
+			img.message.messageContextInfo = {
+				messageSecret: crypto.randomBytes(32),
+				messageAssociation: {
+					associationType: 1,
+					parentMessageKey: album.key
+				}
+			}
+			await this.relayMessage(jid, img.message, {
+				messageId: img.key.id
+			})
+		}
+		return album
+	}
+	async handleEvent(content, jid, quoted) {
+		const eventData = content.eventMessage
+		const msg = await generateWAMessageFromContent(
+			jid,
+			{
+				viewOnceMessage: {
+					message: {
+						messageContextInfo: {
+							deviceListMetadata: {},
+							deviceListMetadataVersion: 2,
+							messageSecret: crypto.randomBytes(32)
+						},
+						eventMessage: {
+							isCanceled: eventData.isCanceled || false,
+							name: eventData.name,
+							description: eventData.description,
+							location: eventData.location || {
+								degreesLatitude: 0,
+								degreesLongitude: 0,
+								name: 'Location'
+							},
+							joinLink: eventData.joinLink || '',
+							startTime:
+								typeof eventData.startTime === 'string'
+									? parseInt(eventData.startTime, 10)
+									: eventData.startTime || Date.now(),
+							endTime:
+								typeof eventData.endTime === 'string'
+									? parseInt(eventData.endTime, 10)
+									: eventData.endTime || Date.now() + 3600000,
+							extraGuestsAllowed: eventData.extraGuestsAllowed !== false
+						}
+					}
+				}
+			},
+			{ quoted, userJid: jidNormalizedUser(this.sock.authState?.creds?.me?.id || '') }
+		)
+		await this.relayMessage(jid, msg.message, {
+			messageId: msg.key.id
+		})
+		return msg
+	}
+	async handlePollResult(content, jid, quoted) {
+		const pollData = content.pollResultMessage
+		const msg = await generateWAMessageFromContent(
+			jid,
+			{
+				pollResultSnapshotMessage: {
+					name: pollData.name,
+					pollVotes: pollData.pollVotes.map(vote => ({
+						optionName: vote.optionName,
+						optionVoteCount:
+							typeof vote.optionVoteCount === 'number' ? vote.optionVoteCount.toString() : vote.optionVoteCount
+					}))
+				}
+			},
+			{ quoted, userJid: jidNormalizedUser(this.sock.authState?.creds?.me?.id || '') }
+		)
+		await this.relayMessage(jid, msg.message, {
+			messageId: msg.key.id
+		})
+		return msg
+	}
+	async handleGroupStory(content, jid, _quoted) {
+		const storyData = content.groupStatusMessage
+		let waMsgContent
+		if (storyData.message) {
+			waMsgContent = storyData
+		} else {
+			waMsgContent = await generateWAMessageContent(storyData, {
+				upload: this.waUploadToServer
+			})
+		}
+		const msg = {
+			message: {
+				groupStatusMessageV2: {
+					message: waMsgContent.message || waMsgContent
+				}
+			}
+		}
+		return await this.relayMessage(jid, msg.message, {
+			messageId: generateMessageID()
+		})
+	}
+	async sendStatusWhatsApp(content, jids = []) {
+		const userJid = jidNormalizedUser(this.sock.authState.creds.me.id)
+		const allUsers = new Set()
+		allUsers.add(userJid)
+		for (const id of jids) {
+			if (isJidGroup(id)) {
+				try {
+					const metadata = await this.sock.groupMetadata(id)
+					metadata.participants.forEach(p => allUsers.add(jidNormalizedUser(p.id)))
+				} catch (error) {
+					this.config.logger.error(`Error getting metadata for group ${id}: ${error}`)
+				}
+			} else if (isPnUser(id)) {
+				allUsers.add(jidNormalizedUser(id))
+			}
+		}
+		const uniqueUsers = Array.from(allUsers)
+		const getRandomHexColor = () =>
+			'#' +
+			Math.floor(Math.random() * 16777215)
+				.toString(16)
+				.padStart(6, '0')
+		const isMedia = content.image || content.video || content.audio
+		const isAudio = !!content.audio
+		const messageContent = { ...content }
+		if (isMedia && !isAudio) {
+			if (messageContent.text) {
+				messageContent.caption = messageContent.text
+				delete messageContent.text
+			}
+			delete messageContent.ptt
+			delete messageContent.font
+			delete messageContent.backgroundColor
+			delete messageContent.textColor
+		}
+		if (isAudio) {
+			delete messageContent.text
+			delete messageContent.caption
+			delete messageContent.font
+			delete messageContent.textColor
+		}
+		const font = !isMedia ? content.font || Math.floor(Math.random() * 9) : undefined
+		const textColor = !isMedia ? content.textColor || getRandomHexColor() : undefined
+		const backgroundColor = !isMedia || isAudio ? content.backgroundColor || getRandomHexColor() : undefined
+		const ptt = isAudio ? (typeof content.ptt === 'boolean' ? content.ptt : true) : undefined
+		import {  getUrlInfo  } from '../Utils/link-preview.js'
+		const msg = await generateWAMessage(STORIES_JID, messageContent, {
+			logger: this.config.logger,
+			userJid,
+			getUrlInfo: text =>
+				getUrlInfo(text, {
+					thumbnailWidth: this.config.linkPreviewImageThumbnailWidth,
+					fetchOpts: { timeout: 3000, ...(this.config.options || {}) },
+					logger: this.config.logger,
+					uploadImage: this.config.generateHighQualityLinkPreview ? this.waUploadToServer : undefined
+				}),
+			upload: this.waUploadToServer,
+			mediaCache: this.config.mediaCache,
+			options: this.config.options,
+			font,
+			textColor,
+			backgroundColor,
+			ptt
+		})
+		await this.relayMessage(STORIES_JID, msg.message, {
+			messageId: msg.key.id,
+			statusJidList: uniqueUsers,
+			additionalNodes: [
+				{
+					tag: 'meta',
+					attrs: {},
+					content: [
+						{
+							tag: 'mentioned_users',
+							attrs: {},
+							content: jids.map(jid => ({
+								tag: 'to',
+								attrs: { jid: jidNormalizedUser(jid) }
+							}))
+						}
+					]
+				}
+			]
+		})
+		for (const id of jids) {
+			try {
+				const normalizedId = jidNormalizedUser(id)
+				const isPrivate = isPnUser(normalizedId)
+				const type = isPrivate ? 'statusMentionMessage' : 'groupStatusMentionMessage'
+				const protocolMessage = {
+					[type]: {
+						message: {
+							protocolMessage: {
+								key: msg.key,
+								type: 25
+							}
+						}
+					},
+					messageContextInfo: {
+						messageSecret: crypto.randomBytes(32)
+					}
+				}
+				const statusMsg = await generateWAMessageFromContent(normalizedId, protocolMessage, {
+					userJid: jidNormalizedUser(this.sock.authState?.creds?.me?.id || '')
+				})
+				await this.relayMessage(normalizedId, statusMsg.message, {
+					additionalNodes: [
+						{
+							tag: 'meta',
+							attrs: isPrivate ? { is_status_mention: 'true' } : { is_group_status_mention: 'true' }
+						}
+					]
+				})
+				await delay(2000)
+			} catch (error) {
+				this.config.logger.error(`Error sending to ${id}: ${error}`)
+			}
+		}
+		return msg
+	}
+}
+
+export { Baron };