@yemo-dev/yebail 1.0.0

package/lib/Signal/lid-mapping.js

@@ -0,0 +1,148 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LIDMappingStore = void 0;
+const WABinary_1 = require("../WABinary");
+class LIDMappingStore {
+    constructor(keys, logger, pnToLIDFunc) {
+        this.mappingCache = new Map();
+        this.keys = keys;
+        this.pnToLIDFunc = pnToLIDFunc;
+        this.logger = logger || { trace: () => { }, debug: () => { }, warn: () => { } };
+    }
+    async storeLIDPNMappings(pairs) {
+        const pairMap = {};
+        for (const { lid, pn } of pairs) {
+            if (!(((0, WABinary_1.isLidUser)(lid) && (0, WABinary_1.isPnUser)(pn)) || ((0, WABinary_1.isPnUser)(lid) && (0, WABinary_1.isLidUser)(pn)))) {
+                this.logger.warn(`Invalid LID-PN mapping: ${lid}, ${pn}`);
+                continue;
+            }
+            const lidDecoded = (0, WABinary_1.jidDecode)(lid);
+            const pnDecoded = (0, WABinary_1.jidDecode)(pn);
+            if (!lidDecoded || !pnDecoded)
+                return;
+            const pnUser = pnDecoded.user;
+            const lidUser = lidDecoded.user;
+            let existingLidUser = this.mappingCache.get(`pn:${pnUser}`);
+            if (!existingLidUser) {
+                const stored = await this.keys.get('lid-mapping', [pnUser]);
+                existingLidUser = stored[pnUser];
+                if (existingLidUser) {
+                    this.mappingCache.set(`pn:${pnUser}`, existingLidUser);
+                    this.mappingCache.set(`lid:${existingLidUser}`, pnUser);
+                }
+            }
+            if (existingLidUser === lidUser) {
+                this.logger.debug({ pnUser, lidUser }, 'LID mapping already exists, skipping');
+                continue;
+            }
+            pairMap[pnUser] = lidUser;
+        }
+        if (!Object.keys(pairMap).length)
+            return;
+        await this.keys.transaction(async () => {
+            for (const [pnUser, lidUser] of Object.entries(pairMap)) {
+                await this.keys.set({
+                    'lid-mapping': {
+                        [pnUser]: lidUser,
+                        [`${lidUser}_reverse`]: pnUser
+                    }
+                });
+                this.mappingCache.set(`pn:${pnUser}`, lidUser);
+                this.mappingCache.set(`lid:${lidUser}`, pnUser);
+            }
+        }, 'lid-mapping');
+    }
+    async getLIDForPN(pn) {
+        const rows = await this.getLIDsForPNs([pn]);
+        return rows?.[0]?.lid || null;
+    }
+    async getLIDsForPNs(pns) {
+        const usyncFetch = {};
+        const successfulPairs = {};
+        for (const pn of pns) {
+            if (!(0, WABinary_1.isPnUser)(pn) && !(0, WABinary_1.isHostedPnUser)(pn))
+                continue;
+            const decoded = (0, WABinary_1.jidDecode)(pn);
+            if (!decoded)
+                continue;
+            const pnUser = decoded.user;
+            let lidUser = this.mappingCache.get(`pn:${pnUser}`);
+            if (!lidUser) {
+                const stored = await this.keys.get('lid-mapping', [pnUser]);
+                lidUser = stored[pnUser];
+                if (lidUser) {
+                    this.mappingCache.set(`pn:${pnUser}`, lidUser);
+                    this.mappingCache.set(`lid:${lidUser}`, pnUser);
+                }
+                else {
+                    const device = decoded.device || 0;
+                    let normalizedPn = (0, WABinary_1.jidNormalizedUser)(pn);
+                    if ((0, WABinary_1.isHostedPnUser)(normalizedPn)) {
+                        normalizedPn = `${pnUser}@s.whatsapp.net`;
+                    }
+                    if (!usyncFetch[normalizedPn]) {
+                        usyncFetch[normalizedPn] = [device];
+                    }
+                    else {
+                        usyncFetch[normalizedPn].push(device);
+                    }
+                    continue;
+                }
+            }
+            lidUser = lidUser.toString();
+            if (!lidUser) {
+                this.logger.warn(`Invalid or empty LID user for PN ${pn}`);
+                return null;
+            }
+            const pnDevice = decoded.device !== undefined ? decoded.device : 0;
+            const deviceSpecificLid = `${lidUser}${pnDevice ? `:${pnDevice}` : ''}@${decoded.server === 'hosted' ? 'hosted.lid' : 'lid'}`;
+            successfulPairs[pn] = { lid: deviceSpecificLid, pn };
+        }
+        if (Object.keys(usyncFetch).length > 0) {
+            const result = await this.pnToLIDFunc?.(Object.keys(usyncFetch));
+            if (result && result.length > 0) {
+                await this.storeLIDPNMappings(result);
+                for (const pair of result) {
+                    const pnDecoded = (0, WABinary_1.jidDecode)(pair.pn);
+                    const pnUser = pnDecoded?.user;
+                    if (!pnUser)
+                        continue;
+                    const lidUser = (0, WABinary_1.jidDecode)(pair.lid)?.user;
+                    if (!lidUser)
+                        continue;
+                    for (const device of usyncFetch[pair.pn] || [0]) {
+                        const deviceSpecificLid = `${lidUser}${device ? `:${device}` : ''}@${device === 99 ? 'hosted.lid' : 'lid'}`;
+                        const deviceSpecificPn = `${pnUser}${device ? `:${device}` : ''}@${device === 99 ? 'hosted' : 's.whatsapp.net'}`;
+                        successfulPairs[deviceSpecificPn] = { lid: deviceSpecificLid, pn: deviceSpecificPn };
+                    }
+                }
+            }
+            else {
+                return null;
+            }
+        }
+        return Object.values(successfulPairs);
+    }
+    async getPNForLID(lid) {
+        if (!(0, WABinary_1.isLidUser)(lid) && !(0, WABinary_1.isHostedLidUser)(lid))
+            return null;
+        const decoded = (0, WABinary_1.jidDecode)(lid);
+        if (!decoded)
+            return null;
+        const lidUser = decoded.user;
+        let pnUser = this.mappingCache.get(`lid:${lidUser}`);
+        if (!pnUser || typeof pnUser !== 'string') {
+            const stored = await this.keys.get('lid-mapping', [`${lidUser}_reverse`]);
+            pnUser = stored[`${lidUser}_reverse`];
+            if (!pnUser || typeof pnUser !== 'string') {
+                this.logger.trace(`No reverse mapping found for LID user: ${lidUser}`);
+                return null;
+            }
+            this.mappingCache.set(`lid:${lidUser}`, pnUser);
+        }
+        const lidDevice = decoded.device !== undefined ? decoded.device : 0;
+        const pnJid = `${pnUser}:${lidDevice}@${decoded.domainType === WABinary_1.WAJIDDomains.HOSTED_LID ? 'hosted' : 's.whatsapp.net'}`;
+        return pnJid;
+    }
+}
+exports.LIDMappingStore = LIDMappingStore;

package/lib/Socket/Client/index.js

@@ -0,0 +1,18 @@
+"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("./types"), exports);
+__exportStar(require("./websocket"), exports);

package/lib/Socket/Client/types.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AbstractSocketClient = void 0;
+const events_1 = require("events");
+class AbstractSocketClient extends events_1.EventEmitter {
+    constructor(url, config) {
+        super();
+        this.url = url;
+        this.config = config;
+        this.setMaxListeners(0);
+    }
+}
+exports.AbstractSocketClient = AbstractSocketClient;

package/lib/Socket/Client/websocket.js

@@ -0,0 +1,72 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebSocketClient = void 0;
+const ws_1 = __importDefault(require("ws"));
+const Defaults_1 = require("../../Defaults");
+const types_1 = require("./types");
+class WebSocketClient extends types_1.AbstractSocketClient {
+    constructor() {
+        super(...arguments);
+        this.socket = null;
+    }
+    get isOpen() {
+        var _a;
+        return ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.OPEN;
+    }
+    get isClosed() {
+        var _a;
+        return this.socket === null || ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.CLOSED;
+    }
+    get isClosing() {
+        var _a;
+        return this.socket === null || ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.CLOSING;
+    }
+    get isConnecting() {
+        var _a;
+        return ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.CONNECTING;
+    }
+    async connect() {
+        var _a, _b;
+        if (this.socket) {
+            return;
+        }
+        this.socket = new ws_1.default(this.url, {
+            origin: Defaults_1.DEFAULT_ORIGIN,
+            headers: (_a = this.config.options) === null || _a === void 0 ? void 0 : _a.headers,
+            handshakeTimeout: this.config.connectTimeoutMs,
+            timeout: this.config.connectTimeoutMs,
+            agent: this.config.agent,
+        });
+        this.socket.setMaxListeners(0);
+        const events = ['close', 'error', 'upgrade', 'message', 'open', 'ping', 'pong', 'unexpected-response'];
+        for (const event of events) {
+            (_b = this.socket) === null || _b === void 0 ? void 0 : _b.on(event, (...args) => this.emit(event, ...args));
+        }
+    }
+    async close() {
+        if (!this.socket) {
+            return;
+        }
+        this.socket.close();
+        this.socket = null;
+    }
+    async restart() {
+        if (this.socket) {
+            await new Promise(resolve => {
+                this.socket.once('close', resolve);
+                this.socket.terminate();
+            });
+            this.socket = null;
+        }
+        await this.connect();
+    }
+    send(str, cb) {
+        var _a;
+        (_a = this.socket) === null || _a === void 0 ? void 0 : _a.send(str, cb);
+        return true;
+    }
+}
+exports.WebSocketClient = WebSocketClient;

package/lib/Socket/business.js

@@ -0,0 +1,368 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.makeBusinessSocket = void 0;
+const business_1 = require("../Utils/business");
+const WABinary_1 = require("../WABinary");
+const generic_utils_1 = require("../WABinary/generic-utils");
+const messages_recv_1 = require("./messages-recv");
+const messages_media_1 = require("../Utils/messages-media");
+const makeBusinessSocket = (config) => {
+    const sock = (0, messages_recv_1.makeMessagesRecvSocket)(config);
+    const { authState, query, waUploadToServer } = sock;
+    const updateBussinesProfile = async (args) => {
+        const node = [];
+        const simpleFields = ['address', 'email', 'description'];
+        for (const key of simpleFields) {
+            if (args[key] !== undefined && args[key] !== null) {
+                node.push({ tag: key, attrs: {}, content: args[key] });
+            }
+        }
+        if (args.websites !== undefined) {
+            node.push(...args.websites.map(website => ({ tag: 'website', attrs: {}, content: website })));
+        }
+        if (args.hours !== undefined) {
+            node.push({
+                tag: 'business_hours',
+                attrs: { timezone: args.hours.timezone },
+                content: args.hours.days.map(dayConfig => {
+                    const base = {
+                        tag: 'business_hours_config',
+                        attrs: {
+                            day_of_week: dayConfig.day,
+                            mode: dayConfig.mode
+                        }
+                    };
+                    if (dayConfig.mode === 'specific_hours') {
+                        return {
+                            ...base,
+                            attrs: {
+                                ...base.attrs,
+                                open_time: dayConfig.openTimeInMinutes,
+                                close_time: dayConfig.closeTimeInMinutes
+                            }
+                        };
+                    }
+                    return base;
+                })
+            });
+        }
+        return query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'set',
+                xmlns: 'w:biz'
+            },
+            content: [
+                {
+                    tag: 'business_profile',
+                    attrs: { v: '3', mutation_type: 'delta' },
+                    content: node
+                }
+            ]
+        });
+    };
+    const updateCoverPhoto = async (photo) => {
+        const { fileSha256, filePath } = await (0, messages_media_1.getRawMediaUploadData)(photo, 'biz-cover-photo', config.logger);
+        const fileSha256B64 = fileSha256.toString('base64');
+        const { meta_hmac, fbid, ts } = await waUploadToServer(filePath, {
+            fileEncSha256B64: fileSha256B64,
+            mediaType: 'biz-cover-photo'
+        });
+        await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'set',
+                xmlns: 'w:biz'
+            },
+            content: [
+                {
+                    tag: 'business_profile',
+                    attrs: { v: '3', mutation_type: 'delta' },
+                    content: [
+                        {
+                            tag: 'cover_photo',
+                            attrs: { id: String(fbid), op: 'update', token: meta_hmac, ts: String(ts) }
+                        }
+                    ]
+                }
+            ]
+        });
+        return fbid;
+    };
+    const removeCoverPhoto = async (id) => {
+        return query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'set',
+                xmlns: 'w:biz'
+            },
+            content: [
+                {
+                    tag: 'business_profile',
+                    attrs: { v: '3', mutation_type: 'delta' },
+                    content: [
+                        {
+                            tag: 'cover_photo',
+                            attrs: { op: 'delete', id }
+                        }
+                    ]
+                }
+            ]
+        });
+    };
+    const getCatalog = async ({ jid, limit, cursor }) => {
+        var _a;
+        jid = jid || ((_a = authState.creds.me) === null || _a === void 0 ? void 0 : _a.id);
+        jid = (0, WABinary_1.jidNormalizedUser)(jid);
+        const queryParamNodes = [
+            {
+                tag: 'limit',
+                attrs: {},
+                content: Buffer.from((limit || 10).toString())
+            },
+            {
+                tag: 'width',
+                attrs: {},
+                content: Buffer.from('100')
+            },
+            {
+                tag: 'height',
+                attrs: {},
+                content: Buffer.from('100')
+            },
+        ];
+        if (cursor) {
+            queryParamNodes.push({
+                tag: 'after',
+                attrs: {},
+                content: cursor
+            });
+        }
+        const result = await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'get',
+                xmlns: 'w:biz:catalog'
+            },
+            content: [
+                {
+                    tag: 'product_catalog',
+                    attrs: {
+                        jid,
+                        'allow_shop_source': 'true'
+                    },
+                    content: queryParamNodes
+                }
+            ]
+        });
+        return (0, business_1.parseCatalogNode)(result);
+    };
+    const getCollections = async (jid, limit = 51) => {
+        var _a;
+        jid = jid || ((_a = authState.creds.me) === null || _a === void 0 ? void 0 : _a.id);
+        jid = (0, WABinary_1.jidNormalizedUser)(jid);
+        const result = await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'get',
+                xmlns: 'w:biz:catalog',
+                'smax_id': '35'
+            },
+            content: [
+                {
+                    tag: 'collections',
+                    attrs: {
+                        'biz_jid': jid,
+                    },
+                    content: [
+                        {
+                            tag: 'collection_limit',
+                            attrs: {},
+                            content: Buffer.from(limit.toString())
+                        },
+                        {
+                            tag: 'item_limit',
+                            attrs: {},
+                            content: Buffer.from(limit.toString())
+                        },
+                        {
+                            tag: 'width',
+                            attrs: {},
+                            content: Buffer.from('100')
+                        },
+                        {
+                            tag: 'height',
+                            attrs: {},
+                            content: Buffer.from('100')
+                        }
+                    ]
+                }
+            ]
+        });
+        return (0, business_1.parseCollectionsNode)(result);
+    };
+    const getOrderDetails = async (orderId, tokenBase64) => {
+        const result = await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'get',
+                xmlns: 'fb:thrift_iq',
+                'smax_id': '5'
+            },
+            content: [
+                {
+                    tag: 'order',
+                    attrs: {
+                        op: 'get',
+                        id: orderId
+                    },
+                    content: [
+                        {
+                            tag: 'image_dimensions',
+                            attrs: {},
+                            content: [
+                                {
+                                    tag: 'width',
+                                    attrs: {},
+                                    content: Buffer.from('100')
+                                },
+                                {
+                                    tag: 'height',
+                                    attrs: {},
+                                    content: Buffer.from('100')
+                                }
+                            ]
+                        },
+                        {
+                            tag: 'token',
+                            attrs: {},
+                            content: Buffer.from(tokenBase64)
+                        }
+                    ]
+                }
+            ]
+        });
+        return (0, business_1.parseOrderDetailsNode)(result);
+    };
+    const productUpdate = async (productId, update) => {
+        update = await (0, business_1.uploadingNecessaryImagesOfProduct)(update, waUploadToServer);
+        const editNode = (0, business_1.toProductNode)(productId, update);
+        const result = await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'set',
+                xmlns: 'w:biz:catalog'
+            },
+            content: [
+                {
+                    tag: 'product_catalog_edit',
+                    attrs: { v: '1' },
+                    content: [
+                        editNode,
+                        {
+                            tag: 'width',
+                            attrs: {},
+                            content: '100'
+                        },
+                        {
+                            tag: 'height',
+                            attrs: {},
+                            content: '100'
+                        }
+                    ]
+                }
+            ]
+        });
+        const productCatalogEditNode = (0, generic_utils_1.getBinaryNodeChild)(result, 'product_catalog_edit');
+        const productNode = (0, generic_utils_1.getBinaryNodeChild)(productCatalogEditNode, 'product');
+        return (0, business_1.parseProductNode)(productNode);
+    };
+    const productCreate = async (create) => {
+        // ensure isHidden is defined
+        create.isHidden = !!create.isHidden;
+        create = await (0, business_1.uploadingNecessaryImagesOfProduct)(create, waUploadToServer);
+        const createNode = (0, business_1.toProductNode)(undefined, create);
+        const result = await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'set',
+                xmlns: 'w:biz:catalog'
+            },
+            content: [
+                {
+                    tag: 'product_catalog_add',
+                    attrs: { v: '1' },
+                    content: [
+                        createNode,
+                        {
+                            tag: 'width',
+                            attrs: {},
+                            content: '100'
+                        },
+                        {
+                            tag: 'height',
+                            attrs: {},
+                            content: '100'
+                        }
+                    ]
+                }
+            ]
+        });
+        const productCatalogAddNode = (0, generic_utils_1.getBinaryNodeChild)(result, 'product_catalog_add');
+        const productNode = (0, generic_utils_1.getBinaryNodeChild)(productCatalogAddNode, 'product');
+        return (0, business_1.parseProductNode)(productNode);
+    };
+    const productDelete = async (productIds) => {
+        const result = await query({
+            tag: 'iq',
+            attrs: {
+                to: WABinary_1.S_WHATSAPP_NET,
+                type: 'set',
+                xmlns: 'w:biz:catalog'
+            },
+            content: [
+                {
+                    tag: 'product_catalog_delete',
+                    attrs: { v: '1' },
+                    content: productIds.map(id => ({
+                        tag: 'product',
+                        attrs: {},
+                        content: [
+                            {
+                                tag: 'id',
+                                attrs: {},
+                                content: Buffer.from(id)
+                            }
+                        ]
+                    }))
+                }
+            ]
+        });
+        const productCatalogDelNode = (0, generic_utils_1.getBinaryNodeChild)(result, 'product_catalog_delete');
+        return {
+            deleted: +((productCatalogDelNode === null || productCatalogDelNode === void 0 ? void 0 : productCatalogDelNode.attrs.deleted_count) || 0)
+        };
+    };
+    return {
+        ...sock,
+        logger: config.logger,
+        getOrderDetails,
+        getCatalog,
+        getCollections,
+        productCreate,
+        productDelete,
+        productUpdate,
+        updateBussinesProfile,
+        updateCoverPhoto,
+        removeCoverPhoto
+    };
+};
+exports.makeBusinessSocket = makeBusinessSocket;