@nuiisweety/baileys 0.1.0 → 0.1.2

package/lib/Utils/rich-message-utils.js

@@ -4,7 +4,13 @@ import { LANGUAGE_KEYWORDS } from '../WABinary/constants.js';
 import { CodeHighlightType, RichSubMessageType } from '../Types/RichType.js';
 import { proto } from '../../WAProto/index.js';
 import { unixTimestampSeconds } from './generics.js';
+
 const NOOP = new Set([]);
+
+/* ─────────────────────────────────────────────────────────────
+   TOKENIZER
+───────────────────────────────────────────────────────────── */
+
 export const tokenizeCode = (code, language = 'javascript') => {
     const keywords = LANGUAGE_KEYWORDS[language] || NOOP;
     const blocks = [];
@@ -13,181 +19,570 @@ export const tokenizeCode = (code, language = 'javascript') => {
     while ((match = LEXER_REGEX.exec(code)) !== null) {
         if (match[1]) {
             blocks.push({ highlightType: CodeHighlightType.COMMENT, codeContent: match[1] });
-        }
-        else if (match[2]) {
+        } else if (match[2]) {
             blocks.push({ highlightType: CodeHighlightType.STRING, codeContent: match[2] });
-        }
-        else if (match[3]) {
+        } else if (match[3]) {
             blocks.push({
                 highlightType: keywords.has(match[3]) ? CodeHighlightType.KEYWORD : CodeHighlightType.METHOD,
                 codeContent: match[3],
             });
-        }
-        else if (match[4]) {
+        } else if (match[4]) {
             blocks.push({
                 highlightType: keywords.has(match[4]) ? CodeHighlightType.KEYWORD : CodeHighlightType.DEFAULT,
                 codeContent: match[4],
             });
-        }
-        else if (match[5]) {
+        } else if (match[5]) {
             blocks.push({ highlightType: CodeHighlightType.NUMBER, codeContent: match[5] });
-        }
-        else {
+        } else {
             blocks.push({ highlightType: CodeHighlightType.DEFAULT, codeContent: match[6] });
         }
     }
     return blocks;
 };
+
+/* ─────────────────────────────────────────────────────────────
+   INCOMING DECODER — parse rich message yang diterima
+───────────────────────────────────────────────────────────── */
+
+/**
+ * Parse sebuah AIRichResponseSubMessage proto menjadi object JS yang mudah dipakai.
+ * Return null kalau tipe tidak dikenal.
+ */
+export const parseRichSubMessage = (submessage) => {
+    if (!submessage) return null;
+    const type = submessage.messageType;
+
+    switch (type) {
+        case RichSubMessageType.TEXT:
+            return {
+                type: 'text',
+                text: submessage.messageText || '',
+            };
+
+        case RichSubMessageType.CODE:
+            return {
+                type: 'code',
+                language: submessage.codeMetadata?.codeLanguage || 'plain',
+                blocks: (submessage.codeMetadata?.codeBlocks || []).map(b => ({
+                    highlight: CodeHighlightType[b.highlightType] ?? 'DEFAULT',
+                    content: b.codeContent || '',
+                })),
+                /** Helper: ambil source code mentah tanpa highlight info */
+                get raw() {
+                    return this.blocks.map(b => b.content).join('');
+                },
+            };
+
+        case RichSubMessageType.TABLE:
+            return {
+                type: 'table',
+                title: submessage.tableMetadata?.title || '',
+                rows: (submessage.tableMetadata?.rows || []).map(row => ({
+                    isHeading: row.isHeading ?? false,
+                    items: row.items || [],
+                })),
+            };
+
+        case RichSubMessageType.GRID_IMAGE:
+            return {
+                type: 'gridImage',
+                gridImageUrl: submessage.gridImageMetadata?.gridImageUrl || null,
+                imageUrls: submessage.gridImageMetadata?.imageUrls || [],
+            };
+
+        case RichSubMessageType.INLINE_IMAGE:
+            return {
+                type: 'inlineImage',
+                imageUrl: submessage.imageMetadata?.imageUrl || null,
+                imageText: submessage.imageMetadata?.imageText || '',
+                alignment: submessage.imageMetadata?.alignment ?? 0,
+                tapLinkUrl: submessage.imageMetadata?.tapLinkUrl || null,
+            };
+
+        case RichSubMessageType.DYNAMIC:
+            return {
+                type: 'dynamic',
+                dynamicType: submessage.dynamicMetadata?.type ?? 0,
+                version: submessage.dynamicMetadata?.version ?? 0,
+                url: submessage.dynamicMetadata?.url || null,
+                loopCount: submessage.dynamicMetadata?.loopCount ?? 0,
+            };
+
+        case RichSubMessageType.MAP:
+            return {
+                type: 'map',
+                centerLatitude: submessage.mapMetadata?.centerLatitude ?? 0,
+                centerLongitude: submessage.mapMetadata?.centerLongitude ?? 0,
+                latitudeDelta: submessage.mapMetadata?.latitudeDelta ?? 0,
+                longitudeDelta: submessage.mapMetadata?.longitudeDelta ?? 0,
+                showInfoList: submessage.mapMetadata?.showInfoList ?? false,
+                annotations: (submessage.mapMetadata?.annotations || []).map(a => ({
+                    number: a.annotationNumber ?? 0,
+                    latitude: a.latitude ?? 0,
+                    longitude: a.longitude ?? 0,
+                    title: a.title || '',
+                    body: a.body || '',
+                })),
+            };
+
+        case RichSubMessageType.LATEX:
+            return {
+                type: 'latex',
+                text: submessage.latexMetadata?.text || '',
+                expressions: (submessage.latexMetadata?.expressions || []).map(e => ({
+                    expression: e.latexExpression || '',
+                    url: e.url || null,
+                    width: e.width ?? 0,
+                    height: e.height ?? 0,
+                })),
+            };
+
+        case RichSubMessageType.CONTENT_ITEMS:
+            return {
+                type: 'contentItems',
+                contentType: submessage.contentItemsMetadata?.contentType ?? 0,
+                items: (submessage.contentItemsMetadata?.itemsMetadata || []).map(item => {
+                    if (item.reelItem) {
+                        return {
+                            kind: 'reel',
+                            title: item.reelItem.title || '',
+                            profileIconUrl: item.reelItem.profileIconUrl || null,
+                            thumbnailUrl: item.reelItem.thumbnailUrl || null,
+                            videoUrl: item.reelItem.videoUrl || null,
+                        };
+                    }
+                    return { kind: 'unknown' };
+                }),
+            };
+
+        default:
+            return { type: 'unknown', raw: submessage };
+    }
+};
+
+/**
+ * Parse seluruh AIRichResponseMessage menjadi array parsed submessages.
+ * Bisa dipanggil langsung dari handler messages.upsert:
+ *
+ *   const inner = normalizeMessageContent(msg.message)
+ *   const parsed = parseRichMessage(inner?.richResponseMessage)
+ */
+export const parseRichMessage = (richResponseMessage) => {
+    if (!richResponseMessage) return null;
+
+    const submessages = (richResponseMessage.submessages || [])
+        .map(parseRichSubMessage)
+        .filter(Boolean);
+
+    let unifiedData = null;
+    if (richResponseMessage.unifiedResponse?.data) {
+        try {
+            const buf = richResponseMessage.unifiedResponse.data;
+            unifiedData = JSON.parse(Buffer.isBuffer(buf) ? buf.toString('utf-8') : Buffer.from(buf).toString('utf-8'));
+        } catch (_) { /* ignore parse error */ }
+    }
+
+    return {
+        messageType: richResponseMessage.messageType ?? 0,
+        submessages,
+        unifiedData,
+    };
+};
+
+/* ─────────────────────────────────────────────────────────────
+   OUTGOING BUILDER — unified response JSON per subtype
+───────────────────────────────────────────────────────────── */
+
+const buildUnifiedSection = (submessage) => {
+    switch (submessage.messageType) {
+        case RichSubMessageType.TEXT:
+            return {
+                view_model: {
+                    primitive: {
+                        text: submessage.messageText,
+                        inline_entities: submessage.inlineEntities || [],
+                        __typename: 'GenAIMarkdownTextUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+
+        case RichSubMessageType.CODE: {
+            const cm = submessage.codeMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        language: cm.codeLanguage,
+                        code_blocks: cm.codeBlocks.map(b => ({
+                            content: b.codeContent,
+                            type: CodeHighlightType[b.highlightType]
+                        })),
+                        __typename: 'GenAICodeUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.TABLE: {
+            const tm = submessage.tableMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        title: tm.title,
+                        rows: tm.rows.map(row => ({
+                            is_header: row.isHeading,
+                            cells: row.items,
+                            markdown_cells: row.items.map(item => ({ text: item }))
+                        })),
+                        __typename: 'GenATableUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.GRID_IMAGE: {
+            const gm = submessage.gridImageMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        grid_image_url: gm.gridImageUrl,
+                        image_urls: gm.imageUrls,
+                        __typename: 'GenAIGridImageUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.INLINE_IMAGE: {
+            const im = submessage.imageMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        image_url: im.imageUrl,
+                        image_text: im.imageText,
+                        alignment: im.alignment ?? 0,
+                        tap_link_url: im.tapLinkUrl || null,
+                        __typename: 'GenAIInlineImageUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.DYNAMIC: {
+            const dm = submessage.dynamicMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        type: dm.type ?? 0,
+                        version: dm.version ?? 0,
+                        url: dm.url,
+                        loop_count: dm.loopCount ?? 0,
+                        __typename: 'GenAIDynamicUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.MAP: {
+            const mm = submessage.mapMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        center_latitude: mm.centerLatitude,
+                        center_longitude: mm.centerLongitude,
+                        latitude_delta: mm.latitudeDelta,
+                        longitude_delta: mm.longitudeDelta,
+                        show_info_list: mm.showInfoList ?? false,
+                        annotations: (mm.annotations || []).map(a => ({
+                            annotation_number: a.annotationNumber,
+                            latitude: a.latitude,
+                            longitude: a.longitude,
+                            title: a.title,
+                            body: a.body
+                        })),
+                        __typename: 'GenAIMapUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.LATEX: {
+            const lm = submessage.latexMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        text: lm.text,
+                        expressions: (lm.expressions || []).map(e => ({
+                            latex_expression: e.latexExpression,
+                            url: e.url,
+                            width: e.width,
+                            height: e.height,
+                            font_height: e.fontHeight,
+                            image_top_padding: e.imageTopPadding,
+                            image_leading_padding: e.imageLeadingPadding,
+                            image_bottom_padding: e.imageBottomPadding,
+                            image_trailing_padding: e.imageTrailingPadding
+                        })),
+                        __typename: 'GenAILatexUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        case RichSubMessageType.CONTENT_ITEMS: {
+            const ci = submessage.contentItemsMetadata;
+            return {
+                view_model: {
+                    primitive: {
+                        content_type: ci.contentType ?? 0,
+                        items_metadata: (ci.itemsMetadata || []).map(item => {
+                            if (item.reelItem) {
+                                return {
+                                    reel_item: {
+                                        title: item.reelItem.title,
+                                        profile_icon_url: item.reelItem.profileIconUrl,
+                                        thumbnail_url: item.reelItem.thumbnailUrl,
+                                        video_url: item.reelItem.videoUrl
+                                    }
+                                };
+                            }
+                            return {};
+                        }),
+                        __typename: 'GenAIContentItemsUXPrimitive'
+                    },
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            };
+        }
+
+        default:
+            return submessage;
+    }
+};
+
 export const toUnified = (submessages) => ({
     response_id: randomUUID(),
-    sections: submessages.map((submessage, index) => {
-        switch (submessage.messageType) {
-            case RichSubMessageType.CODE:
-                const codeMetadata = submessage.codeMetadata;
-                return {
-                    view_model: {
-                        primitive: {
-                            language: codeMetadata.codeLanguage,
-                            code_blocks: codeMetadata.codeBlocks.map((block) => ({ content: block.codeContent, type: CodeHighlightType[block.highlightType] })),
-                            __typename: 'GenAICodeUXPrimitive'
-                        },
-                        __typename: 'GenAISingleLayoutViewModel'
-                    }
-                };
-            case RichSubMessageType.TABLE:
-                const tableMetadata = submessage.tableMetadata;
-                return {
-                    view_model: {
-                        primitive: {
-                            title: tableMetadata.title,
-                            rows: tableMetadata.rows.map((row) => ({
-                                is_header: row.isHeading,
-                                cells: row.items,
-                                markdown_cells: row.items.map((item) => ({ text: item }))
-                            })),
-                            __typename: 'GenATableUXPrimitive'
-                        },
-                        __typename: 'GenAISingleLayoutViewModel'
-                    }
-                };
-            case RichSubMessageType.TEXT:
+    sections: submessages.map(buildUnifiedSection)
+});
+
+/* ─────────────────────────────────────────────────────────────
+   PREPARE HELPERS — builder per tipe submessage
+───────────────────────────────────────────────────────────── */
+
+/** Buat submessage TEXT */
+const makeTextSub = (text, inlineEntities) => ({
+    messageType: RichSubMessageType.TEXT,
+    messageText: text,
+    ...(inlineEntities ? { inlineEntities } : {})
+});
+
+/** Buat submessage CODE dengan auto-tokenize */
+const makeCodeSub = (code, language = 'javascript') => ({
+    messageType: RichSubMessageType.CODE,
+    codeMetadata: {
+        codeLanguage: language,
+        codeBlocks: tokenizeCode(code, language)
+    }
+});
+
+/** Buat submessage TABLE */
+const makeTableSub = (rows, title, noHeading) => ({
+    messageType: RichSubMessageType.TABLE,
+    tableMetadata: {
+        title: title || '',
+        rows: rows.map((items, i) => ({
+            isHeading: !noHeading && i === 0,
+            items
+        }))
+    }
+});
+
+/** Buat submessage GRID_IMAGE */
+const makeGridImageSub = (gridImageUrl, imageUrls = []) => ({
+    messageType: RichSubMessageType.GRID_IMAGE,
+    gridImageMetadata: { gridImageUrl, imageUrls }
+});
+
+/** Buat submessage INLINE_IMAGE */
+const makeInlineImageSub = (imageUrl, imageText = '', alignment = 0, tapLinkUrl = null) => ({
+    messageType: RichSubMessageType.INLINE_IMAGE,
+    imageMetadata: { imageUrl, imageText, alignment, tapLinkUrl }
+});
+
+/** Buat submessage DYNAMIC (animated image/GIF) */
+const makeDynamicSub = (url, dynamicType = 1, version = 1, loopCount = 0) => ({
+    messageType: RichSubMessageType.DYNAMIC,
+    dynamicMetadata: { type: dynamicType, version, url, loopCount }
+});
+
+/** Buat submessage MAP */
+const makeMapSub = (centerLatitude, centerLongitude, options = {}) => ({
+    messageType: RichSubMessageType.MAP,
+    mapMetadata: {
+        centerLatitude,
+        centerLongitude,
+        latitudeDelta: options.latitudeDelta ?? 0.05,
+        longitudeDelta: options.longitudeDelta ?? 0.05,
+        showInfoList: options.showInfoList ?? false,
+        annotations: (options.annotations || []).map((a, i) => ({
+            annotationNumber: a.number ?? i + 1,
+            latitude: a.latitude,
+            longitude: a.longitude,
+            title: a.title || '',
+            body: a.body || ''
+        }))
+    }
+});
+
+/** Buat submessage LATEX */
+const makeLatexSub = (text, expressions = []) => ({
+    messageType: RichSubMessageType.LATEX,
+    latexMetadata: {
+        text,
+        expressions: expressions.map(e => ({
+            latexExpression: e.expression || e.latexExpression || '',
+            url: e.url || null,
+            width: e.width ?? 0,
+            height: e.height ?? 0,
+            fontHeight: e.fontHeight ?? 0,
+            imageTopPadding: e.imageTopPadding ?? 0,
+            imageLeadingPadding: e.imageLeadingPadding ?? 0,
+            imageBottomPadding: e.imageBottomPadding ?? 0,
+            imageTrailingPadding: e.imageTrailingPadding ?? 0
+        }))
+    }
+});
+
+/** Buat submessage CONTENT_ITEMS (carousel reel) */
+const makeContentItemsSub = (items, contentType = 0) => ({
+    messageType: RichSubMessageType.CONTENT_ITEMS,
+    contentItemsMetadata: {
+        contentType,
+        itemsMetadata: items.map(item => {
+            if (item.reelItem || item.kind === 'reel') {
+                const r = item.reelItem || item;
                 return {
-                    view_model: {
-                        primitive: {
-                            text: submessage.messageText,
-                            inline_entities: submessage.inlineEntities || [],
-                            __typename: 'GenAIMarkdownTextUXPrimitive'
-                        },
-                        __typename: 'GenAISingleLayoutViewModel'
+                    reelItem: {
+                        title: r.title || '',
+                        profileIconUrl: r.profileIconUrl || null,
+                        thumbnailUrl: r.thumbnailUrl || null,
+                        videoUrl: r.videoUrl || null
                     }
                 };
-        }
-        return submessage;
-    })
+            }
+            return {};
+        })
+    }
 });
+
+/* ─────────────────────────────────────────────────────────────
+   MAIN BUILDER — prepareRichResponseMessage
+───────────────────────────────────────────────────────────── */
+
 export const prepareRichResponseMessage = (content) => {
-    const { code, contentText, disclaimerText, footerText, headerText, language, links, noHeading, richResponse, table, title } = content;
+    const {
+        code, contentText, disclaimerText, footerText, headerText,
+        language, links, noHeading, richResponse, table, title,
+        // sub-types baru
+        gridImage, inlineImage, dynamic: dynamicContent,
+        map: mapContent, latex, contentItems
+    } = content;
+
     let submessages = [];
+
+    /* ── mode array (richResponse) — multi-section campuran ── */
     if (Array.isArray(richResponse)) {
-        submessages = richResponse.map((submessage) => {
-            if (submessage.text) {
-                return {
-                    messageType: RichSubMessageType.TEXT,
-                    messageText: submessage.text,
-                    inlineEntities: submessage.inlineEntities
-                };
-            }
-            else if (submessage.code) {
-                return {
-                    messageType: RichSubMessageType.CODE,
-                    codeMetadata: {
-                        codeLanguage: submessage.language,
-                        codeBlocks: submessage.code
-                    }
-                };
-            }
-            else if (submessage.table) {
-                return {
-                    messageType: RichSubMessageType.TABLE,
-                    tableMetadata: {
-                        title: submessage.title,
-                        rows: submessage.table
-                    }
-                };
-            }
-            return submessage;
-        });
-    }
-    else {
-        if (headerText) {
-            submessages.push({
-                messageType: RichSubMessageType.TEXT,
-                messageText: headerText
-            });
+        submessages = richResponse.map(sub => {
+            if (sub.text != null)       return makeTextSub(sub.text, sub.inlineEntities);
+            if (sub.code != null)       return makeCodeSub(sub.code, sub.language || 'javascript');
+            if (sub.table != null)      return makeTableSub(sub.table, sub.title, sub.noHeading);
+            if (sub.gridImage != null)  return makeGridImageSub(sub.gridImage.gridImageUrl, sub.gridImage.imageUrls);
+            if (sub.inlineImage != null) return makeInlineImageSub(sub.inlineImage.imageUrl, sub.inlineImage.imageText, sub.inlineImage.alignment, sub.inlineImage.tapLinkUrl);
+            if (sub.dynamic != null)    return makeDynamicSub(sub.dynamic.url, sub.dynamic.type, sub.dynamic.version, sub.dynamic.loopCount);
+            if (sub.map != null)        return makeMapSub(sub.map.centerLatitude, sub.map.centerLongitude, sub.map);
+            if (sub.latex != null)      return makeLatexSub(sub.latex.text, sub.latex.expressions);
+            if (sub.contentItems != null) return makeContentItemsSub(sub.contentItems.items, sub.contentItems.contentType);
+            return sub; // passthrough kalau sudah bentuk proto
+        }).filter(Boolean);
+
+    /* ── mode flat (convenience fields) ── */
+    } else {
+        if (headerText)    submessages.push(makeTextSub(headerText));
+        if (contentText)   submessages.push(makeTextSub(contentText));
+
+        if (code) {
+            submessages.push(makeCodeSub(code, language || 'javascript'));
         }
-        if (contentText) {
-            submessages.push({
-                messageType: RichSubMessageType.TEXT,
-                messageText: contentText
-            });
+
+        if (table) {
+            submessages.push(makeTableSub(table, title, noHeading));
         }
-        if (code) {
-            language ||= 'javascript';
-            submessages.push({
-                messageType: RichSubMessageType.CODE,
-                codeMetadata: {
-                    codeLanguage: language,
-                    codeBlocks: tokenizeCode(code, language)
-                }
-            });
+
+        if (gridImage) {
+            submessages.push(makeGridImageSub(gridImage.gridImageUrl, gridImage.imageUrls));
+        }
+
+        if (inlineImage) {
+            submessages.push(makeInlineImageSub(inlineImage.imageUrl, inlineImage.imageText, inlineImage.alignment, inlineImage.tapLinkUrl));
         }
-        else if (links) {
+
+        if (dynamicContent) {
+            submessages.push(makeDynamicSub(dynamicContent.url, dynamicContent.type, dynamicContent.version, dynamicContent.loopCount));
+        }
+
+        if (mapContent) {
+            submessages.push(makeMapSub(mapContent.centerLatitude, mapContent.centerLongitude, mapContent));
+        }
+
+        if (latex) {
+            submessages.push(makeLatexSub(latex.text, latex.expressions));
+        }
+
+        if (contentItems) {
+            submessages.push(makeContentItemsSub(contentItems.items, contentItems.contentType));
+        }
+
+        /* links — bisa dikombinasi dengan tipe lain di atas */
+        if (links && Array.isArray(links)) {
             links.forEach((linkField, index) => {
                 const prefix = 'SS_' + index;
                 const url = linkField.url || DONATE_URL;
-                const sources = linkField.sources?.map((sourceField) => ({
+                const sources = (linkField.sources || []).map(s => ({
                     source_type: 'THIRD_PARTY',
-                    source_display_name: sourceField.displayName || 'Donate',
-                    source_subtitle: sourceField.subtitle || 'Saweria',
-                    source_url: sourceField.url || url
+                    source_display_name: s.displayName || 'Source',
+                    source_subtitle: s.subtitle || '',
+                    source_url: s.url || url
                 }));
-                submessages.push({
-                    messageType: RichSubMessageType.TEXT,
-                    messageText: linkField.text + ` {{${prefix}}}¹{{/${prefix}}} `,
-                    inlineEntities: [{
-                            key: prefix,
-                            metadata: {
-                                reference_id: index + 1,
-                                reference_url: url,
-                                reference_title: linkField.title || 'For Donation via Saweria',
-                                reference_display_name: linkField.displayName || 'Donation',
-                                sources: sources || [],
-                                __typename: 'GenAISearchCitationItem'
-                            }
-                        }]
-                });
-            });
-        }
-        else if (table) {
-            submessages.push({
-                messageType: RichSubMessageType.TABLE,
-                tableMetadata: {
-                    title,
-                    rows: table.map((items, index) => ({
-                        isHeading: !noHeading && index == 0,
-                        items
-                    }))
-                }
-            });
-        }
-        if (footerText) {
-            submessages.push({
-                messageType: RichSubMessageType.TEXT,
-                messageText: footerText
+                submessages.push(makeTextSub(
+                    linkField.text + ` {{${prefix}}}¹{{/${prefix}}} `,
+                    [{
+                        key: prefix,
+                        metadata: {
+                            reference_id: index + 1,
+                            reference_url: url,
+                            reference_title: linkField.title || url,
+                            reference_display_name: linkField.displayName || url,
+                            sources,
+                            __typename: 'GenAISearchCitationItem'
+                        }
+                    }]
+                ));
             });
         }
+
+        if (footerText)    submessages.push(makeTextSub(footerText));
     }
+
+    /* build unifiedResponse JSON */
     const unified = toUnified(submessages);
+
     const richResponseMessage = proto.AIRichResponseMessage.create({
         submessages,
         messageType: proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD,
@@ -201,19 +596,28 @@ export const prepareRichResponseMessage = (content) => {
             forwardOrigin: 4
         }
     });
+
     const message = wrapToBotForwardedMessage(richResponseMessage);
     const botMetadata = message.messageContextInfo.botMetadata;
+
     if (disclaimerText) {
         botMetadata.messageDisclaimerText = disclaimerText;
     }
     botMetadata.botResponseId = unified.response_id;
+
     return message;
 };
+
+/* ─────────────────────────────────────────────────────────────
+   BOT WRAPPER HELPERS
+───────────────────────────────────────────────────────────── */
+
 export const botMetadataSignature = () => {
     const signature = new Uint8Array(64);
     getRandomValues(signature);
     return signature;
 };
+
 export const botMetadataCertificate = (length = 685) => {
     const certificate = new Uint8Array(length);
     certificate[0] = 48;
@@ -221,6 +625,7 @@ export const botMetadataCertificate = (length = 685) => {
     getRandomValues(certificate.subarray(2));
     return certificate;
 };
+
 export const wrapToBotForwardedMessage = (richResponseMessage) => ({
     messageContextInfo: {
         botMetadata: {
@@ -243,4 +648,4 @@ export const wrapToBotForwardedMessage = (richResponseMessage) => ({
         message: { richResponseMessage }
     }
 });
-//# sourceMappingURL=rich-message-utils.js.map
+//# sourceMappingURL=rich-message-utils.js.map