@blckrose/baileys 2.0.3 → 2.0.5

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

@@ -0,0 +1,392 @@
+/**
+ * Lia@Changes 09-04-26 [WIP]
+ * Adds support for tables and code blocks with richResponseMessage (wrapped inside botForwardedMessage).
+ *
+ * If you use or copy this code, please credit my name or project.
+ * @itsliaaa/baileys
+ */
+import { getRandomValues, randomUUID, randomBytes } from 'crypto';
+import { BOT_RENDERING_CONFIG_METADATA, DONATE_URL, LEXER_REGEX } from '../Defaults/index.js';
+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([]);
+export const tokenizeCode = (code, language = 'javascript') => {
+    const keywords = LANGUAGE_KEYWORDS[language] || NOOP;
+    const blocks = [];
+    LEXER_REGEX.lastIndex = 0;
+    let match;
+    while ((match = LEXER_REGEX.exec(code)) !== null) {
+        if (match[1]) {
+            blocks.push({ highlightType: CodeHighlightType.COMMENT, codeContent: match[1] });
+        }
+        else if (match[2]) {
+            blocks.push({ highlightType: CodeHighlightType.STRING, codeContent: match[2] });
+        }
+        else if (match[3]) {
+            blocks.push({
+                highlightType: keywords.has(match[3]) ? CodeHighlightType.KEYWORD : CodeHighlightType.METHOD,
+                codeContent: match[3],
+            });
+        }
+        else if (match[4]) {
+            blocks.push({
+                highlightType: keywords.has(match[4]) ? CodeHighlightType.KEYWORD : CodeHighlightType.DEFAULT,
+                codeContent: match[4],
+            });
+        }
+        else if (match[5]) {
+            blocks.push({ highlightType: CodeHighlightType.NUMBER, codeContent: match[5] });
+        }
+        else {
+            blocks.push({ highlightType: CodeHighlightType.DEFAULT, codeContent: match[6] });
+        }
+    }
+    return blocks;
+};
+// Lia@Changes 09-04-26 --- Inject buffer into unifiedResponse.data to support proper rendering of rich messages (ex: tables and code blocks)
+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.CONTENT_ITEMS:
+                    return {
+                        view_model: {
+                            primitives: submessage.contentItemsMetadata.itemsMetadata.map((item) => {
+                                const reelItem = item.reelItem
+                                return {
+                                    reels_url: reelItem.videoUrl,
+                                    thumbnail_url: reelItem.thumbnailUrl,
+                                    creator: reelItem.creator || '@itsliaaa/baileys',
+                                    avatar_url: reelItem.profileIconUrl,
+                                    reels_title: reelItem.title,
+                                    likes_count: reelItem.likesCount || 0,
+                                    shares_count: reelItem.sharesCount || 0,
+                                    view_count: reelItem.viewCount || 0,
+                                    reel_source: reelItem.reelSource || 'IG',
+                                    is_verified: reelItem.isVerified || false,
+                                    __typename: 'GenAIReelPrimitive'
+                                }
+                            }),
+                            __typename: 'GenAIHScrollLayoutViewModel'
+                        }
+                    };
+                case RichSubMessageType.LATEX:
+                    const latexMetadata = submessage.latexMetadata;
+                    const item = {
+                        latex_expression: latexMetadata.expressions[0]?.latexExpression,
+                        font_height: latexMetadata.expressions[0]?.fontHeight,
+                        padding: 15,
+                        latex_image: {
+                            url: latexMetadata.expressions[0]?.url,
+                            width: latexMetadata.expressions[0]?.width || 388,
+                            height: latexMetadata.expressions[0]?.height || 160
+                        }
+                    };
+                    return {
+                        view_model: {
+                            primitive: {
+                                item,
+                                ...item,
+                                __typename: 'GenAILatexUXPrimitive'
+                            },
+                            __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: [] })),
+                                __typename: 'GenATableUXPrimitive'
+                            },
+                            __typename: 'GenAISingleLayoutViewModel'
+                        }
+                    };
+                case RichSubMessageType.TEXT:
+                    return {
+                        view_model: {
+                            primitive: {
+                                text: submessage.messageText,
+                                inline_entities: submessage.inlineEntities || [],
+                                __typename: 'GenAIMarkdownTextUXPrimitive'
+                            },
+                            __typename: 'GenAISingleLayoutViewModel'
+                        }
+                    };
+            }
+            return submessage;
+        })
+    });
+// Lia@Note 17-04-26 --- WIP
+export const buildAdditionalBotMetadataContext = (submessages) => {
+    const sources = [];
+    const mediaDetailsMetadataList = [];
+    for (let i = 0; i < submessages.length; i++) {
+        const submessage = submessages[i];
+        switch (submessage.messageType) {
+            case RichSubMessageType.CONTENT_ITEMS:
+                const itemsMetadata = submessage.contentItemsMetadata.itemsMetadata;
+                for (let n = 0; n < itemsMetadata.length; n++) {
+                    const reelItem = itemsMetadata[n].reelItem;
+                    sources.push({
+                        provider: 0,
+                        thumbnailCdnUrl: reelItem.thumbnailUrl,
+                        sourceProviderUrl: reelItem.videoUrl,
+                        sourceQuery: '',
+                        faviconCdnUrl: '',
+                        citationNumber: i + 1,
+                        sourceTitle: reelItem.title
+                    });
+                    mediaDetailsMetadataList.push({
+                        id: randomBytes(32).toString('hex'),
+                        previewMedia: {
+                            fileSha256: '',
+                            mediaKey: '',
+                            fileEncSha256: '',
+                            directPath: '',
+                            mediaKeyTimestamp: unixTimestampSeconds(),
+                            mimetype: 'image/jpeg'
+                        }
+                    });
+                }
+                break;
+            case RichSubMessageType.LATEX:
+                const expressions = submessage.latexMetadata.expressions;
+                for (let n = 0; n < expressions.length; n++) {
+                    const expression = expressions[n];
+                    mediaDetailsMetadataList.push({
+                        id: randomBytes(32).toString('hex'),
+                        previewMedia: {
+                            fileSha256: '',
+                            mediaKey: '',
+                            fileEncSha256: '',
+                            directPath: '',
+                            mediaKeyTimestamp: unixTimestampSeconds(),
+                            mimetype: 'image/jpeg'
+                        }
+                    });
+                }
+                break;
+        }
+    }
+    return { sources, mediaDetailsMetadataList };
+}
+export const prepareRichResponseMessage = (content) => {
+    const { code, contentText, expressions, footerText, headerText, items, language, links, noHeading, richResponse, table, text, title } = content;
+    let submessages = [];
+    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.expressions) {
+                return {
+                    messageType: RichSubMessageType.LATEX,
+                    latexMetadata: {
+                        text: submessage.text,
+                        expressions: submessage.expressions
+                    }
+                };
+            }
+            else if (submessage.items) {
+                return {
+                    messageType: RichSubMessageType.CONTENT_ITEMS,
+                    contentItemsMetadata: {
+                        itemsMetadata: submessage.items
+                    }
+                };
+            }
+            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
+            });
+        }
+        if (contentText) {
+            submessages.push({
+                messageType: RichSubMessageType.TEXT,
+                messageText: contentText
+            });
+        }
+        if (code) {
+            language ??= 'javascript';
+            submessages.push({
+                messageType: RichSubMessageType.CODE,
+                codeMetadata: {
+                    codeLanguage: language,
+                    codeBlocks: tokenizeCode(code, language)
+                }
+            });
+        }
+        else if (expressions) {
+            submessages.push({
+                messageType: RichSubMessageType.LATEX,
+                latexMetadata: {
+                    text,
+                    expressions
+                }
+            });
+        }
+        else if (items) {
+            submessages.push({
+                messageType: RichSubMessageType.CONTENT_ITEMS,
+                contentItemsMetadata: {
+                    itemsMetadata: items.map((item) => ({ reelItem: item })),
+                    contentType: proto.AIRichResponseContentItemsMetadata.ContentType.CAROUSEL
+                }
+            });
+        }
+        else if (links) {
+            links.forEach((linkField, index) => {
+                const prefix = 'SS_' + index;
+                const url = linkField.url || DONATE_URL;
+                const sources = linkField.sources?.map((sourceField) => ({
+                    source_type: 'THIRD_PARTY',
+                    source_display_name: sourceField.displayName || 'Donate',
+                    source_subtitle: sourceField.subtitle || 'Saweria',
+                    source_url: sourceField.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
+            });
+        }
+    }
+    const unified = toUnified(submessages);
+    const message = wrapToBotForwardedMessage({
+        submessages: [],
+        messageType: proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD,
+        unifiedResponse: {
+            data: Buffer.from(JSON.stringify(unified), 'utf-8') // Lia@Note 25-04-26 --- Expects "ArrayBufferLike"
+        },
+        contextInfo: {
+            isForwarded: true,
+            forwardingScore: 1,
+            forwardedAiBotMessageInfo: { botJid: '867051314767696@bot' },
+            forwardOrigin: 4,
+            botMessageSharingInfo: { forwardScore: 1 }
+        }
+    });
+    // Lia@Note 17-04-26 --- TODO: Fill mediaDetailsMetadataList and sources field
+    const { sources, mediaDetailsMetadataList } = buildAdditionalBotMetadataContext(submessages);
+    const botMetadata = message.messageContextInfo.botMetadata;
+    if (sources.length > 0) {
+        botMetadata.richResponseSourcesMetadata = { sources };
+    }
+    if (mediaDetailsMetadataList.length > 0) {
+        botMetadata.unifiedResponseMutation = { mediaDetailsMetadataList };
+    }
+    return message;
+}
+// Lia@Note 17-04-26 --- signature and certificateChain for proofs[] field
+export const botMetadataSignature = () => {
+    const signature = new Uint8Array(64);
+    getRandomValues(signature);
+    return signature;
+}
+export const botMetadataCertificate = (length = 700) => {
+    const certificate = new Uint8Array(length);
+    certificate[0] = 48;
+    certificate[1] = 130;
+    getRandomValues(certificate.subarray(2));
+    return certificate;
+}
+export const wrapToBotForwardedMessage = (richResponseMessage) =>
+    ({
+        messageContextInfo: {
+            botMetadata: {
+                pluginMetadata: {},
+                // Lia@Note 09-04-26 --- TODO: Fill verificationMetadata field
+                verificationMetadata: {
+                    proofs: [
+                        {
+                            certificateChain: [
+                                botMetadataCertificate(684),
+                                botMetadataCertificate(892)
+                            ],
+                            version: 1,
+                            useCase: 1,
+                            signature: botMetadataSignature()
+                        }
+                    ]
+                },
+                botRenderingConfigMetadata: BOT_RENDERING_CONFIG_METADATA
+            }
+        },
+        botForwardedMessage: {
+            message: { richResponseMessage }
+        }
+    });

package/lib/Utils/stanza-ack.js

@@ -0,0 +1,37 @@
+/**
+ * Builds an ACK stanza for a received node.
+ * Pure function -- no I/O, no side effects.
+ *
+ * Mirrors WhatsApp Web's ACK construction:
+ * - WAWebHandleMsgSendAck.sendAck / sendNack
+ * - WAWebCreateNackFromStanza.createNackFromStanza
+ */
+export function buildAckStanza(node, errorCode, meId) {
+    const { tag, attrs } = node;
+    const stanza = {
+        tag: 'ack',
+        attrs: {
+            id: attrs.id,
+            to: attrs.from,
+            class: tag
+        }
+    };
+    if (errorCode) {
+        stanza.attrs.error = errorCode.toString();
+    }
+    if (attrs.participant) {
+        stanza.attrs.participant = attrs.participant;
+    }
+    if (attrs.recipient) {
+        stanza.attrs.recipient = attrs.recipient;
+    }
+    // WA Web always includes type when present: `n.type || DROP_ATTR`
+    if (attrs.type) {
+        stanza.attrs.type = attrs.type;
+    }
+    // WA Web WAWebHandleMsgSendAck.sendAck/sendNack always include `from` for message-class ACKs
+    if (tag === 'message' && meId) {
+        stanza.attrs.from = meId;
+    }
+    return stanza;
+}

package/lib/WABinary/constants.js

@@ -1298,4 +1298,77 @@ for (const [i, DOUBLE_BYTE_TOKEN] of DOUBLE_BYTE_TOKENS.entries()) {
         TOKEN_MAP[element] = { dict: i, index: j };
     }
 }
+
+export const CPP_KEYWORDS = new Set([
+    'alignas', 'alignof', 'and', 'and_eq', 'asm', 'auto', 'bitand', 'bitor', 'bool', 'break', 'case',
+    'catch', 'char', 'class', 'compl', 'concept', 'const', 'consteval', 'constexpr', 'constinit',
+    'const_cast', 'continue', 'co_await', 'co_return', 'co_yield', 'decltype', 'default', 'delete',
+    'do', 'double', 'dynamic_cast', 'else', 'enum', 'explicit', 'export', 'extern', 'false', 'float',
+    'for', 'friend', 'goto', 'if', 'inline', 'int', 'long', 'mutable', 'namespace', 'new', 'noexcept',
+    'not', 'not_eq', 'nullptr', 'operator', 'or', 'or_eq', 'private', 'protected', 'public', 'register',
+    'reinterpret_cast', 'requires', 'return', 'short', 'signed', 'sizeof', 'static', 'static_assert',
+    'static_cast', 'struct', 'switch', 'template', 'this', 'thread_local', 'throw', 'true', 'try',
+    'typedef', 'typeid', 'typename', 'union', 'unsigned', 'using', 'virtual', 'void', 'volatile',
+    'wchar_t', 'while', 'xor', 'xor_eq'
+]);
+export const CSS_KEYWORDS = new Set([
+    'import', 'media', 'font-face', 'keyframes', 'supports', 'charset',
+    'important', 'root', 'hover', 'active', 'focus', 'visited', 'before', 'after',
+    'not', 'nth-child', 'first-child', 'last-child', 'only-child',
+    'none', 'inherit', 'initial', 'unset', 'auto', 'transparent', 'currentcolor'
+]);
+export const GO_KEYWORDS = new Set([
+    'break', 'default', 'func', 'interface', 'select', 'case', 'defer', 'go', 'map', 'struct',
+    'chan', 'else', 'goto', 'package', 'switch', 'const', 'fallthrough', 'if', 'range', 'type',
+    'continue', 'for', 'import', 'return', 'var', 'true', 'false', 'nil'
+]);
+export const HTML_KEYWORDS = new Set([
+    'html', 'head', 'body', 'title', 'meta', 'link', 'script', 'style',
+    'header', 'footer', 'main', 'section', 'article', 'aside', 'nav',
+    'div', 'span', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'p', 'a', 'img',
+    'ul', 'ol', 'li', 'table', 'tr', 'td', 'th', 'thead', 'tbody',
+    'form', 'input', 'button', 'select', 'textarea', 'label', 'option',
+    'canvas', 'svg', 'iframe', 'video', 'audio', 'source'
+]);
+export const JS_KEYWORDS = new Set([
+    'import', 'export', 'from', 'default', 'as',
+    'const', 'let', 'var', 'function', 'class', 'extends', 'new',
+    'return', 'if', 'else', 'for', 'while', 'do', 'switch', 'case', 'break', 'continue',
+    'try', 'catch', 'finally', 'throw',
+    'async', 'await', 'yield',
+    'typeof', 'instanceof', 'in', 'of', 'delete', 'void',
+    'true', 'false', 'null', 'undefined', 'NaN', 'Infinity',
+    'this', 'super', 'static', 'get', 'set',
+    'debugger', 'with'
+]);
+export const PYTHON_KEYWORDS = new Set([
+    'import', 'from', 'as', 'def', 'class', 'return', 'if', 'elif', 'else',
+    'for', 'while', 'break', 'continue', 'try', 'except', 'finally', 'raise',
+    'with', 'yield', 'lambda', 'pass', 'del', 'global', 'nonlocal', 'assert',
+    'True', 'False', 'None', 'and', 'or', 'not', 'in', 'is', 'async', 'await',
+    'self', 'print'
+]);
+export const RUST_KEYWORDS = new Set([
+    'as', 'break', 'const', 'continue', 'crate', 'else', 'enum', 'extern', 'false', 'fn', 'for',
+    'if', 'impl', 'in', 'let', 'loop', 'match', 'mod', 'move', 'mut', 'pub', 'ref', 'return',
+    'self', 'Self', 'static', 'struct', 'super', 'trait', 'true', 'type', 'unsafe', 'use',
+    'where', 'while', 'async', 'await', 'dyn', 'abstract', 'become', 'box', 'do', 'final',
+    'macro', 'override', 'priv', 'typeof', 'unsized', 'virtual', 'yield'
+]);
+export const LANGUAGE_KEYWORDS = {
+    css: CSS_KEYWORDS,
+    html: HTML_KEYWORDS,
+    javascript: JS_KEYWORDS,
+    typescript: JS_KEYWORDS,
+    js: JS_KEYWORDS,
+    ts: JS_KEYWORDS,
+    python: PYTHON_KEYWORDS,
+    py: PYTHON_KEYWORDS,
+    go: GO_KEYWORDS,
+    golang: GO_KEYWORDS,
+    cpp: CPP_KEYWORDS,
+    'c++': CPP_KEYWORDS,
+    rust: RUST_KEYWORDS,
+    rs: RUST_KEYWORDS,
+};
 //# sourceMappingURL=constants.js.map

package/lib/WAUSync/Protocols/USyncUsernameProtocol.js

@@ -0,0 +1,22 @@
+import { assertNodeErrorFree } from '../../WABinary/index.js';
+import { USyncUser } from '../USyncUser.js';
+export class USyncUsernameProtocol {
+    name = 'username';
+    getQueryElement() {
+        return {
+            tag: 'username',
+            attrs: {}
+        };
+    }
+    getUserElement(user) {
+        void user;
+        return null;
+    }
+    parser(node) {
+        if (node.tag === 'username') {
+            assertNodeErrorFree(node);
+            return typeof node.content === 'string' ? node.content : null;
+        }
+        return null;
+    }
+}

package/lib/WAUSync/Protocols/index.js

@@ -3,4 +3,5 @@ export * from './USyncContactProtocol.js';
 export * from './USyncStatusProtocol.js';
 export * from './USyncDisappearingModeProtocol.js';
 export * from './UsyncBotProfileProtocol.js';
-export * from './UsyncLIDProtocol.js';
+export * from './UsyncLIDProtocol.js';
+export * from './USyncUsernameProtocol.js';

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@blckrose/baileys",
   "type": "module",
-  "version": "2.0.3",
+  "version": "2.0.5",
   "description": "A WebSockets library for interacting with WhatsApp Web",
   "keywords": [
     "whatsapp",
@@ -82,4 +82,4 @@
   "engines": {
     "node": ">=20.0.0"
   }
-}
+}