@alannxd/baileys 6.0.5 → 6.0.9

package/lib/Utils/message-composer.js

@@ -0,0 +1,273 @@
+import { proto } from '../../WAProto/index.js'
+import { generateMessageIDV2 } from './generics.js'
+
+const JS_KEYWORDS = new Set([
+    'import', 'export', 'from', 'default', 'as', 'const', 'let', 'var',
+    'function', 'class', 'extends', 'new', 'return', 'if', 'else', 'for',
+    'while', 'do', 'switch', 'case', 'break', 'continue', 'try', 'catch',
+    'finally', 'throw', 'async', 'await', 'yield', 'typeof', 'instanceof',
+    'in', 'of', 'delete', 'void', 'true', 'false', 'null', 'undefined',
+    'NaN', 'Infinity', 'this', 'super', 'static', 'get', 'set', 'debugger', 'with'
+])
+
+const PYTHON_KEYWORDS = new Set([
+    'import', 'from', 'as', 'def', 'class', 'return', 'if', 'elif', 'else',
+    'for', 'while', 'break', 'continue', 'try', 'except', 'finally', 'raise',
+    'with', 'yield', 'lambda', 'pass', 'del', 'global', 'nonlocal', 'assert',
+    'True', 'False', 'None', 'and', 'or', 'not', 'in', 'is', 'async', 'await',
+    'self', 'print'
+])
+
+const LANGUAGE_KEYWORDS = {
+    javascript: JS_KEYWORDS,
+    typescript: JS_KEYWORDS,
+    js: JS_KEYWORDS,
+    ts: JS_KEYWORDS,
+    python: PYTHON_KEYWORDS,
+    py: PYTHON_KEYWORDS
+}
+
+export var CodeHighlightType
+;(function (CodeHighlightType) {
+    CodeHighlightType[(CodeHighlightType['DEFAULT'] = 0)] = 'DEFAULT'
+    CodeHighlightType[(CodeHighlightType['KEYWORD'] = 1)] = 'KEYWORD'
+    CodeHighlightType[(CodeHighlightType['METHOD'] = 2)] = 'METHOD'
+    CodeHighlightType[(CodeHighlightType['STRING'] = 3)] = 'STRING'
+    CodeHighlightType[(CodeHighlightType['NUMBER'] = 4)] = 'NUMBER'
+    CodeHighlightType[(CodeHighlightType['COMMENT'] = 5)] = 'COMMENT'
+})(CodeHighlightType || (CodeHighlightType = {}))
+
+export var RichSubMessageType
+;(function (RichSubMessageType) {
+    RichSubMessageType[(RichSubMessageType['UNKNOWN'] = 0)] = 'UNKNOWN'
+    RichSubMessageType[(RichSubMessageType['GRID_IMAGE'] = 1)] = 'GRID_IMAGE'
+    RichSubMessageType[(RichSubMessageType['TEXT'] = 2)] = 'TEXT'
+    RichSubMessageType[(RichSubMessageType['INLINE_IMAGE'] = 3)] = 'INLINE_IMAGE'
+    RichSubMessageType[(RichSubMessageType['TABLE'] = 4)] = 'TABLE'
+    RichSubMessageType[(RichSubMessageType['CODE'] = 5)] = 'CODE'
+    RichSubMessageType[(RichSubMessageType['DYNAMIC'] = 6)] = 'DYNAMIC'
+    RichSubMessageType[(RichSubMessageType['MAP'] = 7)] = 'MAP'
+    RichSubMessageType[(RichSubMessageType['LATEX'] = 8)] = 'LATEX'
+    RichSubMessageType[(RichSubMessageType['CONTENT_ITEMS'] = 9)] = 'CONTENT_ITEMS'
+})(RichSubMessageType || (RichSubMessageType = {}))
+
+export const tokenizeCode = (codeStr, language = 'javascript') => {
+    const keywords = LANGUAGE_KEYWORDS[language] || JS_KEYWORDS
+    const blocks = []
+    const lines = codeStr.split('\n')
+    for (let li = 0; li < lines.length; li++) {
+        const line = lines[li]
+        const isLast = li === lines.length - 1
+        const nl = isLast? '' : '\n'
+        if (!line.trim()) {
+            blocks.push({ highlightType: CodeHighlightType.DEFAULT, codeContent: line + nl })
+            continue
+        }
+        if (line.trim().startsWith('//') || line.trim().startsWith('#')) {
+            blocks.push({ highlightType: CodeHighlightType.COMMENT, codeContent: line + nl })
+            continue
+        }
+        const regex = /(\/\/.*$|#.*$)|(["'`](?:[^"'`\\]|\\.)*["'`])|(\b\d+(?:\.\d+)?\b)|(\b[a-zA-Z_$][\w$]*\b)|([^\s\w$"'`]+)|(\s+)/g
+        let match
+        const tokens = []
+        while ((match = regex.exec(line))!== null) {
+            const val = match[0]
+            if (match[1]) {
+                tokens.push({ highlightType: CodeHighlightType.COMMENT, codeContent: val })
+            } else if (match[2]) {
+                tokens.push({ highlightType: CodeHighlightType.STRING, codeContent: val })
+            } else if (match[3]) {
+                tokens.push({ highlightType: CodeHighlightType.NUMBER, codeContent: val })
+            } else if (match[4]) {
+                if (keywords.has(val)) {
+                    tokens.push({ highlightType: CodeHighlightType.KEYWORD, codeContent: val })
+                } else {
+                    const after = line.slice(regex.lastIndex).trimStart()
+                    if (after.startsWith('(')) {
+                        tokens.push({ highlightType: CodeHighlightType.METHOD, codeContent: val })
+                    } else {
+                        tokens.push({ highlightType: CodeHighlightType.DEFAULT, codeContent: val })
+                    }
+                }
+            } else {
+                tokens.push({ highlightType: CodeHighlightType.DEFAULT, codeContent: val })
+            }
+        }
+        if (tokens.length === 0) {
+            blocks.push({ highlightType: CodeHighlightType.DEFAULT, codeContent: line + nl })
+            continue
+        }
+        const merged = []
+        for (const t of tokens) {
+            const prev = merged.length > 0? merged[merged.length - 1] : undefined
+            if (prev && prev.highlightType === t.highlightType) {
+                prev.codeContent += t.codeContent
+            } else {
+                merged.push({...t })
+            }
+        }
+        if (merged.length > 0) {
+            merged[merged.length - 1].codeContent += nl
+        }
+        blocks.push(...merged)
+    }
+    return blocks
+}
+
+export const buildRichContextInfo = (quoted, options) => {
+    const ctxInfo = {
+        forwardingScore: 1,
+        isForwarded: true,
+        forwardedAiBotMessageInfo: { botJid: options?.botJid? options.botJid : '867051314767696@bot' },
+        forwardOrigin: 4,
+       ...(options?.mentions? { mentionedJid: options.mentions } : {})
+    }
+    if (quoted?.key) {
+        ctxInfo.stanzaId = quoted.key.id
+        ctxInfo.participant = quoted.key.participant || quoted.sender || quoted.key.remoteJid
+        ctxInfo.quotedMessage = quoted.message
+    }
+    return ctxInfo
+}
+
+export const buildBotForwardedMessage = (submessages, contextInfo, unifiedResponse) => {
+    const richResponse = { messageType: 1, submessages, contextInfo }
+    if (unifiedResponse) {
+        richResponse.unifiedResponse = unifiedResponse
+    }
+    return {
+        botForwardedMessage: {
+            message: { richResponseMessage: richResponse }
+        }
+    }
+}
+
+export const generateTableContent = (title, headers, rows, quoted, options = {}) => {
+    const { footer, headerText } = options
+    const tableRows = [{ items: headers, isHeading: true },...rows.map(row => ({ items: row.map(String) }))]
+    const submessages = []
+    if (headerText) submessages.push({ messageType: 2, messageText: headerText })
+    submessages.push({ messageType: 4, tableMetadata: { title, rows: tableRows } })
+    if (footer) submessages.push({ messageType: 2, messageText: footer })
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export const generateListContent = (title, items, quoted, options = {}) => {
+    const { footer, headerText } = options
+    const tableRows = items.map(item => ({
+        items: Array.isArray(item)? item.map(String) : [String(item)]
+    }))
+    const submessages = []
+    if (headerText) submessages.push({ messageType: 2, messageText: headerText })
+    submessages.push({ messageType: 4, tableMetadata: { title, rows: tableRows } })
+    if (footer) submessages.push({ messageType: 2, messageText: footer })
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export const generateCodeBlockContent = (code, quoted, options = {}) => {
+    const { title, footer, language = 'javascript' } = options
+    const submessages = []
+    if (title) submessages.push({ messageType: 2, messageText: title })
+    submessages.push({
+        messageType: 5,
+        codeMetadata: { codeLanguage: language, codeBlocks: tokenizeCode(code, language) }
+    })
+    if (footer) submessages.push({ messageType: 2, messageText: footer })
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export const generateLatexContent = (quoted, options) => {
+    const { text, expressions, headerText, footer } = options
+    const submessages = []
+    if (headerText) submessages.push({ messageType: 2, messageText: headerText })
+    const latexExpressions = expressions.map(expr => {
+        const entry = {
+            latexExpression: expr.latexExpression,
+            url: expr.url,
+            width: expr.width,
+            height: expr.height
+        }
+        if (expr.fontHeight!== undefined) entry.fontHeight = expr.fontHeight
+        if (expr.imageTopPadding!== undefined) entry.imageTopPadding = expr.imageTopPadding
+        if (expr.imageLeadingPadding!== undefined) entry.imageLeadingPadding = expr.imageLeadingPadding
+        if (expr.imageBottomPadding!== undefined) entry.imageBottomPadding = expr.imageBottomPadding
+        if (expr.imageTrailingPadding!== undefined) entry.imageTrailingPadding = expr.imageTrailingPadding
+        return entry
+    })
+    submessages.push({ messageType: 8, latexMetadata: { text: text || '', expressions: latexExpressions } })
+    if (footer) submessages.push({ messageType: 2, messageText: footer })
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export const generateLatexImageContent = async (quoted, options, uploadFn, renderLatexToPng) => {
+    const { text, expressions, headerText, footer } = options
+    const submessages = []
+    if (headerText) submessages.push({ messageType: 2, messageText: headerText })
+    const latexExpressions = await Promise.all(
+        expressions.map(async expr => {
+            const { buffer, width, height } = await renderLatexToPng(expr.latexExpression)
+            const uploadResult = await uploadFn(buffer, 'image')
+            const imageUrl = uploadResult.url || uploadResult.directPath
+            return { latexExpression: expr.latexExpression, url: imageUrl, width, height }
+        })
+    )
+    submessages.push({ messageType: 8, latexMetadata: { text: text || '', expressions: latexExpressions } })
+    if (footer) submessages.push({ messageType: 2, messageText: footer })
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export const generateLatexInlineImageContent = async (quoted, options, uploadFn, renderLatexToPng) => {
+    const { text, expressions, headerText, footer } = options
+    const submessages = []
+    if (headerText) submessages.push({ messageType: 2, messageText: headerText })
+    if (text) submessages.push({ messageType: 2, messageText: text })
+    for (const expr of expressions) {
+        const { buffer, width, height } = await renderLatexToPng(expr.latexExpression)
+        const uploadResult = await uploadFn(buffer, 'image')
+        const imageUrl = uploadResult.url || uploadResult.directPath
+        submessages.push({
+            messageType: 3,
+            imageMetadata: {
+                imageUrl: { imagePreviewUrl: imageUrl, imageHighResUrl: imageUrl },
+                imageText: expr.latexExpression,
+                alignment: 2
+            }
+        })
+    }
+    if (footer) submessages.push({ messageType: 2, messageText: footer })
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export const captureUnifiedResponse = msg => {
+    const botFwd = msg?.botForwardedMessage?.message
+    if (!botFwd) return null
+    const rich = botFwd.richResponseMessage
+    if (!rich?.unifiedResponse?.data) return null
+    return {
+        unifiedResponse: { data: rich.unifiedResponse.data },
+        submessages: rich.submessages || [],
+        contextInfo: rich.contextInfo || {}
+    }
+}
+
+export const generateUnifiedResponseContent = (quoted, captured) => {
+    const ctxInfo = buildRichContextInfo(quoted)
+    return { message: buildBotForwardedMessage(captured.submessages, ctxInfo, captured.unifiedResponse), messageId: generateMessageIDV2() }
+}
+
+export const generateRichMessageContent = (submessages, quoted, options) => {
+    const ctxInfo = buildRichContextInfo(quoted, options)
+    return { message: buildBotForwardedMessage(submessages, ctxInfo), messageId: generateMessageIDV2() }
+}
+
+export {
+    JS_KEYWORDS,
+    PYTHON_KEYWORDS,
+    LANGUAGE_KEYWORDS
+}

package/lib/Utils/messages.js

@@ -1,5 +1,5 @@
 import { Boom } from '@hapi/boom';
-import { randomBytes } from 'crypto';
+import { randomUUID, randomBytes } from 'crypto';
 import { promises as fs } from 'fs';
 import {} from 'stream';
 import { proto } from '../../WAProto/index.js';
@@ -26,6 +26,133 @@ const MessageTypeProto = {
     document: WAProto.Message.DocumentMessage
 };
 const ButtonType = proto.Message.ButtonsMessage.HeaderType;
+const RICH_RESPONSE_CODE_KEYWORDS = new Set([
+	'break',
+	'case',
+	'catch',
+	'continue',
+	'debugger',
+	'default',
+	'delete',
+	'do',
+	'else',
+	'finally',
+	'for',
+	'function',
+	'if',
+	'in',
+	'instanceof',
+	'new',
+	'return',
+	'switch',
+	'this',
+	'throw',
+	'try',
+	'typeof',
+	'var',
+	'void',
+	'while',
+	'with',
+	'true',
+	'false',
+	'null',
+	'undefined',
+	'NaN',
+	'Infinity',
+	'class',
+	'const',
+	'let',
+	'super',
+	'extends',
+	'export',
+	'import',
+	'yield',
+	'static',
+	'constructor',
+	'of',
+	'async',
+	'await',
+	'get',
+	'set',
+	'implements',
+	'interface',
+	'package',
+	'private',
+	'protected',
+	'public',
+	'enum',
+	'throws',
+	'transient'
+])
+const tokenizeCode = code => {
+	const tokens = []
+	let i = 0
+	const len = code.length
+	while (i < len) {
+		if (/\s/.test(code[i])) {
+			const start = i
+			while (i < len && /\s/.test(code[i])) i++
+			tokens.push({ content: code.slice(start, i), type: 'DEFAULT' })
+			continue
+		}
+		if (code[i] === '"' || code[i] === "'" || code[i] === '`') {
+			const start = i
+			const quote = code[i]
+			i++
+			while (i < len && code[i] !== quote) {
+				if (code[i] === '\\') i++
+				i++
+			}
+			i++
+			tokens.push({ content: code.slice(start, i), type: 'STR' })
+			continue
+		}
+		if (code[i] === '/' && i + 1 < len && code[i + 1] === '/') {
+			const start = i
+			while (i < len && code[i] !== '\n') i++
+			tokens.push({ content: code.slice(start, i), type: 'COMMENT' })
+			continue
+		}
+		if (code[i] === '/' && i + 1 < len && code[i + 1] === '*') {
+			const start = i
+			i += 2
+			while (i + 1 < len && !(code[i] === '*' && code[i + 1] === '/')) i++
+			i += 2
+			tokens.push({ content: code.slice(start, i), type: 'COMMENT' })
+			continue
+		}
+		if (/[0-9]/.test(code[i])) {
+			const start = i
+			while (i < len && /[0-9.]/.test(code[i])) i++
+			tokens.push({ content: code.slice(start, i), type: 'NUMBER' })
+			continue
+		}
+		if (/[a-zA-Z_$]/.test(code[i])) {
+			const start = i
+			while (i < len && /[a-zA-Z0-9_$]/.test(code[i])) i++
+			const word = code.slice(start, i)
+			if (RICH_RESPONSE_CODE_KEYWORDS.has(word)) {
+				tokens.push({ content: word, type: 'KEYWORD' })
+			} else {
+				let j = i
+				while (j < len && /\s/.test(code[j])) j++
+				tokens.push({ content: word, type: j < len && code[j] === '(' ? 'METHOD' : 'DEFAULT' })
+			}
+			continue
+		}
+		tokens.push({ content: code[i], type: 'DEFAULT' })
+		i++
+	}
+	const merged = []
+	for (const t of tokens) {
+		if (merged.length && merged[merged.length - 1].type === 'DEFAULT' && t.type === 'DEFAULT') {
+			merged[merged.length - 1].content += t.content
+		} else {
+			merged.push(t)
+		}
+	}
+	return merged
+}
 /**
  * Uses a regex to test whether the string contains a URL, and returns the URL if it does.
  * @param text eg. hello https://google.com
@@ -271,10 +398,242 @@ export const hasNonNullishProperty = (message, key) => {
 function hasOptionalProperty(obj, key) {
     return typeof obj === 'object' && obj !== null && key in obj && obj[key] !== null;
 }
+const applyContextInfoAndMentions = (interactiveMessage, message) => {
+	if ('contextInfo' in message && !!message.contextInfo) {
+		interactiveMessage.contextInfo = message.contextInfo
+	}
+	if ('mentions' in message && !!message.mentions) {
+		interactiveMessage.contextInfo = {
+			...(interactiveMessage.contextInfo || {}),
+			mentionedJid: message.mentions
+		}
+	}
+}
 export const generateWAMessageContent = async (message, options) => {
     var _a, _b;
     let m = {};
-    if (hasNonNullishProperty(message, 'text')) {
+    if ('interactiveButtons' in message && !!message.interactiveButtons) {
+        const nativeFlow = proto.Message.InteractiveMessage.NativeFlowMessage.fromObject({
+            buttons: message.interactiveButtons,
+            messageParamsJson: message.messageParams ?? "{}"
+        });
+        let interactiveMessage = { nativeFlowMessage: nativeFlow };
+        if ('text' in message) {
+            interactiveMessage.body = { text: message.text };
+        } else if ('caption' in message) {
+            interactiveMessage.body = { text: message.caption };
+        }
+        if ('title' in message && !!message.title) {
+            interactiveMessage.header = {
+                title: message.title,
+                subtitle: message.subtitle || "",
+                hasMediaAttachment: false
+            };
+            let media = null;
+            if ('image' in message && !!message.image) {
+                media = await prepareWAMessageMedia({ image: message.image }, options);
+                interactiveMessage.header.imageMessage = media.imageMessage;
+                if (message.image.caption) {
+                    interactiveMessage.header.imageMessage.caption = message.image.caption;
+                }
+            } else if ('video' in message && !!message.video) {
+                media = await prepareWAMessageMedia({ video: message.video }, options);
+                interactiveMessage.header.videoMessage = media.videoMessage;
+                if (message.video.caption) {
+                    interactiveMessage.header.videoMessage.caption = message.video.caption;
+                }
+            } else if ('gif' in message && !!message.gif) {
+                media = await prepareWAMessageMedia({ video: message.gif }, options);
+                interactiveMessage.header.videoMessage = media.videoMessage;
+                interactiveMessage.header.videoMessage.gifPlayback = true;
+                if (message.gif.caption) {
+                    interactiveMessage.header.videoMessage.caption = message.gif.caption;
+                }
+            } else if ('document' in message && !!message.document) {
+                media = await prepareWAMessageMedia({ document: message.document }, options);
+                interactiveMessage.header.documentMessage = media.documentMessage;
+                let docuR = interactiveMessage.header.documentMessage;
+                docuR.fileName = message.document.fileName || "Document";
+                docuR.mimetype = message.document.mimetype || docuR.mimetype;
+                if (message.document.caption) docuR.caption = message.document.caption;
+                if (message.document.jpegThumbnail) docuR.jpegThumbnail = message.document.jpegThumbnail;
+            } else if ('location' in message && !!message.location) {
+                let mLoc = message.location;
+                interactiveMessage.header.locationMessage = {
+                    degreesLongitude: mLoc.longitude || 0,
+                    degreesLatitude: mLoc.latitude || 0,
+                    name: mLoc.name || null,
+                    address: mLoc.address || null,
+                    url: mLoc.url || null
+                };
+                media = true;
+            } else if ('thumbnail' in message && !!message.thumbnail) {
+                interactiveMessage.header.jpegThumbnail = message.thumbnail;
+            }
+            if (media || ('thumbnail' in message && !!message.thumbnail)) {
+                interactiveMessage.header.hasMediaAttachment = true;
+            }
+        }
+        if ('footer' in message && !!message.footer) {
+            interactiveMessage.footer = { text: message.footer };
+        }
+        applyContextInfoAndMentions(interactiveMessage, message);
+        m = { interactiveMessage };
+    }
+	else if ('buttons' in message && !!message.buttons) {
+        const buttonsMessage = {
+            buttons: message.buttons.map(b => ({ ...b, type: proto.Message.ButtonsMessage.Button.Type.RESPONSE }))
+        };
+        if ('text' in message) {
+            buttonsMessage.contentText = message.text;
+            buttonsMessage.headerType = ButtonType.EMPTY;
+        }
+        else {
+            if ('caption' in message) {
+                buttonsMessage.contentText = message.caption;
+            }
+            const contentType = getContentType(m);
+            const type = contentType.replace('Message', '').toUpperCase();
+            buttonsMessage.headerType = ButtonType[type];
+            Object.assign(buttonsMessage, m);
+        }
+        if ('footer' in message && !!message.footer) {
+            buttonsMessage.footerText = message.footer;
+        }
+        m = { buttonsMessage };
+    }
+    else if ('richResponse' in message) {
+		const {
+			text,
+			code,
+			language = 'javascript',
+			botJid = '867051314767696@bot',
+			table,
+			latex,
+			map,
+			imageUrl,
+			imageUrls,
+			responseId,
+			messageSecret: richSecret
+		} = message.richResponse
+		const sections = []
+		if (text) {
+			sections.push({
+				view_model: {
+					primitive: { text, __typename: 'GenAIMarkdownTextUXPrimitive' },
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (code) {
+			sections.push({
+				view_model: {
+					primitive: {
+						language,
+						code_blocks: tokenizeCode(String(code)),
+						__typename: 'GenAICodeUXPrimitive'
+					},
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (table && Array.isArray(table.rows)) {
+			sections.push({
+				view_model: {
+					primitive: {
+						rows: table.rows.map(row => ({
+							cells: Array.isArray(row) ? row.map(c => ({ text: String(c) })) : row.cells
+						})),
+						__typename: 'GenAITableUXPrimitive'
+					},
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (latex) {
+			const expressions = Array.isArray(latex)
+				? latex.map(e => (typeof e === 'string' ? { expression: e } : e))
+				: [{ expression: String(latex) }]
+			sections.push({
+				view_model: {
+					primitive: { expressions, __typename: 'GenAILatexUXPrimitive' },
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (map) {
+			sections.push({
+				view_model: {
+					primitive: {
+						latitude: map.latitude,
+						longitude: map.longitude,
+						zoom: map.zoom,
+						title: map.title,
+						annotations: map.annotations || [],
+						__typename: 'GenAIMapUXPrimitive'
+					},
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (imageUrl) {
+			sections.push({
+				view_model: {
+					primitive: { url: imageUrl, __typename: 'GenAIInlineImageUXPrimitive' },
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (imageUrls && Array.isArray(imageUrls) && imageUrls.length > 0) {
+			sections.push({
+				view_model: {
+					primitive: {
+						urls: imageUrls.map(u => (typeof u === 'string' ? { url: u } : u)),
+						__typename: 'GenAIGridImageUXPrimitive'
+					},
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		if (!sections.length && !text) {
+			sections.push({
+				view_model: {
+					primitive: { text: '', __typename: 'GenAIMarkdownTextUXPrimitive' },
+					__typename: 'GenAISingleLayoutViewModel'
+				}
+			})
+		}
+		const unifiedData = {
+			response_id: responseId || randomUUID(),
+			sections
+		}
+		return proto.Message.fromObject({
+			messageContextInfo: {
+				deviceListMetadata: {},
+				deviceListMetadataVersion: 2,
+				messageSecret: richSecret || randomBytes(32)
+			},
+			botForwardedMessage: {
+				message: {
+					richResponseMessage: {
+						submessages: [],
+						messageType: 1,
+						unifiedResponse: { data: Buffer.from(JSON.stringify(unifiedData)) },
+						contextInfo: {
+							forwardingScore: 2,
+							isForwarded: true,
+							forwardedAiBotMessageInfo: { botJid },
+							botMessageSharingInfo: {
+								botEntryPointOrigin: 1,
+								forwardScore: 2
+							}
+						}
+					}
+				}
+			}
+		})
+	}
+    else if (hasNonNullishProperty(message, 'text')) {
         const extContent = { text: message.text };
         let urlInfo = message.linkPreview;
         if (typeof urlInfo === 'undefined') {
@@ -317,28 +676,6 @@ export const generateWAMessageContent = async (message, options) => {
             m.contactsArrayMessage = WAProto.Message.ContactsArrayMessage.create(message.contacts);
         }
     }
-    else if (hasNonNullishProperty(message, 'buttons')) {
-        const buttonsMessage = {
-            buttons: message.buttons.map(b => ({ ...b, type: proto.Message.ButtonsMessage.Button.Type.RESPONSE }))
-        };
-        if ('text' in message) {
-            buttonsMessage.contentText = message.text;
-            buttonsMessage.headerType = ButtonType.EMPTY;
-        }
-        else {
-            if ('caption' in message) {
-                buttonsMessage.contentText = message.caption;
-            }
-            const contentType = getContentType(m);
-            const type = contentType.replace('Message', '').toUpperCase();
-            buttonsMessage.headerType = ButtonType[type];
-            Object.assign(buttonsMessage, m);
-        }
-        if ('footer' in message && !!message.footer) {
-            buttonsMessage.footerText = message.footer;
-        }
-        m = { buttonsMessage };
-    }
     else if (hasNonNullishProperty(message, 'location')) {
         m.locationMessage = WAProto.Message.LocationMessage.create(message.location);
     }