@nuiisweety/baileys 0.1.3 → 0.1.5

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

@@ -19,6 +19,22 @@ const buildLatexUrl = (expression) => {
     return `https://latex.codecogs.com/png.image?\dpi{150}\bg{white}${encoded}`;
 };
 
+/* ─────────────────────────────────────────────────────────────
+   IMAGE URL HELPER
+   Konversi objek AIRichResponseImageURL ke snake_case JSON
+   agar WA client bisa baca dengan benar~
+───────────────────────────────────────────────────────────── */
+const toImageUrlJson = (imgUrl) => {
+    if (!imgUrl) return null;
+    // support both plain string and object
+    if (typeof imgUrl === 'string') return { image_preview_url: imgUrl, image_high_res_url: imgUrl, source_url: null };
+    return {
+        image_preview_url: imgUrl.imagePreviewUrl || imgUrl.image_preview_url || null,
+        image_high_res_url: imgUrl.imageHighResUrl || imgUrl.image_high_res_url || null,
+        source_url: imgUrl.sourceUrl || imgUrl.source_url || null
+    };
+};
+
 /* ─────────────────────────────────────────────────────────────
    TOKENIZER
 ───────────────────────────────────────────────────────────── */
@@ -259,8 +275,8 @@ const buildUnifiedSection = (submessage) => {
             return {
                 view_model: {
                     primitive: {
-                        grid_image_url: gm.gridImageUrl,
-                        image_urls: gm.imageUrls,
+                        grid_image_url: toImageUrlJson(gm.gridImageUrl),
+                        image_urls: (gm.imageUrls || []).map(toImageUrlJson),
                         __typename: 'GenAIGridImageUXPrimitive'
                     },
                     __typename: 'GenAISingleLayoutViewModel'
@@ -273,7 +289,7 @@ const buildUnifiedSection = (submessage) => {
             return {
                 view_model: {
                     primitive: {
-                        image_url: im.imageUrl,
+                        image_url: toImageUrlJson(im.imageUrl),
                         image_text: im.imageText,
                         alignment: im.alignment ?? 0,
                         tap_link_url: im.tapLinkUrl || null,
@@ -416,16 +432,35 @@ const makeTableSub = (rows, title, noHeading) => ({
     }
 });
 
+/** Normalize input ke format AIRichResponseImageURL */
+const normalizeImageUrl = (u) => {
+    if (!u) return null;
+    if (typeof u === 'string') return { imagePreviewUrl: u, imageHighResUrl: u, sourceUrl: null };
+    return {
+        imagePreviewUrl: u.imagePreviewUrl || u.image_preview_url || null,
+        imageHighResUrl: u.imageHighResUrl || u.image_high_res_url || null,
+        sourceUrl: u.sourceUrl || u.source_url || null
+    };
+};
+
 /** Buat submessage GRID_IMAGE */
 const makeGridImageSub = (gridImageUrl, imageUrls = []) => ({
     messageType: RichSubMessageType.GRID_IMAGE,
-    gridImageMetadata: { gridImageUrl, imageUrls }
+    gridImageMetadata: {
+        gridImageUrl: normalizeImageUrl(gridImageUrl),
+        imageUrls: (imageUrls || []).map(normalizeImageUrl)
+    }
 });
 
 /** Buat submessage INLINE_IMAGE */
 const makeInlineImageSub = (imageUrl, imageText = '', alignment = 0, tapLinkUrl = null) => ({
     messageType: RichSubMessageType.INLINE_IMAGE,
-    imageMetadata: { imageUrl, imageText, alignment, tapLinkUrl }
+    imageMetadata: {
+        imageUrl: normalizeImageUrl(imageUrl),
+        imageText,
+        alignment,
+        tapLinkUrl
+    }
 });
 
 /** Buat submessage DYNAMIC (animated image/GIF) */
@@ -499,6 +534,231 @@ const makeContentItemsSub = (items, contentType = 0) => ({
     }
 });
 
+/** Buat section SOURCE (search result links) — hanya unified, tidak ada submessage proto */
+const makeSourceSection = (sources = []) => {
+    const list = Array.isArray(sources) ? sources : [sources];
+    return {
+        view_model: {
+            primitive: {
+                sources: list.map(s => ({
+                    source_url: s.url ?? s.source_url ?? '',
+                    source_title: s.title ?? s.source_title ?? '',
+                    source_display_name: s.display_name ?? s.displayName ?? s.title ?? '',
+                    source_subtitle: s.subtitle ?? s.source_subtitle ?? '',
+                    source_type: s.source_type ?? 'THIRD_PARTY',
+                    favicon: {
+                        url: s.favicon ?? s.faviconCDNURL ?? '',
+                        width: 16,
+                        height: 16
+                    }
+                })),
+                __typename: 'GenAISearchResultPrimitive'
+            },
+            __typename: 'GenAISingleLayoutViewModel'
+        }
+    };
+};
+
+/** Buat submessage + unified section untuk IMAGE (grid image style) */
+const makeImageSub = (imageUrl) => {
+    const urls = Array.isArray(imageUrl) ? imageUrl : [imageUrl];
+    const imageUrls = urls.map(u => ({
+        imagePreviewUrl: u,
+        imageHighResUrl: u,
+        sourceUrl: null
+    }));
+    return {
+        sub: {
+            messageType: RichSubMessageType.GRID_IMAGE,
+            gridImageMetadata: {
+                gridImageUrl: normalizeImageUrl(urls[0]),
+                imageUrls: imageUrls.map(normalizeImageUrl)
+            }
+        },
+        sections: imageUrls.map(({ imagePreviewUrl }) => ({
+            view_model: {
+                primitive: {
+                    media: { url: imagePreviewUrl, mime_type: 'image/png' },
+                    imagine_type: 'IMAGE',
+                    status: { status: 'READY' },
+                    __typename: 'GenAIImaginePrimitive'
+                },
+                __typename: 'GenAISingleLayoutViewModel'
+            }
+        }))
+    };
+};
+
+/** Buat submessage + unified section untuk VIDEO */
+const makeVideoSub = (videoUrl) => {
+    const urls = Array.isArray(videoUrl) ? videoUrl : [videoUrl];
+    const parsed = urls.map(item => {
+        const [url, duration = '0'] = item.split('|');
+        return { url, duration: Number(duration) || 0 };
+    });
+    return {
+        sub: {
+            messageType: RichSubMessageType.TEXT,
+            messageText: '[ CANNOT_LOAD_VIDEO ]'
+        },
+        sections: parsed.map(({ url, duration }) => ({
+            view_model: {
+                primitive: {
+                    media: { url, mime_type: 'video/mp4', duration },
+                    imagine_type: 'ANIMATE',
+                    status: { status: 'READY' },
+                    __typename: 'GenAIImaginePrimitive'
+                },
+                __typename: 'GenAISingleLayoutViewModel'
+            }
+        }))
+    };
+};
+
+/** Buat submessage + unified section untuk REELS (HScroll) */
+const makeReelsSub = (reelsItems = []) => {
+    const list = Array.isArray(reelsItems) ? reelsItems : [reelsItems];
+    return {
+        sub: {
+            messageType: RichSubMessageType.CONTENT_ITEMS,
+            contentItemsMetadata: {
+                contentType: 1,
+                itemsMetadata: list.map(item => ({
+                    reelItem: {
+                        title: item.username ?? item.title ?? '',
+                        profileIconUrl: item.profileIconUrl ?? item.profile_url ?? null,
+                        thumbnailUrl: item.thumbnailUrl ?? item.thumbnail ?? null,
+                        videoUrl: item.videoUrl ?? item.url ?? null
+                    }
+                }))
+            }
+        },
+        section: {
+            view_model: {
+                primitives: list.map(item => ({
+                    reels_url: item.videoUrl ?? item.url ?? '',
+                    thumbnail_url: item.thumbnailUrl ?? item.thumbnail ?? '',
+                    creator: item.username ?? item.title ?? '',
+                    avatar_url: item.profileIconUrl ?? item.profile_url ?? '',
+                    reels_title: item.reels_title ?? item.title ?? '',
+                    likes_count: item.likes_count ?? item.like ?? 0,
+                    shares_count: item.shares_count ?? item.share ?? 0,
+                    view_count: item.view_count ?? item.view ?? 0,
+                    reel_source: item.reel_source ?? item.source ?? 'IG',
+                    is_verified: !!(item.is_verified || item.verified),
+                    __typename: 'GenAIReelPrimitive'
+                })),
+                __typename: 'GenAIHScrollLayoutViewModel'
+            }
+        }
+    };
+};
+
+/** Buat unified section untuk PRODUCT (single: Single layout, array: HScroll) */
+const makeProductSection = (data) => {
+    const items = Array.isArray(data) ? data : [data];
+    const primitives = items.map(item => ({
+        title: item.title ?? '',
+        brand: item.brand ?? '',
+        price: item.price ?? '',
+        sale_price: item.sale_price ?? item.salePrice ?? '',
+        product_url: item.product_url ?? item.url ?? '',
+        image: { url: item.image_url ?? item.image ?? '' },
+        additional_images: [{ url: item.icon_url ?? item.icon ?? '' }],
+        __typename: 'GenAIProductItemCardPrimitive'
+    }));
+    const isMultiple = Array.isArray(data);
+    return {
+        sub: {
+            messageType: RichSubMessageType.TEXT,
+            messageText: '[ CANNOT_LOAD_PRODUCT ]'
+        },
+        section: isMultiple
+            ? {
+                view_model: {
+                    primitives,
+                    __typename: 'GenAIHScrollLayoutViewModel'
+                }
+            }
+            : {
+                view_model: {
+                    primitive: primitives[0],
+                    __typename: 'GenAISingleLayoutViewModel'
+                }
+            }
+    };
+};
+
+/** Buat unified section untuk POST (HScroll) */
+const makePostSection = (data) => {
+    const posts = Array.isArray(data) ? data : [data];
+    const primitives = posts.map(p => ({
+        title: p.title ?? '',
+        subtitle: p.subtitle ?? '',
+        username: p.username ?? '',
+        profile_picture_url: p.profile_picture_url ?? p.profile_url ?? '',
+        is_verified: !!(p.is_verified || p.verified),
+        thumbnail_url: p.thumbnail_url ?? p.thumbnail ?? '',
+        post_caption: p.post_caption ?? p.caption ?? '',
+        likes_count: p.likes_count ?? p.like ?? 0,
+        comments_count: p.comments_count ?? p.comment ?? 0,
+        shares_count: p.shares_count ?? p.share ?? 0,
+        post_url: p.post_url ?? p.url ?? '',
+        post_deeplink: p.post_deeplink ?? p.deeplink ?? '',
+        source_app: p.source_app ?? p.source ?? 'INSTAGRAM',
+        footer_label: p.footer_label ?? p.footer ?? '',
+        footer_icon: p.footer_icon ?? p.icon ?? '',
+        is_carousel: posts.length > 1,
+        orientation: p.orientation ?? 'LANDSCAPE',
+        post_type: p.post_type ?? 'VIDEO',
+        __typename: 'GenAIPostPrimitive'
+    }));
+    return {
+        sub: {
+            messageType: RichSubMessageType.TEXT,
+            messageText: '[ CANNOT_LOAD_POST ]'
+        },
+        section: {
+            view_model: {
+                primitives,
+                __typename: 'GenAIHScrollLayoutViewModel'
+            }
+        }
+    };
+};
+
+/** Buat submessage + unified section untuk TIP (metadata text) */
+const makeTipSub = (text) => ({
+    sub: {
+        messageType: RichSubMessageType.TEXT,
+        messageText: text
+    },
+    section: {
+        view_model: {
+            primitive: {
+                text,
+                __typename: 'GenAIMetadataTextPrimitive'
+            },
+            __typename: 'GenAISingleLayoutViewModel'
+        }
+    }
+});
+
+/** Buat unified section untuk SUGGEST (ActionRow pill buttons) — tidak ada submessage proto */
+const makeSuggestSection = (suggestion) => {
+    const suggestions = Array.isArray(suggestion) ? suggestion : [suggestion];
+    return {
+        view_model: {
+            primitives: suggestions.map(text => ({
+                prompt_text: text,
+                prompt_type: 'SUGGESTED_PROMPT',
+                __typename: 'GenAIFollowUpSuggestionPillPrimitive'
+            })),
+            __typename: 'GenAIActionRowLayoutViewModel'
+        }
+    };
+};
+
 /* ─────────────────────────────────────────────────────────────
    MAIN BUILDER — prepareRichResponseMessage
 ───────────────────────────────────────────────────────────── */
@@ -507,29 +767,54 @@ export const prepareRichResponseMessage = (content) => {
     const {
         code, contentText, disclaimerText, footerText, headerText,
         language, links, noHeading, richResponse, table, title,
-        // sub-types baru
+        // sub-types lama
         gridImage, inlineImage, dynamic: dynamicContent,
-        map: mapContent, latex, contentItems
+        map: mapContent, latex, contentItems,
+        // sub-types baru
+        image, video, reels, source, product, post, tip, suggest
     } = content;
 
     let submessages = [];
+    // extraSections: section-section unified yang tidak punya proto submessage 1:1
+    // (source, product, post, suggest) atau punya struktur multi-section (image, video, reels)
+    let extraSections = null;
+
+    /**
+     * Helper: push ke submessages dan optionally ke extraSections.
+     * Untuk tipe yang punya make*Sub yang mengembalikan { sub, sections[] } atau { sub, section }
+     */
+    const pushRich = (built) => {
+        if (!built) return;
+        if (built.sub) submessages.push(built.sub);
+        if (!extraSections) extraSections = [];
+        if (built.sections) extraSections.push(...built.sections);
+        else if (built.section) extraSections.push(built.section);
+    };
 
     /* ── mode array (richResponse) — multi-section campuran ── */
     if (Array.isArray(richResponse)) {
-        // headerText dan footerText tetap dihormati meskipun pakai mode array
         if (headerText) submessages.push(makeTextSub(headerText));
-        submessages.push(...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));
+        richResponse.forEach(sub => {
+            if (sub.text != null)         { submessages.push(makeTextSub(sub.text, sub.inlineEntities)); return; }
+            if (sub.code != null)         { submessages.push(makeCodeSub(sub.code, sub.language || 'javascript')); return; }
+            if (sub.table != null)        { submessages.push(makeTableSub(sub.table, sub.title, sub.noHeading)); return; }
+            if (sub.gridImage != null)    { submessages.push(makeGridImageSub(sub.gridImage.gridImageUrl, sub.gridImage.imageUrls)); return; }
+            if (sub.inlineImage != null)  { submessages.push(makeInlineImageSub(sub.inlineImage.imageUrl, sub.inlineImage.imageText, sub.inlineImage.alignment, sub.inlineImage.tapLinkUrl)); return; }
+            if (sub.dynamic != null)      { submessages.push(makeDynamicSub(sub.dynamic.url, sub.dynamic.type, sub.dynamic.version, sub.dynamic.loopCount)); return; }
+            if (sub.map != null)          { submessages.push(makeMapSub(sub.map.centerLatitude, sub.map.centerLongitude, sub.map)); return; }
+            if (sub.latex != null)        { submessages.push(makeLatexSub(sub.latex.text, sub.latex.expressions)); return; }
+            if (sub.contentItems != null) { submessages.push(makeContentItemsSub(sub.contentItems.items, sub.contentItems.contentType)); return; }
+            // tipe baru
+            if (sub.image != null)    { pushRich(makeImageSub(sub.image)); return; }
+            if (sub.video != null)    { pushRich(makeVideoSub(sub.video)); return; }
+            if (sub.reels != null)    { pushRich(makeReelsSub(sub.reels)); return; }
+            if (sub.source != null)   { if (!extraSections) extraSections = []; extraSections.push(makeSourceSection(sub.source)); return; }
+            if (sub.product != null)  { pushRich(makeProductSection(sub.product)); return; }
+            if (sub.post != null)     { pushRich(makePostSection(sub.post)); return; }
+            if (sub.tip != null)      { pushRich(makeTipSub(sub.tip)); return; }
+            if (sub.suggest != null)  { if (!extraSections) extraSections = []; extraSections.push(makeSuggestSection(sub.suggest)); return; }
+            submessages.push(sub); // passthrough kalau sudah bentuk proto
+        });
         if (footerText) submessages.push(makeTextSub(footerText));
 
     /* ── mode flat (convenience fields) ── */
@@ -537,37 +822,24 @@ export const prepareRichResponseMessage = (content) => {
         if (headerText)    submessages.push(makeTextSub(headerText));
         if (contentText)   submessages.push(makeTextSub(contentText));
 
-        if (code) {
-            submessages.push(makeCodeSub(code, language || 'javascript'));
-        }
-
-        if (table) {
-            submessages.push(makeTableSub(table, title, noHeading));
-        }
-
-        if (gridImage) {
-            submessages.push(makeGridImageSub(gridImage.gridImageUrl, gridImage.imageUrls));
-        }
-
-        if (inlineImage) {
-            submessages.push(makeInlineImageSub(inlineImage.imageUrl, inlineImage.imageText, inlineImage.alignment, inlineImage.tapLinkUrl));
-        }
-
-        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));
-        }
+        if (code)          submessages.push(makeCodeSub(code, language || 'javascript'));
+        if (table)         submessages.push(makeTableSub(table, title, noHeading));
+        if (gridImage)     submessages.push(makeGridImageSub(gridImage.gridImageUrl, gridImage.imageUrls));
+        if (inlineImage)   submessages.push(makeInlineImageSub(inlineImage.imageUrl, inlineImage.imageText, inlineImage.alignment, inlineImage.tapLinkUrl));
+        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));
+
+        // ── tipe baru ──
+        if (image)    pushRich(makeImageSub(image));
+        if (video)    pushRich(makeVideoSub(video));
+        if (reels)    pushRich(makeReelsSub(reels));
+        if (source)   { if (!extraSections) extraSections = []; extraSections.push(makeSourceSection(source)); }
+        if (product)  pushRich(makeProductSection(product));
+        if (post)     pushRich(makePostSection(post));
+        if (tip)      pushRich(makeTipSub(tip));
+        if (suggest)  { if (!extraSections) extraSections = []; extraSections.push(makeSuggestSection(suggest)); }
 
         /* links — bisa dikombinasi dengan tipe lain di atas */
         if (links && Array.isArray(links)) {
@@ -597,11 +869,14 @@ export const prepareRichResponseMessage = (content) => {
             });
         }
 
-        if (footerText)    submessages.push(makeTextSub(footerText));
+        if (footerText) submessages.push(makeTextSub(footerText));
     }
 
-    /* build unifiedResponse JSON */
-    const unified = toUnified(submessages);
+    /* build unifiedResponse JSON — gabungkan sections dari submessages + extraSections */
+    const baseUnified = toUnified(submessages);
+    const unified = extraSections && extraSections.length > 0
+        ? { ...baseUnified, sections: [...baseUnified.sections, ...extraSections] }
+        : baseUnified;
 
     const richResponseMessage = proto.AIRichResponseMessage.create({
         submessages,