@vanzxy/baileys 1.6.2 → 1.6.4

package/lib/Utils/button-helper-utils.js

@@ -0,0 +1,314 @@
+// Evernight AI — lib/Utils/button-helper-utils.js / Creator : Vanzxy🍃
+// Ported from @queenanya/baileys (Bail-master fork) src/addons/message-utils.ts
+// Original credit inside that file: "Ported from @ryuu-reinzz/button-helper v2.2.5"
+// Converted TS -> JS ESM to match @vanzxy/baileys compiled-lib conventions.
+
+import { randomBytes } from 'crypto';
+import { generateWAMessage, generateWAMessageFromContent, normalizeMessageContent } from './messages.js';
+import { getUrlFromDirectPath } from './messages-media.js';
+import { unixTimestampSeconds } from './generics.js';
+import { getBinaryNodeChild, isJidGroup, isJidNewsletter, jidNormalizedUser, S_WHATSAPP_NET } from '../WABinary/index.js';
+import { QueryIds, XWAPaths } from '../Types/Mex.js';
+
+// ─────────────────────────────────────────────────────────────────────────────
+// SECTION 1 — Message Type Detection
+// ─────────────────────────────────────────────────────────────────────────────
+
+/** @param {import('../../WAProto/index.js').proto.IMessage} message */
+export function getMediaType(message) {
+	if (message.imageMessage) return 'image';
+	if (message.videoMessage) return message.videoMessage.gifPlayback ? 'gif' : 'video';
+	if (message.audioMessage) return message.audioMessage.ptt ? 'ptt' : 'audio';
+	if (message.contactMessage) return 'vcard';
+	if (message.documentMessage) return 'document';
+	if (message.contactsArrayMessage) return 'contact_array';
+	if (message.liveLocationMessage) return 'livelocation';
+	if (message.stickerMessage) return 'sticker';
+	if (message.listMessage) return 'list';
+	if (message.listResponseMessage) return 'list_response';
+	if (message.buttonsResponseMessage) return 'buttons_response';
+	if (message.orderMessage) return 'order';
+	if (message.productMessage) return 'product';
+	if (message.interactiveResponseMessage) return 'native_flow_response';
+	if (message.groupInviteMessage) return 'url';
+	return '';
+}
+
+/** @param {import('../../WAProto/index.js').proto.IMessage} message */
+export function getMessageType(message) {
+	const normalizedMessage = normalizeMessageContent(message);
+	if (!normalizedMessage) return 'text';
+	if (normalizedMessage.reactionMessage || normalizedMessage.encReactionMessage) return 'reaction';
+	if (
+		normalizedMessage.pollCreationMessage ||
+		normalizedMessage.pollCreationMessageV2 ||
+		normalizedMessage.pollCreationMessageV3 ||
+		normalizedMessage.pollUpdateMessage
+	)
+		return 'poll';
+	if (normalizedMessage.eventMessage) return 'event';
+	if (getMediaType(normalizedMessage) !== '') return 'media';
+	return 'text';
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// SECTION 2 — Button / Biz Node Helpers
+// ─────────────────────────────────────────────────────────────────────────────
+
+/** @param {import('../../WAProto/index.js').proto.IMessage} message */
+export function getButtonType(message) {
+	const inner = message.viewOnceMessageV2Extension?.message || message;
+	if (inner.listMessage) return 'list';
+	if (inner.buttonsMessage) return 'buttons';
+	if (inner.interactiveMessage?.nativeFlowMessage) return 'native_flow';
+	if (inner.interactiveMessage?.carouselMessage) return 'native_flow';
+	if (message.viewOnceMessage?.message?.interactiveMessage?.carouselMessage) return 'native_flow';
+	if (message.viewOnceMessageV2?.message?.interactiveMessage?.carouselMessage) return 'native_flow';
+	if (message.viewOnceMessage?.message?.interactiveMessage?.nativeFlowMessage) return 'native_flow';
+	if (message.viewOnceMessageV2?.message?.interactiveMessage?.nativeFlowMessage) return 'native_flow';
+	if (message.viewOnceMessageV2Extension?.message?.interactiveMessage?.nativeFlowMessage) return 'native_flow';
+	if (message.viewOnceMessageV2Extension?.message?.interactiveMessage?.carouselMessage) return 'native_flow';
+	return undefined;
+}
+
+/**
+ * Derive the binary node(s) that must accompany a button/interactive/list
+ * message so it renders correctly across WA Messenger + WA Business,
+ * Android + iOS.
+ * @param {import('../../WAProto/index.js').proto.IMessage} message
+ * @returns {import('../WABinary/index.js').BinaryNode}
+ */
+export function getButtonArgs(message) {
+	const inner = message.viewOnceMessageV2Extension?.message || message;
+	const nativeFlow =
+		inner.interactiveMessage?.nativeFlowMessage ||
+		message.viewOnceMessage?.message?.interactiveMessage?.nativeFlowMessage ||
+		message.viewOnceMessageV2?.message?.interactiveMessage?.nativeFlowMessage;
+	const carouselMessage =
+		inner.interactiveMessage?.carouselMessage ||
+		message.viewOnceMessage?.message?.interactiveMessage?.carouselMessage ||
+		message.viewOnceMessageV2?.message?.interactiveMessage?.carouselMessage;
+
+	const firstButtonName =
+		nativeFlow?.buttons?.[0]?.name ||
+		carouselMessage?.cards?.[0]?.nativeFlowMessage?.buttons?.[0]?.name;
+	const nativeFlowSpecials = [
+		'mpm',
+		'cta_catalog',
+		'send_location',
+		'call_permission_request',
+		'wa_payment_transaction_details',
+		'automated_greeting_message_view_catalog'
+	];
+	const ts = unixTimestampSeconds().toString();
+	if (nativeFlow && (firstButtonName === 'review_and_pay' || firstButtonName === 'payment_info')) {
+		return {
+			tag: 'biz',
+			attrs: { native_flow_name: firstButtonName === 'review_and_pay' ? 'order_details' : firstButtonName }
+		};
+	}
+
+	if (nativeFlow && nativeFlowSpecials.includes(firstButtonName ?? '')) {
+		return {
+			tag: 'biz',
+			attrs: { actual_actors: '2', host_storage: '2', privacy_mode_ts: ts },
+			content: [
+				{
+					tag: 'interactive',
+					attrs: { type: 'native_flow', v: '1' },
+					content: [{ tag: 'native_flow', attrs: { v: '2', name: firstButtonName } }]
+				},
+				{ tag: 'quality_control', attrs: { source_type: 'third_party' } }
+			]
+		};
+	}
+
+	if (nativeFlow || carouselMessage || message.buttonsMessage) {
+		return {
+			tag: 'biz',
+			attrs: {},
+			content: [
+				{
+					tag: 'interactive',
+					attrs: { type: 'native_flow', v: '1' },
+					content: [{ tag: 'native_flow', attrs: { v: '9', name: 'mixed' } }]
+				}
+			]
+		};
+	}
+
+	if (inner.listMessage) {
+		return { tag: 'biz', attrs: {}, content: [{ tag: 'list', attrs: { v: '2', type: 'product_list' } }] };
+	}
+
+	return { tag: 'biz', attrs: {} };
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// SECTION 3 — WS Extras (mentions, media, MD patch, album, socket extras)
+// ─────────────────────────────────────────────────────────────────────────────
+
+/** Build contextInfo for @mention or @all. */
+export const buildMentionContextInfo = (message) => {
+	if (message.mentionAll) return { contextInfo: { nonJidMentions: 1 } };
+	if (message.mentions?.length) return { contextInfo: { mentionedJid: message.mentions } };
+	return { contextInfo: {} };
+};
+
+/** Extract embedded media from buttons/interactive message (top-level or header-nested). */
+export const extractFromButtonsMessage = (msg) => {
+	const header = typeof msg.header === 'object' && msg.header !== null;
+	if (header ? msg.header?.imageMessage : msg.imageMessage)
+		return { imageMessage: header ? msg.header.imageMessage : msg.imageMessage };
+	if (header ? msg.header?.videoMessage : msg.videoMessage)
+		return { videoMessage: header ? msg.header.videoMessage : msg.videoMessage };
+	if (header ? msg.header?.documentMessage : msg.documentMessage)
+		return { documentMessage: header ? msg.header.documentMessage : msg.documentMessage };
+	return null;
+};
+
+/** Normalise media input: string → { url }, Buffer → as-is, others → as-is. */
+export const normalizeMediaInput = (media) => {
+	if (!media) return media;
+	if (Buffer.isBuffer(media)) return media;
+	if (typeof media === 'string') return { url: media };
+	return media;
+};
+
+/** Wrap buttons/template/list/interactive in viewOnceMessageV2Extension for MD clients. */
+export const patchMessageForMdIfRequired = (message) => {
+	if (
+		message?.buttonsMessage ||
+		message?.templateMessage ||
+		message?.listMessage ||
+		message?.interactiveMessage?.nativeFlowMessage
+	) {
+		return {
+			viewOnceMessageV2Extension: {
+				message: {
+					messageContextInfo: { deviceListMetadataVersion: 2, deviceListMetadata: {} },
+					...message
+				}
+			}
+		};
+	}
+
+	return message;
+};
+
+/**
+ * Build and relay an album (multi-image/video) message.
+ * @param {string} jid
+ * @param {Array<object>} albums
+ * @param {{ userJid: string, suki: { relayMessage: Function, waUploadToServer: Function } }} options
+ * @returns {Promise<Array<object>>} Array of individual media WAMessage objects
+ * @example
+ * const items = await prepareAlbumMessageContent(jid, [
+ *     { image: { url: 'https://...' }, caption: 'Photo 1' },
+ *     { video: { url: 'https://...' }, caption: 'Video 1' }
+ * ], { userJid: sock.user.id, suki: sock })
+ */
+export const prepareAlbumMessageContent = async (jid, albums, options) => {
+	const messages = [];
+
+	const albumMsg = generateWAMessageFromContent(
+		jid,
+		{
+			albumMessage: {
+				expectedImageCount: albums.filter((item) => 'image' in item).length,
+				expectedVideoCount: albums.filter((item) => 'video' in item).length
+			}
+		},
+		{ userJid: options.userJid }
+	);
+
+	await options.suki.relayMessage(jid, albumMsg.message, { messageId: albumMsg.key.id });
+
+	for (const media of albums) {
+		let mediaMsg;
+		const uploadFn = async (encFilePath, opts) => {
+			const res = await options.suki.waUploadToServer(encFilePath, {
+				...opts,
+				newsletter: isJidNewsletter(jid)
+			});
+			return {
+				mediaUrl: res.url ?? '',
+				directPath: res.directPath ?? '',
+				handle: res.handle,
+				mediaKey: res.mediaKey,
+				fileEncSha256: res.fileEncSha256,
+				fileSha256: res.fileSha256,
+				fileLength: res.fileLength
+			};
+		};
+
+		const sharedOpts = { userJid: options.userJid, upload: uploadFn };
+
+		if ('image' in media && media.image) mediaMsg = await generateWAMessage(jid, media, sharedOpts);
+		else if ('video' in media && media.video) mediaMsg = await generateWAMessage(jid, media, sharedOpts);
+
+		if (mediaMsg) {
+			mediaMsg.message.messageContextInfo = {
+				messageSecret: randomBytes(32),
+				messageAssociation: { associationType: 1, parentMessageKey: albumMsg.key }
+			};
+			messages.push(mediaMsg);
+		}
+	}
+
+	return messages;
+};
+
+/**
+ * Addon factory for socket-level message extras (profilePictureUrl, getEphemeralGroup).
+ * @param {{ query: Function, newsletterWMexQuery?: Function }} ctx
+ */
+export const makeMessageExtrasAddon = (ctx) => {
+	const { query, newsletterWMexQuery } = ctx;
+
+	/**
+	 * Fetch profile picture URL for any JID, including newsletters.
+	 * @example
+	 * const url = await sock.profilePictureUrl('1234567890@s.whatsapp.net')
+	 */
+	const profilePictureUrl = async (jid) => {
+		if (isJidNewsletter(jid) && newsletterWMexQuery) {
+			const node = await newsletterWMexQuery(undefined, QueryIds.METADATA, {
+				input: { key: jid, type: 'JID', view_role: 'GUEST' },
+				fetch_viewer_metadata: true,
+				fetch_full_image: true,
+				fetch_creation_time: true
+			});
+			const resultStr = getBinaryNodeChild(node, 'result')?.content?.toString();
+			if (!resultStr) return null;
+
+			const metadata = JSON.parse(resultStr).data[XWAPaths.xwa2_newsletter_metadata];
+			return getUrlFromDirectPath(metadata?.thread_metadata?.picture?.direct_path || '');
+		}
+
+		const result = await query({
+			tag: 'iq',
+			attrs: { target: jidNormalizedUser(jid), to: S_WHATSAPP_NET, type: 'get', xmlns: 'w:profile:picture' },
+			content: [{ tag: 'picture', attrs: { type: 'image', query: 'url' }, content: undefined }]
+		});
+		return getBinaryNodeChild(result, 'picture')?.attrs?.url || null;
+	};
+
+	/**
+	 * Query the ephemeral (disappearing messages) timer for a group.
+	 * @returns {Promise<number|string>} timer in seconds, or 0 if not set
+	 * @example
+	 * const timer = await sock.getEphemeralGroup('120363xxx@g.us')
+	 */
+	const getEphemeralGroup = async (jid) => {
+		if (!isJidGroup(jid)) throw new TypeError('Jid should originate from a group!');
+		const result = await query({
+			tag: 'iq',
+			attrs: { id: `ephemeral-${Date.now()}`, to: jid, type: 'get', xmlns: 'w:g2' },
+			content: [{ tag: 'query', attrs: { request: 'interactive' }, content: undefined }]
+		});
+		return getBinaryNodeChild(getBinaryNodeChild(result, 'group'), 'ephemeral')?.attrs?.expiration || 0;
+	};
+
+	return { profilePictureUrl, getEphemeralGroup };
+};