@mrlegendbot/baileys 1.2.5 → 1.2.7

package/lib/Utils/pre-key-manager.js

@@ -1,77 +1,106 @@
 import PQueue from 'p-queue';
+/**
+ * Manages pre-key operations with proper concurrency control
+ */
 export class PreKeyManager {
-  constructor(store, logger) {
-    this.store = store;
-    this.logger = logger;
-    this.queues = new Map();
-  }
-  getQueue(keyType) {
-    if (!this.queues.has(keyType)) {
-      this.queues.set(keyType, new PQueue({
-        concurrency: 1
-      }));
+    constructor(store, logger) {
+        this.store = store;
+        this.logger = logger;
+        this.queues = new Map();
     }
-    return this.queues.get(keyType);
-  }
-  async processOperations(data, keyType, transactionCache, mutations, isInTransaction) {
-    const keyData = data[keyType];
-    if (!keyData) return;
-    return this.getQueue(keyType).add(async () => {
-      transactionCache[keyType] = transactionCache[keyType] || {};
-      mutations[keyType] = mutations[keyType] || {};
-      const deletions = [];
-      const updates = {};
-      for (const keyId in keyData) {
-        if (keyData[keyId] === null) {
-          deletions.push(keyId);
-        } else {
-          updates[keyId] = keyData[keyId];
+    /**
+     * Get or create a queue for a specific key type
+     */
+    getQueue(keyType) {
+        if (!this.queues.has(keyType)) {
+            this.queues.set(keyType, new PQueue({ concurrency: 1 }));
         }
-      }
-      if (Object.keys(updates).length > 0) {
-        Object.assign(transactionCache[keyType], updates);
-        Object.assign(mutations[keyType], updates);
-      }
-      if (deletions.length > 0) {
-        await this.processDeletions(keyType, deletions, transactionCache, mutations, isInTransaction);
-      }
-    });
-  }
-  async processDeletions(keyType, ids, transactionCache, mutations, isInTransaction) {
-    if (isInTransaction) {
-      for (const keyId of ids) {
-        if (transactionCache[keyType]?.[keyId]) {
-          transactionCache[keyType][keyId] = null;
-          mutations[keyType][keyId] = null;
-        } else {
-          this.logger.warn(`Skipping deletion of non-existent ${keyType} in transaction: ${keyId}`);
+        return this.queues.get(keyType);
+    }
+    /**
+     * Process pre-key operations (updates and deletions)
+     */
+    async processOperations(data, keyType, transactionCache, mutations, isInTransaction) {
+        const keyData = data[keyType];
+        if (!keyData)
+            return;
+        return this.getQueue(keyType).add(async () => {
+            // Ensure structures exist
+            transactionCache[keyType] = transactionCache[keyType] || {};
+            mutations[keyType] = mutations[keyType] || {};
+            // Separate deletions from updates
+            const deletions = [];
+            const updates = {};
+            for (const keyId in keyData) {
+                if (keyData[keyId] === null) {
+                    deletions.push(keyId);
+                }
+                else {
+                    updates[keyId] = keyData[keyId];
+                }
+            }
+            // Process updates (no validation needed)
+            if (Object.keys(updates).length > 0) {
+                Object.assign(transactionCache[keyType], updates);
+                Object.assign(mutations[keyType], updates);
+            }
+            // Process deletions with validation
+            if (deletions.length > 0) {
+                await this.processDeletions(keyType, deletions, transactionCache, mutations, isInTransaction);
+            }
+        });
+    }
+    /**
+     * Process deletions with validation
+     */
+    async processDeletions(keyType, ids, transactionCache, mutations, isInTransaction) {
+        if (isInTransaction) {
+            // In transaction, only allow deletion if key exists in cache
+            for (const keyId of ids) {
+                if (transactionCache[keyType]?.[keyId]) {
+                    transactionCache[keyType][keyId] = null;
+                    mutations[keyType][keyId] = null;
+                }
+                else {
+                    this.logger.warn(`Skipping deletion of non-existent ${keyType} in transaction: ${keyId}`);
+                }
+            }
         }
-      }
-    } else {
-      const existingKeys = await this.store.get(keyType, ids);
-      for (const keyId of ids) {
-        if (existingKeys[keyId]) {
-          transactionCache[keyType][keyId] = null;
-          mutations[keyType][keyId] = null;
-        } else {
-          this.logger.warn(`Skipping deletion of non-existent ${keyType}: ${keyId}`);
+        else {
+            // Outside transaction, validate against store
+            const existingKeys = await this.store.get(keyType, ids);
+            for (const keyId of ids) {
+                if (existingKeys[keyId]) {
+                    transactionCache[keyType][keyId] = null;
+                    mutations[keyType][keyId] = null;
+                }
+                else {
+                    this.logger.warn(`Skipping deletion of non-existent ${keyType}: ${keyId}`);
+                }
+            }
         }
-      }
     }
-  }
-  async validateDeletions(data, keyType) {
-    const keyData = data[keyType];
-    if (!keyData) return;
-    return this.getQueue(keyType).add(async () => {
-      const deletionIds = Object.keys(keyData).filter(id => keyData[id] === null);
-      if (deletionIds.length === 0) return;
-      const existingKeys = await this.store.get(keyType, deletionIds);
-      for (const keyId of deletionIds) {
-        if (!existingKeys[keyId]) {
-          this.logger.warn(`Skipping deletion of non-existent ${keyType}: ${keyId}`);
-          delete data[keyType][keyId];
-        }
-      }
-    });
-  }
+    /**
+     * Validate and process pre-key deletions outside transactions
+     */
+    async validateDeletions(data, keyType) {
+        const keyData = data[keyType];
+        if (!keyData)
+            return;
+        return this.getQueue(keyType).add(async () => {
+            // Find all deletion requests
+            const deletionIds = Object.keys(keyData).filter(id => keyData[id] === null);
+            if (deletionIds.length === 0)
+                return;
+            // Validate deletions
+            const existingKeys = await this.store.get(keyType, deletionIds);
+            for (const keyId of deletionIds) {
+                if (!existingKeys[keyId]) {
+                    this.logger.warn(`Skipping deletion of non-existent ${keyType}: ${keyId}`);
+                    delete data[keyType][keyId];
+                }
+            }
+        });
+    }
 }
+//# sourceMappingURL=pre-key-manager.js.map

package/lib/Utils/process-message.d.ts

@@ -0,0 +1,60 @@
+import { proto } from '../../WAProto/index.js';
+import type { AuthenticationCreds, BaileysEventEmitter, CacheStore, SignalKeyStoreWithTransaction, SignalRepositoryWithLIDStore, SocketConfig, WAMessage, WAMessageKey } from '../Types/index.js';
+import type { ILogger } from './logger.js';
+type ProcessMessageContext = {
+    shouldProcessHistoryMsg: boolean;
+    placeholderResendCache?: CacheStore;
+    creds: AuthenticationCreds;
+    keyStore: SignalKeyStoreWithTransaction;
+    ev: BaileysEventEmitter;
+    logger?: ILogger;
+    options: RequestInit;
+    signalRepository: SignalRepositoryWithLIDStore;
+    getMessage: SocketConfig['getMessage'];
+};
+/** Cleans a received message to further processing */
+export declare const cleanMessage: (message: WAMessage, meId: string, meLid: string) => void;
+export declare const isRealMessage: (message: WAMessage) => boolean;
+export declare const shouldIncrementChatUnread: (message: WAMessage) => boolean;
+/**
+ * Get the ID of the chat from the given key.
+ * Typically -- that'll be the remoteJid, but for broadcasts, it'll be the participant
+ */
+export declare const getChatId: ({ remoteJid, participant, fromMe }: WAMessageKey) => string;
+type PollContext = {
+    /** normalised jid of the person that created the poll */
+    pollCreatorJid: string;
+    /** ID of the poll creation message */
+    pollMsgId: string;
+    /** poll creation message enc key */
+    pollEncKey: Uint8Array;
+    /** jid of the person that voted */
+    voterJid: string;
+};
+type EventContext = {
+    /** normalised jid of the person that created the event */
+    eventCreatorJid: string;
+    /** ID of the event creation message */
+    eventMsgId: string;
+    /** event creation message enc key */
+    eventEncKey: Uint8Array;
+    /** jid of the person that responded */
+    responderJid: string;
+};
+/**
+ * Decrypt a poll vote
+ * @param vote encrypted vote
+ * @param ctx additional info about the poll required for decryption
+ * @returns list of SHA256 options
+ */
+export declare function decryptPollVote({ encPayload, encIv }: proto.Message.IPollEncValue, { pollCreatorJid, pollMsgId, pollEncKey, voterJid }: PollContext): proto.Message.PollVoteMessage;
+/**
+ * Decrypt an event response
+ * @param response encrypted event response
+ * @param ctx additional info about the event required for decryption
+ * @returns event response message
+ */
+export declare function decryptEventResponse({ encPayload, encIv }: proto.Message.IPollEncValue, { eventCreatorJid, eventMsgId, eventEncKey, responderJid }: EventContext): proto.Message.EventResponseMessage;
+declare const processMessage: (message: WAMessage, { shouldProcessHistoryMsg, placeholderResendCache, ev, creds, signalRepository, keyStore, logger, options, getMessage }: ProcessMessageContext) => Promise<void>;
+export default processMessage;
+//# sourceMappingURL=process-message.d.ts.map