@kyyinfinite/lumina 1.0.5 → 1.0.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kyyinfinite/lumina",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "description": "Lumina — Modern WhatsApp framework built on top of Baileys. Hide the protocol complexity, expose a fluent API.",
   "type": "module",
   "main": "./src/index.js",

package/src/builders/ai-rich.js

@@ -1,61 +1,43 @@
-
 import { applyContentFields } from './base.js'
 import { ensureObjectOrArray, ensureString, ensureStringArray } from '../utils/validator.js'
 import { ValidationError } from '../errors.js'
-
 import { extractInlineEntities } from '../parsers/inline-entity.js'
 import { tokenizeCode } from '../parsers/code-tokenizer.js'
 import { toTableMetadata } from '../parsers/table-metadata.js'
-
 import {
-  markdownTextPrimitive,
-  codePrimitive,
-  tablePrimitive,
-  searchResultPrimitive,
-  reelPrimitive,
-  imaginePrimitive,
-  productCardPrimitive,
-  postPrimitive,
-  metadataTextPrimitive,
-  followUpSuggestionPillPrimitive,
-  shapeSourceEntry,
-  shapeReelEntry,
+  markdownTextPrimitive, codePrimitive, tablePrimitive, searchResultPrimitive,
+  reelPrimitive, imaginePrimitive, productCardPrimitive, postPrimitive,
+  metadataTextPrimitive, followUpSuggestionPillPrimitive, shapeSourceEntry, shapeReelEntry,
 } from '../proto/primitives.js'
 import { singleLayout, hscrollLayout, actionRowLayout } from '../proto/layouts.js'
 import { assembleRichResponse } from '../proto/rich-response.js'
 import { MessageType, ImagineType, LayoutKind } from '../proto/enums.js'
 
-
 export class AIRichBuilder {
+  #conn
+  #proto
+  #media
+
   constructor(conn, proto, media) {
     applyContentFields(this)
     this.#conn = conn
     this.#proto = proto
     this.#media = media
-
     this._submessages = []
     this._sections = []
     this._richResponseSources = []
-
     this._forwarded = true
     this._notification = false
     this._includesUnifiedResponse = true
     this._includesSubmessages = true
-    this._quoted
-    this._quotedParticipant
-  }
-
-
-
-  forwarded(v = true) {
-    this._forwarded = v
-    return this
+    this._quoted = undefined
+    this._quotedParticipant = undefined
   }
 
-  notification(v) {
-    this._notification = v
-    return this
-  }
+  forwarded(v = true) { this._forwarded = v; return this }
+  notification(v) { this._notification = v; return this }
+  includesUnifiedResponse(v = true) { this._includesUnifiedResponse = v; return this }
+  includesSubmessages(v = true) { this._includesSubmessages = v; return this }
 
   quoted(quoted, quotedParticipant) {
     this._quoted = quoted
@@ -63,16 +45,6 @@ export class AIRichBuilder {
     return this
   }
 
-  includesUnifiedResponse(v = true) {
-    this._includesUnifiedResponse = v
-    return this
-  }
-
-  includesSubmessages(v = true) {
-    this._includesSubmessages = v
-    return this
-  }
-
   submessage(msg) {
     const items = Array.isArray(msg) ? msg : [msg]
     items.forEach((m) => ensureObjectOrArray(m, 'submessage'))
@@ -80,42 +52,23 @@ export class AIRichBuilder {
     return this
   }
 
-
   async text(text, opts = {}) {
     ensureString(text, 'text')
     const { hyperlink = true, citation = true, latex = true } = opts
-
-    const { text: rewritten, metadata } = extractInlineEntities(text, {
-      hyperlink,
-      citation,
-      latex,
-    })
-
-    this._submessages.push({
-      messageType: MessageType.TEXT,
-      messageText: rewritten,
-    })
-
-    this._sections.push(
-      singleLayout(markdownTextPrimitive(rewritten, metadata.length ? metadata : undefined)),
-    )
+    const { text: rewritten, metadata } = extractInlineEntities(text, { hyperlink, citation, latex })
+    this._submessages.push({ messageType: MessageType.TEXT, messageText: rewritten })
+    this._sections.push(singleLayout(markdownTextPrimitive(rewritten, metadata.length ? metadata : undefined)))
     return this
   }
 
   async code(language, code) {
     ensureString(language, 'language')
     ensureString(code, 'code')
-
     const { codeBlock, unifiedBlocks } = tokenizeCode(code, language)
-
     this._submessages.push({
       messageType: MessageType.CODE,
-      codeMetadata: {
-        codeLanguage: language,
-        codeBlocks: codeBlock,
-      },
+      codeMetadata: { codeLanguage: language, codeBlocks: codeBlock },
     })
-
     this._sections.push(singleLayout(codePrimitive(language, unifiedBlocks)))
     return this
   }
@@ -126,34 +79,22 @@ export class AIRichBuilder {
     }
     const { title = '', hyperlink = true, citation = true, latex = true } = opts
     const meta = toTableMetadata(table, { title, hyperlink, citation, latex })
-
     this._submessages.push({
       messageType: MessageType.TABLE,
       tableMetadata: { title: meta.title, rows: meta.rows },
     })
-
     this._sections.push(singleLayout(tablePrimitive(meta.unifiedRows)))
     return this
   }
 
   async image(source, opts = {}) {
-    const { resolveUrl = false } = opts
     const list = Array.isArray(source) ? source : [source]
-
     const resolved = await Promise.all(
       list.map(async (s) => {
-        const url = await this.#media.resolve(s, {
-          mediaType: 'image',
-          strategy: resolveUrl ? 'auto' : 'auto',
-        })
-        return {
-          imagePreviewUrl: url,
-          imageHighResUrl: url,
-          sourceUrl: url,
-        }
+        const url = await this.#media.resolve(s, { mediaType: 'image', strategy: 'auto' })
+        return { imagePreviewUrl: url, imageHighResUrl: url, sourceUrl: url }
       }),
     )
-
     this._submessages.push({
       messageType: MessageType.RICH_RESPONSE,
       gridImageMetadata: {
@@ -161,15 +102,9 @@ export class AIRichBuilder {
         imageUrls: resolved,
       },
     })
-
     for (const r of resolved) {
       this._sections.push(
-        singleLayout(
-          imaginePrimitive(
-            { url: r.imagePreviewUrl, mime_type: 'image/png' },
-            ImagineType.IMAGE,
-          ),
-        ),
+        singleLayout(imaginePrimitive({ url: r.imagePreviewUrl, mime_type: 'image/png' }, ImagineType.IMAGE)),
       )
     }
     return this
@@ -177,22 +112,15 @@ export class AIRichBuilder {
 
   async video(source, opts = {}) {
     const { autoFill = true } = opts
-    const isObjectVideo = (v) => v && typeof v === 'object' && v.url
+    const isObjVideo = (v) => v && typeof v === 'object' && v.url
     const items = Array.isArray(source) ? source : [source]
 
-    this._submessages.push({
-      messageType: MessageType.TEXT,
-      messageText: '[ CANNOT_LOAD_VIDEO ]',
-    })
+    this._submessages.push({ messageType: MessageType.TEXT, messageText: '[ CANNOT_LOAD_VIDEO ]' })
 
     await Promise.all(
       items.map(async (item) => {
-        const isObj = isObjectVideo(item)
-        const url = await this.#media.resolve(
-          isObj ? item.url : item,
-          { mediaType: 'video', strategy: 'auto' },
-        )
-
+        const isObj = isObjVideo(item)
+        const url = await this.#media.resolve(isObj ? item.url : item, { mediaType: 'video', strategy: 'auto' })
         let fileLength = isObj && item.file_length != null ? item.file_length : 0
         let duration = isObj && item.duration != null ? item.duration : 0
         let thumbnailB64 = null
@@ -204,26 +132,18 @@ export class AIRichBuilder {
               fileLength = buf.length
               duration = this.#media.duration(buf)
               thumbnailB64 = await this.#media.videoThumbnail(buf, {
-                time: 0,
-                result: 'base64',
-                resizeOutput: true,
-                width: 300,
-                height: 300,
+                time: 0, result: 'base64', resizeOutput: true, width: 300, height: 300,
               })
             }
-          } catch {
-          }
+          } catch {}
         }
 
         const mimeType = isObj ? item.mime_type ?? 'video/mp4' : 'video/mp4'
-
         this._sections.push(
           singleLayout(
             imaginePrimitive(
               { url, mime_type: mimeType, file_length: fileLength, duration },
-              ImagineType.ANIMATE,
-              'READY',
-              thumbnailB64 || undefined,
+              ImagineType.ANIMATE, 'READY', thumbnailB64 || undefined,
             ),
           ),
         )
@@ -236,12 +156,7 @@ export class AIRichBuilder {
     if (!Array.isArray(sources)) {
       throw new ValidationError('sources must be an array', { code: 'LUMINA_VALIDATION_SOURCE' })
     }
-
-    const normalised =
-      sources.length && typeof sources[0] === 'string'
-        ? [sources]
-        : sources
-
+    const normalised = sources.length && typeof sources[0] === 'string' ? [sources] : sources
     const entries = await Promise.all(
       normalised.map(async (entry) => {
         const [iconUrl, url, text] = entry
@@ -251,7 +166,6 @@ export class AIRichBuilder {
         return shapeSourceEntry({ iconUrl: resolvedIcon, url, text })
       }),
     )
-
     this._sections.push(singleLayout(searchResultPrimitive(entries)))
     return this
   }
@@ -259,27 +173,17 @@ export class AIRichBuilder {
   async reels(items) {
     ensureObjectOrArray(items, 'reels')
     const list = Array.isArray(items) ? items : [items]
-
     const resolved = await Promise.all(
       list.map(async (item) => {
-        const [avatar, thumb] = await Promise.all([
-          item.profileIconUrl ?? item.profile_url ?? item.profile
-            ? this.#media.resolve(item.profileIconUrl ?? item.profile_url ?? item.profile, {
-                mediaType: 'image',
-                strategy: 'auto',
-              })
-            : Promise.resolve(''),
-          item.thumbnailUrl ?? item.thumbnail
-            ? this.#media.resolve(item.thumbnailUrl ?? item.thumbnail, {
-                mediaType: 'image',
-                strategy: 'auto',
-              })
-            : Promise.resolve(''),
+        const avatarSrc = item.profileIconUrl ?? item.profile_url ?? item.profile
+        const thumbSrc = item.thumbnailUrl ?? item.thumbnail
+        const [avatar, thumbnail] = await Promise.all([
+          avatarSrc ? this.#media.resolve(avatarSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
+          thumbSrc ? this.#media.resolve(thumbSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
         ])
-        return { ...item, avatar, thumbnail: thumb }
+        return { ...item, avatar, thumbnail }
       }),
     )
-
     this._submessages.push({
       messageType: MessageType.REELS,
       contentItemsMetadata: {
@@ -294,7 +198,6 @@ export class AIRichBuilder {
         })),
       },
     })
-
     resolved.forEach((item, idx) => {
       this._richResponseSources.push({
         provider: 'LUMINA',
@@ -306,110 +209,62 @@ export class AIRichBuilder {
         sourceTitle: item.username ?? '',
       })
     })
-
-    this._sections.push(
-      hscrollLayout(resolved.map((item) => reelPrimitive(shapeReelEntry(item)))),
-    )
+    this._sections.push(hscrollLayout(resolved.map((item) => reelPrimitive(shapeReelEntry(item)))))
     return this
   }
 
   async product(item) {
     ensureObjectOrArray(item, 'product')
     const list = Array.isArray(item) ? item : [item]
-
-    this._submessages.push({
-      messageType: MessageType.TEXT,
-      messageText: '[ CANNOT_LOAD_PRODUCT ]',
-    })
-
+    this._submessages.push({ messageType: MessageType.TEXT, messageText: '[ CANNOT_LOAD_PRODUCT ]' })
     const products = await Promise.all(
       list.map(async (p) => {
+        const imgSrc = p.image_url ?? p.image
+        const iconSrc = p.icon_url ?? p.icon
         const [imageUrl, iconUrl] = await Promise.all([
-          p.image_url ?? p.image
-            ? this.#media.resolve(p.image_url ?? p.image, { mediaType: 'image', strategy: 'auto' })
-            : Promise.resolve(''),
-          p.icon_url ?? p.icon
-            ? this.#media.resolve(p.icon_url ?? p.icon, { mediaType: 'image', strategy: 'auto' })
-            : Promise.resolve(''),
+          imgSrc ? this.#media.resolve(imgSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
+          iconSrc ? this.#media.resolve(iconSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
         ])
         return {
-          title: p.title,
-          brand: p.brand,
-          price: p.price,
-          sale_price: p.sale_price,
+          title: p.title, brand: p.brand, price: p.price, sale_price: p.sale_price,
           product_url: p.product_url ?? p.url,
           image: { url: imageUrl },
           additional_images: [{ url: iconUrl }],
         }
       }),
     )
-
     const primitives = products.map((p) => productCardPrimitive(p))
-    this._sections.push(
-      list.length === 1 ? singleLayout(primitives[0]) : hscrollLayout(primitives),
-    )
+    this._sections.push(list.length === 1 ? singleLayout(primitives[0]) : hscrollLayout(primitives))
     return this
   }
 
   async post(item) {
     ensureObjectOrArray(item, 'post')
     const list = Array.isArray(item) ? item : [item]
-
-    this._submessages.push({
-      messageType: MessageType.TEXT,
-      messageText: '[ CANNOT_LOAD_POST ]',
-    })
-
+    this._submessages.push({ messageType: MessageType.TEXT, messageText: '[ CANNOT_LOAD_POST ]' })
     const isCarousel = list.length > 1
-
     const primitives = await Promise.all(
       list.map(async (p) => {
+        const profileSrc = p.profile_picture_url ?? p.profile_url ?? p.profile
+        const thumbSrc = p.thumbnail_url ?? p.thumbnail
+        const footerSrc = p.footer_icon ?? p.icon
         const [profileUrl, thumbUrl, footerIcon] = await Promise.all([
-          p.profile_picture_url ?? p.profile_url ?? p.profile
-            ? this.#media.resolve(p.profile_picture_url ?? p.profile_url ?? p.profile, {
-                mediaType: 'image',
-                strategy: 'auto',
-              })
-            : Promise.resolve(''),
-          p.thumbnail_url ?? p.thumbnail
-            ? this.#media.resolve(p.thumbnail_url ?? p.thumbnail, {
-                mediaType: 'image',
-                strategy: 'auto',
-              })
-            : Promise.resolve(''),
-          p.footer_icon ?? p.icon
-            ? this.#media.resolve(p.footer_icon ?? p.icon, {
-                mediaType: 'image',
-                strategy: 'auto',
-              })
-            : Promise.resolve(''),
+          profileSrc ? this.#media.resolve(profileSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
+          thumbSrc ? this.#media.resolve(thumbSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
+          footerSrc ? this.#media.resolve(footerSrc, { mediaType: 'image', strategy: 'auto' }) : Promise.resolve(''),
         ])
-
-        return postPrimitive(
-          {
-            title: p.title ?? '',
-            subtitle: p.subtitle ?? '',
-            username: p.username ?? '',
-            profile_picture_url: profileUrl,
-            is_verified: !!(p.is_verified ?? p.verified),
-            thumbnail_url: thumbUrl,
-            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: footerIcon,
-            orientation: p.orientation ?? 'LANDSCAPE',
-            post_type: p.post_type ?? 'VIDEO',
-          },
-          isCarousel,
-        )
+        return postPrimitive({
+          title: p.title ?? '', subtitle: p.subtitle ?? '', username: p.username ?? '',
+          profile_picture_url: profileUrl, is_verified: !!(p.is_verified ?? p.verified),
+          thumbnail_url: thumbUrl, 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: footerIcon,
+          orientation: p.orientation ?? 'LANDSCAPE', post_type: p.post_type ?? 'VIDEO',
+        }, isCarousel)
       }),
     )
-
     this._sections.push(hscrollLayout(primitives))
     return this
   }
@@ -423,15 +278,12 @@ export class AIRichBuilder {
 
   async suggest(suggestion, opts = {}) {
     const { scroll = true, layout } = opts
-
     if (typeof suggestion === 'string') {
       suggestion = [suggestion]
     } else {
       ensureStringArray(suggestion, 'suggestion')
     }
-
     const pills = suggestion.map((text) => followUpSuggestionPillPrimitive(text))
-
     let kind
     if (layout) {
       kind = layout
@@ -440,7 +292,6 @@ export class AIRichBuilder {
     } else {
       kind = scroll ? LayoutKind.HSCROLL : LayoutKind.ACTION_ROW
     }
-
     if (kind === LayoutKind.SINGLE) {
       this._sections.push(singleLayout(pills[0], { __typename: 'GenAIUnifiedResponseSection' }))
     } else if (kind === LayoutKind.HSCROLL) {
@@ -451,7 +302,6 @@ export class AIRichBuilder {
     return this
   }
 
-
   async build(opts = {}) {
     return assembleRichResponse({
       title: this._title,