@ubedsockets/baileys 1.6.6 → 1.6.8

package/lib/Socket/Client/websocket.js

@@ -1,54 +1,111 @@
-import WebSocket from 'ws';
-import { DEFAULT_ORIGIN } from '../../Defaults/index.js';
-import { AbstractSocketClient } from './types.js';
-export class WebSocketClient extends AbstractSocketClient {
+"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;
+        // queue & dispatch variables for throttling outgoing messages to avoid rate limits
+        this._queue = [];
+        this._isDispatching = false;
+        this._lastDispatch = 0;
+        this._minSendIntervalMs = (Defaults_1.DEFAULT_CONNECTION_CONFIG && Defaults_1.DEFAULT_CONNECTION_CONFIG.minSendIntervalMs) || 50;
     }
     get isOpen() {
-        return this.socket?.readyState === WebSocket.OPEN;
+        var _a;
+        return ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.OPEN;
     }
     get isClosed() {
-        return this.socket === null || this.socket?.readyState === WebSocket.CLOSED;
+        var _a;
+        return this.socket === null || ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.CLOSED;
     }
     get isClosing() {
-        return this.socket === null || this.socket?.readyState === WebSocket.CLOSING;
+        var _a;
+        return this.socket === null || ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.CLOSING;
     }
     get isConnecting() {
-        return this.socket?.readyState === WebSocket.CONNECTING;
+        var _a;
+        return ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === ws_1.default.CONNECTING;
     }
-    connect() {
+    async connect() {
+        var _a, _b;
         if (this.socket) {
             return;
         }
-        this.socket = new WebSocket(this.url, {
-            origin: DEFAULT_ORIGIN,
-            headers: this.config.options?.headers,
+        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
+            agent: this.config.agent,
         });
         this.socket.setMaxListeners(0);
         const events = ['close', 'error', 'upgrade', 'message', 'open', 'ping', 'pong', 'unexpected-response'];
         for (const event of events) {
-            this.socket?.on(event, (...args) => this.emit(event, ...args));
+            (_b = this.socket) === null || _b === void 0 ? void 0 : _b.on(event, (...args) => this.emit(event, ...args));
         }
     }
     async close() {
         if (!this.socket) {
             return;
         }
-        const closePromise = new Promise(resolve => {
-            this.socket?.once('close', resolve);
-        });
         this.socket.close();
-        await closePromise;
         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) {
-        this.socket?.send(str, cb);
-        return Boolean(this.socket);
+        // throttle sends to reduce rate-limit likelihood
+        const doSend = () => {
+            var _a;
+            (_a = this.socket) === null || _a === void 0 ? void 0 : _a.send(str, cb);
+            return Boolean(this.socket);
+        };
+        this._queue.push(doSend);
+        this._dispatch();
+        return true;
+    }
+
+    _dispatch() {
+        if (this._isDispatching) {
+            return;
+        }
+        this._isDispatching = true;
+        const tick = () => {
+            const now = Date.now();
+            if (this._queue.length === 0) {
+                this._isDispatching = false;
+                return;
+            }
+            const delta = now - this._lastDispatch;
+            const wait = Math.max(0, this._minSendIntervalMs - delta);
+            setTimeout(() => {
+                const fn = this._queue.shift();
+                this._lastDispatch = Date.now();
+                try {
+                    fn && fn();
+                }
+                catch (_err) {
+                    // ignore send errors here; they'll surface elsewhere
+                }
+                tick();
+            }, wait);
+        };
+        tick();
     }
 }
-//# sourceMappingURL=websocket.js.map
+exports.WebSocketClient = WebSocketClient;

package/lib/Socket/business.js

@@ -1,134 +1,17 @@
-import { getRawMediaUploadData } from '../Utils/index.js';
-import { parseCatalogNode, parseCollectionsNode, parseOrderDetailsNode, parseProductNode, toProductNode, uploadingNecessaryImagesOfProduct } from '../Utils/business.js';
-import { jidNormalizedUser, S_WHATSAPP_NET } from '../WABinary/index.js';
-import { getBinaryNodeChild } from '../WABinary/generic-utils.js';
-import { makeMessagesRecvSocket } from './messages-recv.js';
-export const makeBusinessSocket = (config) => {
-    const sock = makeMessagesRecvSocket(config);
+"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 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'];
-        node.push(...simpleFields
-            .filter(key => args[key] !== undefined && args[key] !== null)
-            .map(key => ({
-            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;
-                })
-            });
-        }
-        const result = await query({
-            tag: 'iq',
-            attrs: {
-                to: S_WHATSAPP_NET,
-                type: 'set',
-                xmlns: 'w:biz'
-            },
-            content: [
-                {
-                    tag: 'business_profile',
-                    attrs: {
-                        v: '3',
-                        mutation_type: 'delta'
-                    },
-                    content: node
-                }
-            ]
-        });
-        return result;
-    };
-    const updateCoverPhoto = async (photo) => {
-        const { fileSha256, filePath } = await getRawMediaUploadData(photo, 'biz-cover-photo');
-        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: 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 await query({
-            tag: 'iq',
-            attrs: {
-                to: 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 }) => {
-        jid = jid || authState.creds.me?.id;
-        jid = jidNormalizedUser(jid);
+        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',
@@ -144,7 +27,7 @@ export const makeBusinessSocket = (config) => {
                 tag: 'height',
                 attrs: {},
                 content: Buffer.from('100')
-            }
+            },
         ];
         if (cursor) {
             queryParamNodes.push({
@@ -156,7 +39,7 @@ export const makeBusinessSocket = (config) => {
         const result = await query({
             tag: 'iq',
             attrs: {
-                to: S_WHATSAPP_NET,
+                to: WABinary_1.S_WHATSAPP_NET,
                 type: 'get',
                 xmlns: 'w:biz:catalog'
             },
@@ -165,30 +48,31 @@ export const makeBusinessSocket = (config) => {
                     tag: 'product_catalog',
                     attrs: {
                         jid,
-                        allow_shop_source: 'true'
+                        'allow_shop_source': 'true'
                     },
                     content: queryParamNodes
                 }
             ]
         });
-        return parseCatalogNode(result);
+        return (0, business_1.parseCatalogNode)(result);
     };
     const getCollections = async (jid, limit = 51) => {
-        jid = jid || authState.creds.me?.id;
-        jid = jidNormalizedUser(jid);
+        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: S_WHATSAPP_NET,
+                to: WABinary_1.S_WHATSAPP_NET,
                 type: 'get',
                 xmlns: 'w:biz:catalog',
-                smax_id: '35'
+                'smax_id': '35'
             },
             content: [
                 {
                     tag: 'collections',
                     attrs: {
-                        biz_jid: jid
+                        'biz_jid': jid,
                     },
                     content: [
                         {
@@ -215,16 +99,16 @@ export const makeBusinessSocket = (config) => {
                 }
             ]
         });
-        return parseCollectionsNode(result);
+        return (0, business_1.parseCollectionsNode)(result);
     };
     const getOrderDetails = async (orderId, tokenBase64) => {
         const result = await query({
             tag: 'iq',
             attrs: {
-                to: S_WHATSAPP_NET,
+                to: WABinary_1.S_WHATSAPP_NET,
                 type: 'get',
                 xmlns: 'fb:thrift_iq',
-                smax_id: '5'
+                'smax_id': '5'
             },
             content: [
                 {
@@ -259,15 +143,15 @@ export const makeBusinessSocket = (config) => {
                 }
             ]
         });
-        return parseOrderDetailsNode(result);
+        return (0, business_1.parseOrderDetailsNode)(result);
     };
     const productUpdate = async (productId, update) => {
-        update = await uploadingNecessaryImagesOfProduct(update, waUploadToServer);
-        const editNode = toProductNode(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: S_WHATSAPP_NET,
+                to: WABinary_1.S_WHATSAPP_NET,
                 type: 'set',
                 xmlns: 'w:biz:catalog'
             },
@@ -291,19 +175,19 @@ export const makeBusinessSocket = (config) => {
                 }
             ]
         });
-        const productCatalogEditNode = getBinaryNodeChild(result, 'product_catalog_edit');
-        const productNode = getBinaryNodeChild(productCatalogEditNode, 'product');
-        return parseProductNode(productNode);
+        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 uploadingNecessaryImagesOfProduct(create, waUploadToServer);
-        const createNode = toProductNode(undefined, create);
+        create = await (0, business_1.uploadingNecessaryImagesOfProduct)(create, waUploadToServer);
+        const createNode = (0, business_1.toProductNode)(undefined, create);
         const result = await query({
             tag: 'iq',
             attrs: {
-                to: S_WHATSAPP_NET,
+                to: WABinary_1.S_WHATSAPP_NET,
                 type: 'set',
                 xmlns: 'w:biz:catalog'
             },
@@ -327,15 +211,15 @@ export const makeBusinessSocket = (config) => {
                 }
             ]
         });
-        const productCatalogAddNode = getBinaryNodeChild(result, 'product_catalog_add');
-        const productNode = getBinaryNodeChild(productCatalogAddNode, 'product');
-        return parseProductNode(productNode);
+        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: S_WHATSAPP_NET,
+                to: WABinary_1.S_WHATSAPP_NET,
                 type: 'set',
                 xmlns: 'w:biz:catalog'
             },
@@ -357,9 +241,9 @@ export const makeBusinessSocket = (config) => {
                 }
             ]
         });
-        const productCatalogDelNode = getBinaryNodeChild(result, 'product_catalog_delete');
+        const productCatalogDelNode = (0, generic_utils_1.getBinaryNodeChild)(result, 'product_catalog_delete');
         return {
-            deleted: +(productCatalogDelNode?.attrs.deleted_count || 0)
+            deleted: +((productCatalogDelNode === null || productCatalogDelNode === void 0 ? void 0 : productCatalogDelNode.attrs.deleted_count) || 0)
         };
     };
     return {
@@ -370,10 +254,7 @@ export const makeBusinessSocket = (config) => {
         getCollections,
         productCreate,
         productDelete,
-        productUpdate,
-        updateBussinesProfile,
-        updateCoverPhoto,
-        removeCoverPhoto
+        productUpdate
     };
 };
-//# sourceMappingURL=business.js.map
+exports.makeBusinessSocket = makeBusinessSocket;