@baileysmod-ubed/baileys 0.3.17

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

@@ -0,0 +1,370 @@
+/**
+ * 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. AND DO NOT CHANGE THIS NOTE
+ */
+import { getRandomValues, randomUUID } from 'crypto';
+import { 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, uuid) => ({
+    response_id: uuid || 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 {};
+            case RichSubMessageType.INLINE_IMAGE:
+                return {};
+            case RichSubMessageType.LATEX:
+                return {};
+            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:
+                return {
+                    view_model: {
+                        primitive: {
+                            text: submessage.messageText,
+                            inline_entities: submessage.inlineEntities || [],
+                            __typename: 'GenAIMarkdownTextUXPrimitive'
+                        },
+                        __typename: 'GenAISingleLayoutViewModel'
+                    }
+                };
+        }
+        return submessage;
+    })
+});
+export const prepareRichResponseMessage = (content) => {
+    const { alignment, code, contentText, disclaimerText, footerText, headerText, imageText, inlineImage, inlineVideo, items, language, latex, links, noHeading, posts, products, suggested, richResponse, table, tapLinkUrl, 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.items) {
+                return {
+                    messageType: RichSubMessageType.CONTENT_ITEMS,
+                    contentItemsMetadata: {
+                        itemsMetadata: submessage.items,
+                        contentType: proto.AIRichResponseContentItemsMetadata.ContentType.CAROUSEL
+                    }
+                };
+            }
+            else if (submessage.inlineImage) {
+                return {
+                    messageType: RichSubMessageType.INLINE_IMAGE,
+                    imageMetadata: {
+                        imageUrl: submessage.inlineImage,
+                        imageText: submessage.imageText,
+                        alignment: submessage.alignment,
+                        tapLinkUrl: submessage.tapLinkUrl
+                    }
+                };
+            }
+            else if (submessage.inlineVideo) {
+                return {
+                    messageType: RichSubMessageType.TEXT,
+                    messageText: 'INLINE_VIDEO'
+                };
+            }
+            else if (submessage.latex) {
+                return {
+                    messageType: RichSubMessageType.LATEX,
+                    latexMetadata: {
+                        text: submessage.text,
+                        expressions: submessage.latex
+                    }
+                };
+            }
+            else if (submessage.posts) {
+                return {
+                    messageType: RichSubMessageType.TEXT,
+                    messageText: 'POSTS'
+                };
+            }
+            else if (submessage.products) {
+                return {
+                    messageType: RichSubMessageType.TEXT,
+                    messageText: 'PRODUCTS'
+                };
+            }
+            else if (submessage.suggested) {
+                return {
+                    messageType: RichSubMessageType.TEXT,
+                    messageText: 'SUGGESTED_PROMPT'
+                };
+            }
+            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)
+                }
+            });
+        }
+        if (items) {
+            submessages.push({
+                messageType: RichSubMessageType.CONTENT_ITEMS,
+                contentItemsMetadata: {
+                    itemsMetadata: items,
+                    contentType: proto.AIRichResponseContentItemsMetadata.ContentType.CAROUSEL
+                }
+            });
+        }
+        if (inlineImage) {
+            submessages.push({
+                messageType: RichSubMessageType.INLINE_IMAGE,
+                imageMetadata: {
+                    imageUrl: inlineImage,
+                    imageText,
+                    alignment,
+                    tapLinkUrl
+                }
+            });
+        }
+        if (inlineVideo) {
+            submessages.push({
+                messageType: RichSubMessageType.TEXT,
+                messageText: 'INLINE_VIDEO'
+            });
+        }
+        if (latex) {
+            submessages.push({
+                messageType: RichSubMessageType.LATEX,
+                latexMetadata: {
+                    text,
+                    expressions: latex
+                }
+            });
+        }
+        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'
+                            }
+                        }]
+                });
+            });
+        }
+        if (posts) {
+            submessages.push({
+                messageType: RichSubMessageType.TEXT,
+                messageText: 'POSTS'
+            });
+        }
+        if (products) {
+            submessages.push({
+                messageType: RichSubMessageType.TEXT,
+                messageText: 'PRODUCTS'
+            });
+        }
+        if (suggested) {
+            submessages.push({
+                messageType: RichSubMessageType.TEXT,
+                messageText: 'SUGGESTED_PROMPT'
+            });
+        }
+        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 uuid = randomUUID();
+    const unified = toUnified(submessages, uuid);
+    const richResponseMessage = proto.AIRichResponseMessage.create({
+        submessages,
+        messageType: proto.AIRichResponseMessageType.AI_RICH_RESPONSE_TYPE_STANDARD,
+        unifiedResponse: {
+            data: Buffer.from(JSON.stringify(unified)) // Lia@Note 25-04-26 --- Expects "ArrayBufferLike"
+        },
+        contextInfo: {
+            isForwarded: true,
+            forwardingScore: 1,
+            forwardedAiBotMessageInfo: { botJid: '867051314767696@bot' },
+            forwardOrigin: 4
+        }
+    });
+    const message = wrapToBotForwardedMessage(richResponseMessage);
+    const botMetadata = message.messageContextInfo.botMetadata;
+    // Lia@Note 13-05-26 --- Add disclaimer text on richResponseMessage
+    if (disclaimerText) {
+        botMetadata.messageDisclaimerText = disclaimerText;
+    }
+    // Lia@Note 15-05-26 --- Add responseId from unified directly to botMetadata
+    botMetadata.botResponseId = uuid;
+    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 = 685) => {
+    const certificate = new Uint8Array(length);
+    certificate[0] = 48;
+    certificate[1] = 130;
+    getRandomValues(certificate.subarray(2));
+    return certificate;
+};
+export const wrapToBotForwardedMessage = (richResponseMessage) => ({
+    messageContextInfo: {
+        botMetadata: {
+            // Lia@Note 09-04-26 --- TODO: Fill verificationMetadata field
+            verificationMetadata: {
+                proofs: [
+                    {
+                        certificateChain: [
+                            botMetadataCertificate(),
+                            botMetadataCertificate(892)
+                        ],
+                        version: 1,
+                        useCase: 1,
+                        signature: botMetadataSignature()
+                    }
+                ]
+            }
+        }
+    },
+    botForwardedMessage: {
+        message: { richResponseMessage }
+    }
+});
+//# sourceMappingURL=rich-message-utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"rich-message-utils.js","sourceRoot":"","sources":["../../src/Utils/rich-message-utils.js"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AAC/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AACrD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,CAAC;AACzB,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,IAAI,EAAE,QAAQ,GAAG,YAAY,EAAE,EAAE;IAC1D,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC;IACrD,MAAM,MAAM,GAAG,EAAE,CAAC;IAClB,WAAW,CAAC,SAAS,GAAG,CAAC,CAAC;IAC1B,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC/C,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACrF,CAAC;aACI,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,iBAAiB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;aACI,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC;gBACR,aAAa,EAAE,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM;gBAC5F,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;aACxB,CAAC,CAAC;QACP,CAAC;aACI,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC;gBACR,aAAa,EAAE,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,OAAO;gBAC7F,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;aACxB,CAAC,CAAC;QACP,CAAC;aACI,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,iBAAiB,CAAC,MAAM,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;aACI,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACrF,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC,CAAC;AACF,6IAA6I;AAC7I,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,WAAW,EAAE,IAAI,EAAE,EAAE,CAC3C,CAAC;IACG,WAAW,EAAE,IAAI,IAAI,UAAU,EAAE;IACjC,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;QAC5C,QAAQ,UAAU,CAAC,WAAW,EAAE,CAAC;YAC7B,KAAK,kBAAkB,CAAC,IAAI;gBACxB,MAAM,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;gBAC7C,OAAO;oBACH,UAAU,EAAE;wBACR,SAAS,EAAE;4BACP,QAAQ,EAAE,YAAY,CAAC,YAAY;4BACnC,WAAW,EAAE,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,WAAW,EAAE,IAAI,EAAE,iBAAiB,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;4BACnI,UAAU,EAAE,sBAAsB;yBACrC;wBACD,UAAU,EAAE,4BAA4B;qBAC3C;iBACJ,CAAC;YACN,KAAK,kBAAkB,CAAC,aAAa;gBACjC,OAAO,EAAE,CAAC;YACd,KAAK,kBAAkB,CAAC,YAAY;gBAChC,OAAO,EAAE,CAAC;YACd,KAAK,kBAAkB,CAAC,KAAK;gBACzB,OAAO,EAAE,CAAC;YACd,KAAK,kBAAkB,CAAC,KAAK;gBACzB,MAAM,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;gBAC/C,OAAO;oBACH,UAAU,EAAE;wBACR,SAAS,EAAE;4BACP,KAAK,EAAE,aAAa,CAAC,KAAK;4BAC1B,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gCACnC,SAAS,EAAE,GAAG,CAAC,SAAS;gCACxB,KAAK,EAAE,GAAG,CAAC,KAAK;gCAChB,cAAc,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;6BAC5D,CAAC,CAAC;4BACH,UAAU,EAAE,sBAAsB;yBACrC;wBACD,UAAU,EAAE,4BAA4B;qBAC3C;iBACJ,CAAC;YACN,KAAK,kBAAkB,CAAC,IAAI;gBACxB,OAAO;oBACH,UAAU,EAAE;wBACR,SAAS,EAAE;4BACP,IAAI,EAAE,UAAU,CAAC,WAAW;4BAC5B,eAAe,EAAE,UAAU,CAAC,cAAc,IAAI,EAAE;4BAChD,UAAU,EAAE,8BAA8B;yBAC7C;wBACD,UAAU,EAAE,4BAA4B;qBAC3C;iBACJ,CAAC;QACV,CAAC;QACD,OAAO,UAAU,CAAC;IACtB,CAAC,CAAC;CACL,CAAC,CAAC;AACP,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,OAAO,EAAE,EAAE;IAClD,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAC5O,IAAI,WAAW,GAAG,EAAE,CAAC;IACrB,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;QAC9B,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YAC1C,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gBAClB,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,WAAW,EAAE,UAAU,CAAC,IAAI;oBAC5B,cAAc,EAAE,UAAU,CAAC,cAAc;iBAC5C,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gBACvB,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,YAAY,EAAE;wBACV,YAAY,EAAE,UAAU,CAAC,QAAQ;wBACjC,UAAU,EAAE,UAAU,CAAC,IAAI;qBAC9B;iBACJ,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACxB,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,aAAa;oBAC7C,oBAAoB,EAAE;wBAClB,aAAa,EAAE,UAAU,CAAC,KAAK;wBAC/B,WAAW,EAAE,KAAK,CAAC,kCAAkC,CAAC,WAAW,CAAC,QAAQ;qBAC7E;iBACJ,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC9B,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,YAAY;oBAC5C,aAAa,EAAE;wBACX,QAAQ,EAAE,UAAU,CAAC,WAAW;wBAChC,SAAS,EAAE,UAAU,CAAC,SAAS;wBAC/B,SAAS,EAAE,UAAU,CAAC,SAAS;wBAC/B,UAAU,EAAE,UAAU,CAAC,UAAU;qBACpC;iBACJ,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC9B,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,WAAW,EAAE,cAAc;iBAC9B,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACxB,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,KAAK;oBACrC,aAAa,EAAE;wBACX,IAAI,EAAE,UAAU,CAAC,IAAI;wBACrB,WAAW,EAAE,UAAU,CAAC,KAAK;qBAChC;iBACJ,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACxB,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,WAAW,EAAE,OAAO;iBACvB,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;gBAC3B,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,WAAW,EAAE,UAAU;iBAC1B,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;gBAC5B,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,WAAW,EAAE,kBAAkB;iBAClC,CAAC;YACN,CAAC;iBACI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACxB,OAAO;oBACH,WAAW,EAAE,kBAAkB,CAAC,KAAK;oBACrC,aAAa,EAAE;wBACX,KAAK,EAAE,UAAU,CAAC,KAAK;wBACvB,IAAI,EAAE,UAAU,CAAC,KAAK;qBACzB;iBACJ,CAAC;YACN,CAAC;YACD,OAAO,UAAU,CAAC;QACtB,CAAC,CAAC,CAAC;IACP,CAAC;SACI,CAAC;QACF,IAAI,UAAU,EAAE,CAAC;YACb,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,UAAU;aAC1B,CAAC,CAAC;QACP,CAAC;QACD,IAAI,WAAW,EAAE,CAAC;YACd,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,WAAW;aAC3B,CAAC,CAAC;QACP,CAAC;QACD,IAAI,IAAI,EAAE,CAAC;YACP,QAAQ,KAAK,YAAY,CAAC;YAC1B,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,YAAY,EAAE;oBACV,YAAY,EAAE,QAAQ;oBACtB,UAAU,EAAE,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC;iBAC3C;aACJ,CAAC,CAAC;QACP,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACR,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,aAAa;gBAC7C,oBAAoB,EAAE;oBAClB,aAAa,EAAE,KAAK;oBACpB,WAAW,EAAE,KAAK,CAAC,kCAAkC,CAAC,WAAW,CAAC,QAAQ;iBAC7E;aACJ,CAAC,CAAC;QACP,CAAC;QACD,IAAI,WAAW,EAAE,CAAC;YACd,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,YAAY;gBAC5C,aAAa,EAAE;oBACX,QAAQ,EAAE,WAAW;oBACrB,SAAS;oBACT,SAAS;oBACT,UAAU;iBACb;aACJ,CAAC,CAAC;QACP,CAAC;QACD,IAAI,WAAW,EAAE,CAAC;YACd,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,cAAc;aAC9B,CAAC,CAAC;QACP,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACR,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,KAAK;gBACrC,aAAa,EAAE;oBACX,IAAI;oBACJ,WAAW,EAAE,KAAK;iBACrB;aACJ,CAAC,CAAC;QACP,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACR,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE;gBAC/B,MAAM,MAAM,GAAG,KAAK,GAAG,KAAK,CAAC;gBAC7B,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC;gBACxC,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;oBACrD,WAAW,EAAE,aAAa;oBAC1B,mBAAmB,EAAE,WAAW,CAAC,WAAW,IAAI,QAAQ;oBACxD,eAAe,EAAE,WAAW,CAAC,QAAQ,IAAI,SAAS;oBAClD,UAAU,EAAE,WAAW,CAAC,GAAG,IAAI,GAAG;iBACrC,CAAC,CAAC,CAAC;gBACJ,WAAW,CAAC,IAAI,CAAC;oBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;oBACpC,WAAW,EAAE,SAAS,CAAC,IAAI,GAAG,MAAM,MAAM,SAAS,MAAM,KAAK;oBAC9D,cAAc,EAAE,CAAC;4BACb,GAAG,EAAE,MAAM;4BACX,QAAQ,EAAE;gCACN,YAAY,EAAE,KAAK,GAAG,CAAC;gCACvB,aAAa,EAAE,GAAG;gCAClB,eAAe,EAAE,SAAS,CAAC,KAAK,IAAI,0BAA0B;gCAC9D,sBAAsB,EAAE,SAAS,CAAC,WAAW,IAAI,UAAU;gCAC3D,OAAO,EAAE,OAAO,IAAI,EAAE;gCACtB,UAAU,EAAE,yBAAyB;6BACxC;yBACJ,CAAC;iBACL,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;QACP,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACR,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,OAAO;aACvB,CAAC,CAAC;QACP,CAAC;QACD,IAAI,QAAQ,EAAE,CAAC;YACX,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,UAAU;aAC1B,CAAC,CAAC;QACP,CAAC;QACD,IAAI,SAAS,EAAE,CAAC;YACZ,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,kBAAkB;aAClC,CAAC,CAAC;QACP,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACR,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,KAAK;gBACrC,aAAa,EAAE;oBACX,KAAK;oBACL,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;wBAC/B,SAAS,EAAE,CAAC,SAAS,IAAI,KAAK,IAAI,CAAC;wBACnC,KAAK;qBACR,CAAC,CAAC;iBACN;aACJ,CAAC,CAAC;QACP,CAAC;QACD,IAAI,UAAU,EAAE,CAAC;YACb,WAAW,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,kBAAkB,CAAC,IAAI;gBACpC,WAAW,EAAE,UAAU;aAC1B,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IACD,MAAM,IAAI,GAAG,UAAU,EAAE,CAAC;IAC1B,MAAM,OAAO,GAAG,SAAS,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC7C,MAAM,mBAAmB,GAAG,KAAK,CAAC,qBAAqB,CAAC,MAAM,CAAC;QAC3D,WAAW;QACX,WAAW,EAAE,KAAK,CAAC,yBAAyB,CAAC,8BAA8B;QAC3E,eAAe,EAAE;YACb,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,kDAAkD;SAChG;QACD,WAAW,EAAE;YACT,WAAW,EAAE,IAAI;YACjB,eAAe,EAAE,CAAC;YAClB,yBAAyB,EAAE,EAAE,MAAM,EAAE,qBAAqB,EAAE;YAC5D,aAAa,EAAE,CAAC;SACnB;KACJ,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,yBAAyB,CAAC,mBAAmB,CAAC,CAAC;IAC/D,MAAM,WAAW,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC;IAC3D,mEAAmE;IACnE,IAAI,cAAc,EAAE,CAAC;QACjB,WAAW,CAAC,qBAAqB,GAAG,cAAc,CAAC;IACvD,CAAC;IACD,4EAA4E;IAC5E,WAAW,CAAC,aAAa,GAAG,IAAI,CAAC;IACjC,OAAO,OAAO,CAAC;AACnB,CAAC,CAAA;AACD,0EAA0E;AAC1E,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAG,EAAE;IACrC,MAAM,SAAS,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,CAAC;IACrC,eAAe,CAAC,SAAS,CAAC,CAAC;IAC3B,OAAO,SAAS,CAAC;AACrB,CAAC,CAAA;AACD,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,EAAE;IACnD,MAAM,WAAW,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;IAC3C,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;IACpB,WAAW,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;IACrB,eAAe,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,OAAO,WAAW,CAAC;AACvB,CAAC,CAAA;AACD,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,mBAAmB,EAAE,EAAE,CAC7D,CAAC;IACG,kBAAkB,EAAE;QAChB,WAAW,EAAE;YACT,8DAA8D;YAC9D,oBAAoB,EAAE;gBAClB,MAAM,EAAE;oBACJ;wBACI,gBAAgB,EAAE;4BACd,sBAAsB,EAAE;4BACxB,sBAAsB,CAAC,GAAG,CAAC;yBAC9B;wBACD,OAAO,EAAE,CAAC;wBACV,OAAO,EAAE,CAAC;wBACV,SAAS,EAAE,oBAAoB,EAAE;qBACpC;iBACJ;aACJ;SACJ;KACJ;IACD,mBAAmB,EAAE;QACjB,OAAO,EAAE,EAAE,mBAAmB,EAAE;KACnC;CACJ,CAAC,CAAC"}

package/lib/Utils/signal.d.ts

@@ -0,0 +1,83 @@
+export function createSignalIdentity(wid: any, accountSignatureKey: any): {
+    identifier: {
+        name: any;
+        deviceId: number;
+    };
+    identifierKey: any;
+};
+export function getPreKeys({ get }: {
+    get: any;
+}, min: any, limit: any): Promise<any>;
+export function generateOrGetPreKeys(creds: any, range: any): {
+    newPreKeys: {};
+    lastPreKeyId: number;
+    preKeysRange: any[];
+};
+export function xmppSignedPreKey(key: any): {
+    tag: string;
+    attrs: {};
+    content: {
+        tag: string;
+        attrs: {};
+        content: any;
+    }[];
+};
+export function xmppPreKey(pair: any, id: any): {
+    tag: string;
+    attrs: {};
+    content: {
+        tag: string;
+        attrs: {};
+        content: any;
+    }[];
+};
+export function extractE2ESessionFromRetryReceipt(receipt: any): {
+    registrationId: number | undefined;
+    identityKey: any;
+    signedPreKey: {
+        keyId: number | undefined;
+        publicKey: any;
+        signature: any;
+    };
+    preKey: {
+        keyId: number | undefined;
+        publicKey: any;
+    } | undefined;
+} | null;
+export function parseAndInjectE2ESessions(node: any, repository: any): Promise<void>;
+export function extractDeviceJids(result: any, myJid: any, myLid: any, excludeZeroDevices: any): {
+    user: any;
+    device: any;
+    domainType: any;
+    server: any;
+}[];
+export function getNextPreKeys({ creds, keys }: {
+    creds: any;
+    keys: any;
+}, count: any): Promise<{
+    update: {
+        nextPreKeyId: number;
+        firstUnuploadedPreKeyId: number;
+    };
+    preKeys: any;
+}>;
+export function getNextPreKeysNode(state: any, count: any): Promise<{
+    update: {
+        nextPreKeyId: number;
+        firstUnuploadedPreKeyId: number;
+    };
+    node: {
+        tag: string;
+        attrs: {
+            xmlns: string;
+            type: string;
+            to: string;
+        };
+        content: {
+            tag: string;
+            attrs: {};
+            content: any;
+        }[];
+    };
+}>;
+//# sourceMappingURL=signal.d.ts.map

package/lib/Utils/signal.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"signal.d.ts","sourceRoot":"","sources":["../../src/Utils/signal.js"],"names":[],"mappings":"AAWO;;;;;;EAKN;AACM;;uCAMN;AACM;;;;EAeN;AACM;;;;;;;;EAQL;AACK;;;;;;;;EAOL;AAEK;;;;;;;;;;;;SAwCN;AACM,qFAqCN;AACM;;;;;IA4BN;AAKM;;;;;;;;;GASN;AACM;;;;;;;;;;;;;;;;;;GAmBN"}

package/lib/Utils/signal.js

@@ -0,0 +1,201 @@
+import { KEY_BUNDLE_TYPE } from '../Defaults/index.js';
+import { assertNodeErrorFree, getBinaryNodeChild, getBinaryNodeChildBuffer, getBinaryNodeChildren, getBinaryNodeChildUInt, getServerFromDomainType, jidDecode, S_WHATSAPP_NET, WAJIDDomains } from '../WABinary/index.js';
+import { Curve, generateSignalPubKey } from './crypto.js';
+import { encodeBigEndian } from './generics.js';
+function chunk(array, size) {
+    const chunks = [];
+    for (let i = 0; i < array.length; i += size) {
+        chunks.push(array.slice(i, i + size));
+    }
+    return chunks;
+}
+export const createSignalIdentity = (wid, accountSignatureKey) => {
+    return {
+        identifier: { name: wid, deviceId: 0 },
+        identifierKey: generateSignalPubKey(accountSignatureKey)
+    };
+};
+export const getPreKeys = async ({ get }, min, limit) => {
+    const idList = [];
+    for (let id = min; id < limit; id++) {
+        idList.push(id.toString());
+    }
+    return get('pre-key', idList);
+};
+export const generateOrGetPreKeys = (creds, range) => {
+    const avaliable = creds.nextPreKeyId - creds.firstUnuploadedPreKeyId;
+    const remaining = range - avaliable;
+    const lastPreKeyId = creds.nextPreKeyId + remaining - 1;
+    const newPreKeys = {};
+    if (remaining > 0) {
+        for (let i = creds.nextPreKeyId; i <= lastPreKeyId; i++) {
+            newPreKeys[i] = Curve.generateKeyPair();
+        }
+    }
+    return {
+        newPreKeys,
+        lastPreKeyId,
+        preKeysRange: [creds.firstUnuploadedPreKeyId, range]
+    };
+};
+export const xmppSignedPreKey = (key) => ({
+    tag: 'skey',
+    attrs: {},
+    content: [
+        { tag: 'id', attrs: {}, content: encodeBigEndian(key.keyId, 3) },
+        { tag: 'value', attrs: {}, content: key.keyPair.public },
+        { tag: 'signature', attrs: {}, content: key.signature }
+    ]
+});
+export const xmppPreKey = (pair, id) => ({
+    tag: 'key',
+    attrs: {},
+    content: [
+        { tag: 'id', attrs: {}, content: encodeBigEndian(id, 3) },
+        { tag: 'value', attrs: {}, content: pair.public }
+    ]
+});
+const isValidUInt = (n) => typeof n === 'number' && Number.isInteger(n);
+export const extractE2ESessionFromRetryReceipt = (receipt) => {
+    const keysNode = getBinaryNodeChild(receipt, 'keys');
+    if (!keysNode)
+        return null;
+    const typeBuf = getBinaryNodeChildBuffer(keysNode, 'type');
+    if (!typeBuf || typeBuf.length !== 1 || typeBuf[0] !== KEY_BUNDLE_TYPE[0])
+        return null;
+    const identity = getBinaryNodeChildBuffer(keysNode, 'identity');
+    const skey = getBinaryNodeChild(keysNode, 'skey');
+    if (!identity || identity.length !== 32 || !skey)
+        return null;
+    const registrationId = getBinaryNodeChildUInt(receipt, 'registration', 4);
+    if (!isValidUInt(registrationId))
+        return null;
+    const signedPubKey = getBinaryNodeChildBuffer(skey, 'value');
+    const signedSig = getBinaryNodeChildBuffer(skey, 'signature');
+    const signedKeyId = getBinaryNodeChildUInt(skey, 'id', 3);
+    if (!signedPubKey || signedPubKey.length !== 32 || !signedSig || !isValidUInt(signedKeyId)) {
+        return null;
+    }
+    const preKeyNode = getBinaryNodeChild(keysNode, 'key');
+    let preKey;
+    if (preKeyNode) {
+        const preKeyPub = getBinaryNodeChildBuffer(preKeyNode, 'value');
+        const preKeyId = getBinaryNodeChildUInt(preKeyNode, 'id', 3);
+        if (!preKeyPub || preKeyPub.length !== 32 || !isValidUInt(preKeyId)) {
+            return null;
+        }
+        preKey = { keyId: preKeyId, publicKey: generateSignalPubKey(preKeyPub) };
+    }
+    return {
+        registrationId,
+        identityKey: generateSignalPubKey(identity),
+        signedPreKey: {
+            keyId: signedKeyId,
+            publicKey: generateSignalPubKey(signedPubKey),
+            signature: signedSig
+        },
+        preKey
+    };
+};
+export const parseAndInjectE2ESessions = async (node, repository) => {
+    const extractKey = (key) => key
+        ? {
+            keyId: getBinaryNodeChildUInt(key, 'id', 3),
+            publicKey: generateSignalPubKey(getBinaryNodeChildBuffer(key, 'value')),
+            signature: getBinaryNodeChildBuffer(key, 'signature')
+        }
+        : undefined;
+    const nodes = getBinaryNodeChildren(getBinaryNodeChild(node, 'list'), 'user');
+    for (const node of nodes) {
+        assertNodeErrorFree(node);
+    }
+    // Most of the work in repository.injectE2ESession is CPU intensive, not IO
+    // So Promise.all doesn't really help here,
+    // but blocks even loop if we're using it inside keys.transaction, and it makes it "sync" actually
+    // This way we chunk it in smaller parts and between those parts we can yield to the event loop
+    // It's rare case when you need to E2E sessions for so many users, but it's possible
+    const chunkSize = 100;
+    const chunks = chunk(nodes, chunkSize);
+    for (const nodesChunk of chunks) {
+        for (const node of nodesChunk) {
+            const signedKey = getBinaryNodeChild(node, 'skey');
+            const key = getBinaryNodeChild(node, 'key');
+            const identity = getBinaryNodeChildBuffer(node, 'identity');
+            const jid = node.attrs.jid;
+            const registrationId = getBinaryNodeChildUInt(node, 'registration', 4);
+            await repository.injectE2ESession({
+                jid,
+                session: {
+                    registrationId: registrationId,
+                    identityKey: generateSignalPubKey(identity),
+                    signedPreKey: extractKey(signedKey),
+                    preKey: extractKey(key)
+                }
+            });
+        }
+    }
+};
+export const extractDeviceJids = (result, myJid, myLid, excludeZeroDevices) => {
+    const { user: myUser, device: myDevice } = jidDecode(myJid);
+    const extracted = [];
+    for (const userResult of result) {
+        const { devices, id } = userResult;
+        const decoded = jidDecode(id), { user, server } = decoded;
+        let { domainType } = decoded;
+        const deviceList = devices?.deviceList;
+        if (!Array.isArray(deviceList))
+            continue;
+        for (const { id: device, keyIndex, isHosted } of deviceList) {
+            if ((!excludeZeroDevices || device !== 0) && // if zero devices are not-excluded, or device is non zero
+                ((myUser !== user && myLid !== user) || myDevice !== device) && // either different user or if me user, not this device
+                (device === 0 || !!keyIndex) // ensure that "key-index" is specified for "non-zero" devices, produces a bad req otherwise
+            ) {
+                if (isHosted) {
+                    domainType = domainType === WAJIDDomains.LID ? WAJIDDomains.HOSTED_LID : WAJIDDomains.HOSTED;
+                }
+                extracted.push({
+                    user,
+                    device,
+                    domainType,
+                    server: getServerFromDomainType(server, domainType)
+                });
+            }
+        }
+    }
+    return extracted;
+};
+/**
+ * get the next N keys for upload or processing
+ * @param count number of pre-keys to get or generate
+ */
+export const getNextPreKeys = async ({ creds, keys }, count) => {
+    const { newPreKeys, lastPreKeyId, preKeysRange } = generateOrGetPreKeys(creds, count);
+    const update = {
+        nextPreKeyId: Math.max(lastPreKeyId + 1, creds.nextPreKeyId),
+        firstUnuploadedPreKeyId: Math.max(creds.firstUnuploadedPreKeyId, lastPreKeyId + 1)
+    };
+    await keys.set({ 'pre-key': newPreKeys });
+    const preKeys = await getPreKeys(keys, preKeysRange[0], preKeysRange[0] + preKeysRange[1]);
+    return { update, preKeys };
+};
+export const getNextPreKeysNode = async (state, count) => {
+    const { creds } = state;
+    const { update, preKeys } = await getNextPreKeys(state, count);
+    const node = {
+        tag: 'iq',
+        attrs: {
+            xmlns: 'encrypt',
+            type: 'set',
+            to: S_WHATSAPP_NET
+        },
+        content: [
+            { tag: 'registration', attrs: {}, content: encodeBigEndian(creds.registrationId) },
+            { tag: 'type', attrs: {}, content: KEY_BUNDLE_TYPE },
+            { tag: 'identity', attrs: {}, content: creds.signedIdentityKey.public },
+            { tag: 'list', attrs: {}, content: Object.keys(preKeys).map(k => xmppPreKey(preKeys[+k], +k)) },
+            xmppSignedPreKey(creds.signedPreKey)
+        ]
+    };
+    return { update, node };
+};
+//# sourceMappingURL=signal.js.map