supunmd-bail 2.0.2 → 2.1.3

package/lib/Utils/event-buffer.js

@@ -1,8 +1,14 @@
-import EventEmitter from 'events';
-import { WAMessageStatus } from '../Types/index.js';
-import { trimUndefined } from './generics.js';
-import { updateMessageWithReaction, updateMessageWithReceipt } from './messages.js';
-import { isRealMessage, shouldIncrementChatUnread } from './process-message.js';
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.makeEventBuffer = void 0;
+const events_1 = __importDefault(require("events"));
+const Types_1 = require("../Types");
+const generics_1 = require("./generics");
+const messages_1 = require("./messages");
+const process_message_1 = require("./process-message");
 const BUFFERABLE_EVENT = [
     'messaging-history.set',
     'chats.upsert',
@@ -15,22 +21,19 @@ const BUFFERABLE_EVENT = [
     'messages.delete',
     'messages.reaction',
     'message-receipt.update',
-    'groups.update'
+    'groups.update',
 ];
 const BUFFERABLE_EVENT_SET = new Set(BUFFERABLE_EVENT);
 /**
  * The event buffer logically consolidates different events into a single event
  * making the data processing more efficient.
+ * @param ev the baileys event emitter
  */
-export const makeEventBuffer = (logger) => {
-    const ev = new EventEmitter();
+const makeEventBuffer = (logger) => {
+    const ev = new events_1.default();
     const historyCache = new Set();
     let data = makeBufferData();
-    let isBuffering = false;
-    let bufferTimeout = null;
-    let bufferCount = 0;
-    const MAX_HISTORY_CACHE_SIZE = 10000; // Limit the history cache size to prevent memory bloat
-    const BUFFER_TIMEOUT_MS = 30000; // 30 seconds
+    let buffersInProgress = 0;
     // take the generic event and fire it as a baileys event
     ev.on('event', (map) => {
         for (const event in map) {
@@ -38,44 +41,25 @@ export const makeEventBuffer = (logger) => {
         }
     });
     function buffer() {
-        if (!isBuffering) {
-            logger.debug('Event buffer activated');
-            isBuffering = true;
-            bufferCount++;
-            // Auto-flush after a timeout to prevent infinite buffering
-            if (bufferTimeout) {
-                clearTimeout(bufferTimeout);
-            }
-            bufferTimeout = setTimeout(() => {
-                if (isBuffering) {
-                    logger.warn('Buffer timeout reached, auto-flushing');
-                    flush();
-                }
-            }, BUFFER_TIMEOUT_MS);
-        }
-        else {
-            bufferCount++;
-        }
+        buffersInProgress += 1;
     }
-    function flush() {
-        if (!isBuffering) {
+    function flush(force = false) {
+        // no buffer going on
+        if (!buffersInProgress) {
             return false;
         }
-        logger.debug({ bufferCount }, 'Flushing event buffer');
-        isBuffering = false;
-        bufferCount = 0;
-        // Clear timeout
-        if (bufferTimeout) {
-            clearTimeout(bufferTimeout);
-            bufferTimeout = null;
-        }
-        // Clear history cache if it exceeds the max size
-        if (historyCache.size > MAX_HISTORY_CACHE_SIZE) {
-            logger.debug({ cacheSize: historyCache.size }, 'Clearing history cache');
-            historyCache.clear();
+        if (!force) {
+            // reduce the number of buffers in progress
+            buffersInProgress -= 1;
+            // if there are still some buffers going on
+            // then we don't flush now
+            if (buffersInProgress) {
+                return false;
+            }
         }
         const newData = makeBufferData();
         const chatUpdates = Object.values(data.chatUpdates);
+        // gather the remaining conditional events so we re-queue them
         let conditionalChatUpdatesLeft = 0;
         for (const update of chatUpdates) {
             if (update.conditional) {
@@ -103,14 +87,14 @@ export const makeEventBuffer = (logger) => {
             };
         },
         emit(event, evData) {
-            if (isBuffering && BUFFERABLE_EVENT_SET.has(event)) {
+            if (buffersInProgress && BUFFERABLE_EVENT_SET.has(event)) {
                 append(data, historyCache, event, evData, logger);
                 return true;
             }
             return ev.emit('event', { [event]: evData });
         },
         isBuffering() {
-            return isBuffering;
+            return buffersInProgress > 0;
         },
         buffer,
         flush,
@@ -119,33 +103,19 @@ export const makeEventBuffer = (logger) => {
                 buffer();
                 try {
                     const result = await work(...args);
-                    // If this is the only buffer, flush after a small delay
-                    if (bufferCount === 1) {
-                        setTimeout(() => {
-                            if (isBuffering && bufferCount === 1) {
-                                flush();
-                            }
-                        }, 100); // Small delay to allow nested buffers
-                    }
                     return result;
                 }
-                catch (error) {
-                    throw error;
-                }
                 finally {
-                    bufferCount = Math.max(0, bufferCount - 1);
-                    if (bufferCount === 0) {
-                        // Auto-flush when no other buffers are active
-                        setTimeout(flush, 100);
-                    }
+                    flush();
                 }
             };
         },
         on: (...args) => ev.on(...args),
         off: (...args) => ev.off(...args),
-        removeAllListeners: (...args) => ev.removeAllListeners(...args)
+        removeAllListeners: (...args) => ev.removeAllListeners(...args),
     };
 };
+exports.makeEventBuffer = makeEventBuffer;
 const makeBufferData = () => {
     return {
         historySets: {
@@ -168,27 +138,25 @@ const makeBufferData = () => {
         groupUpdates: {}
     };
 };
-function append(data, historyCache, event, 
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-eventData, logger) {
+function append(data, historyCache, event, eventData, logger) {
+    var _a, _b, _c;
     switch (event) {
         case 'messaging-history.set':
             for (const chat of eventData.chats) {
-                const id = chat.id || '';
-                const existingChat = data.historySets.chats[id];
+                const existingChat = data.historySets.chats[chat.id];
                 if (existingChat) {
                     existingChat.endOfHistoryTransferType = chat.endOfHistoryTransferType;
                 }
-                if (!existingChat && !historyCache.has(id)) {
-                    data.historySets.chats[id] = chat;
-                    historyCache.add(id);
+                if (!existingChat && !historyCache.has(chat.id)) {
+                    data.historySets.chats[chat.id] = chat;
+                    historyCache.add(chat.id);
                     absorbingChatUpdate(chat);
                 }
             }
             for (const contact of eventData.contacts) {
                 const existingContact = data.historySets.contacts[contact.id];
                 if (existingContact) {
-                    Object.assign(existingContact, trimUndefined(contact));
+                    Object.assign(existingContact, (0, generics_1.trimUndefined)(contact));
                 }
                 else {
                     const historyContactId = `c:${contact.id}`;
@@ -208,19 +176,15 @@ eventData, logger) {
                 }
             }
             data.historySets.empty = false;
-            data.historySets.syncType = eventData.syncType;
-            data.historySets.progress = eventData.progress;
-            data.historySets.peerDataRequestSessionId = eventData.peerDataRequestSessionId;
             data.historySets.isLatest = eventData.isLatest || data.historySets.isLatest;
             break;
         case 'chats.upsert':
             for (const chat of eventData) {
-                const id = chat.id || '';
-                let upsert = data.chatUpserts[id];
-                if (id && !upsert) {
-                    upsert = data.historySets.chats[id];
+                let upsert = data.chatUpserts[chat.id];
+                if (!upsert) {
+                    upsert = data.historySets[chat.id];
                     if (upsert) {
-                        logger.debug({ chatId: id }, 'absorbed chat upsert in chat set');
+                        logger.debug({ chatId: chat.id }, 'absorbed chat upsert in chat set');
                     }
                 }
                 if (upsert) {
@@ -228,11 +192,11 @@ eventData, logger) {
                 }
                 else {
                     upsert = chat;
-                    data.chatUpserts[id] = upsert;
+                    data.chatUpserts[chat.id] = upsert;
                 }
                 absorbingChatUpdate(upsert);
-                if (data.chatDeletes.has(id)) {
-                    data.chatDeletes.delete(id);
+                if (data.chatDeletes.has(chat.id)) {
+                    data.chatDeletes.delete(chat.id);
                 }
             }
             break;
@@ -292,14 +256,14 @@ eventData, logger) {
                     }
                 }
                 if (upsert) {
-                    upsert = Object.assign(upsert, trimUndefined(contact));
+                    upsert = Object.assign(upsert, (0, generics_1.trimUndefined)(contact));
                 }
                 else {
                     upsert = contact;
                     data.contactUpserts[contact.id] = upsert;
                 }
                 if (data.contactUpdates[contact.id]) {
-                    upsert = Object.assign(data.contactUpdates[contact.id], trimUndefined(contact));
+                    upsert = Object.assign(data.contactUpdates[contact.id], (0, generics_1.trimUndefined)(contact));
                     delete data.contactUpdates[contact.id];
                 }
             }
@@ -324,7 +288,7 @@ eventData, logger) {
             const { messages, type } = eventData;
             for (const message of messages) {
                 const key = stringifyMessageKey(message.key);
-                let existing = data.messageUpserts[key]?.message;
+                let existing = (_a = data.messageUpserts[key]) === null || _a === void 0 ? void 0 : _a.message;
                 if (!existing) {
                     existing = data.historySets.messages[key];
                     if (existing) {
@@ -345,7 +309,9 @@ eventData, logger) {
                 else {
                     data.messageUpserts[key] = {
                         message,
-                        type: type === 'notify' || data.messageUpserts[key]?.type === 'notify' ? 'notify' : type
+                        type: type === 'notify' || ((_b = data.messageUpserts[key]) === null || _b === void 0 ? void 0 : _b.type) === 'notify'
+                            ? 'notify'
+                            : type
                     };
                 }
             }
@@ -354,13 +320,13 @@ eventData, logger) {
             const msgUpdates = eventData;
             for (const { key, update } of msgUpdates) {
                 const keyStr = stringifyMessageKey(key);
-                const existing = data.historySets.messages[keyStr] || data.messageUpserts[keyStr]?.message;
+                const existing = data.historySets.messages[keyStr] || ((_c = data.messageUpserts[keyStr]) === null || _c === void 0 ? void 0 : _c.message);
                 if (existing) {
                     Object.assign(existing, update);
                     // if the message was received & read by us
                     // the chat counter must have been incremented
                     // so we need to decrement it
-                    if (update.status === WAMessageStatus.READ && !key.fromMe) {
+                    if (update.status === Types_1.WAMessageStatus.READ && !key.fromMe) {
                         decrementChatReadCounterIfMsgDidUnread(existing);
                     }
                 }
@@ -398,11 +364,12 @@ eventData, logger) {
                 const keyStr = stringifyMessageKey(key);
                 const existing = data.messageUpserts[keyStr];
                 if (existing) {
-                    updateMessageWithReaction(existing.message, reaction);
+                    (0, messages_1.updateMessageWithReaction)(existing.message, reaction);
                 }
                 else {
-                    data.messageReactions[keyStr] = data.messageReactions[keyStr] || { key, reactions: [] };
-                    updateMessageWithReaction(data.messageReactions[keyStr], reaction);
+                    data.messageReactions[keyStr] = data.messageReactions[keyStr]
+                        || { key, reactions: [] };
+                    (0, messages_1.updateMessageWithReaction)(data.messageReactions[keyStr], reaction);
                 }
             }
             break;
@@ -412,11 +379,12 @@ eventData, logger) {
                 const keyStr = stringifyMessageKey(key);
                 const existing = data.messageUpserts[keyStr];
                 if (existing) {
-                    updateMessageWithReceipt(existing.message, receipt);
+                    (0, messages_1.updateMessageWithReceipt)(existing.message, receipt);
                 }
                 else {
-                    data.messageReceipts[keyStr] = data.messageReceipts[keyStr] || { key, userReceipt: [] };
-                    updateMessageWithReceipt(data.messageReceipts[keyStr], receipt);
+                    data.messageReceipts[keyStr] = data.messageReceipts[keyStr]
+                        || { key, userReceipt: [] };
+                    (0, messages_1.updateMessageWithReceipt)(data.messageReceipts[keyStr], receipt);
                 }
             }
             break;
@@ -434,7 +402,7 @@ eventData, logger) {
             throw new Error(`"${event}" cannot be buffered`);
     }
     function absorbingChatUpdate(existing) {
-        const chatId = existing.id || '';
+        const chatId = existing.id;
         const update = data.chatUpdates[chatId];
         if (update) {
             const conditionMatches = update.conditional ? update.conditional(data) : true;
@@ -455,10 +423,10 @@ eventData, logger) {
         // if the message has already been marked read by us
         const chatId = message.key.remoteJid;
         const chat = data.chatUpdates[chatId] || data.chatUpserts[chatId];
-        if (isRealMessage(message) &&
-            shouldIncrementChatUnread(message) &&
-            typeof chat?.unreadCount === 'number' &&
-            chat.unreadCount > 0) {
+        if ((0, process_message_1.isRealMessage)(message, '')
+            && (0, process_message_1.shouldIncrementChatUnread)(message)
+            && typeof (chat === null || chat === void 0 ? void 0 : chat.unreadCount) === 'number'
+            && chat.unreadCount > 0) {
             logger.debug({ chatId: chat.id }, 'decrementing chat counter');
             chat.unreadCount -= 1;
             if (chat.unreadCount === 0) {
@@ -474,10 +442,7 @@ function consolidateEvents(data) {
             chats: Object.values(data.historySets.chats),
             messages: Object.values(data.historySets.messages),
             contacts: Object.values(data.historySets.contacts),
-            syncType: data.historySets.syncType,
-            progress: data.historySets.progress,
-            isLatest: data.historySets.isLatest,
-            peerDataRequestSessionId: data.historySets.peerDataRequestSessionId
+            isLatest: data.historySets.isLatest
         };
     }
     const chatUpsertList = Object.values(data.chatUpserts);
@@ -531,10 +496,12 @@ function consolidateEvents(data) {
     return map;
 }
 function concatChats(a, b) {
-    if (b.unreadCount === null && // neutralize unread counter
-        a.unreadCount < 0) {
-        a.unreadCount = undefined;
-        b.unreadCount = undefined;
+    if (b.unreadCount === null) {
+        // neutralize unread counter
+        if (a.unreadCount < 0) {
+            a.unreadCount = undefined;
+            b.unreadCount = undefined;
+        }
     }
     if (typeof a.unreadCount === 'number' && typeof b.unreadCount === 'number') {
         b = { ...b };
@@ -545,4 +512,3 @@ function concatChats(a, b) {
     return Object.assign(a, b);
 }
 const stringifyMessageKey = (key) => `${key.remoteJid},${key.id},${key.fromMe ? '1' : '0'}`;
-//# sourceMappingURL=event-buffer.js.map

package/lib/Utils/generics.d.ts

@@ -1,17 +1,24 @@
-import { proto } from '../../WAProto/index.js';
-import type { BaileysEventEmitter, BaileysEventMap, ConnectionState, WACallUpdateType, WAMessageKey, WAVersion } from '../Types/index.js';
-import { type BinaryNode } from '../WABinary/index.js';
-export declare const BufferJSON: {
-    replacer: (k: any, value: any) => any;
-    reviver: (_: any, value: any) => any;
+/// <reference types="node" />
+import { AxiosRequestConfig } from 'axios';
+import { Logger } from 'pino';
+import { proto } from '../../WAProto';
+import { BaileysEventEmitter, BaileysEventMap, WACallUpdateType, WAVersion } from '../Types';
+import { BinaryNode } from '../WABinary';
+export declare const Browsers: {
+    ubuntu: (browser: any) => [string, string, string];
+    macOS: (browser: any) => [string, string, string];
+    baileys: (browser: any) => [string, string, string];
+    windows: (browser: any) => [string, string, string];
+    /** The appropriate browser based on your OS & release */
+    appropriate: (browser: any) => [string, string, string];
 };
-export declare const getKeyAuthor: (key: WAMessageKey | undefined | null, meId?: string) => string;
-export declare const writeRandomPadMax16: (msg: Uint8Array) => Buffer<ArrayBuffer>;
-export declare const unpadRandomMax16: (e: Uint8Array | Buffer) => Uint8Array<ArrayBuffer>;
-export declare const generateParticipantHashV2: (participants: string[]) => string;
-export declare const encodeWAMessage: (message: proto.IMessage) => Buffer<ArrayBuffer>;
+export declare const getPlatformId: (browser: string) => any;
+export declare const getKeyAuthor: (key: proto.IMessageKey | undefined | null, meId?: string) => string;
+export declare const writeRandomPadMax16: (msg: Uint8Array) => Buffer;
+export declare const unpadRandomMax16: (e: Uint8Array | Buffer) => Uint8Array;
+export declare const encodeWAMessage: (message: proto.IMessage) => Buffer;
 export declare const generateRegistrationId: () => number;
-export declare const encodeBigEndian: (e: number, t?: number) => Uint8Array<ArrayBuffer>;
+export declare const encodeBigEndian: (e: number, t?: number) => Uint8Array;
 export declare const toNumber: (t: Long | number | null | undefined) => number;
 /** unix timestamp of a date in seconds */
 export declare const unixTimestampSeconds: (date?: Date) => number;
@@ -30,33 +37,34 @@ export declare const delayCancellable: (ms: number) => {
 export declare function promiseTimeout<T>(ms: number | undefined, promise: (resolve: (v: T) => void, reject: (error: any) => void) => void): Promise<T>;
 export declare const generateMessageIDV2: (userId?: string) => string;
 export declare const generateMessageID: () => string;
-export declare function bindWaitForEvent<T extends keyof BaileysEventMap>(ev: BaileysEventEmitter, event: T): (check: (u: BaileysEventMap[T]) => Promise<boolean | undefined>, timeoutMs?: number) => Promise<void>;
-export declare const bindWaitForConnectionUpdate: (ev: BaileysEventEmitter) => (check: (u: Partial<ConnectionState>) => Promise<boolean | undefined>, timeoutMs?: number) => Promise<void>;
+export declare function bindWaitForEvent<T extends keyof BaileysEventMap>(ev: BaileysEventEmitter, event: T): (check: (u: BaileysEventMap[T]) => boolean | undefined, timeoutMs?: number) => Promise<void>;
+export declare const bindWaitForConnectionUpdate: (ev: BaileysEventEmitter) => (check: (u: Partial<import("../Types").ConnectionState>) => boolean | undefined, timeoutMs?: number) => Promise<void>;
+export declare const printQRIfNecessaryListener: (ev: BaileysEventEmitter, logger: Logger) => void;
 /**
  * utility that fetches latest baileys version from the master branch.
  * Use to ensure your WA connection is always on the latest version
  */
-export declare const fetchLatestBaileysVersion: (options?: RequestInit) => Promise<{
+export declare const fetchLatestBaileysVersion: (options?: AxiosRequestConfig<any>) => Promise<{
     version: WAVersion;
     isLatest: boolean;
     error?: undefined;
 } | {
     version: WAVersion;
     isLatest: boolean;
-    error: unknown;
+    error: any;
 }>;
 /**
  * A utility that fetches the latest web version of whatsapp.
  * Use to ensure your WA connection is always on the latest version
  */
-export declare const fetchLatestWaWebVersion: (options?: RequestInit) => Promise<{
+export declare const fetchLatestWaWebVersion: (options: AxiosRequestConfig<any>) => Promise<{
     version: WAVersion;
     isLatest: boolean;
     error?: undefined;
 } | {
     version: WAVersion;
     isLatest: boolean;
-    error: unknown;
+    error: any;
 }>;
 /** unique message tag prefix for MD clients */
 export declare const generateMdTagPrefix: () => string;
@@ -64,7 +72,7 @@ export declare const generateMdTagPrefix: () => string;
  * Given a type of receipt, returns what the new status of the message should be
  * @param type type from receipt
  */
-export declare const getStatusFromReceiptType: (type: string | undefined) => proto.WebMessageInfo.Status | undefined;
+export declare const getStatusFromReceiptType: (type: string | undefined) => proto.WebMessageInfo.Status;
 /**
  * Stream errors generally provide a reason, map that to a baileys DisconnectReason
  * @param reason the string reason given, eg. "conflict"
@@ -79,12 +87,6 @@ export declare const getCodeFromWSError: (error: Error) => number;
  * Is the given platform WA business
  * @param platform AuthenticationCreds.platform
  */
-export declare const isWABusinessPlatform: (platform: string) => platform is "smba" | "smbi";
-export declare function trimUndefined(obj: {
-    [_: string]: any;
-}): {
-    [_: string]: any;
-};
+export declare const isWABusinessPlatform: (platform: string) => boolean;
+export declare function trimUndefined(obj: any): any;
 export declare function bytesToCrockford(buffer: Buffer): string;
-export declare function encodeNewsletterMessage(message: proto.IMessage): Uint8Array;
-//# sourceMappingURL=generics.d.ts.map