@celestial-community/baileys 0.0.1-security → 1.0.0

package/lib/WAM/encode.js

@@ -0,0 +1,149 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.encodeWAM = void 0
+const constants_1 = require('./constants')
+const getHeaderBitLength = (key) => (key < 256 ? 2 : 3)
+const encodeWAM = (binaryInfo) => {
+    binaryInfo.buffer = []
+    encodeWAMHeader(binaryInfo)
+    encodeEvents(binaryInfo)
+    console.log(binaryInfo.buffer)
+    const totalSize = binaryInfo.buffer.map((a) => a.length).reduce((a, b) => a + b)
+    const buffer = Buffer.alloc(totalSize)
+    let offset = 0
+    binaryInfo.buffer.forEach((buffer_) => {
+        buffer_.copy(buffer, offset)
+        offset += buffer_.length
+    })
+    return buffer
+}
+exports.encodeWAM = encodeWAM
+function encodeWAMHeader(binaryInfo) {
+    const headerBuffer = Buffer.alloc(8) // starting buffer
+    headerBuffer.write('WAM', 0, 'utf8')
+    headerBuffer.writeUInt8(binaryInfo.protocolVersion, 3)
+    headerBuffer.writeUInt8(1, 4) //random flag
+    headerBuffer.writeUInt16BE(binaryInfo.sequence, 5)
+    headerBuffer.writeUInt8(0, 7) // regular channel
+    binaryInfo.buffer.push(headerBuffer)
+}
+function encodeGlobalAttributes(binaryInfo, globals) {
+    for (const [key, _value] of Object.entries(globals)) {
+        const id = constants_1.WEB_GLOBALS.find(
+            (a) => (a === null || a === void 0 ? void 0 : a.name) === key
+        ).id
+        let value = _value
+        if (typeof value === 'boolean') {
+            value = value ? 1 : 0
+        }
+        binaryInfo.buffer.push(serializeData(id, value, constants_1.FLAG_GLOBAL))
+    }
+}
+function encodeEvents(binaryInfo) {
+    for (const [name, { props, globals }] of binaryInfo.events.map((a) => Object.entries(a)[0])) {
+        encodeGlobalAttributes(binaryInfo, globals)
+        const event = constants_1.WEB_EVENTS.find((a) => a.name === name)
+        const props_ = Object.entries(props)
+        let extended = false
+        for (const [, value] of props_) {
+            extended || (extended = value !== null)
+        }
+        const eventFlag = extended
+            ? constants_1.FLAG_EVENT
+            : constants_1.FLAG_EVENT | constants_1.FLAG_EXTENDED
+        binaryInfo.buffer.push(serializeData(event.id, -event.weight, eventFlag))
+        for (let i = 0; i < props_.length; i++) {
+            const [key, _value] = props_[i]
+            const id = event.props[key][0]
+            extended = i < props_.length - 1
+            let value = _value
+            if (typeof value === 'boolean') {
+                value = value ? 1 : 0
+            }
+            const fieldFlag = extended
+                ? constants_1.FLAG_EVENT
+                : constants_1.FLAG_FIELD | constants_1.FLAG_EXTENDED
+            binaryInfo.buffer.push(serializeData(id, value, fieldFlag))
+        }
+    }
+}
+function serializeData(key, value, flag) {
+    const bufferLength = getHeaderBitLength(key)
+    let buffer
+    let offset = 0
+    if (value === null) {
+        if (flag === constants_1.FLAG_GLOBAL) {
+            buffer = Buffer.alloc(bufferLength)
+            offset = serializeHeader(buffer, offset, key, flag)
+            return buffer
+        }
+    } else if (typeof value === 'number' && Number.isInteger(value)) {
+        // is number
+        if (value === 0 || value === 1) {
+            buffer = Buffer.alloc(bufferLength)
+            offset = serializeHeader(buffer, offset, key, flag | ((value + 1) << 4))
+            return buffer
+        } else if (-128 <= value && value < 128) {
+            buffer = Buffer.alloc(bufferLength + 1)
+            offset = serializeHeader(buffer, offset, key, flag | (3 << 4))
+            buffer.writeInt8(value, offset)
+            return buffer
+        } else if (-32768 <= value && value < 32768) {
+            buffer = Buffer.alloc(bufferLength + 2)
+            offset = serializeHeader(buffer, offset, key, flag | (4 << 4))
+            buffer.writeInt16LE(value, offset)
+            return buffer
+        } else if (-2147483648 <= value && value < 2147483648) {
+            buffer = Buffer.alloc(bufferLength + 4)
+            offset = serializeHeader(buffer, offset, key, flag | (5 << 4))
+            buffer.writeInt32LE(value, offset)
+            return buffer
+        } else {
+            buffer = Buffer.alloc(bufferLength + 8)
+            offset = serializeHeader(buffer, offset, key, flag | (7 << 4))
+            buffer.writeDoubleLE(value, offset)
+            return buffer
+        }
+    } else if (typeof value === 'number') {
+        // is float
+        buffer = Buffer.alloc(bufferLength + 8)
+        offset = serializeHeader(buffer, offset, key, flag | (7 << 4))
+        buffer.writeDoubleLE(value, offset)
+        return buffer
+    } else if (typeof value === 'string') {
+        // is string
+        const utf8Bytes = Buffer.byteLength(value, 'utf8')
+        if (utf8Bytes < 256) {
+            buffer = Buffer.alloc(bufferLength + 1 + utf8Bytes)
+            offset = serializeHeader(buffer, offset, key, flag | (8 << 4))
+            buffer.writeUint8(utf8Bytes, offset++)
+        } else if (utf8Bytes < 65536) {
+            buffer = Buffer.alloc(bufferLength + 2 + utf8Bytes)
+            offset = serializeHeader(buffer, offset, key, flag | (9 << 4))
+            buffer.writeUInt16LE(utf8Bytes, offset)
+            offset += 2
+        } else {
+            buffer = Buffer.alloc(bufferLength + 4 + utf8Bytes)
+            offset = serializeHeader(buffer, offset, key, flag | (10 << 4))
+            buffer.writeUInt32LE(utf8Bytes, offset)
+            offset += 4
+        }
+        buffer.write(value, offset, 'utf8')
+        return buffer
+    }
+    throw 'missing'
+}
+function serializeHeader(buffer, offset, key, flag) {
+    if (key < 256) {
+        buffer.writeUInt8(flag, offset)
+        offset += 1
+        buffer.writeUInt8(key, offset)
+        offset += 1
+    } else {
+        buffer.writeUInt8(flag | constants_1.FLAG_BYTE, offset)
+        offset += 1
+        buffer.writeUInt16LE(key, offset)
+        offset += 2
+    }
+    return offset
+}

package/lib/WAM/index.js

@@ -0,0 +1,32 @@
+'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) {
+              if (k2 === undefined) k2 = k
+              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('./constants'), exports)
+__exportStar(require('./encode'), exports)
+__exportStar(require('./BinaryInfo'), exports)

package/lib/WAUSync/Protocols/USyncContactProtocol.js

@@ -0,0 +1,37 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncContactProtocol = void 0
+const WABinary_1 = require('../../WABinary')
+class USyncContactProtocol {
+    constructor() {
+        this.name = 'contact'
+    }
+    getQueryElement() {
+        return {
+            tag: 'contact',
+            attrs: {}
+        }
+    }
+    getUserElement(user) {
+        //TODO: Implement type / username fields (not yet supported)
+        return {
+            tag: 'contact',
+            attrs: {},
+            content: user.phone
+        }
+    }
+    parser(node) {
+        var _a
+        if (node.tag === 'contact') {
+            ;(0, WABinary_1.assertNodeErrorFree)(node)
+            return (
+                ((_a = node === null || node === void 0 ? void 0 : node.attrs) === null ||
+                _a === void 0
+                    ? void 0
+                    : _a.type) === 'in'
+            )
+        }
+        return false
+    }
+}
+exports.USyncContactProtocol = USyncContactProtocol

package/lib/WAUSync/Protocols/USyncDeviceProtocol.js

@@ -0,0 +1,71 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncDeviceProtocol = void 0
+const WABinary_1 = require('../../WABinary')
+class USyncDeviceProtocol {
+    constructor() {
+        this.name = 'devices'
+    }
+    getQueryElement() {
+        return {
+            tag: 'devices',
+            attrs: {
+                version: '2'
+            }
+        }
+    }
+    getUserElement(/* user: USyncUser */) {
+        //TODO: Implement device phashing, ts and expectedTs
+        //TODO: if all are not present, return null <- current behavior
+        //TODO: otherwise return a node w tag 'devices' w those as attrs
+        return null
+    }
+    parser(node) {
+        const deviceList = []
+        let keyIndex = undefined
+        if (node.tag === 'devices') {
+            ;(0, WABinary_1.assertNodeErrorFree)(node)
+            const deviceListNode = (0, WABinary_1.getBinaryNodeChild)(node, 'device-list')
+            const keyIndexNode = (0, WABinary_1.getBinaryNodeChild)(node, 'key-index-list')
+            if (
+                Array.isArray(
+                    deviceListNode === null || deviceListNode === void 0
+                        ? void 0
+                        : deviceListNode.content
+                )
+            ) {
+                for (const { tag, attrs } of deviceListNode.content) {
+                    const id = +attrs.id
+                    const keyIndex = +attrs['key-index']
+                    if (tag === 'device') {
+                        deviceList.push({
+                            id,
+                            keyIndex,
+                            isHosted: !!(attrs['is_hosted'] && attrs['is_hosted'] === 'true')
+                        })
+                    }
+                }
+            }
+            if (
+                (keyIndexNode === null || keyIndexNode === void 0 ? void 0 : keyIndexNode.tag) ===
+                'key-index-list'
+            ) {
+                keyIndex = {
+                    timestamp: +keyIndexNode.attrs['ts'],
+                    signedKeyIndex:
+                        keyIndexNode === null || keyIndexNode === void 0
+                            ? void 0
+                            : keyIndexNode.content,
+                    expectedTimestamp: keyIndexNode.attrs['expected_ts']
+                        ? +keyIndexNode.attrs['expected_ts']
+                        : undefined
+                }
+            }
+        }
+        return {
+            deviceList,
+            keyIndex
+        }
+    }
+}
+exports.USyncDeviceProtocol = USyncDeviceProtocol

package/lib/WAUSync/Protocols/USyncDisappearingModeProtocol.js

@@ -0,0 +1,32 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncDisappearingModeProtocol = void 0
+const WABinary_1 = require('../../WABinary')
+class USyncDisappearingModeProtocol {
+    constructor() {
+        this.name = 'disappearing_mode'
+    }
+    getQueryElement() {
+        return {
+            tag: 'disappearing_mode',
+            attrs: {}
+        }
+    }
+    getUserElement() {
+        return null
+    }
+    parser(node) {
+        if (node.tag === 'status') {
+            ;(0, WABinary_1.assertNodeErrorFree)(node)
+            const duration = +(node === null || node === void 0 ? void 0 : node.attrs.duration)
+            const setAt = new Date(
+                +((node === null || node === void 0 ? void 0 : node.attrs.t) || 0) * 1000
+            )
+            return {
+                duration,
+                setAt
+            }
+        }
+    }
+}
+exports.USyncDisappearingModeProtocol = USyncDisappearingModeProtocol

package/lib/WAUSync/Protocols/USyncStatusProtocol.js

@@ -0,0 +1,49 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncStatusProtocol = void 0
+const WABinary_1 = require('../../WABinary')
+class USyncStatusProtocol {
+    constructor() {
+        this.name = 'status'
+    }
+    getQueryElement() {
+        return {
+            tag: 'status',
+            attrs: {}
+        }
+    }
+    getUserElement() {
+        return null
+    }
+    parser(node) {
+        var _a, _b, _c
+        if (node.tag === 'status') {
+            ;(0, WABinary_1.assertNodeErrorFree)(node)
+            let status =
+                (_b =
+                    (_a = node === null || node === void 0 ? void 0 : node.content) === null ||
+                    _a === void 0
+                        ? void 0
+                        : _a.toString()) !== null && _b !== void 0
+                    ? _b
+                    : null
+            const setAt = new Date(
+                +((node === null || node === void 0 ? void 0 : node.attrs.t) || 0) * 1000
+            )
+            if (!status) {
+                if (+((_c = node.attrs) === null || _c === void 0 ? void 0 : _c.code) === 401) {
+                    status = ''
+                } else {
+                    status = null
+                }
+            } else if (typeof status === 'string' && status.length === 0) {
+                status = null
+            }
+            return {
+                status,
+                setAt
+            }
+        }
+    }
+}
+exports.USyncStatusProtocol = USyncStatusProtocol

package/lib/WAUSync/Protocols/USyncUsernameProtocol.js

@@ -0,0 +1,23 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncUsernameProtocol = void 0
+
+class USyncUsernameProtocol {
+    name() {
+        return 'username'
+    }
+    getQueryDetails() {
+        return { tag: 'username', attrs: {} }
+    }
+    getUserDetails(user) {
+        if (!user.username) return []
+        return [{ tag: 'username', attrs: {}, content: user.username }]
+    }
+    parseUserResult(node) {
+        const usernameNode = node.content?.find?.((n) => n.tag === 'username')
+        if (!usernameNode) return {}
+        return { username: usernameNode.content?.toString?.() || usernameNode.attrs?.value }
+    }
+}
+
+exports.USyncUsernameProtocol = USyncUsernameProtocol

package/lib/WAUSync/Protocols/UsyncBotProfileProtocol.js

@@ -0,0 +1,58 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncBotProfileProtocol = void 0
+const WABinary_1 = require('../../WABinary')
+class USyncBotProfileProtocol {
+    constructor() {
+        this.name = 'bot'
+    }
+    getQueryElement() {
+        return {
+            tag: 'bot',
+            attrs: {},
+            content: [{ tag: 'profile', attrs: { v: '1' } }]
+        }
+    }
+    getUserElement(user) {
+        return {
+            tag: 'bot',
+            attrs: {},
+            content: [{ tag: 'profile', attrs: { persona_id: user.personaId } }]
+        }
+    }
+    parser(node) {
+        const botNode = (0, WABinary_1.getBinaryNodeChild)(node, 'bot')
+        const profile = (0, WABinary_1.getBinaryNodeChild)(botNode, 'profile')
+        const commandsNode = (0, WABinary_1.getBinaryNodeChild)(profile, 'commands')
+        const promptsNode = (0, WABinary_1.getBinaryNodeChild)(profile, 'prompts')
+        const commands = []
+        const prompts = []
+        for (const command of (0, WABinary_1.getBinaryNodeChildren)(commandsNode, 'command')) {
+            commands.push({
+                name: (0, WABinary_1.getBinaryNodeChildString)(command, 'name'),
+                description: (0, WABinary_1.getBinaryNodeChildString)(command, 'description')
+            })
+        }
+        for (const prompt of (0, WABinary_1.getBinaryNodeChildren)(promptsNode, 'prompt')) {
+            prompts.push(
+                `${(0, WABinary_1.getBinaryNodeChildString)(prompt, 'emoji')} ${(0, WABinary_1.getBinaryNodeChildString)(prompt, 'text')}`
+            )
+        }
+        return {
+            isDefault: !!(0, WABinary_1.getBinaryNodeChild)(profile, 'default'),
+            jid: node.attrs.jid,
+            name: (0, WABinary_1.getBinaryNodeChildString)(profile, 'name'),
+            attributes: (0, WABinary_1.getBinaryNodeChildString)(profile, 'attributes'),
+            description: (0, WABinary_1.getBinaryNodeChildString)(profile, 'description'),
+            category: (0, WABinary_1.getBinaryNodeChildString)(profile, 'category'),
+            personaId: profile.attrs['persona_id'],
+            commandsDescription: (0, WABinary_1.getBinaryNodeChildString)(
+                commandsNode,
+                'description'
+            ),
+            commands,
+            prompts
+        }
+    }
+}
+exports.USyncBotProfileProtocol = USyncBotProfileProtocol

package/lib/WAUSync/Protocols/UsyncLIDProtocol.js

@@ -0,0 +1,24 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncLIDProtocol = void 0
+class USyncLIDProtocol {
+    constructor() {
+        this.name = 'lid'
+    }
+    getQueryElement() {
+        return {
+            tag: 'lid',
+            attrs: {}
+        }
+    }
+    getUserElement() {
+        return null
+    }
+    parser(node) {
+        if (node.tag === 'lid') {
+            return node.attrs.val
+        }
+        return null
+    }
+}
+exports.USyncLIDProtocol = USyncLIDProtocol

package/lib/WAUSync/Protocols/index.js

@@ -0,0 +1,34 @@
+'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) {
+              if (k2 === undefined) k2 = k
+              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('./USyncDeviceProtocol'), exports)
+__exportStar(require('./USyncContactProtocol'), exports)
+__exportStar(require('./USyncStatusProtocol'), exports)
+__exportStar(require('./USyncDisappearingModeProtocol'), exports)
+__exportStar(require('./USyncUsernameProtocol'), exports)

package/lib/WAUSync/USyncQuery.js

@@ -0,0 +1,99 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncQuery = void 0
+const WABinary_1 = require('../WABinary')
+const UsyncBotProfileProtocol_1 = require('./Protocols/UsyncBotProfileProtocol')
+const UsyncLIDProtocol_1 = require('./Protocols/UsyncLIDProtocol')
+const Protocols_1 = require('./Protocols')
+class USyncQuery {
+    constructor() {
+        this.protocols = []
+        this.users = []
+        this.context = 'interactive'
+        this.mode = 'query'
+    }
+    withMode(mode) {
+        this.mode = mode
+        return this
+    }
+    withContext(context) {
+        this.context = context
+        return this
+    }
+    withUser(user) {
+        this.users.push(user)
+        return this
+    }
+    parseUSyncQueryResult(result) {
+        if (result.attrs.type !== 'result') {
+            return
+        }
+        const protocolMap = Object.fromEntries(
+            this.protocols.map((protocol) => {
+                return [protocol.name, protocol.parser]
+            })
+        )
+        const queryResult = {
+            // TODO: implement errors etc.
+            list: [],
+            sideList: []
+        }
+        const usyncNode = (0, WABinary_1.getBinaryNodeChild)(result, 'usync')
+        //TODO: implement error backoff, refresh etc.
+        //TODO: see if there are any errors in the result node
+        //const resultNode = getBinaryNodeChild(usyncNode, 'result')
+        const listNode = (0, WABinary_1.getBinaryNodeChild)(usyncNode, 'list')
+        if (
+            Array.isArray(listNode === null || listNode === void 0 ? void 0 : listNode.content) &&
+            typeof listNode !== 'undefined'
+        ) {
+            queryResult.list = listNode.content.map((node) => {
+                const id = node === null || node === void 0 ? void 0 : node.attrs.jid
+                const data = Array.isArray(node === null || node === void 0 ? void 0 : node.content)
+                    ? Object.fromEntries(
+                          node.content
+                              .map((content) => {
+                                  const protocol = content.tag
+                                  const parser = protocolMap[protocol]
+                                  if (parser) {
+                                      return [protocol, parser(content)]
+                                  } else {
+                                      return [protocol, null]
+                                  }
+                              })
+                              .filter(([, b]) => b !== null)
+                      )
+                    : {}
+                return { ...data, id }
+            })
+        }
+        //TODO: implement side list
+        //const sideListNode = getBinaryNodeChild(usyncNode, 'side_list')
+        return queryResult
+    }
+    withDeviceProtocol() {
+        this.protocols.push(new Protocols_1.USyncDeviceProtocol())
+        return this
+    }
+    withContactProtocol() {
+        this.protocols.push(new Protocols_1.USyncContactProtocol())
+        return this
+    }
+    withStatusProtocol() {
+        this.protocols.push(new Protocols_1.USyncStatusProtocol())
+        return this
+    }
+    withDisappearingModeProtocol() {
+        this.protocols.push(new Protocols_1.USyncDisappearingModeProtocol())
+        return this
+    }
+    withBotProfileProtocol() {
+        this.protocols.push(new UsyncBotProfileProtocol_1.USyncBotProfileProtocol())
+        return this
+    }
+    withLIDProtocol() {
+        this.protocols.push(new UsyncLIDProtocol_1.USyncLIDProtocol())
+        return this
+    }
+}
+exports.USyncQuery = USyncQuery

package/lib/WAUSync/USyncUser.js

@@ -0,0 +1,26 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.USyncUser = void 0
+class USyncUser {
+    withId(id) {
+        this.id = id
+        return this
+    }
+    withLid(lid) {
+        this.lid = lid
+        return this
+    }
+    withPhone(phone) {
+        this.phone = phone
+        return this
+    }
+    withType(type) {
+        this.type = type
+        return this
+    }
+    withPersonaId(personaId) {
+        this.personaId = personaId
+        return this
+    }
+}
+exports.USyncUser = USyncUser

package/lib/WAUSync/USyncUsernameProtocol.d.ts

@@ -0,0 +1,9 @@
+import { USyncQueryProtocol } from '../../Types/USync'
+import { BinaryNode } from '../../WABinary'
+
+export declare class USyncUsernameProtocol implements USyncQueryProtocol {
+    name: string
+    getQueryElement(): BinaryNode
+    getUserElement(user: any): BinaryNode | null
+    parser(node: BinaryNode): string | null
+}