@mrlegendbot/baileys 1.2.1 → 1.2.4

package/lib/Utils/auth-utils.js

@@ -4,299 +4,262 @@ import { AsyncLocalStorage } from 'async_hooks';
 import { Mutex } from 'async-mutex';
 import { randomBytes } from 'crypto';
 import PQueue from 'p-queue';
+import { v4 as uuidv4 } from 'uuid';
 import { DEFAULT_CACHE_TTLS } from '../Defaults/index.js';
 import { Curve, signedKeyPair } from './crypto.js';
 import { delay, generateRegistrationId } from './generics.js';
 import { PreKeyManager } from './pre-key-manager.js';
-/**
- * Adds caching capability to a SignalKeyStore
- * @param store the store to add caching to
- * @param logger to log trace events
- * @param _cache cache store to use
- */
 export function makeCacheableSignalKeyStore(store, logger, _cache) {
-    const cache = _cache ||
-        new NodeCache({
-            stdTTL: DEFAULT_CACHE_TTLS.SIGNAL_STORE, // 5 minutes
-            useClones: false,
-            deleteOnExpire: true
-        });
-    // Mutex for protecting cache operations
-    const cacheMutex = new Mutex();
-    function getUniqueId(type, id) {
-        return `${type}.${id}`;
-    }
-    return {
-        async get(type, ids) {
-            return cacheMutex.runExclusive(async () => {
-                const data = {};
-                const idsToFetch = [];
-                for (const id of ids) {
-                    const item = (await cache.get(getUniqueId(type, id)));
-                    if (typeof item !== 'undefined') {
-                        data[id] = item;
-                    }
-                    else {
-                        idsToFetch.push(id);
-                    }
-                }
-                if (idsToFetch.length) {
-                    logger?.trace({ items: idsToFetch.length }, 'loading from store');
-                    const fetched = await store.get(type, idsToFetch);
-                    for (const id of idsToFetch) {
-                        const item = fetched[id];
-                        if (item) {
-                            data[id] = item;
-                            // eslint-disable-next-line @typescript-eslint/no-unnecessary-type-assertion
-                            await cache.set(getUniqueId(type, id), item);
-                        }
-                    }
-                }
-                return data;
-            });
-        },
-        async set(data) {
-            return cacheMutex.runExclusive(async () => {
-                let keys = 0;
-                for (const type in data) {
-                    for (const id in data[type]) {
-                        await cache.set(getUniqueId(type, id), data[type][id]);
-                        keys += 1;
-                    }
-                }
-                logger?.trace({ keys }, 'updated cache');
-                await store.set(data);
-            });
-        },
-        async clear() {
-            await cache.flushAll();
-            await store.clear?.();
+  const cache = _cache || new NodeCache({
+    stdTTL: DEFAULT_CACHE_TTLS.SIGNAL_STORE,
+    useClones: false,
+    deleteOnExpire: true
+  });
+  const cacheMutex = new Mutex();
+  function getUniqueId(type, id) {
+    return `${type}.${id}`;
+  }
+  return {
+    async get(type, ids) {
+      return cacheMutex.runExclusive(async () => {
+        const data = {};
+        const idsToFetch = [];
+        for (const id of ids) {
+          const item = await cache.get(getUniqueId(type, id));
+          if (typeof item !== 'undefined') {
+            data[id] = item;
+          } else {
+            idsToFetch.push(id);
+          }
         }
-    };
-}
-/**
- * Adds DB-like transaction capability to the SignalKeyStore
- * Uses AsyncLocalStorage for automatic context management
- * @param state the key store to apply this capability to
- * @param logger logger to log events
- * @returns SignalKeyStore with transaction capability
- */
-export const addTransactionCapability = (state, logger, { maxCommitRetries, delayBetweenTriesMs }) => {
-    const txStorage = new AsyncLocalStorage();
-    // Queues for concurrency control (keyed by signal data type - bounded set)
-    const keyQueues = new Map();
-    // Transaction mutexes with reference counting for cleanup
-    const txMutexes = new Map();
-    const txMutexRefCounts = new Map();
-    // Pre-key manager for specialized operations
-    const preKeyManager = new PreKeyManager(state, logger);
-    /**
-     * Get or create a queue for a specific key type
-     */
-    function getQueue(key) {
-        if (!keyQueues.has(key)) {
-            keyQueues.set(key, new PQueue({ concurrency: 1 }));
+        if (idsToFetch.length) {
+          logger?.trace({
+            items: idsToFetch.length
+          }, 'loading from store');
+          const fetched = await store.get(type, idsToFetch);
+          for (const id of idsToFetch) {
+            const item = fetched[id];
+            if (item) {
+              data[id] = item;
+              await cache.set(getUniqueId(type, id), item);
+            }
+          }
         }
-        return keyQueues.get(key);
-    }
-    /**
-     * Get or create a transaction mutex
-     */
-    function getTxMutex(key) {
-        if (!txMutexes.has(key)) {
-            txMutexes.set(key, new Mutex());
-            txMutexRefCounts.set(key, 0);
+        return data;
+      });
+    },
+    async set(data) {
+      return cacheMutex.runExclusive(async () => {
+        let keys = 0;
+        for (const type in data) {
+          for (const id in data[type]) {
+            await cache.set(getUniqueId(type, id), data[type][id]);
+            keys += 1;
+          }
         }
-        return txMutexes.get(key);
+        logger?.trace({
+          keys
+        }, 'updated cache');
+        await store.set(data);
+      });
+    },
+    async clear() {
+      await cache.flushAll();
+      await store.clear?.();
     }
-    /**
-     * Acquire a reference to a transaction mutex
-     */
-    function acquireTxMutexRef(key) {
-        const count = txMutexRefCounts.get(key) ?? 0;
-        txMutexRefCounts.set(key, count + 1);
+  };
+}
+export const addTransactionCapability = (state, logger, {
+  maxCommitRetries,
+  delayBetweenTriesMs
+}) => {
+  const txStorage = new AsyncLocalStorage();
+  const keyQueues = new Map();
+  const txMutexes = new Map();
+  const txMutexRefCounts = new Map();
+  const preKeyManager = new PreKeyManager(state, logger);
+  function getQueue(key) {
+    if (!keyQueues.has(key)) {
+      keyQueues.set(key, new PQueue({
+        concurrency: 1
+      }));
     }
-    /**
-     * Release a reference to a transaction mutex and cleanup if no longer needed
-     */
-    function releaseTxMutexRef(key) {
-        const count = (txMutexRefCounts.get(key) ?? 1) - 1;
-        txMutexRefCounts.set(key, count);
-        // Cleanup if no more references and mutex is not locked
-        if (count <= 0) {
-            const mutex = txMutexes.get(key);
-            if (mutex && !mutex.isLocked()) {
-                txMutexes.delete(key);
-                txMutexRefCounts.delete(key);
-            }
-        }
+    return keyQueues.get(key);
+  }
+  function getTxMutex(key) {
+    if (!txMutexes.has(key)) {
+      txMutexes.set(key, new Mutex());
+      txMutexRefCounts.set(key, 0);
     }
-    /**
-     * Check if currently in a transaction
-     */
-    function isInTransaction() {
-        return !!txStorage.getStore();
+    return txMutexes.get(key);
+  }
+  function acquireTxMutexRef(key) {
+    const count = txMutexRefCounts.get(key) ?? 0;
+    txMutexRefCounts.set(key, count + 1);
+  }
+  function releaseTxMutexRef(key) {
+    const count = (txMutexRefCounts.get(key) ?? 1) - 1;
+    txMutexRefCounts.set(key, count);
+    if (count <= 0) {
+      const mutex = txMutexes.get(key);
+      if (mutex && !mutex.isLocked()) {
+        txMutexes.delete(key);
+        txMutexRefCounts.delete(key);
+      }
     }
-    /**
-     * Commit transaction with retries
-     */
-    async function commitWithRetry(mutations) {
-        if (Object.keys(mutations).length === 0) {
-            logger.trace('no mutations in transaction');
-            return;
-        }
-        logger.trace('committing transaction');
-        for (let attempt = 0; attempt < maxCommitRetries; attempt++) {
-            try {
-                await state.set(mutations);
-                logger.trace({ mutationCount: Object.keys(mutations).length }, 'committed transaction');
-                return;
-            }
-            catch (error) {
-                const retriesLeft = maxCommitRetries - attempt - 1;
-                logger.warn(`failed to commit mutations, retries left=${retriesLeft}`);
-                if (retriesLeft === 0) {
-                    throw error;
-                }
-                await delay(delayBetweenTriesMs);
-            }
+  }
+  function isInTransaction() {
+    return !!txStorage.getStore();
+  }
+  async function commitWithRetry(mutations) {
+    if (Object.keys(mutations).length === 0) {
+      logger.trace('no mutations in transaction');
+      return;
+    }
+    logger.trace('committing transaction');
+    for (let attempt = 0; attempt < maxCommitRetries; attempt++) {
+      try {
+        await state.set(mutations);
+        logger.trace({
+          mutationCount: Object.keys(mutations).length
+        }, 'committed transaction');
+        return;
+      } catch (error) {
+        const retriesLeft = maxCommitRetries - attempt - 1;
+        logger.warn(`failed to commit mutations, retries left=${retriesLeft}`);
+        if (retriesLeft === 0) {
+          throw error;
         }
+        await delay(delayBetweenTriesMs);
+      }
     }
-    return {
-        get: async (type, ids) => {
-            const ctx = txStorage.getStore();
-            if (!ctx) {
-                // No transaction - direct read without exclusive lock for concurrency
-                return state.get(type, ids);
-            }
-            // In transaction - check cache first
-            const cached = ctx.cache[type] || {};
-            const missing = ids.filter(id => !(id in cached));
-            if (missing.length > 0) {
-                ctx.dbQueries++;
-                logger.trace({ type, count: missing.length }, 'fetching missing keys in transaction');
-                const fetched = await getTxMutex(type).runExclusive(() => state.get(type, missing));
-                // Update cache
-                ctx.cache[type] = ctx.cache[type] || {};
-                Object.assign(ctx.cache[type], fetched);
-            }
-            // Return requested ids from cache
-            const result = {};
-            for (const id of ids) {
-                const value = ctx.cache[type]?.[id];
-                if (value !== undefined && value !== null) {
-                    result[id] = value;
-                }
-            }
-            return result;
-        },
-        set: async (data) => {
-            const ctx = txStorage.getStore();
-            if (!ctx) {
-                // No transaction - direct write with queue protection
-                const types = Object.keys(data);
-                // Process pre-keys with validation
-                for (const type_ of types) {
-                    const type = type_;
-                    if (type === 'pre-key') {
-                        await preKeyManager.validateDeletions(data, type);
-                    }
-                }
-                // Write all data in parallel
-                await Promise.all(types.map(type => getQueue(type).add(async () => {
-                    const typeData = { [type]: data[type] };
-                    await state.set(typeData);
-                })));
-                return;
-            }
-            // In transaction - update cache and mutations
-            logger.trace({ types: Object.keys(data) }, 'caching in transaction');
-            for (const key_ in data) {
-                const key = key_;
-                // Ensure structures exist
-                ctx.cache[key] = ctx.cache[key] || {};
-                ctx.mutations[key] = ctx.mutations[key] || {};
-                // Special handling for pre-keys
-                if (key === 'pre-key') {
-                    await preKeyManager.processOperations(data, key, ctx.cache, ctx.mutations, true);
-                }
-                else {
-                    // Normal key types
-                    Object.assign(ctx.cache[key], data[key]);
-                    Object.assign(ctx.mutations[key], data[key]);
-                }
-            }
-        },
-        isInTransaction,
-        transaction: async (work, key) => {
-            const existing = txStorage.getStore();
-            // Nested transaction - reuse existing context
-            if (existing) {
-                logger.trace('reusing existing transaction context');
-                return work();
-            }
-            // New transaction - acquire mutex and create context
-            const mutex = getTxMutex(key);
-            acquireTxMutexRef(key);
-            try {
-                return await mutex.runExclusive(async () => {
-                    const ctx = {
-                        cache: {},
-                        mutations: {},
-                        dbQueries: 0
-                    };
-                    logger.trace('entering transaction');
-                    try {
-                        const result = await txStorage.run(ctx, work);
-                        // Commit mutations
-                        await commitWithRetry(ctx.mutations);
-                        logger.trace({ dbQueries: ctx.dbQueries }, 'transaction completed');
-                        return result;
-                    }
-                    catch (error) {
-                        logger.error({ error }, 'transaction failed, rolling back');
-                        throw error;
-                    }
-                });
-            }
-            finally {
-                releaseTxMutexRef(key);
-            }
+  }
+  return {
+    get: async (type, ids) => {
+      const ctx = txStorage.getStore();
+      if (!ctx) {
+        return state.get(type, ids);
+      }
+      const cached = ctx.cache[type] || {};
+      const missing = ids.filter(id => !(id in cached));
+      if (missing.length > 0) {
+        ctx.dbQueries++;
+        logger.trace({
+          type,
+          count: missing.length
+        }, 'fetching missing keys in transaction');
+        const fetched = await getTxMutex(type).runExclusive(() => state.get(type, missing));
+        ctx.cache[type] = ctx.cache[type] || {};
+        Object.assign(ctx.cache[type], fetched);
+      }
+      const result = {};
+      for (const id of ids) {
+        const value = ctx.cache[type]?.[id];
+        if (value !== undefined && value !== null) {
+          result[id] = value;
         }
-    };
-};
-/**
- * Returns the authenticated user's JID, or throws a Boom-401 if creds are not yet authenticated.
- * Use this anywhere we'd otherwise reach for `creds.me!.id` to fail fast with a descriptive error.
- */
-export const assertMeId = (creds) => {
-    const id = creds.me?.id;
-    if (!id) {
-        throw new Boom('Cannot proceed: socket is not authenticated yet (creds.me.id is missing)', { statusCode: 401 });
+      }
+      return result;
+    },
+    set: async data => {
+      const ctx = txStorage.getStore();
+      if (!ctx) {
+        const types = Object.keys(data);
+        for (const type_ of types) {
+          const type = type_;
+          if (type === 'pre-key') {
+            await preKeyManager.validateDeletions(data, type);
+          }
+        }
+        await Promise.all(types.map(type => getQueue(type).add(async () => {
+          const typeData = {
+            [type]: data[type]
+          };
+          await state.set(typeData);
+        })));
+        return;
+      }
+      logger.trace({
+        types: Object.keys(data)
+      }, 'caching in transaction');
+      for (const key_ in data) {
+        const key = key_;
+        ctx.cache[key] = ctx.cache[key] || {};
+        ctx.mutations[key] = ctx.mutations[key] || {};
+        if (key === 'pre-key') {
+          await preKeyManager.processOperations(data, key, ctx.cache, ctx.mutations, true);
+        } else {
+          Object.assign(ctx.cache[key], data[key]);
+          Object.assign(ctx.mutations[key], data[key]);
+        }
+      }
+    },
+    isInTransaction,
+    transaction: async (work, key) => {
+      const existing = txStorage.getStore();
+      if (existing) {
+        logger.trace('reusing existing transaction context');
+        return work();
+      }
+      const mutex = getTxMutex(key);
+      acquireTxMutexRef(key);
+      try {
+        return await mutex.runExclusive(async () => {
+          const ctx = {
+            cache: {},
+            mutations: {},
+            dbQueries: 0
+          };
+          logger.trace('entering transaction');
+          try {
+            const result = await txStorage.run(ctx, work);
+            await commitWithRetry(ctx.mutations);
+            logger.trace({
+              dbQueries: ctx.dbQueries
+            }, 'transaction completed');
+            return result;
+          } catch (error) {
+            logger.error({
+              error
+            }, 'transaction failed, rolling back');
+            throw error;
+          }
+        });
+      } finally {
+        releaseTxMutexRef(key);
+      }
     }
-    return id;
+  };
+};
+export const assertMeId = creds => {
+  const id = creds.me?.id;
+  if (!id) {
+    throw new Boom('Cannot proceed: socket is not authenticated yet (creds.me.id is missing)', {
+      statusCode: 401
+    });
+  }
+  return id;
 };
 export const initAuthCreds = () => {
-    const identityKey = Curve.generateKeyPair();
-    return {
-        noiseKey: Curve.generateKeyPair(),
-        pairingEphemeralKeyPair: Curve.generateKeyPair(),
-        signedIdentityKey: identityKey,
-        signedPreKey: signedKeyPair(identityKey, 1),
-        registrationId: generateRegistrationId(),
-        advSecretKey: randomBytes(32).toString('base64'),
-        processedHistoryMessages: [],
-        nextPreKeyId: 1,
-        firstUnuploadedPreKeyId: 1,
-        accountSyncCounter: 0,
-        accountSettings: {
-            unarchiveChats: false
-        },
-        registered: false,
-        pairingCode: undefined,
-        lastPropHash: undefined,
-        routingInfo: undefined,
-        additionalData: undefined
-    };
+  const identityKey = Curve.generateKeyPair();
+  return {
+    noiseKey: Curve.generateKeyPair(),
+    pairingEphemeralKeyPair: Curve.generateKeyPair(),
+    signedIdentityKey: identityKey,
+    signedPreKey: signedKeyPair(identityKey, 1),
+    registrationId: generateRegistrationId(),
+    advSecretKey: randomBytes(32).toString('base64'),
+    processedHistoryMessages: [],
+    nextPreKeyId: 1,
+    firstUnuploadedPreKeyId: 1,
+    accountSyncCounter: 0,
+    accountSettings: {
+      unarchiveChats: false
+    },
+    registered: false,
+    pairingCode: undefined,
+    lastPropHash: undefined,
+    routingInfo: undefined,
+    additionalData: undefined
+  };
 };
-//# sourceMappingURL=auth-utils.js.map

package/lib/Utils/browser-utils.js

@@ -1,29 +1,26 @@
 import { platform, release } from 'os';
-import { proto } from '../../WAProto/index.js';
+import { proto } from '../../WAProto/compiler.js';
 const PLATFORM_MAP = {
-    aix: 'AIX',
-    darwin: 'Mac OS',
-    win32: 'Windows',
-    android: 'Android',
-    freebsd: 'FreeBSD',
-    openbsd: 'OpenBSD',
-    sunos: 'Solaris',
-    linux: undefined,
-    haiku: undefined,
-    cygwin: undefined,
-    netbsd: undefined
+  aix: 'AIX',
+  darwin: 'Mac OS',
+  win32: 'Windows',
+  android: 'Android',
+  freebsd: 'FreeBSD',
+  openbsd: 'OpenBSD',
+  sunos: 'Solaris',
+  linux: undefined,
+  haiku: undefined,
+  cygwin: undefined,
+  netbsd: undefined
 };
 export const Browsers = {
-    ubuntu: browser => ['Ubuntu', browser, '22.04.4'],
-    macOS: browser => ['Mac OS', browser, '14.4.1'],
-    baileys: browser => ['Baileys', browser, '6.5.0'],
-    windows: browser => ['Windows', browser, '10.0.22631'],
-    android: browser => [browser, 'Android', ''],
-    /** The appropriate browser based on your OS & release */
-    appropriate: browser => [PLATFORM_MAP[platform()] || 'Ubuntu', browser, release()]
+  ubuntu: browser => ['Ubuntu', browser, '22.04.4'],
+  macOS: browser => ['Mac OS', browser, '14.4.1'],
+  baileys: browser => ['Baileys', browser, '6.5.0'],
+  windows: browser => ['Windows', browser, '10.0.22631'],
+  appropriate: browser => [PLATFORM_MAP[platform()] || 'Ubuntu', browser, release()]
 };
-export const getPlatformId = (browser) => {
-    const platformType = proto.DeviceProps.PlatformType[browser.toUpperCase()];
-    return platformType ? platformType.toString() : '1'; //chrome
+export const getPlatformId = browser => {
+  const platformType = proto.DeviceProps.PlatformType[browser.toUpperCase()];
+  return platformType ? platformType.toString() : '1';
 };
-//# sourceMappingURL=browser-utils.js.map