@prototypevip/baileys 0.0.1

package/lib/Utils/process-message.js

@@ -0,0 +1,648 @@
+import { Boom } from "@hapi/boom";
+import { proto } from "../../WAProto/index.js";
+import { WAMessageStubType } from "../Types/index.js";
+import { getContentType, normalizeMessageContent } from "../Utils/messages.js";
+import {
+  areJidsSameUser,
+  isHostedLidUser,
+  isHostedPnUser,
+  isJidBroadcast,
+  isJidStatusBroadcast,
+  isLidUser,
+  jidDecode,
+  jidEncode,
+  jidNormalizedUser,
+} from "../WABinary/index.js";
+import { aesDecryptGCM, hmacSign } from "./crypto.js";
+import { getKeyAuthor, toNumber } from "./generics.js";
+import { downloadAndProcessHistorySyncNotification } from "./history.js";
+import {
+  buildMergedTcTokenIndexWrite,
+  resolveTcTokenJid,
+} from "./tc-token-utils.js";
+import { normalizeMessageIdentity } from "./normalize-message-identity.js";
+const REAL_MSG_STUB_TYPES = new Set([
+  WAMessageStubType.CALL_MISSED_GROUP_VIDEO,
+  WAMessageStubType.CALL_MISSED_GROUP_VOICE,
+  WAMessageStubType.CALL_MISSED_VIDEO,
+  WAMessageStubType.CALL_MISSED_VOICE,
+]);
+const REAL_MSG_REQ_ME_STUB_TYPES = new Set([
+  WAMessageStubType.GROUP_PARTICIPANT_ADD,
+]);
+async function storeTcTokensFromHistorySync(
+  chats,
+  signalRepository,
+  keyStore,
+  logger,
+) {
+  const getLIDForPN = signalRepository.lidMapping.getLIDForPN.bind(
+    signalRepository.lidMapping,
+  );
+  const candidates = [];
+  for (const chat of chats) {
+    const ts = chat.tcTokenTimestamp ? toNumber(chat.tcTokenTimestamp) : 0;
+    if (chat.tcToken?.length && ts > 0) {
+      const jid = jidNormalizedUser(chat.id);
+      const storageJid = await resolveTcTokenJid(jid, getLIDForPN);
+      candidates.push({
+        storageJid,
+        token: Buffer.from(chat.tcToken),
+        ts,
+        senderTs: chat.tcTokenSenderTimestamp
+          ? toNumber(chat.tcTokenSenderTimestamp)
+          : undefined,
+      });
+    }
+  }
+  if (!candidates.length) {
+    return;
+  }
+  const jids = candidates.map((c) => c.storageJid);
+  const existing = await keyStore.get("tctoken", jids);
+  const entries = {};
+  for (const c of candidates) {
+    const existingEntry = existing[c.storageJid];
+    const existingTs = existingEntry?.timestamp
+      ? Number(existingEntry.timestamp)
+      : 0;
+    if (existingTs > 0 && existingTs >= c.ts) {
+      continue;
+    }
+    entries[c.storageJid] = {
+      ...existingEntry,
+      token: c.token,
+      timestamp: String(c.ts),
+      ...(c.senderTs !== undefined ? { senderTimestamp: c.senderTs } : {}),
+    };
+  }
+  if (Object.keys(entries).length) {
+    logger?.debug(
+      { count: Object.keys(entries).length },
+      "storing tctokens from history sync",
+    );
+    try {
+      const indexWrite = await buildMergedTcTokenIndexWrite(
+        keyStore,
+        Object.keys(entries),
+      );
+      await keyStore.set({ tctoken: { ...entries, ...indexWrite } });
+    } catch (err) {
+      logger?.warn({ err }, "failed to store tctokens from history sync");
+    }
+  }
+}
+/** Cleans a received message to further processing */
+export const cleanMessage = (message, meId, meLid) => {
+  if (
+    isHostedPnUser(message.key.remoteJid) ||
+    isHostedLidUser(message.key.remoteJid)
+  ) {
+    message.key.remoteJid = jidEncode(
+      jidDecode(message.key?.remoteJid)?.user,
+      isHostedPnUser(message.key.remoteJid) ? "s.whatsapp.net" : "lid",
+    );
+  } else {
+    message.key.remoteJid = jidNormalizedUser(message.key.remoteJid);
+  }
+  if (
+    isHostedPnUser(message.key.participant) ||
+    isHostedLidUser(message.key.participant)
+  ) {
+    message.key.participant = jidEncode(
+      jidDecode(message.key.participant)?.user,
+      isHostedPnUser(message.key.participant) ? "s.whatsapp.net" : "lid",
+    );
+  } else {
+    message.key.participant = jidNormalizedUser(message.key.participant);
+  }
+  const content = normalizeMessageContent(message.message);
+  if (content?.reactionMessage) {
+    normaliseKey(content.reactionMessage.key);
+  }
+  if (content?.pollUpdateMessage) {
+    normaliseKey(content.pollUpdateMessage.pollCreationMessageKey);
+  }
+  function normaliseKey(msgKey) {
+    if (!message.key.fromMe) {
+      msgKey.fromMe = !msgKey.fromMe
+        ? areJidsSameUser(msgKey.participant || msgKey.remoteJid, meId) ||
+          areJidsSameUser(msgKey.participant || msgKey.remoteJid, meLid)
+        : msgKey.fromMe;
+    }
+  }
+  if (
+    isJidBroadcast(message.key.remoteJid) ||
+    isJidStatusBroadcast(message.key.remoteJid)
+  ) {
+    message.key.participant = jidNormalizedUser(
+      message.key.participant || message.participant,
+    );
+  }
+};
+export function decryptEventResponse(
+  { encPayload, encIv },
+  { eventCreatorJid, eventMsgId, eventEncKey, responderJid },
+) {
+  const sign = Buffer.concat([
+    toBinary(eventMsgId),
+    toBinary(eventCreatorJid),
+    toBinary(responderJid),
+    toBinary("Event Response"),
+    new Uint8Array([1]),
+  ]);
+  const key0 = hmacSign(eventEncKey, new Uint8Array(32), "sha256");
+  const decKey = hmacSign(sign, key0, "sha256");
+  const aad = toBinary(`${eventMsgId}\u0000${responderJid}`);
+  const decrypted = aesDecryptGCM(encPayload, decKey, encIv, aad);
+  return proto.Message.EventResponseMessage.decode(decrypted);
+  function toBinary(txt) {
+    return Buffer.from(txt);
+  }
+}
+export const isRealMessage = (message) => {
+  const normalizedContent = normalizeMessageContent(message.message);
+  const hasSomeContent = !!getContentType(normalizedContent);
+  return (
+    (!!normalizedContent ||
+      REAL_MSG_STUB_TYPES.has(message.messageStubType) ||
+      REAL_MSG_REQ_ME_STUB_TYPES.has(message.messageStubType)) &&
+    hasSomeContent &&
+    !normalizedContent?.protocolMessage &&
+    !normalizedContent?.reactionMessage &&
+    !normalizedContent?.pollUpdateMessage
+  );
+};
+export const shouldIncrementChatUnread = (message) =>
+  !message.key.fromMe && !message.messageStubType;
+
+const processMessage = async (
+  message,
+  {
+    shouldProcessHistoryMsg,
+    placeholderResendCache,
+    ev,
+    creds,
+    signalRepository,
+    keyStore,
+    logger,
+    options,
+    getMessage,
+  },
+) => {
+  const meId = creds.me.id;
+  const { accountSettings } = creds;
+  const chat = { id: jidNormalizedUser(getChatId(message.key)) };
+  const isRealMsg = isRealMessage(message);
+  if (isRealMsg) {
+    chat.messages = [{ message }];
+    chat.conversationTimestamp = toNumber(message.messageTimestamp);
+    if (shouldIncrementChatUnread(message)) {
+      chat.unreadCount = (chat.unreadCount || 0) + 1;
+    }
+  }
+  const content = normalizeMessageContent(message.message);
+  if (
+    (isRealMsg || content?.reactionMessage?.key?.fromMe) &&
+    accountSettings?.unarchiveChats
+  ) {
+    chat.archived = false;
+    chat.readOnly = false;
+  }
+  const protocolMsg = content?.protocolMessage;
+  if (protocolMsg) {
+    const SELF_ONLY_TYPES = new Set([
+      proto.Message.ProtocolMessage.Type.HISTORY_SYNC_NOTIFICATION,
+      proto.Message.ProtocolMessage.Type.APP_STATE_SYNC_KEY_SHARE,
+      proto.Message.ProtocolMessage.Type.LID_MIGRATION_MAPPING_SYNC,
+      proto.Message.ProtocolMessage.Type
+        .PEER_DATA_OPERATION_REQUEST_RESPONSE_MESSAGE,
+    ]);
+    if (
+      protocolMsg.type !== null &&
+      protocolMsg.type !== undefined &&
+      SELF_ONLY_TYPES.has(protocolMsg.type) &&
+      !message.key.fromMe
+    ) {
+      logger?.warn(
+        {
+          msgId: message.key.id,
+          type: protocolMsg.type,
+          from: message.key.participant || message.key.remoteJid,
+        },
+        "dropping spoofed self-only protocolMessage from non-self origin",
+      );
+      return;
+    }
+    switch (protocolMsg.type) {
+      case proto.Message.ProtocolMessage.Type.HISTORY_SYNC_NOTIFICATION:
+        const histNotification = protocolMsg.historySyncNotification;
+        const process = shouldProcessHistoryMsg;
+        const isLatest = !creds.processedHistoryMessages?.length;
+        logger?.info(
+          {
+            histNotification,
+            process,
+            id: message.key.id,
+            isLatest,
+          },
+          "got history notification",
+        );
+        if (process) {
+          if (
+            histNotification.syncType !==
+            proto.HistorySync.HistorySyncType.ON_DEMAND
+          ) {
+            ev.emit("creds.update", {
+              processedHistoryMessages: [
+                ...(creds.processedHistoryMessages || []),
+                {
+                  key: message.key,
+                  messageTimestamp: message.messageTimestamp,
+                },
+              ],
+            });
+          }
+          const data = await downloadAndProcessHistorySyncNotification(
+            histNotification,
+            options,
+            logger,
+          );
+          if (data.lidPnMappings?.length) {
+            logger?.debug(
+              { count: data.lidPnMappings.length },
+              "processing LID-PN mappings from history sync",
+            );
+            await signalRepository.lidMapping
+              .storeLIDPNMappings(data.lidPnMappings)
+              .catch((err) =>
+                logger?.warn(
+                  { err },
+                  "failed to store LID-PN mappings from history sync",
+                ),
+              );
+          }
+          await storeTcTokensFromHistorySync(
+            data.chats,
+            signalRepository,
+            keyStore,
+            logger,
+          );
+          if (data.nctSalt?.length) {
+            ev.emit("creds.update", { nctSalt: data.nctSalt });
+          }
+          ev.emit("messaging-history.set", {
+            ...data,
+            isLatest:
+              histNotification.syncType !==
+              proto.HistorySync.HistorySyncType.ON_DEMAND
+                ? isLatest
+                : undefined,
+            chunkOrder: histNotification.chunkOrder,
+            peerDataRequestSessionId: histNotification.peerDataRequestSessionId,
+          });
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.APP_STATE_SYNC_KEY_SHARE:
+        const keys = protocolMsg.appStateSyncKeyShare.keys;
+        if (keys?.length) {
+          let newAppStateSyncKeyId = "";
+          await keyStore.transaction(async () => {
+            const newKeys = [];
+            for (const { keyData, keyId } of keys) {
+              const strKeyId = Buffer.from(keyId.keyId).toString("base64");
+              newKeys.push(strKeyId);
+              await keyStore.set({
+                "app-state-sync-key": { [strKeyId]: keyData },
+              });
+              newAppStateSyncKeyId = strKeyId;
+            }
+            logger?.info(
+              { newAppStateSyncKeyId, newKeys },
+              "injecting new app state sync keys",
+            );
+          }, meId);
+          ev.emit("creds.update", { myAppStateKeyId: newAppStateSyncKeyId });
+        } else {
+          logger?.info({ protocolMsg }, "recv app state sync with 0 keys");
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.REVOKE:
+        ev.emit("messages.update", [
+          {
+            key: {
+              ...message.key,
+              id: protocolMsg.key.id,
+            },
+            update: {
+              message: null,
+              messageStubType: WAMessageStubType.REVOKE,
+              key: message.key,
+            },
+          },
+        ]);
+        break;
+      case proto.Message.ProtocolMessage.Type.EPHEMERAL_SETTING:
+        Object.assign(chat, {
+          ephemeralSettingTimestamp: toNumber(message.messageTimestamp),
+          ephemeralExpiration: protocolMsg.ephemeralExpiration || null,
+        });
+        break;
+      case proto.Message.ProtocolMessage.Type
+        .PEER_DATA_OPERATION_REQUEST_RESPONSE_MESSAGE:
+        const response = protocolMsg.peerDataOperationRequestResponseMessage;
+        if (response) {
+          const peerDataOperationResult =
+            response.peerDataOperationResult || [];
+          for (const result of peerDataOperationResult) {
+            const retryResponse = result?.placeholderMessageResendResponse;
+            if (!retryResponse?.webMessageInfoBytes) {
+              continue;
+            }
+            try {
+              const webMessageInfo = proto.WebMessageInfo.decode(
+                retryResponse.webMessageInfoBytes,
+              );
+              const msgId = webMessageInfo.key?.id;
+              const cachedData = msgId
+                ? await placeholderResendCache?.get(msgId)
+                : undefined;
+              if (msgId) {
+                await placeholderResendCache?.del(msgId);
+              }
+              let finalMsg;
+              if (cachedData && typeof cachedData === "object") {
+                cachedData.message = webMessageInfo.message;
+                if (webMessageInfo.messageTimestamp) {
+                  cachedData.messageTimestamp = webMessageInfo.messageTimestamp;
+                }
+                finalMsg = cachedData;
+              } else {
+                finalMsg = webMessageInfo;
+              }
+              logger?.debug(
+                { msgId, requestId: response.stanzaId },
+                "received placeholder resend",
+              );
+              ev.emit("messages.upsert", {
+                messages: [normalizeMessageIdentity(finalMsg)],
+                type: "notify",
+                requestId: response.stanzaId,
+              });
+            } catch (err) {
+              logger?.warn(
+                { err, stanzaId: response.stanzaId },
+                "failed to decode placeholder resend response",
+              );
+            }
+          }
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.MESSAGE_EDIT:
+        ev.emit("messages.update", [
+          {
+            key: { ...message.key, id: protocolMsg.key?.id },
+            update: {
+              message: {
+                editedMessage: {
+                  message: protocolMsg.editedMessage,
+                },
+              },
+              messageTimestamp: protocolMsg.timestampMs
+                ? Math.floor(toNumber(protocolMsg.timestampMs) / 1000)
+                : message.messageTimestamp,
+            },
+          },
+        ]);
+        break;
+      case proto.Message.ProtocolMessage.Type.GROUP_MEMBER_LABEL_CHANGE:
+        const labelAssociationMsg = protocolMsg.memberLabel;
+        if (labelAssociationMsg?.label) {
+          ev.emit("group.member-tag.update", {
+            groupId: chat.id,
+            label: labelAssociationMsg.label,
+            participant: message.key.participant,
+            participantAlt: message.key.participantAlt,
+            messageTimestamp: Number(message.messageTimestamp),
+          });
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.LID_MIGRATION_MAPPING_SYNC:
+        const encodedPayload =
+          protocolMsg.lidMigrationMappingSyncMessage?.encodedMappingPayload;
+        const { pnToLidMappings, chatDbMigrationTimestamp } =
+          proto.LIDMigrationMappingSyncPayload.decode(encodedPayload);
+        logger?.debug(
+          { pnToLidMappings, chatDbMigrationTimestamp },
+          "got lid mappings and chat db migration timestamp",
+        );
+        const pairs = [];
+        for (const { pn, latestLid, assignedLid } of pnToLidMappings) {
+          const lid = latestLid || assignedLid;
+          pairs.push({ lid: `${lid}@lid`, pn: `${pn}@s.whatsapp.net` });
+        }
+        await signalRepository.lidMapping.storeLIDPNMappings(pairs);
+        if (pairs.length) {
+          for (const { pn, lid } of pairs) {
+            await signalRepository.migrateSession(pn, lid);
+          }
+        }
+    }
+  } else if (content?.reactionMessage) {
+    const reaction = {
+      ...content.reactionMessage,
+      key: message.key,
+    };
+    ev.emit("messages.reaction", [
+      {
+        reaction,
+        key: content.reactionMessage?.key,
+      },
+    ]);
+  } else if (content?.encEventResponseMessage) {
+    const encEventResponse = content.encEventResponseMessage;
+    const creationMsgKey = encEventResponse.eventCreationMessageKey;
+    const eventMsg = await getMessage(creationMsgKey);
+    if (eventMsg) {
+      try {
+        const meIdNormalised = jidNormalizedUser(meId);
+        const eventCreatorKey =
+          creationMsgKey.participant || creationMsgKey.remoteJid;
+        const eventCreatorPn = isLidUser(eventCreatorKey)
+          ? await signalRepository.lidMapping.getPNForLID(eventCreatorKey)
+          : eventCreatorKey;
+        const eventCreatorJid = getKeyAuthor(
+          {
+            remoteJid: jidNormalizedUser(eventCreatorPn),
+            fromMe: meIdNormalised === eventCreatorPn,
+          },
+          meIdNormalised,
+        );
+        const responderJid = getKeyAuthor(message.key, meIdNormalised);
+        const eventEncKey = eventMsg?.messageContextInfo?.messageSecret;
+        if (!eventEncKey) {
+          logger?.warn(
+            { creationMsgKey },
+            "event response: missing messageSecret for decryption",
+          );
+        } else {
+          const responseMsg = decryptEventResponse(encEventResponse, {
+            eventEncKey,
+            eventCreatorJid,
+            eventMsgId: creationMsgKey.id,
+            responderJid,
+          });
+          const eventResponse = {
+            eventResponseMessageKey: message.key,
+            senderTimestampMs: responseMsg.timestampMs,
+            response: responseMsg,
+          };
+          ev.emit("messages.update", [
+            {
+              key: creationMsgKey,
+              update: {
+                eventResponses: [eventResponse],
+              },
+            },
+          ]);
+        }
+      } catch (err) {
+        logger?.warn(
+          { err, creationMsgKey },
+          "failed to decrypt event response",
+        );
+      }
+    } else {
+      logger?.warn(
+        { creationMsgKey },
+        "event creation message not found, cannot decrypt response",
+      );
+    }
+  } else if (message.messageStubType) {
+    const jid = message.key?.remoteJid;
+    let participants;
+    const emitParticipantsUpdate = (action) =>
+      ev.emit("group-participants.update", {
+        id: jid,
+        author: message.key.participant,
+        authorPn: message.key.participantAlt,
+        participants: participants,
+        action: action,
+      });
+    const emitGroupUpdate = (update) => {
+      ev.emit("groups.update", [
+        {
+          id: jid,
+          ...update,
+          author: message.key.participant ?? undefined,
+          authorPn: message.key.participantAlt,
+        },
+      ]);
+    };
+    const participantsIncludesMe = () =>
+      participants.find((jid) =>
+        areJidsSameUser(meId, jid.phoneNumber || jid.lid),
+      );
+    switch (message.messageStubType) {
+      case WAMessageStubType.GROUP_PARTICIPANT_CHANGE_NUMBER:
+        participants =
+          message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("modify");
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_LEAVE:
+      case WAMessageStubType.GROUP_PARTICIPANT_REMOVE:
+        participants =
+          message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("remove");
+        if (participantsIncludesMe()) {
+          chat.readOnly = true;
+        }
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_ADD:
+      case WAMessageStubType.GROUP_PARTICIPANT_INVITE:
+      case WAMessageStubType.GROUP_PARTICIPANT_ADD_REQUEST_JOIN:
+        participants =
+          message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        if (participantsIncludesMe()) {
+          chat.readOnly = false;
+        }
+        emitParticipantsUpdate("add");
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_DEMOTE:
+        participants =
+          message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("demote");
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_PROMOTE:
+        participants =
+          message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("promote");
+        break;
+      case WAMessageStubType.GROUP_CHANGE_ANNOUNCE:
+        const announceValue = message.messageStubParameters?.[0];
+        emitGroupUpdate({
+          announce: announceValue === "true" || announceValue === "on",
+        });
+        break;
+      case WAMessageStubType.GROUP_CHANGE_RESTRICT:
+        const restrictValue = message.messageStubParameters?.[0];
+        emitGroupUpdate({
+          restrict: restrictValue === "true" || restrictValue === "on",
+        });
+        break;
+    }
+  }
+  return { chat };
+};
+/**
+ * Get the ID of the chat from the given key.
+ * Typically -- that'll be the remoteJid, but for broadcasts, it'll be the participant
+ * (مضافة من الرسمية: التعامل الصحيح مع الـ broadcast lists بدل remoteJid فقط)
+ */
+export const getChatId = ({ remoteJid, participant, fromMe }) => {
+  if (!remoteJid) {
+    throw new Boom("Cannot derive chat id: message key is missing remoteJid", {
+      data: { remoteJid, participant, fromMe },
+    });
+  }
+  if (isJidBroadcast(remoteJid) && !isJidStatusBroadcast(remoteJid) && !fromMe) {
+    if (!participant) {
+      throw new Boom(
+        "Cannot derive chat id: broadcast message key is missing participant",
+        { data: { remoteJid, fromMe } },
+      );
+    }
+    return participant;
+  }
+  return remoteJid;
+};
+
+/**
+ * Decrypt a poll vote (مضافة من الرسمية: جينتوكي كان بيطبّع مفتاح رسالة التصويت
+ * بس مكنش بيفك تشفيره فعليًا)
+ * @param vote encrypted vote
+ * @param ctx additional info about the poll required for decryption
+ * @returns list of SHA256 options
+ */
+export function decryptPollVote(
+  { encPayload, encIv },
+  { pollCreatorJid, pollMsgId, pollEncKey, voterJid },
+) {
+  const sign = Buffer.concat([
+    toBinary(pollMsgId),
+    toBinary(pollCreatorJid),
+    toBinary(voterJid),
+    toBinary("Poll Vote"),
+    new Uint8Array([1]),
+  ]);
+  const key0 = hmacSign(pollEncKey, new Uint8Array(32), "sha256");
+  const decKey = hmacSign(sign, key0, "sha256");
+  const aad = toBinary(`${pollMsgId}\u0000${voterJid}`);
+  const decrypted = aesDecryptGCM(encPayload, decKey, encIv, aad);
+  return proto.Message.PollVoteMessage.decode(decrypted);
+  function toBinary(txt) {
+    return Buffer.from(txt);
+  }
+}
+
+export { processMessage };