@zervonex/baileys 3.0.7 → 3.0.9

package/lib/Builders/AIRich.js

@@ -3,10 +3,10 @@ import { proto } from '../../WAProto/index.js'
 import {
   generateMessageIDV2, generateVerificationMetadata, waitAllPromises, extractIE, tokenizeCode, generateWAMessageFromContent, generateWAMessage,
   makeWhatsAppFlowButton, metaTyping, sendMetaComposited, PlanningStepStatus, buildSteps, replayPlanning, replayPlanningOnly, buildReasoningSteps, buildSearchSteps,
-  mixedSteps, buildBotRichResponse, BotCapabilityType, generateTableContent, generateListContent, generateCodeBlockContent,
+  mixedSteps, buildBotRichResponse, BotCapabilityType, RichSubMessageType, generateTableContent, generateListContent, generateCodeBlockContent,
   generateLatexContent, captureUnifiedResponse, generateUnifiedResponseContent
 } from '../Utils/index.js'
-import { isJidGroup, isJidMetaAI, jidNormalizedUser } from '../WABinary/index.js'
+import { isJidGroup, isJidMetaAI, jidNormalizedUser, getBizBinaryNode } from '../WABinary/index.js'
 import { BOT_RENDERING_CONFIG_METADATA } from '../Defaults/index.js'
 import { randomBytes } from 'crypto'
 
@@ -60,6 +60,18 @@ const mkProduct = d => ({ __typename: 'GenAIProductItemCardPrimitive', title: d.
 const mkSearchResult = d => ({ __typename: 'GenAISearchResultPrimitive', source_url: d.url ?? d.source_url ?? '', source_display_name: d.title ?? d.source_display_name ?? '', source_type: d.source_type ?? 'web', source_subtitle: d.subtitle ?? d.source_subtitle ?? '', favicon: d.favicon ?? null })
 const mkMap = ({ latitude, longitude, name = '', address = '' }) => ({ __typename: 'GenAIMapPrimitive', latitude, longitude, name, address })
 
+const _mp4Duration = (buf, silent = true) => {
+  try { if (!Buffer.isBuffer(buf) || buf.length < 8) { if (silent) return 0; throw new Error('Invalid buffer') }; let o = 0; while (o < buf.length - 8) { const sz = buf.readUInt32BE(o); if (sz < 8 || o + sz > buf.length) { if (silent) return 0; throw new Error('Bad atom') }; if (buf.toString('ascii', o + 4, o + 8) === 'moov') { let m = o + 8; const me = o + sz; while (m < me - 8) { const cs = buf.readUInt32BE(m); if (cs < 8 || m + cs > me) break; if (buf.toString('ascii', m + 4, m + 8) === 'mvhd') { const v = buf.readUInt8(m + 8); if (v === 0) { const ts = buf.readUInt32BE(m + 20); const dur = buf.readUInt32BE(m + 24); return ts ? dur / ts : 0 }; if (v === 1) { const ts = buf.readUInt32BE(m + 32); const dur = Number(buf.readBigUInt64BE(m + 36)); return ts ? dur / ts : 0 } }; m += cs } }; o += sz }; return 0 }
+  catch { return 0 }
+}
+
+const _mp4Preview = async (buf, time, silent = true) => {
+  const fail = e => { if (!silent) throw e; return null }; if (!Buffer.isBuffer(buf) || !buf.length) return fail(new Error('Empty buffer'))
+  try {
+    const { Readable, PassThrough } = await import('stream'); const { default: ffmpeg } = await import('fluent-ffmpeg'); const src = new Readable({ read() { } }); src.push(buf); src.push(null); const out = new PassThrough(); const chunks = []; out.on('data', c => chunks.push(c)); const t = time ?? Math.max(0.5, Math.min(_mp4Duration(buf) * 0.3, 10)); await new Promise((res, rej) => { out.on('end', res); out.on('error', rej); ffmpeg(src).outputOptions([`-ss ${t}`, '-vframes 1', '-vcodec png', '-f image2pipe']).on('error', rej).pipe(out, { end: true }) }); const b = Buffer.concat(chunks); return b.length ? b.toString('base64') : fail(new Error('Empty output'))
+  } catch (e) { return fail(e) }
+}
+
 const mkRichMap = ({ centerLatitude, centerLongitude, latitudeDelta, longitudeDelta, annotations = [], showInfoList = false }) => {
   if (centerLatitude == null || centerLongitude == null) throw new TypeError('addRichMap requires centerLatitude and centerLongitude')
   const submessage = { messageType: proto.AIRichResponseSubMessageType.AI_RICH_RESPONSE_MAP, mapMetadata: { centerLatitude, centerLongitude, ...(latitudeDelta != null ? { latitudeDelta } : {}), ...(longitudeDelta != null ? { longitudeDelta } : {}), showInfoList: !!showInfoList, annotations: annotations.map((a, i) => ({ annotationNumber: a.annotationNumber ?? i + 1, latitude: a.latitude, longitude: a.longitude, title: a.title ?? '', body: a.body ?? '' })) } }
@@ -177,8 +189,8 @@ export const prepareRichGenerationMessage = ({ text = '', mediaType = 'video', u
   const sections = []
   if (text) sections.push({ view_model: { primitive: { text: String(text), inline_entities: [], __typename: 'GenAIMarkdownTextUXPrimitive' }, __typename: 'GenAISingleLayoutViewModel' } })
   sections.push({ view_model: { primitive: buildImaginePrimitive({ mediaType, url, thumbnail, mimeType, fileLength, duration, status, estimatedMs }), __typename: 'GenAISingleLayoutViewModel', id: itemId } })
-  const submessages = text ? [{ messageType: 2, messageText: String(text) }] : []
-  const richResponseMessage = proto.AIRichResponseMessage.create({ submessages, messageType: 1, unifiedResponse: { data: Buffer.from(JSON.stringify({ response_id: responseId, sections })) }, contextInfo: { isForwarded: true, forwardingScore: 1, forwardedAiBotMessageInfo: { botJid: DEFAULT_BOT_JID }, forwardOrigin: 4 } })
+  const submessages = text ? [{ messageType: RichSubMessageType.AI_RICH_RESPONSE_TEXT, messageText: String(text) }] : []
+  const richResponseMessage = proto.AIRichResponseMessage.create({ submessages, messageType: proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD, unifiedResponse: { data: Buffer.from(JSON.stringify({ response_id: responseId, sections })) }, contextInfo: { isForwarded: true, forwardingScore: 1, forwardedAiBotMessageInfo: { botJid: DEFAULT_BOT_JID }, forwardOrigin: 4 } })
   const message = wrapToBotForwardedMessage(richResponseMessage, verificationMetadata)
   message.messageContextInfo.botMetadata.botResponseId = responseId
   return { message, responseId, itemId }
@@ -189,6 +201,7 @@ export class AIRich {
   #responseId = null; #dynamic = true; #signedVerificationMetadata = null
   #contextInfoExtra = {}; #embeddedScreens = []; #suggestedPrompts = null
   #botSources = null; #pttTranscript = null
+  #footerSections = []; #responseMeta = {}; #botMetadataExtra = {}
 
   constructor(client, opts = {}) {
     if (!client) throw new Error('Socket client is required')
@@ -210,17 +223,28 @@ export class AIRich {
   setBotJid(jid) { this.#botJid = jid; return this }
   setResponseId(id) { this.#responseId = id; this.#dynamic = false; return this }
   setContextInfo(ci) { this.#contextInfoExtra = ci; return this }
+  setResponseMeta(meta) { if (!meta || typeof meta !== 'object') throw new TypeError('responseMeta must be a plain object'); this.#responseMeta = { ...this.#responseMeta, ...meta }; return this }
+  setBotMetadata(extra) { if (!extra || typeof extra !== 'object') throw new TypeError('botMetadata must be a plain object'); this.#botMetadataExtra = { ...this.#botMetadataExtra, ...extra }; return this }
+  addFooterSection(section) { if (!section || typeof section !== 'object') throw new TypeError('footer section must be a plain object'); this.#footerSections.push(section); return this }
+  clearFooterSections() { this.#footerSections = []; return this }
 
   addText(text, opts = {}) { const { text: t, inline_entities } = extractIE(text); return this.#add(single(mkMarkdown(t, inline_entities)), null, opts) }
-  addCode(language, content, opts = {}) { const { codeBlocks, unified_codeBlock } = tokenizeCode(content, language); const submessage = { messageType: 5, codeMetadata: { codeLanguage: language, codeBlocks } }; return this.#add(single(mkCode(content, language, unified_codeBlock)), submessage, opts) }
+  addCode(language, content, opts = {}) { const { codeBlocks, unified_codeBlock } = tokenizeCode(content, language); const submessage = { messageType: RichSubMessageType.AI_RICH_RESPONSE_CODE, codeMetadata: { codeLanguage: language, codeBlocks } }; return this.#add(single(mkCode(content, language, unified_codeBlock)), submessage, opts) }
   addTable(rows, title = '', opts = {}) { return this.#add(single(mkTable(rows, title)), null, opts) }
   addImage(input, opts = {}) { const resolved = this.#uploadMedia(input, 'image').then(url => single(mkImage(url))); return this.#add(resolved, null, opts) }
-  addVideo(url, opts = {}) { const resolved = this.#uploadMedia(url, 'video').then(u => single(mkVideo(u))); return this.#add(resolved, null, opts) }
+  addVideo(url, { autoFill = false, mimeType = 'video/mp4', ...opts } = {}) {
+    const resolved = this.#uploadMedia(url, 'video').then(async finalUrl => { if (!autoFill) return single(mkVideo(finalUrl)); const buf = await fetch(finalUrl).then(r => r.arrayBuffer()).then(b => Buffer.from(b)).catch(() => null); const duration = buf ? _mp4Duration(buf) : 0; const thumbnail = buf ? await _mp4Preview(buf, null, true) : null; return single({ __typename: 'GenAIImaginePrimitive', imagine_type: 'ANIMATE', status: { status: 'READY' }, media: { url: finalUrl, mime_type: mimeType, file_length: buf?.length ?? 0, duration }, thumbnail: thumbnail ? { raw_media: thumbnail } : undefined }) }); return this.#add(resolved, null, opts)
+  }
   addGrid(items, opts = {}) { if (!Array.isArray(items) || items.length < 2) throw new Error('addGrid requires at least 2 items'); return this.#add(grid(items.map(i => mkImage(i))), null, opts) }
 
   addVideoGrid(videos, opts = {}) {
     if (!Array.isArray(videos) || videos.length < 2) throw new Error('addVideoGrid requires at least 2 videos')
-    const prims = videos.map(v => { const url = typeof v === 'string' ? v : (v?.url ?? v?.videoUrl ?? ''); const item = typeof v === 'string' ? {} : v; return { __typename: 'GenAIVideoPrimitive', reels_url: url, reels_title: item.title, thumbnail_url: item.thumbnailUrl ?? item.thumbnail, creator: item.creator, avatar_url: item.avatarUrl, likes_count: item.likes, comments_count: item.comments, shares_count: item.shares, is_verified: item.isVerified, video_delivery_response: { progressive_urls: (item.progressiveUrls ?? []).map(u => ({ progressive_url: u, __typename: 'GenAIProgressiveUrlResponse' })), dash_manifests: (item.dashManifests ?? []).map(m => ({ manifest_xml: m, __typename: 'GenAIDashManifestResponse' })), __typename: 'GenAIVideoDeliveryResponse' } } })
+    const prims = videos.map((v, i) => {
+      const item = typeof v === 'string' ? {} : v
+      const url = typeof v === 'string' ? v : (v?.url ?? v?.videoUrl ?? '')
+      if (!url) throw new Error(`VideoGrid item ${i}: video requires a URL`)
+      return { __typename: 'GenAIVideoPrimitive', reels_url: url, reels_title: item.title, thumbnail_url: item.thumbnailUrl ?? item.thumbnail, creator: item.creator, avatar_url: item.avatarUrl, likes_count: item.likes, comments_count: item.comments, shares_count: item.shares, is_verified: item.isVerified, video_delivery_response: { progressive_urls: (item.progressiveUrls ?? []).map(u => ({ progressive_url: u, __typename: 'GenAIProgressiveUrlResponse' })), dash_manifests: (item.dashManifests ?? []).map(m => ({ manifest_xml: m, __typename: 'GenAIDashManifestResponse' })), __typename: 'GenAIVideoDeliveryResponse' } }
+    })
     return this.#add(grid(prims), null, opts)
   }
 
@@ -282,15 +306,15 @@ export class AIRich {
   addVerticalAlbum(items, opts = {}) {
     if (!Array.isArray(items) || items.length < 2) throw new Error('addVerticalAlbum requires at least 2 items')
     const expiration = String(Date.now() + 30 * 24 * 60 * 60 * 1000)
-    const primitives = items.map(item => {
+    const primitives = items.map((item, i) => {
       const kind = item?.type || (item?.video ? 'video' : 'image')
       if (kind === 'video') {
-        const v = item.video || item
-        return buildImaginePrimitive({ mediaType: 'video', url: typeof v === 'string' ? v : (v?.url || ''), mimeType: v?.mimeType || 'video/mp4', fileLength: v?.fileLength || 0, duration: v?.duration || 0, status: 'READY' })
+        const v = item.video || item.data || item
+        return buildImaginePrimitive({ mediaType: 'video', url: typeof v === 'string' ? v : (v?.url || ''), mimeType: v?.mimeType ?? v?.mime_type ?? 'video/mp4', fileLength: v?.fileLength || 0, duration: v?.duration || 0, status: 'READY' })
       }
       const img = typeof (item?.image || item) === 'string' ? { url: item?.image || item } : (item?.image || item || {})
-      const url = img.url || ''
-      if (!url) throw new Error('Each vertical album image item requires a URL')
+      const url = img?.imageHighResUrl ?? img?.highResUrl ?? img?.fullUrl ?? img?.imagePreviewUrl ?? img?.previewUrl ?? img?.url ?? ''
+      if (!url) throw new Error(`VerticalAlbum item ${i}: image requires a URL`)
       return { __typename: 'GenAIImagePrimitive', preview_image: { url, url_fallback: url, mime_type: img.mimeType || 'image/jpeg', expiration_timestamp_ms: expiration, width: img.width || 600, height: img.height || 400 }, full_image: { url, url_fallback: url, mime_type: img.mimeType || 'image/jpeg', expiration_timestamp_ms: expiration, width: img.width || 600, height: img.height || 400 }, dark_mode_preview_image: { url, url_fallback: url, mime_type: img.mimeType || 'image/jpeg', expiration_timestamp_ms: expiration, width: img.width || 600, height: img.height || 400 }, dark_mode_full_image: { url, url_fallback: url, mime_type: img.mimeType || 'image/jpeg', expiration_timestamp_ms: expiration, width: img.width || 600, height: img.height || 400 }, asset_query_status: 'FETCHED', display_mode: 'FULL_WIDTH', full_width: true }
     })
     return this.#add({ view_model: { primitives, display_mode: 'FULL_WIDTH', full_width: true, __typename: 'GenAIVStackLayoutViewModel' } }, null, opts)
@@ -302,6 +326,28 @@ export class AIRich {
     return this.#add(single(mkMeta(`🎤 _${transcript}_`)), null, opts)
   }
 
+  // ─── Node introspection & mutation ────────────────────────────────────────
+  hasId(id) { return typeof id === 'string' && id in this.#idIndex }
+  getIds() { return this.#nodes.map(n => n.id) }
+  peek(id) { const idx = this.#idIndex[id]; return idx !== undefined ? { ...this.#nodes[idx] } : null }
+  assignId(index, newId) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.#nodes.length) throw new RangeError(`Node index ${index} is out of range`)
+    if (newId in this.#idIndex) throw new Error(`Id "${newId}" already exists`)
+    const node = this.#nodes[index]
+    delete this.#idIndex[node.id]
+    node.id = newId
+    this.#idIndex[newId] = index
+    return this
+  }
+  delete(id) {
+    const idx = this.#idIndex[id]
+    if (idx === undefined) throw new Error(`Node id "${id}" not found`)
+    this.#nodes.splice(idx, 1)
+    this.#idIndex = {}
+    this.#nodes.forEach((n, i) => { this.#idIndex[n.id] = i })
+    return this
+  }
+
   loadFrom(msg) {
     const rich = msg?.botForwardedMessage?.message?.richResponseMessage ?? msg?.richResponseMessage
     if (!rich) throw new Error('Not an AIRich message')
@@ -324,11 +370,12 @@ export class AIRich {
     const submessages = resolved.map(n => n.submessage).filter(Boolean)
     const embeddedScreens = await waitAllPromises(this.#embeddedScreens)
     const responseId = this.#dynamic ? crypto.randomUUID() : (this.#responseId ?? crypto.randomUUID())
-    const unifiedData = JSON.stringify({ response_id: responseId, sections, ...(embeddedScreens.length ? { embedded_screens: embeddedScreens } : {}) })
+    const footerSections = await waitAllPromises(this.#footerSections)
+    const unifiedData = JSON.stringify({ response_id: responseId, ...this.#responseMeta, sections, ...(footerSections.length ? { footer_sections: footerSections } : {}), ...(embeddedScreens.length ? { embedded_screens: embeddedScreens } : {}) })
     const contextInfo = { isForwarded: true, forwardOrigin: 4, forwardingScore: 1, forwardedAiBotMessageInfo: { botJid }, participant: '13135550002@s.whatsapp.net', remoteJid: 'status@broadcast', quotedMessage: { protocolMessage: { type: 25 } }, ...this.#contextInfoExtra, ...(quoted ? { quotedMessage: quoted.message, stanzaId: quoted.key?.id, participant: quotedParticipant ?? quoted.key?.participant ?? quoted.key?.remoteJid, remoteJid: quoted.key?.remoteJid } : {}) }
     const verification = this.#signedVerificationMetadata ?? generateVerificationMetadata()
     const { sources, mediaDetailsMetadataList } = buildAdditionalBotMetadata(submessages)
-    const richResponseMessage = proto.AIRichResponseMessage.create({ messageType: 1, submessages, unifiedResponse: { data: Buffer.from(unifiedData) }, originalRecipientMetadata: { data: Buffer.from(unifiedData) }, contextInfo })
+    const richResponseMessage = proto.AIRichResponseMessage.create({ messageType: proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD, submessages, unifiedResponse: { data: Buffer.from(unifiedData) }, originalRecipientMetadata: { data: Buffer.from(unifiedData) }, contextInfo })
     const message = wrapToBotForwardedMessage(richResponseMessage, verification)
     const botMeta = message.messageContextInfo.botMetadata
     botMeta.botResponseId = responseId
@@ -340,6 +387,7 @@ export class AIRich {
     if (mediaDetailsMetadataList.length) botMeta.unifiedResponseMutation = { mediaDetailsMetadataList }
     if (this.#suggestedPrompts) botMeta.suggestedPromptMetadata = this.#suggestedPrompts
     if (this.#pttTranscript) botMeta.pttPromptMetadata = { transcript: this.#pttTranscript }
+    if (Object.keys(this.#botMetadataExtra).length) Object.assign(botMeta, this.#botMetadataExtra)
     botMeta.capabilities = { richResponseUnifiedResponse: true, richResponseEmbeddedScreens: true, richResponseInlineLinksEnabled: true, richResponseUrBloksEnabled: true, richResponseUrImagine: true, richResponseUrReasoning: true }
     return await await generateWAMessageFromContent(jid, message, { userJid: this.#client.user?.id, messageId })
   }
@@ -347,14 +395,11 @@ export class AIRich {
   async send(jid, opts = {}) {
     const { additionalNodes = [], ...buildOpts } = opts
     const msg = await this.build(jid, buildOpts)
+    const bizNode = getBizBinaryNode(msg.message)
     const relayFn = this.#client.relayRichMessage ?? this.#client.relayMessage
     await relayFn(jid, msg, {
       ...opts,
-      additionalNodes: [
-        { tag: 'bot', attrs: { biz_bot: '1' }, content: undefined },
-        { tag: 'biz', attrs: {}, content: [{ tag: 'interactive', attrs: { type: 'native_flow', v: '1' }, content: [{ tag: 'native_flow', attrs: { v: '9', name: 'mixed' } }] }] },
-        ...(opts.additionalNodes ?? [])
-      ]
+      additionalNodes: [...(bizNode ? [bizNode] : []), ...(opts.additionalNodes ?? [])]
     })
     return msg
   }
@@ -370,31 +415,17 @@ export class AIRich {
 }
 
 export function aiRichFromObject(client, jid, data, opts = {}) {
-  const r = new AIRich(client, { botJid: data.botJid })
-  if (data.title) r.setTitle(data.title)
+  const { _r, ...sendOpts } = opts
+  const r = _r ?? new AIRich(client, { botJid: data.botJid })
+  const isChild = !!_r
+  if (!isChild && (data.title ?? data.disclaimerText)) r.setTitle(data.title ?? data.disclaimerText)
   if (data.parts) {
     for (const p of data.parts) {
-      if (p.type === 'text') r.addText(p.content ?? p.text)
-      if (p.type === 'code') r.addCode(p.language ?? '', p.content ?? p.code)
-      if (p.type === 'table') r.addTable(p.table, p.title)
-      if (p.type === 'image') r.addImage(p.url ?? p.image)
-      if (p.type === 'grid') r.addGrid(p.items ?? p.images)
-      if (p.type === 'videoGrid') r.addVideoGrid(p.items ?? p.videos)
-      if (p.type === 'imagine') r.addImagine(p)
-      if (p.type === 'sources') r.addSource(p.sources)
-      if (p.type === 'divider') r.addDivider()
-      if (p.type === 'spacer') r.addSpacer(p.spacing)
-      if (p.type === 'tip') r.addTip(p.content ?? p.text)
-      if (p.type === 'meta') r.addMetadata(p.content ?? p.text)
-      if (p.type === 'foa') r.addFOAText(p.content ?? p.text)
-      if (p.type === 'html') r.addHtml(p.html, p.trustedSources)
-      if (p.type === 'richHtml') r.addRichHtml(p.tabs ?? p.html, p.title)
-      if (p.type === 'copy') r.addCopyAction(p.text, p.label, p.alignment)
-      if (p.type === 'social') r.addSocialProfile(p)
-      if (p.type === 'imagine') r.addImagine(p)
-      if (p.type === 'rawSections') r.addRawSections(p.sections, p.submessages)
+      const code = Array.isArray(p.code) ? p.code.map(c => c.codeContent ?? c.content ?? '').join('\n') : p.code
+      const table = Array.isArray(p.table) ? p.table.map(r => Array.isArray(r) ? r : (r.items ?? r.cells ?? [])) : p.table
+      aiRichFromObject(client, jid, { ...p, ...(code ? { code } : {}), ...(table ? { table } : {}) }, { ...sendOpts, _r: r })
     }
-    return r.send(jid, opts)
+    return isChild ? r : r.send(jid, sendOpts)
   }
   if (data.texts) data.texts.forEach(t => r.addText(t))
   else if (data.text) r.addText(data.text)
@@ -426,7 +457,7 @@ export function aiRichFromObject(client, jid, data, opts = {}) {
   if (data.metadata ?? data.meta) r.addMetadata(data.metadata ?? data.meta)
   if (data.footer) r.addFOAText(data.footer)
   if (data.rawSections) r.addRawSections(data.rawSections, data.rawSubmessages)
-  return r.send(jid, opts)
+  return isChild ? r : r.send(jid, sendOpts)
 }
 
 const stripDevice = jid => String(jid || '').replace(/:\d+$/, '')
@@ -484,7 +515,7 @@ export function attachAIRich(result, config) {
   result.captureAiRich = msg => {
     const rich = msg?.botForwardedMessage?.message?.richResponseMessage ?? msg?.richResponseMessage
     if (!rich?.unifiedResponse?.data) return null
-    return { submessages: rich.submessages ?? [], sections: JSON.parse(Buffer.from(rich.unifiedResponse.data, 'base64').toString()), contextInfo: rich.contextInfo ?? {}, messageType: rich.messageType ?? 1 }
+    return { submessages: rich.submessages ?? [], sections: JSON.parse(Buffer.isBuffer(rich.unifiedResponse.data) ? rich.unifiedResponse.data.toString() : Buffer.from(rich.unifiedResponse.data, 'base64').toString()), contextInfo: rich.contextInfo ?? {}, messageType: rich.messageType ?? proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD }
   }
 
   result.relayAiRich = (jid, captured, opts = {}) => {
@@ -524,7 +555,7 @@ export function attachAIRich(result, config) {
   }
   result.sendBotPlanning = async (jid, content = {}, opts = {}) => {
     const { text = 'Execution plan', steps = [], queryPlan = true, botJid = META_AI_BOT_JID } = content
-    const submessages = [{ messageType: proto.AIRichResponseSubMessageType.AI_RICH_RESPONSE_TEXT, messageText: text }, ...steps.map(s => ({ messageType: proto.AIRichResponseSubMessageType.AI_RICH_RESPONSE_TEXT, messageText: typeof s === 'string' ? s : `${s.title || s.name || 'Step'}${s.body ? `: ${s.body}` : ''}` }))]
+    const submessages = [{ messageType: RichSubMessageType.AI_RICH_RESPONSE_TEXT, messageText: text }, ...steps.map(s => ({ messageType: RichSubMessageType.AI_RICH_RESPONSE_TEXT, messageText: typeof s === 'string' ? s : `${s.title || s.name || 'Step'}${s.body ? `: ${s.body}` : ''}` }))]
     const wrapped = buildBotRichResponse({ botJid, submessages, capabilities: [BotCapabilityType.AGENTIC_PLANNING, ...(queryPlan ? [BotCapabilityType.QUERY_PLAN] : [])] })
     const full = await await generateWAMessageFromContent(jid, wrapped, { userJid: sock.user?.id, messageId: opts.messageId || generateMessageIDV2(sock.user?.id) })
     return relayRichMessage(jid, full, opts)
@@ -534,7 +565,7 @@ export function attachAIRich(result, config) {
     if (!Array.isArray(steps) || steps.length === 0) throw new Error('sendBotPlan requires at least one step')
     const statusEnum = proto.BotProgressIndicatorMetadata.BotPlanningStepMetadata.PlanningStepStatus
     const normalizedSteps = steps.map((s, i) => ({ statusTitle: String(s.title || s.name || `Step ${i + 1}`), statusBody: String(s.body || s.description || ''), status: s.status === 'executing' ? statusEnum.EXECUTING : (s.status === 'finished' || s.status === 'completed') ? statusEnum.FINISHED : statusEnum.PLANNED, isReasoning: s.isReasoning === true, isEnhancedSearch: s.isEnhancedSearch === true }))
-    const wrapped = { messageContextInfo: { botMetadata: { capabilityMetadata: { capabilities: [proto.BotCapabilityMetadata.BotCapabilityType.AGENTIC_PLANNING, proto.BotCapabilityMetadata.BotCapabilityType.QUERY_PLAN] }, progressIndicatorMetadata: { progressDescription: title, stepsMetadata: normalizedSteps, ...(estimatedCompletionTime ? { estimatedCompletionTime } : {}) } } }, botForwardedMessage: { message: { richResponseMessage: { messageType: 1, submessages: [{ messageType: 2, messageText: text }], contextInfo: { isForwarded: true, forwardingScore: 1, forwardedAiBotMessageInfo: { botJid }, forwardOrigin: 4 } } } } }
+    const wrapped = { messageContextInfo: { botMetadata: { capabilityMetadata: { capabilities: [proto.BotCapabilityMetadata.BotCapabilityType.AGENTIC_PLANNING, proto.BotCapabilityMetadata.BotCapabilityType.QUERY_PLAN] }, progressIndicatorMetadata: { progressDescription: title, stepsMetadata: normalizedSteps, ...(estimatedCompletionTime ? { estimatedCompletionTime } : {}) } } }, botForwardedMessage: { message: { richResponseMessage: { messageType: proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD, submessages: [{ messageType: RichSubMessageType.AI_RICH_RESPONSE_TEXT, messageText: text }], contextInfo: { isForwarded: true, forwardingScore: 1, forwardedAiBotMessageInfo: { botJid }, forwardOrigin: 4 } } } } }
     const full = await await generateWAMessageFromContent(jid, wrapped, { userJid: sock.user?.id, messageId: opts.messageId || generateMessageIDV2(sock.user?.id) })
     await sock.relayMessage(jid, full.message, { messageId: full.key.id })
     return full

package/lib/Builders/base.js

@@ -1,6 +1,6 @@
 import crypto from 'crypto'
 import { proto } from '../../WAProto/index.js'
-import { generateWAMessage, generateWAMessageFromContent, getUrlInfo, prepareWAMessageMedia } from '../Utils/index.js'
+import { generateWAMessage, generateWAMessageFromContent, getUrlInfo, prepareWAMessageMedia, resolveNativeFlowButtons } from '../Utils/index.js'
 
 export const getRandomHex = () => '#' + Array.from(crypto.getRandomValues(new Uint8Array(3)), b => b.toString(16).padStart(2, '0')).join('')
 
@@ -100,7 +100,7 @@ export async function handlePollResult(sock, jid, p) {
         pollResultSnapshotMessage: proto.Message.PollResultSnapshotMessage.create({
             name: p.name,
             pollVotes: (p.pollVotes ?? []).map(v => proto.Message.PollResultSnapshotMessage.PollVote.create({ optionName: v.optionName, optionVoteCount: String(v.optionVoteCount ?? 0) })),
-            contextInfo: proto.ContextInfo.create({ isForwarded: true, forwardingScore: 1, forwardedNewsletterMessageInfo: proto.ContextInfo.ForwardedNewsletterMessageInfo.create({ newsletterName: p.newsletter?.newsletterName ?? 'Newsletter', newsletterJid: p.newsletter?.newsletterJid ?? '120363399602691477@newsletter', serverMessageId: 1000, contentType: 'UPDATE' }) })
+            contextInfo: proto.ContextInfo.create({ isForwarded: true, forwardingScore: 1 })
         })
     }, { userJid: sock.user?.id })
     await sock.relayMessage(jid, msg.message, { messageId: msg.key.id })
@@ -109,6 +109,16 @@ export async function handlePollResult(sock, jid, p) {
 
 export const q = quoted => quoted ? { quoted } : {}
 
+export async function handleNativeFlow(sock, jid, content, opts = {}) {
+    const rawButtons = content.nativeFlow
+    const resolved = resolveNativeFlowButtons(rawButtons)
+    const offerParams = content.offerText ? { limited_time_offer: { text: content.offerText, url: content.offerUrl || '', copy_code: content.offerCode || '', expiration_time: content.offerExpiration ? Math.floor(content.offerExpiration / 1000) : 0 } } : {}
+    const messageParams = { ...offerParams }
+    if (content.optionText) messageParams.bottom_sheet = { in_thread_buttons_limit: 1, divider_indices: Array.from({ length: resolved.length }, (_, i) => i), list_title: content.optionTitle || content.optionText, button_title: content.optionText }
+    if (content.interactiveAsTemplate) return sock.sendMessage(jid, { ...content, nativeFlow: undefined, interactiveButtons: undefined, templateButtons: resolved.map(b => { const p = JSON.parse(b.buttonParamsJson ?? '{}'); return b.name === 'cta_url' ? { text: p.display_text, url: p.url } : b.name === 'cta_call' ? { text: p.display_text, call: p.phone_number } : { text: p.display_text, id: p.id } }) }, opts)
+    return sock.sendMessage(jid, { ...content, nativeFlow: undefined, interactiveButtons: resolved, ...(Object.keys(messageParams).length ? { messageParamsJson: JSON.stringify(messageParams) } : {}) }, opts)
+}
+
 export const PRIMITIVE_SHORTCUTS = {
     sendText: sock => (jid, text, opts = {}) => sock.sendMessage(jid, { text }, opts),
     sendImage: sock => (jid, image, caption = '', opts = {}) => sock.sendMessage(jid, { image, caption }, opts),
@@ -126,5 +136,6 @@ export const HANDLER_SHORTCUTS = {
     sendStatusMentionMessage: (sock, config) => (jid, data) => handleStatusMention(sock, jid, norm(data), config),
     sendStatusMentions: (sock, config) => (jid, data) => handleStatusMention(sock, jid, norm(data), config),
     sendPollResultMessage: sock => (jid, data) => handlePollResult(sock, jid, norm(data)),
-    sendGroupStatusMessage: (sock, config) => (jid, content, opts = {}) => handleGroupStatus(sock, jid, norm(content), config)
+    sendGroupStatusMessage: (sock, config) => (jid, content, opts = {}) => handleGroupStatus(sock, jid, norm(content), config),
+    sendNativeFlow: sock => (jid, content, opts = {}) => handleNativeFlow(sock, jid, norm(content), opts)
 }