zapo-js 0.3.0 → 1.0.1

package/dist/message/{content.js → encode/content.js}

@@ -1,16 +1,156 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.getContentType = getContentType;
+exports.resolveMessageTarget = resolveMessageTarget;
 exports.isSendMediaMessage = isSendMediaMessage;
+exports.isSendTextMessage = isSendTextMessage;
+exports.isSendReactionMessage = isSendReactionMessage;
+exports.isSendRevokeMessage = isSendRevokeMessage;
+exports.isSendPinMessage = isSendPinMessage;
+exports.isSendKeepMessage = isSendKeepMessage;
+exports.isSendPollMessage = isSendPollMessage;
+exports.isSendPollVoteMessage = isSendPollVoteMessage;
+exports.isSendEventMessage = isSendEventMessage;
+exports.isSendEventResponseMessage = isSendEventResponseMessage;
+exports.isSendAddonCryptoMessage = isSendAddonCryptoMessage;
+exports.supportsViewOnce = supportsViewOnce;
+exports.wrapAsViewOnce = wrapAsViewOnce;
 exports.unwrapMessage = unwrapMessage;
 exports.resolveMessageTypeAttr = resolveMessageTypeAttr;
+exports.resolveDecryptFailAttr = resolveDecryptFailAttr;
 exports.needsSecretPersistence = needsSecretPersistence;
 exports.resolveEditAttr = resolveEditAttr;
 exports.resolveEncMediaType = resolveEncMediaType;
+exports.resolveButtonAddonKind = resolveButtonAddonKind;
 exports.resolveMetaAttrs = resolveMetaAttrs;
-const _proto_1 = require("../proto.js");
-const constants_1 = require("../protocol/constants");
+const _proto_1 = require("../../proto");
+const constants_1 = require("../../protocol/constants");
+/**
+ * Returns the content-type key of a message - `'conversation'`,
+ * `'imageMessage'`, `'extendedTextMessage'`, etc. - or `undefined` for an empty
+ * message. `senderKeyDistributionMessage` is skipped so group messages report
+ * their real payload type rather than the piggy-backed sender-key.
+ */
+function getContentType(content) {
+    if (!content)
+        return undefined;
+    const key = Object.keys(content).find((k) => (k === 'conversation' || k.includes('Message')) && k !== 'senderKeyDistributionMessage');
+    return key;
+}
+/**
+ * Normalizes a reply/edit/reaction/revoke/pin target into a {@link WaMessageKey}:
+ * an explicit key is returned as-is, while a received `message` event
+ * ({@link WaMessageRef}, identified by its `rawNode`) yields its pre-computed
+ * `key`.
+ */
+function resolveMessageTarget(ref) {
+    if (!('rawNode' in ref)) {
+        return ref;
+    }
+    const key = ref.key;
+    if (key === undefined || typeof key.id !== 'string') {
+        throw new Error('message event used as a target is missing key.id');
+    }
+    return key;
+}
 function isSendMediaMessage(content) {
-    return !!content && typeof content === 'object' && 'type' in content && 'media' in content;
+    if (!content || typeof content !== 'object' || !('type' in content)) {
+        return false;
+    }
+    if ('media' in content) {
+        return true;
+    }
+    return content.type === 'sticker-pack' && 'stickers' in content;
+}
+function isSendTextMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'text' &&
+        'text' in content);
+}
+function isSendReactionMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'reaction' &&
+        'target' in content);
+}
+function isSendRevokeMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'revoke' &&
+        'target' in content);
+}
+function isSendPinMessage(content) {
+    if (!content || typeof content !== 'object' || !('type' in content) || !('target' in content)) {
+        return false;
+    }
+    const type = content.type;
+    return type === 'pin' || type === 'unpin';
+}
+function isSendKeepMessage(content) {
+    if (!content || typeof content !== 'object' || !('type' in content) || !('target' in content)) {
+        return false;
+    }
+    const type = content.type;
+    return type === 'keep' || type === 'unkeep';
+}
+function isSendPollMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'poll' &&
+        'name' in content &&
+        'options' in content);
+}
+function isSendPollVoteMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'poll-vote' &&
+        'poll' in content &&
+        'selectedOptionNames' in content);
+}
+function isSendEventMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'event' &&
+        'name' in content &&
+        'startTime' in content);
+}
+function isSendEventResponseMessage(content) {
+    return (!!content &&
+        typeof content === 'object' &&
+        'type' in content &&
+        content.type === 'event-response' &&
+        'event' in content &&
+        'response' in content);
+}
+function isSendAddonCryptoMessage(content) {
+    return isSendPollVoteMessage(content) || isSendEventResponseMessage(content);
+}
+const VIEW_ONCE_MEDIA_FIELDS = [
+    'imageMessage',
+    'videoMessage',
+    'ptvMessage',
+    'audioMessage'
+];
+function supportsViewOnce(message) {
+    return VIEW_ONCE_MEDIA_FIELDS.some((field) => message[field]);
+}
+function wrapAsViewOnce(message) {
+    if (message.viewOnceMessage || message.viewOnceMessageV2) {
+        return message;
+    }
+    for (const field of VIEW_ONCE_MEDIA_FIELDS) {
+        const media = message[field];
+        if (media)
+            return { ...message, [field]: { ...media, viewOnce: true } };
+    }
+    return message;
 }
 function unwrapMessage(message) {
     let msg = message;
@@ -75,6 +215,35 @@ function resolveMessageTypeAttr(message) {
     return constants_1.WA_STANZA_MSG_TYPES.MEDIA;
 }
 const REVOKED_REACTION_TEXT = '';
+function resolveDecryptFailAttr(message) {
+    const msg = unwrapMessage(message);
+    const secretEncType = msg.secretEncryptedMessage?.secretEncType;
+    if (msg.reactionMessage ||
+        msg.encReactionMessage ||
+        (msg.pollUpdateMessage &&
+            msg.pollUpdateMessage.vote !== null &&
+            msg.pollUpdateMessage.vote !== undefined) ||
+        msg.keepInChatMessage ||
+        msg.editedMessage ||
+        msg.pinInChatMessage ||
+        msg.encCommentMessage ||
+        msg.encEventResponseMessage ||
+        secretEncType === _proto_1.proto.Message.SecretEncryptedMessage.SecretEncType.EVENT_EDIT ||
+        secretEncType === _proto_1.proto.Message.SecretEncryptedMessage.SecretEncType.POLL_EDIT ||
+        secretEncType === _proto_1.proto.Message.SecretEncryptedMessage.SecretEncType.POLL_ADD_OPTION ||
+        msg.messageHistoryNotice) {
+        return 'hide';
+    }
+    if (msg.protocolMessage) {
+        const protocolType = msg.protocolMessage.type;
+        if (protocolType === _proto_1.proto.Message.ProtocolMessage.Type.EPHEMERAL_SYNC_RESPONSE ||
+            protocolType === _proto_1.proto.Message.ProtocolMessage.Type.REQUEST_WELCOME_MESSAGE ||
+            msg.protocolMessage.editedMessage) {
+            return 'hide';
+        }
+    }
+    return undefined;
+}
 function needsSecretPersistence(message) {
     const msg = unwrapMessage(message);
     return !!(msg.pollCreationMessage ||
@@ -83,14 +252,18 @@ function needsSecretPersistence(message) {
         msg.pollCreationMessageV4 ||
         msg.pollCreationMessageV5 ||
         msg.pollCreationMessageV6 ||
-        msg.eventMessage);
+        msg.eventMessage ||
+        // botMetadata sits on the top-level MessageContextInfo even when the
+        // prompt body is wrapped (e.g. in `botInvokeMessage`); check both.
+        message.messageContextInfo?.botMetadata ||
+        msg.messageContextInfo?.botMetadata);
 }
-function resolveEditAttr(message, subtype) {
+function resolveEditAttr(message) {
     const msg = unwrapMessage(message);
     if (msg.protocolMessage) {
         const protocolType = msg.protocolMessage.type;
         if (protocolType === _proto_1.proto.Message.ProtocolMessage.Type.REVOKE) {
-            return subtype === 'admin_revoke'
+            return msg.protocolMessage.key?.fromMe === false
                 ? constants_1.WA_EDIT_ATTRS.ADMIN_REVOKE
                 : constants_1.WA_EDIT_ATTRS.SENDER_REVOKE;
         }
@@ -133,6 +306,8 @@ function resolveEncMediaType(message) {
         return constants_1.WA_ENC_MEDIA_TYPES.IMAGE;
     if (msg.stickerMessage)
         return constants_1.WA_ENC_MEDIA_TYPES.STICKER;
+    if (msg.stickerPackMessage)
+        return constants_1.WA_ENC_MEDIA_TYPES.STICKER_PACK;
     if (msg.locationMessage) {
         return msg.locationMessage.isLive
             ? constants_1.WA_ENC_MEDIA_TYPES.LIVE_LOCATION
@@ -175,6 +350,14 @@ function resolveEncMediaType(message) {
     }
     return null;
 }
+function resolveButtonAddonKind(message) {
+    const msg = unwrapMessage(message);
+    if (msg.listMessage)
+        return 'list';
+    if (msg.buttonsMessage || msg.interactiveMessage?.nativeFlowMessage)
+        return 'interactive';
+    return null;
+}
 function resolveMetaAttrs(message) {
     const msg = unwrapMessage(message);
     let polltype;
@@ -204,7 +387,9 @@ function resolveMetaAttrs(message) {
         _proto_1.proto.Message.SecretEncryptedMessage.SecretEncType.EVENT_EDIT) {
         eventType = constants_1.WA_EVENT_META_TYPES.EDIT;
     }
-    if (message.viewOnceMessage || message.viewOnceMessageV2) {
+    if (message.viewOnceMessage ||
+        message.viewOnceMessageV2 ||
+        VIEW_ONCE_MEDIA_FIELDS.some((field) => msg[field]?.viewOnce)) {
         viewOnce = 'true';
     }
     if (!polltype && !eventType && !viewOnce) {

package/dist/message/encode/media-payload.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveMediaPayload = resolveMediaPayload;
+const content_1 = require("../encode/content");
+const primitives_1 = require("../../util/primitives");
+function buildPayload(mediaType, fields) {
+    if (!fields.directPath || !fields.mediaKey) {
+        return null;
+    }
+    const mediaKey = fields.mediaKey instanceof Uint8Array ? fields.mediaKey : null;
+    if (!mediaKey) {
+        return null;
+    }
+    const fileSha256 = fields.fileSha256 instanceof Uint8Array ? fields.fileSha256 : undefined;
+    const fileEncSha256 = fields.fileEncSha256 instanceof Uint8Array ? fields.fileEncSha256 : undefined;
+    return {
+        mediaType,
+        directPath: fields.directPath,
+        mediaKey,
+        fileSha256,
+        fileEncSha256,
+        mimetype: fields.mimetype ?? undefined,
+        fileLength: fields.fileLength ? (0, primitives_1.longToNumber)(fields.fileLength) : undefined
+    };
+}
+function resolveMediaPayload(message) {
+    if (!message)
+        return null;
+    const msg = (0, content_1.unwrapMessage)(message);
+    if (msg.imageMessage)
+        return buildPayload('image', msg.imageMessage);
+    if (msg.videoMessage) {
+        return buildPayload(msg.videoMessage.gifPlayback ? 'gif' : 'video', msg.videoMessage);
+    }
+    if (msg.audioMessage) {
+        return buildPayload(msg.audioMessage.ptt ? 'ptt' : 'audio', msg.audioMessage);
+    }
+    if (msg.documentMessage)
+        return buildPayload('document', msg.documentMessage);
+    if (msg.stickerMessage)
+        return buildPayload('sticker', msg.stickerMessage);
+    if (msg.ptvMessage)
+        return buildPayload('ptv', msg.ptvMessage);
+    return null;
+}

package/dist/message/{padding.js → encode/padding.js}

@@ -2,11 +2,17 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.writeRandomPadMax16 = writeRandomPadMax16;
 exports.unpadPkcs7 = unpadPkcs7;
-const _crypto_1 = require("../crypto/index.js");
+const _crypto_1 = require("../../crypto");
 const RANDOM_PAD_MAX_16_MASK = 0x0f;
+const PAD_SEED_BATCH_SIZE = 1024;
+let padSeedBuffer = null;
+let padSeedCursor = 0;
 async function writeRandomPadMax16(message) {
-    const padSeed = await (0, _crypto_1.randomBytesAsync)(1);
-    const padLength = (padSeed[0] & RANDOM_PAD_MAX_16_MASK) + 1;
+    if (!padSeedBuffer || padSeedCursor >= padSeedBuffer.length) {
+        padSeedBuffer = await (0, _crypto_1.randomBytesAsync)(PAD_SEED_BATCH_SIZE);
+        padSeedCursor = 0;
+    }
+    const padLength = (padSeedBuffer[padSeedCursor++] & RANDOM_PAD_MAX_16_MASK) + 1;
     const out = new Uint8Array(message.length + padLength);
     out.set(message, 0);
     out.fill(padLength, message.length);

package/dist/message/index.js

@@ -17,6 +17,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.WaMessageClient = void 0;
 var WaMessageClient_1 = require("./WaMessageClient");
 Object.defineProperty(exports, "WaMessageClient", { enumerable: true, get: function () { return WaMessageClient_1.WaMessageClient; } });
-__exportStar(require("./addon-crypto"), exports);
+__exportStar(require("./crypto/addon-crypto"), exports);
 __exportStar(require("./types"), exports);
-__exportStar(require("./use-case-secret"), exports);
+__exportStar(require("./crypto/use-case-secret"), exports);

package/dist/message/kinds/bot.js

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.genBotMsgSecret = genBotMsgSecret;
+exports.deriveBotChunkKey = deriveBotChunkKey;
+exports.decryptBotChunk = decryptBotChunk;
+exports.attachBotMetadata = attachBotMetadata;
+exports.attachBotThread = attachBotThread;
+exports.extractInvokedBotJid = extractInvokedBotJid;
+exports.buildBotInvokeProtoCopy = buildBotInvokeProtoCopy;
+const _crypto_1 = require("../../crypto");
+const addon_crypto_1 = require("../crypto/addon-crypto");
+const use_case_secret_1 = require("../crypto/use-case-secret");
+const _proto_1 = require("../../proto");
+const bot_1 = require("../../protocol/bot");
+const jid_1 = require("../../protocol/jid");
+const bytes_1 = require("../../util/bytes");
+function genBotMsgSecret(messageSecret) {
+    const ikm = (0, use_case_secret_1.assertMessageSecret)(messageSecret, 'bot parent message secret');
+    return (0, _crypto_1.hkdf)(ikm, null, bot_1.WA_BOT_HKDF_INFO, bot_1.WA_BOT_MSG_SECRET_BYTES);
+}
+function deriveBotChunkKey(input) {
+    if (!input.saltId)
+        throw new Error('bot chunk salt id must be a non-empty string');
+    if (!input.targetSenderJid)
+        throw new Error('bot chunk target sender jid must be a non-empty string');
+    if (!input.authorJid)
+        throw new Error('bot chunk author jid must be a non-empty string');
+    const info = bytes_1.TEXT_ENCODER.encode(input.saltId + input.targetSenderJid + input.authorJid);
+    return (0, _crypto_1.hkdf)((0, bytes_1.toBytesView)(input.botMsgSecret), null, info, bot_1.WA_BOT_MSG_SECRET_BYTES);
+}
+function decryptBotChunk(input) {
+    const botMsgSecret = genBotMsgSecret(input.parentMessageSecret);
+    const key = deriveBotChunkKey({
+        botMsgSecret,
+        saltId: input.saltId,
+        targetSenderJid: input.targetSenderJid,
+        authorJid: input.authorJid
+    });
+    return (0, _crypto_1.aesGcmDecrypt)(key, (0, bytes_1.toBytesView)(input.encIv), (0, bytes_1.toBytesView)(input.encPayload), (0, addon_crypto_1.buildAddonAdditionalData)(input.saltId, input.authorJid));
+}
+function attachBotMetadata(message, input) {
+    if (!input.personaId && !input.invokerJid && !input.capabilities) {
+        return message;
+    }
+    const botMetadata = {};
+    if (input.personaId)
+        botMetadata.personaId = input.personaId;
+    if (input.invokerJid)
+        botMetadata.invokerJid = input.invokerJid;
+    if (input.capabilities && input.capabilities.length > 0) {
+        botMetadata.capabilityMetadata = {
+            capabilities: input.capabilities
+        };
+    }
+    return {
+        ...message,
+        messageContextInfo: {
+            ...(message.messageContextInfo ?? {}),
+            botMetadata
+        }
+    };
+}
+function attachBotThread(message, input) {
+    const ctx = message.messageContextInfo ?? {};
+    const existingBotMetadata = ctx.botMetadata ?? {};
+    const threadType = input.threadType ?? _proto_1.proto.AIThreadInfo.AIThreadClientInfo.AIThreadType.DEFAULT;
+    return {
+        ...message,
+        messageContextInfo: {
+            ...ctx,
+            threadId: [
+                {
+                    threadType: _proto_1.proto.ThreadID.ThreadType.AI_THREAD,
+                    threadKey: {
+                        remoteJid: input.remoteJid,
+                        fromMe: true,
+                        id: input.aiThreadId
+                    }
+                }
+            ],
+            botMetadata: {
+                ...existingBotMetadata,
+                botThreadInfo: { clientInfo: { type: threadType } }
+            }
+        }
+    };
+}
+function extractInvokedBotJid(message) {
+    const inner = message.botInvokeMessage?.message;
+    if (!inner)
+        return null;
+    // Mention metadata lives on whichever submessage carries the body – text
+    // prompts on extendedTextMessage, media prompts on the caption-bearing body.
+    const mentioned = inner.extendedTextMessage?.contextInfo?.mentionedJid ??
+        inner.imageMessage?.contextInfo?.mentionedJid ??
+        inner.videoMessage?.contextInfo?.mentionedJid ??
+        inner.documentMessage?.contextInfo?.mentionedJid;
+    if (!mentioned)
+        return null;
+    for (const jid of mentioned) {
+        if ((0, jid_1.isBotJid)(jid))
+            return jid;
+    }
+    return null;
+}
+// Mirrors `WAWebE2EProtoGenerator.updateBotInvokeMsgProtoCopyForCapi`, minus
+// the pushname rewrite (raw mentions are accepted by Meta AI).
+function buildBotInvokeProtoCopy(message, botMessageSecret) {
+    const { messageContextInfo, ...rest } = message;
+    if (!messageContextInfo && !botMessageSecret)
+        return rest;
+    const nextContextInfo = {
+        ...(messageContextInfo ?? {}),
+        messageSecret: undefined
+    };
+    if (botMessageSecret) {
+        nextContextInfo.botMessageSecret = botMessageSecret;
+    }
+    return { ...rest, messageContextInfo: nextContextInfo };
+}