@celestial-community/baileys 0.0.1-security → 1.0.0

package/lib/Utils/identity-change-handler.js

@@ -0,0 +1,72 @@
+'use strict'
+
+Object.defineProperty(exports, '__esModule', { value: true })
+
+const WABinary_1 = require('../WABinary')
+const generics_1 = require('./generics')
+
+async function handleIdentityChange(node, ctx) {
+    const from = node.attrs.from
+    if (!from) {
+        return { action: 'invalid_notification' }
+    }
+
+    const identityNode = WABinary_1.getBinaryNodeChild(node, 'identity')
+    if (!identityNode) {
+        return { action: 'no_identity_node' }
+    }
+
+    ctx.logger.info({ jid: from }, 'identity changed')
+
+    const decoded = WABinary_1.jidDecode(from)
+    if (decoded?.device && decoded.device !== 0) {
+        ctx.logger.debug(
+            { jid: from, device: decoded.device },
+            'ignoring identity change from companion device'
+        )
+        return { action: 'skipped_companion_device', device: decoded.device }
+    }
+
+    const isSelfPrimary =
+        ctx.meId &&
+        (WABinary_1.areJidsSameUser(from, ctx.meId) ||
+            (ctx.meLid && WABinary_1.areJidsSameUser(from, ctx.meLid)))
+    if (isSelfPrimary) {
+        ctx.logger.info({ jid: from }, 'self primary identity changed')
+        return { action: 'skipped_self_primary' }
+    }
+
+    if (ctx.debounceCache.get(from)) {
+        ctx.logger.debug({ jid: from }, 'skipping identity assert (debounced)')
+        return { action: 'debounced' }
+    }
+
+    ctx.debounceCache.set(from, true)
+
+    const isOfflineNotification = !generics_1.isStringNullOrEmpty(node.attrs.offline)
+    const hasExistingSession = await ctx.validateSession(from)
+
+    if (!hasExistingSession.exists) {
+        ctx.logger.debug({ jid: from }, 'no old session, skipping session refresh')
+        return { action: 'skipped_no_session' }
+    }
+
+    ctx.logger.debug({ jid: from }, 'old session exists, will refresh session')
+
+    if (isOfflineNotification) {
+        ctx.logger.debug({ jid: from }, 'skipping session refresh during offline processing')
+        return { action: 'skipped_offline' }
+    }
+
+    try {
+        await ctx.assertSessions([from], true)
+        return { action: 'session_refreshed' }
+    } catch (error) {
+        ctx.logger.warn({ error, jid: from }, 'failed to assert sessions after identity change')
+        return { action: 'session_refresh_failed', error }
+    }
+}
+
+module.exports = {
+    handleIdentityChange
+}

package/lib/Utils/index.js

@@ -0,0 +1,83 @@
+'use strict'
+var __createBinding =
+    (this && this.__createBinding) ||
+    (Object.create
+        ? function (o, m, k, k2) {
+              if (k2 === undefined) k2 = k
+              var desc = Object.getOwnPropertyDescriptor(m, k)
+              if (!desc || ('get' in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+                  desc = {
+                      enumerable: true,
+                      get: function () {
+                          return m[k]
+                      }
+                  }
+              }
+              Object.defineProperty(o, k2, desc)
+          }
+        : function (o, m, k, k2) {
+              o[k2] = m[k]
+          })
+var __exportStar =
+    (this && this.__exportStar) ||
+    function (m, exports) {
+        for (var p in m)
+            if (p !== 'default' && !Object.prototype.hasOwnProperty.call(exports, p))
+                __createBinding(exports, m, p)
+    }
+Object.defineProperty(exports, '__esModule', { value: true })
+__exportStar(require('./generics'), exports)
+__exportStar(require('./decode-wa-message'), exports)
+__exportStar(require('./messages'), exports)
+__exportStar(require('./messages-media'), exports)
+__exportStar(require('./validate-connection'), exports)
+__exportStar(require('./crypto'), exports)
+__exportStar(require('./signal'), exports)
+__exportStar(require('./noise-handler'), exports)
+__exportStar(require('./history'), exports)
+__exportStar(require('./chat-utils'), exports)
+__exportStar(require('./lt-hash'), exports)
+__exportStar(require('./auth-utils'), exports)
+__exportStar(require('./baileys-event-stream'), exports)
+__exportStar(require('./use-multi-file-auth-state'), exports)
+__exportStar(require('./link-preview'), exports)
+__exportStar(require('./event-buffer'), exports)
+__exportStar(require('./process-message'), exports)
+__exportStar(require('./message-composer'), exports)
+__exportStar(require('./scheduling'), exports)
+__exportStar(require('./anti-delete'), exports)
+__exportStar(require('./interactive-message'), exports)
+__exportStar(require('./chat-control'), exports)
+__exportStar(require('./message-search'), exports)
+__exportStar(require('./vCard'), exports)
+__exportStar(require('./rich-message-utils'), exports)
+__exportStar(require('./status-posting'), exports)
+__exportStar(require('./templates'), exports)
+__exportStar(require('./message-retry-manager'), exports)
+__exportStar(require('./identity-change-handler'), exports)
+__exportStar(require('./use-single-file-auth-state'), exports)
+__exportStar(require('./use-mongo-file-auth-state'), exports)
+__exportStar(require('./logger'), exports)
+__exportStar(require('./rate-limiter'), exports)
+__exportStar(require('./human-entropy'), exports)
+__exportStar(require('./warmup'), exports)
+__exportStar(require('./delivery-tracker'), exports)
+__exportStar(require('./group-operation-guard'), exports)
+__exportStar(require('./device-fingerprint'), exports)
+__exportStar(require('./session-fingerprint'), exports)
+__exportStar(require('./legitimacy-signal-injector'), exports)
+__exportStar(require('./reconnect-throttle'), exports)
+__exportStar(require('./stealth-connect'), exports)
+__exportStar(require('./health-monitor'), exports)
+__exportStar(require('./timelock-guard'), exports)
+__exportStar(require('./reply-ratio-guard'), exports)
+__exportStar(require('./contact-graph-warmer'), exports)
+__exportStar(require('./presence-choreographer'), exports)
+__exportStar(require('./retry-reason-tracker'), exports)
+__exportStar(require('./session-stability'), exports)
+__exportStar(require("./antiban-presets"), exports);
+__exportStar(require("./state-manager"), exports);
+__exportStar(require("./antiban-profiles"), exports);
+__exportStar(require("./anti-ban"), exports);
+__exportStar(require("./pre-key-manager"), exports);
+__exportStar(require("./use-sqlite-auth-state"), exports);

package/lib/Utils/interactive-message.d.ts

@@ -0,0 +1,139 @@
+import { proto } from '../../WAProto/index.js'
+import type { WAMediaUpload, WAMessageContent } from '../Types'
+import type { MediaGenerationOptions } from '../Types/Message'
+
+export interface InteractiveButton {
+    name: string
+    buttonId: string
+    displayText: string
+}
+
+export interface InteractiveButtonReply extends InteractiveButton {
+    index?: number
+}
+
+export interface QuickReplyButton {
+    id: string
+    displayText: string
+}
+
+export interface UrlButton {
+    displayText: string
+    url: string
+    merchantUrl?: string
+}
+
+export interface CallButton {
+    displayText: string
+    phoneNumber: string
+}
+
+export interface CopyButton {
+    displayText: string
+    copyCode: string
+}
+
+export interface ListSection {
+    title: string
+    rows: ListRow[]
+}
+
+export interface ListRow {
+    rowId: string
+    title: string
+    description?: string
+}
+
+export interface InteractiveListMessage {
+    title: string
+    buttonText: string
+    sections: ListSection[]
+    footer?: string
+    description?: string
+}
+
+export interface InteractiveButtonsMessage {
+    title?: string
+    body: string
+    footer?: string
+    buttons: InteractiveButton[]
+    headerImage?: WAMediaUpload
+    headerVideo?: WAMediaUpload
+    headerDocument?: WAMediaUpload & { filename?: string }
+}
+
+export interface TemplateButton {
+    index: number
+    quickReplyButton?: QuickReplyButton
+    urlButton?: UrlButton
+    callButton?: CallButton
+}
+
+export interface TemplateMessage {
+    namespace?: string
+    title?: string
+    body: string
+    footer?: string
+    buttons: TemplateButton[]
+    headerImage?: WAMediaUpload
+    headerVideo?: WAMediaUpload
+    headerLocation?: {
+        latitude: number
+        longitude: number
+        name?: string
+        address?: string
+    }
+}
+
+export declare const generateInteractiveButtonMessage: (content: InteractiveButtonsMessage, options?: MediaGenerationOptions) => Promise<WAMessageContent>
+export declare const generateInteractiveListMessage: (content: InteractiveListMessage) => WAMessageContent
+export declare const generateTemplateMessage: (content: TemplateMessage, options?: MediaGenerationOptions) => Promise<WAMessageContent>
+export declare const generateNativeFlowMessage: (body: string, buttons: Array<{
+    name: string
+    buttonParamsJson: string
+}>, options?: {
+    footer?: string
+    header?: {
+        title?: string
+        subtitle?: string
+        hasMediaAttachment?: boolean
+    }
+}) => WAMessageContent
+export declare const generateCopyCodeButton: (body: string, copyCode: string, displayText?: string, options?: {
+    footer?: string
+}) => WAMessageContent
+export declare const generateUrlButtonMessage: (body: string, buttons: Array<{
+    displayText: string
+    url: string
+    merchantUrl?: string
+}>, options?: {
+    footer?: string
+    title?: string
+}) => WAMessageContent
+export declare const generateQuickReplyButtons: (body: string, buttons: Array<{
+    id: string
+    displayText: string
+}>, options?: {
+    footer?: string
+    title?: string
+}) => WAMessageContent
+export declare const generateCombinedButtons: (body: string, buttons: Array<{
+    type: 'url'
+    displayText: string
+    url: string
+} | {
+    type: 'reply'
+    displayText: string
+    id: string
+} | {
+    type: 'copy'
+    displayText: string
+    copyCode: string
+} | {
+    type: 'call'
+    displayText: string
+    phoneNumber: string
+}>, options?: {
+    footer?: string
+    title?: string
+}) => WAMessageContent

package/lib/Utils/interactive-message.js

@@ -0,0 +1,273 @@
+'use strict'
+
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.generateCombinedButtons =
+    exports.generateQuickReplyButtons =
+    exports.generateUrlButtonMessage =
+    exports.generateCopyCodeButton =
+    exports.generateNativeFlowMessage =
+    exports.generateTemplateMessage =
+    exports.generateInteractiveListMessage =
+    exports.generateInteractiveButtonMessage =
+        void 0
+
+const WAProto_1 = require('../../WAProto')
+const messages_1 = require('./messages')
+
+const generateInteractiveButtonMessage = async (content, options) => {
+    const buttons = content.buttons.map((btn, idx) => ({
+        buttonId: btn.buttonId || `btn-${idx}`,
+        buttonText: { displayText: btn.displayText },
+        type: WAProto_1.proto.Message.ButtonsMessage.Button.Type.RESPONSE,
+        nativeFlowInfo: undefined
+    }))
+
+    const buttonsMessage = {
+        contentText: content.body,
+        footerText: content.footer,
+        buttons,
+        headerType: WAProto_1.proto.Message.ButtonsMessage.HeaderType.EMPTY
+    }
+
+    if (content.headerImage && options) {
+        const media = await (0, messages_1.prepareWAMessageMedia)(
+            { image: content.headerImage },
+            options
+        )
+        buttonsMessage.imageMessage = media.imageMessage
+        buttonsMessage.headerType = WAProto_1.proto.Message.ButtonsMessage.HeaderType.IMAGE
+    } else if (content.headerVideo && options) {
+        const media = await (0, messages_1.prepareWAMessageMedia)(
+            { video: content.headerVideo },
+            options
+        )
+        buttonsMessage.videoMessage = media.videoMessage
+        buttonsMessage.headerType = WAProto_1.proto.Message.ButtonsMessage.HeaderType.VIDEO
+    } else if (content.headerDocument && options) {
+        const media = await (0, messages_1.prepareWAMessageMedia)(
+            {
+                document: content.headerDocument,
+                mimetype: 'application/pdf',
+                fileName: content.headerDocument.filename
+            },
+            options
+        )
+        buttonsMessage.documentMessage = media.documentMessage
+        buttonsMessage.headerType = WAProto_1.proto.Message.ButtonsMessage.HeaderType.DOCUMENT
+    } else if (content.title) {
+        buttonsMessage.text = content.title
+        buttonsMessage.headerType = WAProto_1.proto.Message.ButtonsMessage.HeaderType.TEXT
+    }
+
+    return { buttonsMessage }
+}
+exports.generateInteractiveButtonMessage = generateInteractiveButtonMessage
+
+const generateInteractiveListMessage = (content) => {
+    const sections = content.sections.map((section) => ({
+        title: section.title,
+        rows: section.rows.map((row) => ({
+            rowId: row.rowId,
+            title: row.title,
+            description: row.description || ''
+        }))
+    }))
+
+    return {
+        listMessage: {
+            title: content.title,
+            description: content.description || '',
+            buttonText: content.buttonText,
+            footerText: content.footer || '',
+            listType: WAProto_1.proto.Message.ListMessage.ListType.SINGLE_SELECT,
+            sections
+        }
+    }
+}
+exports.generateInteractiveListMessage = generateInteractiveListMessage
+
+const generateTemplateMessage = async (content, options) => {
+    const hydratedButtons = content.buttons.map((btn) => {
+        const hydratedBtn = {
+            index: btn.index
+        }
+
+        if (btn.quickReplyButton) {
+            hydratedBtn.quickReplyButton = {
+                displayText: btn.quickReplyButton.displayText,
+                id: btn.quickReplyButton.id
+            }
+        } else if (btn.urlButton) {
+            hydratedBtn.urlButton = {
+                displayText: btn.urlButton.displayText,
+                url: btn.urlButton.url
+            }
+        } else if (btn.callButton) {
+            hydratedBtn.callButton = {
+                displayText: btn.callButton.displayText,
+                phoneNumber: btn.callButton.phoneNumber
+            }
+        }
+
+        return hydratedBtn
+    })
+
+    const hydratedTemplate = {
+        hydratedContentText: content.body,
+        hydratedFooterText: content.footer,
+        hydratedButtons
+    }
+
+    if (content.title) {
+        hydratedTemplate.hydratedTitleText = content.title
+    }
+
+    if (content.headerImage && options) {
+        const media = await (0, messages_1.prepareWAMessageMedia)(
+            { image: content.headerImage },
+            options
+        )
+        hydratedTemplate.imageMessage = media.imageMessage
+    } else if (content.headerVideo && options) {
+        const media = await (0, messages_1.prepareWAMessageMedia)(
+            { video: content.headerVideo },
+            options
+        )
+        hydratedTemplate.videoMessage = media.videoMessage
+    } else if (content.headerLocation) {
+        hydratedTemplate.locationMessage = {
+            degreesLatitude: content.headerLocation.latitude,
+            degreesLongitude: content.headerLocation.longitude,
+            name: content.headerLocation.name,
+            address: content.headerLocation.address
+        }
+    }
+
+    return {
+        templateMessage: {
+            hydratedTemplate
+        }
+    }
+}
+exports.generateTemplateMessage = generateTemplateMessage
+
+const generateNativeFlowMessage = (body, buttons, options) => {
+    const nativeFlowButtons = buttons.map((btn, idx) => ({
+        name: btn.name,
+        buttonParamsJson: btn.buttonParamsJson
+    }))
+
+    return {
+        interactiveMessage: {
+            body: { text: body },
+            footer: options?.footer ? { text: options.footer } : undefined,
+            header: options?.header
+                ? {
+                      title: options.header.title,
+                      subtitle: options.header.subtitle,
+                      hasMediaAttachment: options.header.hasMediaAttachment || false
+                  }
+                : undefined,
+            nativeFlowMessage: {
+                buttons: nativeFlowButtons,
+                messageParamsJson: ''
+            }
+        }
+    }
+}
+exports.generateNativeFlowMessage = generateNativeFlowMessage
+
+const generateCopyCodeButton = (body, copyCode, displayText = 'Copy Code', options) => {
+    return (0, exports.generateNativeFlowMessage)(
+        body,
+        [
+            {
+                name: 'cta_copy',
+                buttonParamsJson: JSON.stringify({
+                    display_text: displayText,
+                    copy_code: copyCode
+                })
+            }
+        ],
+        options
+    )
+}
+exports.generateCopyCodeButton = generateCopyCodeButton
+
+const generateUrlButtonMessage = (body, buttons, options) => {
+    const nativeButtons = buttons.map((btn) => ({
+        name: 'cta_url',
+        buttonParamsJson: JSON.stringify({
+            display_text: btn.displayText,
+            url: btn.url,
+            merchant_url: btn.merchantUrl
+        })
+    }))
+
+    return (0, exports.generateNativeFlowMessage)(body, nativeButtons, {
+        footer: options?.footer,
+        header: options?.title ? { title: options.title } : undefined
+    })
+}
+exports.generateUrlButtonMessage = generateUrlButtonMessage
+
+const generateQuickReplyButtons = (body, buttons, options) => {
+    const nativeButtons = buttons.map((btn) => ({
+        name: 'quick_reply',
+        buttonParamsJson: JSON.stringify({
+            display_text: btn.displayText,
+            id: btn.id
+        })
+    }))
+
+    return (0, exports.generateNativeFlowMessage)(body, nativeButtons, {
+        footer: options?.footer,
+        header: options?.title ? { title: options.title } : undefined
+    })
+}
+exports.generateQuickReplyButtons = generateQuickReplyButtons
+
+const generateCombinedButtons = (body, buttons, options) => {
+    const nativeButtons = buttons.map((btn) => {
+        switch (btn.type) {
+            case 'url':
+                return {
+                    name: 'cta_url',
+                    buttonParamsJson: JSON.stringify({
+                        display_text: btn.displayText,
+                        url: btn.url
+                    })
+                }
+            case 'reply':
+                return {
+                    name: 'quick_reply',
+                    buttonParamsJson: JSON.stringify({
+                        display_text: btn.displayText,
+                        id: btn.id
+                    })
+                }
+            case 'copy':
+                return {
+                    name: 'cta_copy',
+                    buttonParamsJson: JSON.stringify({
+                        display_text: btn.displayText,
+                        copy_code: btn.copyCode
+                    })
+                }
+            case 'call':
+                return {
+                    name: 'cta_call',
+                    buttonParamsJson: JSON.stringify({
+                        display_text: btn.displayText,
+                        phone_number: btn.phoneNumber
+                    })
+                }
+        }
+    })
+
+    return (0, exports.generateNativeFlowMessage)(body, nativeButtons, {
+        footer: options?.footer,
+        header: options?.title ? { title: options.title } : undefined
+    })
+}
+exports.generateCombinedButtons = generateCombinedButtons