@skyzopedia/baileys-mod 5.0.8 → 6.0.1

package/lib/Utils/decode-wa-message.js

@@ -1,267 +1,198 @@
-//=======================================================//
-import { areJidsSameUser, isHostedLidUser, isHostedPnUser, isJidBroadcast, isJidGroup, isJidMetaAI, isJidNewsletter, isJidStatusBroadcast, isLidUser, isPnUser } from "../WABinary/index.js";
-import { unpadRandomMax16 } from "./generics.js";
-import { proto } from "../../WAProto/index.js";
-import { Boom } from "@hapi/boom";
-//=======================================================//
-export const getDecryptionJid = async (sender, repository) => {
-  if (isLidUser(sender) || isHostedLidUser(sender)) {
-    return sender;
-  }
-  const mapped = await repository.lidMapping.getLIDForPN(sender);
-  return mapped || sender;
-};
-//=======================================================//
-const storeMappingFromEnvelope = async (stanza, sender, repository, decryptionJid, logger) => {
-  const { senderAlt } = extractAddressingContext(stanza);
-  if (senderAlt && isLidUser(senderAlt) && isPnUser(sender) && decryptionJid === sender) {
-    try {
-      await repository.lidMapping.storeLIDPNMappings([{ lid: senderAlt, pn: sender }]);
-      await repository.migrateSession(sender, senderAlt);
-      logger.debug({ sender, senderAlt }, "Stored LID mapping from envelope");
-    }
-    catch (error) {
-      logger.warn({ sender, senderAlt, error }, "Failed to store LID mapping");
-    }
-  }
-};
-//=======================================================//
-export const NO_MESSAGE_FOUND_ERROR_TEXT = "Message absent from node";
-export const MISSING_KEYS_ERROR_TEXT = "Key used already or never filled";
-//=======================================================//
-export const DECRYPTION_RETRY_CONFIG = {
-  maxRetries: 3,
-  baseDelayMs: 100,
-  sessionRecordErrors: ["No session record", "SessionError: No session record"]
-};
-//=======================================================//
-export const NACK_REASONS = {
-  ParsingError: 487,
-  UnrecognizedStanza: 488,
-  UnrecognizedStanzaClass: 489,
-  UnrecognizedStanzaType: 490,
-  InvalidProtobuf: 491,
-  InvalidHostedCompanionStanza: 493,
-  MissingMessageSecret: 495,
-  SignalErrorOldCounter: 496,
-  MessageDeletedOnPeer: 499,
-  UnhandledError: 500,
-  UnsupportedAdminRevoke: 550,
-  UnsupportedLIDGroup: 551,
-  DBOperationFailed: 552
-};
-//=======================================================//
-export const extractAddressingContext = (stanza) => {
-  let senderAlt;
-  let recipientAlt;
-  const sender = stanza.attrs.participant || stanza.attrs.from;
-  const addressingMode = stanza.attrs.addressing_mode || (sender?.endsWith("lid") ? "lid" : "pn");
-  if (addressingMode === "lid") {
-    senderAlt = stanza.attrs.participant_pn || stanza.attrs.sender_pn || stanza.attrs.peer_recipient_pn;
-    recipientAlt = stanza.attrs.recipient_pn;
-  }
-  else {
-    senderAlt = stanza.attrs.participant_lid || stanza.attrs.sender_lid || stanza.attrs.peer_recipient_lid;
-    recipientAlt = stanza.attrs.recipient_lid;
-  }
-  return {
-    addressingMode,
-    senderAlt,
-    recipientAlt
-  };
-};
-//=======================================================//
-export function decodeMessageNode(stanza, meId, meLid) {
-  let msgType;
-  let chatId;
-  let author;
-  let fromMe = false;
-  const msgId = stanza.attrs.id;
-  const from = stanza.attrs.from;
-  const participant = stanza.attrs.participant;
-  const recipient = stanza.attrs.recipient;
-  const addressingContext = extractAddressingContext(stanza);
-  const isMe = (jid) => areJidsSameUser(jid, meId);
-  const isMeLid = (jid) => areJidsSameUser(jid, meLid);
-  if (isPnUser(from) || isLidUser(from) || isHostedLidUser(from) || isHostedPnUser(from)) {
-    if (recipient && !isJidMetaAI(recipient)) {
-      if (!isMe(from) && !isMeLid(from)) {
-        throw new Boom("receipient present, but msg not from me", { data: stanza });
-      }
-      if (isMe(from) || isMeLid(from)) {
-        fromMe = true;
-      }
-      chatId = recipient;
-    }
-    else {
-      chatId = from;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.decryptMessageNode = exports.decodeMessageNode = void 0;
+const boom_1 = require("@hapi/boom");
+const WAProto_1 = require("../../WAProto");
+const WABinary_1 = require("../WABinary");
+const generics_1 = require("./generics");
+const NO_MESSAGE_FOUND_ERROR_TEXT = 'Message absent from node';
+/**
+ * Decode the received node as a message.
+ * @note this will only parse the message, not decrypt it
+ */
+function decodeMessageNode(stanza, meId, meLid) {
+    var _a, _b;
+    let msgType;
+    let chatId;
+    let author;
+    const msgId = stanza.attrs.id;
+    const from = stanza.attrs.from;
+    const participant = stanza.attrs.participant;
+    const recipient = stanza.attrs.recipient;
+    const isMe = (jid) => (0, WABinary_1.areJidsSameUser)(jid, meId);
+    const isMeLid = (jid) => (0, WABinary_1.areJidsSameUser)(jid, meLid);
+    if ((0, WABinary_1.isJidUser)(from)) {
+        if (recipient) {
+            if (!isMe(from)) {
+                throw new boom_1.Boom('receipient present, but msg not from me', { data: stanza });
+            }
+            chatId = recipient;
+        }
+        else {
+            chatId = from;
+        }
+        msgType = 'chat';
+        author = from;
     }
-    msgType = "chat";
-    author = from;
-  }
-  else if (isJidGroup(from)) {
-    if (!participant) {
-      throw new Boom("No participant in group message");
+    else if ((0, WABinary_1.isLidUser)(from)) {
+        if (recipient) {
+            if (!isMeLid(from)) {
+                throw new boom_1.Boom('receipient present, but msg not from me', { data: stanza });
+            }
+            chatId = recipient;
+        }
+        else {
+            chatId = from;
+        }
+        msgType = 'chat';
+        author = from;
     }
-    if (isMe(participant) || isMeLid(participant)) {
-      fromMe = true;
+    else if ((0, WABinary_1.isJidGroup)(from)) {
+        if (!participant) {
+            throw new boom_1.Boom('No participant in group message');
+        }
+        msgType = 'group';
+        author = participant;
+        chatId = from;
     }
-    msgType = "group";
-    author = participant;
-    chatId = from;
-  }
-  else if (isJidBroadcast(from)) {
-    if (!participant) {
-      throw new Boom("No participant in group message");
+    else if ((0, WABinary_1.isJidBroadcast)(from)) {
+        if (!participant) {
+            throw new boom_1.Boom('No participant in group message');
+        }
+        const isParticipantMe = isMe(participant);
+        if ((0, WABinary_1.isJidStatusBroadcast)(from)) {
+            msgType = isParticipantMe ? 'direct_peer_status' : 'other_status';
+        }
+        else {
+            msgType = isParticipantMe ? 'peer_broadcast' : 'other_broadcast';
+        }
+        chatId = from;
+        author = participant;
     }
-    const isParticipantMe = isMe(participant);
-    if (isJidStatusBroadcast(from)) {
-      msgType = isParticipantMe ? "direct_peer_status" : "other_status";
+    else if ((0, WABinary_1.isJidNewsLetter)(from)) {
+        msgType = 'newsletter';
+        author = from;
+        chatId = from;
     }
     else {
-      msgType = isParticipantMe ? "peer_broadcast" : "other_broadcast";
+        throw new boom_1.Boom('Unknown message type', { data: stanza });
+    }
+    const fromMe = (0, WABinary_1.isJidNewsLetter)(from) ? !!((_a = stanza.attrs) === null || _a === void 0 ? void 0 : _a.is_sender) : ((0, WABinary_1.isLidUser)(from) ? isMeLid : isMe)(stanza.attrs.participant || stanza.attrs.from);
+    const pushname = stanza.attrs.notify;
+    const key = {
+        remoteJid: chatId,
+        fromMe,
+        id: msgId,
+        participant
+    };
+    const fullMessage = {
+        key,
+        messageTimestamp: +stanza.attrs.t,
+        pushName: pushname,
+        broadcast: (0, WABinary_1.isJidBroadcast)(from)
+    };
+    if (msgType === 'newsletter') {
+        fullMessage.newsletterServerId = +((_b = stanza.attrs) === null || _b === void 0 ? void 0 : _b.server_id);
     }
-    fromMe = isParticipantMe;
-    chatId = from;
-    author = participant;
-  }
-  else if (isJidNewsletter(from)) {
-    msgType = "newsletter";
-    chatId = from;
-    author = from;
-    if (isMe(from) || isMeLid(from)) {
-      fromMe = true;
+    if (key.fromMe) {
+        fullMessage.status = WAProto_1.proto.WebMessageInfo.Status.SERVER_ACK;
     }
-  }
-  else {
-    throw new Boom("Unknown message type", { data: stanza });
-  }
-  const pushname = stanza?.attrs?.notify;
-  const key = {
-    remoteJid: chatId,
-    remoteJidAlt: !isJidGroup(chatId) ? addressingContext.senderAlt : undefined,
-    fromMe,
-    id: msgId,
-    participant,
-    participantAlt: isJidGroup(chatId) ? addressingContext.senderAlt : undefined,
-    addressingMode: addressingContext.addressingMode,
-    ...(msgType === "newsletter" && stanza.attrs.server_id ? { server_id: stanza.attrs.server_id } : {})
-  };
-  const fullMessage = {
-    key,
-    messageTimestamp: +stanza.attrs.t,
-    pushName: pushname,
-    broadcast: isJidBroadcast(from)
-  };
-  if (key.fromMe) {
-    fullMessage.status = proto.WebMessageInfo.Status.SERVER_ACK;
-  }
-  return {
-    fullMessage,
-    author,
-    sender: msgType === "chat" ? author : chatId
-  };
+    return {
+        fullMessage,
+        author,
+        sender: msgType === 'chat' ? author : chatId
+    };
 }
-//=======================================================//
-export const decryptMessageNode = (stanza, meId, meLid, repository, logger) => {
-  const { fullMessage, author, sender } = decodeMessageNode(stanza, meId, meLid);
-  return {
-    fullMessage,
-    category: stanza.attrs.category,
-    author,
-    async decrypt() {
-      let decryptables = 0;
-      if (Array.isArray(stanza.content)) {
-        for (const { tag, attrs, content } of stanza.content) {
-          if (tag === "verified_name" && content instanceof Uint8Array) {
-            const cert = proto.VerifiedNameCertificate.decode(content);
-            const details = proto.VerifiedNameCertificate.Details.decode(cert.details);
-            fullMessage.verifiedBizName = details.verifiedName;
-          }
-          if (tag === "unavailable" && attrs.type === "view_once") {
-            fullMessage.key.isViewOnce = true;
-          }
-          if (tag !== "enc" && tag !== "plaintext") {
-            continue;
-          }
-          if (!(content instanceof Uint8Array)) {
-            continue;
-          }
-          decryptables += 1;
-          let msgBuffer;
-          const decryptionJid = await getDecryptionJid(author, repository);
-          if (tag !== "plaintext") {
-            await storeMappingFromEnvelope(stanza, author, repository, decryptionJid, logger);
-          }
-          try {
-            const e2eType = tag === "plaintext" ? "plaintext" : attrs.type;
-            switch (e2eType) {
-              case "skmsg":
-                msgBuffer = await repository.decryptGroupMessage({
-                  group: sender,
-                  authorJid: author,
-                  msg: content
-                });
-                break;
-              case "pkmsg":
-              case "msg":
-                msgBuffer = await repository.decryptMessage({
-                  jid: decryptionJid,
-                  type: e2eType,
-                  ciphertext: content
-                });
-                break;
-              case "plaintext":
-                msgBuffer = content;
-                break;
-              default:
-                throw new Error(`Unknown e2e type: ${e2eType}`);
+exports.decodeMessageNode = decodeMessageNode;
+const decryptMessageNode = (stanza, meId, meLid, repository, logger) => {
+    const { fullMessage, author, sender } = decodeMessageNode(stanza, meId, meLid);
+    return {
+        fullMessage,
+        category: stanza.attrs.category,
+        author,
+        async decrypt() {
+            var _a;
+            let decryptables = 0;
+            async function processSenderKeyDistribution(msg) {
+                if (msg.senderKeyDistributionMessage) {
+                    try {
+                        await repository.processSenderKeyDistributionMessage({
+                            authorJid: author,
+                            item: msg.senderKeyDistributionMessage
+                        });
+                    }
+                    catch (err) {
+                        logger.error({ key: fullMessage.key, err }, 'failed to process senderKeyDistribution');
+                    }
+                }
             }
-            let msg = proto.Message.decode(e2eType !== "plaintext" ? unpadRandomMax16(msgBuffer) : msgBuffer);
-            msg = msg.deviceSentMessage?.message || msg;
-            if (msg.senderKeyDistributionMessage) {
-              try {
-                await repository.processSenderKeyDistributionMessage({
-                  authorJid: author,
-                  item: msg.senderKeyDistributionMessage
-                });
-              }
-              catch (err) {
-                logger.error({ key: fullMessage.key, err }, "failed to process sender key distribution message");
-              }
+            if ((0, WABinary_1.isJidNewsLetter)(fullMessage.key.remoteJid)) {
+                const node = (0, WABinary_1.getBinaryNodeChild)(stanza, 'plaintext');
+                const msg = WAProto_1.proto.Message.decode(node === null || node === void 0 ? void 0 : node.content);
+                await processSenderKeyDistribution(msg);
+                fullMessage.message = msg;
+                decryptables += 1;
             }
-            if (fullMessage.message) {
-              Object.assign(fullMessage.message, msg);
+            else if (Array.isArray(stanza.content)) {
+                for (const { tag, attrs, content } of stanza.content) {
+                    if (tag === 'verified_name' && content instanceof Uint8Array) {
+                        const cert = WAProto_1.proto.VerifiedNameCertificate.decode(content);
+                        const details = WAProto_1.proto.VerifiedNameCertificate.Details.decode(cert.details);
+                        fullMessage.verifiedBizName = details.verifiedName;
+                    }
+                    if (tag !== 'enc') {
+                        continue;
+                    }
+                    if (!(content instanceof Uint8Array)) {
+                        continue;
+                    }
+                    decryptables += 1;
+                    let msgBuffer;
+                    try {
+                        const e2eType = attrs.type;
+                        switch (e2eType) {
+                            case 'skmsg':
+                                msgBuffer = await repository.decryptGroupMessage({
+                                    group: sender,
+                                    authorJid: author,
+                                    msg: content
+                                });
+                                break;
+                            case 'pkmsg':
+                            case 'msg':
+                                const user = (0, WABinary_1.isJidUser)(sender) ? sender : author;
+                                msgBuffer = await repository.decryptMessage({
+                                    jid: user,
+                                    type: e2eType,
+                                    ciphertext: content
+                                });
+                                break;
+                            default:
+                                throw new Error(`Unknown e2e type: ${e2eType}`);
+                        }
+                        let msg = WAProto_1.proto.Message.decode((0, generics_1.unpadRandomMax16)(msgBuffer));
+                        msg = ((_a = msg.deviceSentMessage) === null || _a === void 0 ? void 0 : _a.message) || msg;
+                        await processSenderKeyDistribution(msg);
+                        if (fullMessage.message) {
+                            Object.assign(fullMessage.message, msg);
+                        }
+                        else {
+                            fullMessage.message = msg;
+                        }
+                    }
+                    catch (err) {
+                        logger.error({ key: fullMessage.key, err }, 'failed to decrypt message');
+                        fullMessage.messageStubType = WAProto_1.proto.WebMessageInfo.StubType.CIPHERTEXT;
+                        fullMessage.messageStubParameters = [err.message];
+                    }
+                }
             }
-            else {
-              fullMessage.message = msg;
+            // if nothing was found to decrypt
+            if (!decryptables) {
+                fullMessage.messageStubType = WAProto_1.proto.WebMessageInfo.StubType.CIPHERTEXT;
+                fullMessage.messageStubParameters = [NO_MESSAGE_FOUND_ERROR_TEXT, JSON.stringify(stanza, generics_1.BufferJSON.replacer)];
             }
-          }
-          catch (err) {
-            const errorContext = {
-              key: fullMessage.key,
-              err,
-              messageType: tag === "plaintext" ? "plaintext" : attrs.type,
-              sender,
-              author,
-              isSessionRecordError: isSessionRecordError(err)
-            };
-            logger.error(errorContext, "failed to decrypt message");
-            fullMessage.messageStubType = proto.WebMessageInfo.StubType.CIPHERTEXT;
-            fullMessage.messageStubParameters = [err.message.toString()];
-          }
         }
-      }
-      if (!decryptables) {
-        fullMessage.messageStubType = proto.WebMessageInfo.StubType.CIPHERTEXT;
-        fullMessage.messageStubParameters = [NO_MESSAGE_FOUND_ERROR_TEXT];
-      }
-    }
-  };
+    };
 };
-//=======================================================//
-function isSessionRecordError(error) {
-  const errorMessage = error?.message || error?.toString() || "";
-  return DECRYPTION_RETRY_CONFIG.sessionRecordErrors.some(errorPattern => errorMessage.includes(errorPattern));
-}
-//=======================================================//
+exports.decryptMessageNode = decryptMessageNode;

package/lib/Utils/event-buffer.d.ts

@@ -0,0 +1,35 @@
+import { Logger } from 'pino';
+import { BaileysEventEmitter, BaileysEventMap } from '../Types';
+/**
+ * A map that contains a list of all events that have been triggered
+ *
+ * Note, this can contain different type of events
+ * this can make processing events extremely efficient -- since everything
+ * can be done in a single transaction
+ */
+type BaileysEventData = Partial<BaileysEventMap>;
+type BaileysBufferableEventEmitter = BaileysEventEmitter & {
+    /** Use to process events in a batch */
+    process(handler: (events: BaileysEventData) => void | Promise<void>): (() => void);
+    /**
+     * starts buffering events, call flush() to release them
+     * */
+    buffer(): void;
+    /** buffers all events till the promise completes */
+    createBufferedFunction<A extends any[], T>(work: (...args: A) => Promise<T>): ((...args: A) => Promise<T>);
+    /**
+     * flushes all buffered events
+     * @param force if true, will flush all data regardless of any pending buffers
+     * @returns returns true if the flush actually happened, otherwise false
+     */
+    flush(force?: boolean): boolean;
+    /** is there an ongoing buffer */
+    isBuffering(): boolean;
+};
+/**
+ * The event buffer logically consolidates different events into a single event
+ * making the data processing more efficient.
+ * @param ev the baileys event emitter
+ */
+export declare const makeEventBuffer: (logger: Logger) => BaileysBufferableEventEmitter;
+export {};