@realvare/based 2.5.1

package/lib/Utils/process-message.js

@@ -0,0 +1,404 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getChatId = exports.shouldIncrementChatUnread = exports.isRealMessage = exports.cleanMessage = void 0;
+exports.decryptPollVote = decryptPollVote;
+const WAProto_1 = require("../../WAProto");
+const Types_1 = require("../Types");
+const messages_1 = require("../Utils/messages");
+const WABinary_1 = require("../WABinary");
+const crypto_1 = require("./crypto");
+const generics_1 = require("./generics");
+const history_1 = require("./history");
+const REAL_MSG_STUB_TYPES = new Set([
+    Types_1.WAMessageStubType.CALL_MISSED_GROUP_VIDEO,
+    Types_1.WAMessageStubType.CALL_MISSED_GROUP_VOICE,
+    Types_1.WAMessageStubType.CALL_MISSED_VIDEO,
+    Types_1.WAMessageStubType.CALL_MISSED_VOICE
+]);
+const REAL_MSG_REQ_ME_STUB_TYPES = new Set([
+    Types_1.WAMessageStubType.GROUP_PARTICIPANT_ADD
+]);
+/** Cleans a received message to further processing */
+const cleanMessage = (message, meId) => {
+    // ensure remoteJid and participant doesn't have device or agent in it
+    // normalize JIDs but catch errors to avoid throwing on invalid LIDs/JIDs
+    try {
+        message.key.remoteJid = (0, WABinary_1.jidNormalizedUser)(message.key.remoteJid);
+    }
+    catch (_e) {
+        // if normalization fails, retain original remoteJid
+    }
+    if (message.key.participant) {
+        try {
+            message.key.participant = (0, WABinary_1.jidNormalizedUser)(message.key.participant);
+        }
+        catch (_e) {
+            // ignore if can't normalize participant
+        }
+    }
+    const content = (0, messages_1.normalizeMessageContent)(message.message);
+    // if the message has a reaction, ensure fromMe & remoteJid are from our perspective
+    if (content === null || content === void 0 ? void 0 : content.reactionMessage) {
+        normaliseKey(content.reactionMessage.key);
+    }
+    if (content === null || content === void 0 ? void 0 : content.pollUpdateMessage) {
+        normaliseKey(content.pollUpdateMessage.pollCreationMessageKey);
+    }
+    function normaliseKey(msgKey) {
+        // if the reaction is from another user
+        // we've to correctly map the key to this user's perspective
+        if (!message.key.fromMe) {
+            // if the sender believed the message being reacted to is not from them
+            // we've to correct the key to be from them, or some other participant
+            msgKey.fromMe = !msgKey.fromMe
+                ? (0, WABinary_1.areJidsSameUser)(msgKey.participant || msgKey.remoteJid, meId)
+                // if the message being reacted to, was from them
+                // fromMe automatically becomes false
+                : false;
+            // set the remoteJid to being the same as the chat the message came from
+            msgKey.remoteJid = message.key.remoteJid;
+            // set participant of the message
+            msgKey.participant = msgKey.participant || message.key.participant;
+        }
+    }
+};
+exports.cleanMessage = cleanMessage;
+const isRealMessage = (message, meId) => {
+    var _a;
+    const normalizedContent = (0, messages_1.normalizeMessageContent)(message.message);
+    const hasSomeContent = !!(0, messages_1.getContentType)(normalizedContent);
+    return (!!normalizedContent
+        || REAL_MSG_STUB_TYPES.has(message.messageStubType)
+        || (REAL_MSG_REQ_ME_STUB_TYPES.has(message.messageStubType)
+            && ((_a = message.messageStubParameters) === null || _a === void 0 ? void 0 : _a.some(p => (0, WABinary_1.areJidsSameUser)(meId, p)))))
+        && hasSomeContent
+        && !(normalizedContent === null || normalizedContent === void 0 ? void 0 : normalizedContent.protocolMessage)
+        && !(normalizedContent === null || normalizedContent === void 0 ? void 0 : normalizedContent.reactionMessage)
+        && !(normalizedContent === null || normalizedContent === void 0 ? void 0 : normalizedContent.pollUpdateMessage);
+};
+exports.isRealMessage = isRealMessage;
+const shouldIncrementChatUnread = (message) => (!message.key.fromMe && !message.messageStubType);
+exports.shouldIncrementChatUnread = shouldIncrementChatUnread;
+/**
+ * Get the ID of the chat from the given key.
+ * Typically -- that'll be the remoteJid, but for broadcasts, it'll be the participant
+ */
+const getChatId = ({ remoteJid, participant, fromMe }) => {
+    if ((0, WABinary_1.isJidBroadcast)(remoteJid)
+        && !(0, WABinary_1.isJidStatusBroadcast)(remoteJid)
+        && !fromMe) {
+        return participant;
+    }
+    return remoteJid;
+};
+exports.getChatId = getChatId;
+/**
+ * Decrypt a poll vote
+ * @param vote encrypted vote
+ * @param ctx additional info about the poll required for decryption
+ * @returns list of SHA256 options
+ */
+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 = (0, crypto_1.hmacSign)(pollEncKey, new Uint8Array(32), 'sha256');
+    const decKey = (0, crypto_1.hmacSign)(sign, key0, 'sha256');
+    const aad = toBinary(`${pollMsgId}\u0000${voterJid}`);
+    const decrypted = (0, crypto_1.aesDecryptGCM)(encPayload, decKey, encIv, aad);
+    return WAProto_1.proto.Message.PollVoteMessage.decode(decrypted);
+    function toBinary(txt) {
+        return Buffer.from(txt);
+    }
+}
+const processMessage = async (message, { shouldProcessHistoryMsg, placeholderResendCache, ev, creds, keyStore, logger, options, getMessage }) => {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s;
+    const meId = creds.me.id;
+    const { accountSettings } = creds;
+    const chat = { id: (0, WABinary_1.jidNormalizedUser)((0, exports.getChatId)(message.key)) };
+    const isRealMsg = (0, exports.isRealMessage)(message, meId);
+    if (isRealMsg) {
+        chat.messages = [{ message }];
+        chat.conversationTimestamp = (0, generics_1.toNumber)(message.messageTimestamp);
+        // only increment unread count if not CIPHERTEXT and from another person
+        if ((0, exports.shouldIncrementChatUnread)(message)) {
+            chat.unreadCount = (chat.unreadCount || 0) + 1;
+        }
+    }
+    const content = (0, messages_1.normalizeMessageContent)(message.message);
+    const senderId = message.key.participant || message.key.remoteJid;
+    if ((0, WABinary_1.isLidUser)(senderId)) {
+        const jid = (0, WABinary_1.lidToJid)(senderId);
+        if (message.key.participant) {
+            message.key.participant = jid;
+        }
+        else {
+            message.key.remoteJid = jid;
+        }
+    }
+    const mJids = content && content.contextInfo && content.contextInfo.mentionedJid ? content.contextInfo.mentionedJid : [];
+    for (let i = 0; i < mJids.length; i++) {
+        if ((0, WABinary_1.isLidUser)(mJids[i])) {
+            mJids[i] = (0, WABinary_1.lidToJid)(mJids[i]);
+        }
+    }
+    if (content && content.contextInfo && content.contextInfo.participant && (0, WABinary_1.isLidUser)(content.contextInfo.participant)) {
+        content.contextInfo.participant = (0, WABinary_1.lidToJid)(content.contextInfo.participant);
+    }
+    // unarchive chat if it's a real message, or someone reacted to our message
+    // and we've the unarchive chats setting on
+    if ((isRealMsg || ((_b = (_a = content === null || content === void 0 ? void 0 : content.reactionMessage) === null || _a === void 0 ? void 0 : _a.key) === null || _b === void 0 ? void 0 : _b.fromMe))
+        && (accountSettings === null || accountSettings === void 0 ? void 0 : accountSettings.unarchiveChats)) {
+        chat.archived = false;
+        chat.readOnly = false;
+    }
+    const protocolMsg = content === null || content === void 0 ? void 0 : content.protocolMessage;
+    if (protocolMsg) {
+        switch (protocolMsg.type) {
+            case WAProto_1.proto.Message.ProtocolMessage.Type.HISTORY_SYNC_NOTIFICATION:
+                const histNotification = protocolMsg.historySyncNotification;
+                const process = shouldProcessHistoryMsg;
+                const isLatest = !((_c = creds.processedHistoryMessages) === null || _c === void 0 ? void 0 : _c.length);
+                logger === null || logger === void 0 ? void 0 : logger.info({
+                    histNotification,
+                    process,
+                    id: message.key.id,
+                    isLatest,
+                }, 'got history notification');
+                if (process) {
+                    if (histNotification.syncType !== WAProto_1.proto.HistorySync.HistorySyncType.ON_DEMAND) {
+                        ev.emit('creds.update', {
+                            processedHistoryMessages: [
+                                ...(creds.processedHistoryMessages || []),
+                                { key: message.key, messageTimestamp: message.messageTimestamp }
+                            ]
+                        });
+                    }
+                    const data = await (0, history_1.downloadAndProcessHistorySyncNotification)(histNotification, options);
+                    ev.emit('messaging-history.set', {
+                        ...data,
+                        isLatest: histNotification.syncType !== WAProto_1.proto.HistorySync.HistorySyncType.ON_DEMAND
+                            ? isLatest
+                            : undefined,
+                        peerDataRequestSessionId: histNotification.peerDataRequestSessionId
+                    });
+                }
+                break;
+            case WAProto_1.proto.Message.ProtocolMessage.Type.APP_STATE_SYNC_KEY_SHARE:
+                const keys = protocolMsg.appStateSyncKeyShare.keys;
+                if (keys === null || keys === void 0 ? void 0 : 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 === null || logger === void 0 ? void 0 : logger.info({ newAppStateSyncKeyId, newKeys }, 'injecting new app state sync keys');
+                    });
+                    ev.emit('creds.update', { myAppStateKeyId: newAppStateSyncKeyId });
+                }
+                else {
+                    logger === null || logger === void 0 ? void 0 : logger.info({ protocolMsg }, 'recv app state sync with 0 keys');
+                }
+                break;
+            case WAProto_1.proto.Message.ProtocolMessage.Type.REVOKE:
+                ev.emit('messages.update', [
+                    {
+                        key: {
+                            ...message.key,
+                            id: protocolMsg.key.id
+                        },
+                        update: { message: null, messageStubType: Types_1.WAMessageStubType.REVOKE, key: message.key }
+                    }
+                ]);
+                break;
+            case WAProto_1.proto.Message.ProtocolMessage.Type.EPHEMERAL_SETTING:
+                Object.assign(chat, {
+                    ephemeralSettingTimestamp: (0, generics_1.toNumber)(message.messageTimestamp),
+                    ephemeralExpiration: protocolMsg.ephemeralExpiration || null
+                });
+                break;
+            case WAProto_1.proto.Message.ProtocolMessage.Type.PEER_DATA_OPERATION_REQUEST_RESPONSE_MESSAGE:
+                const response = protocolMsg.peerDataOperationRequestResponseMessage;
+                if (response) {
+                    placeholderResendCache === null || placeholderResendCache === void 0 ? void 0 : placeholderResendCache.del(response.stanzaId);
+                    // TODO: IMPLEMENT HISTORY SYNC ETC (sticker uploads etc.).
+                    const { peerDataOperationResult } = response;
+                    for (const result of peerDataOperationResult) {
+                        const { placeholderMessageResendResponse: retryResponse } = result;
+                        if (retryResponse) {
+                            const webMessageInfo = WAProto_1.proto.WebMessageInfo.decode(retryResponse.webMessageInfoBytes);
+                            // wait till another upsert event is available, don't want it to be part of the PDO response message
+                            setTimeout(() => {
+                                ev.emit('messages.upsert', {
+                                    messages: [webMessageInfo],
+                                    type: 'notify',
+                                    requestId: response.stanzaId
+                                });
+                            }, 500);
+                        }
+                    }
+                }
+            case WAProto_1.proto.Message.ProtocolMessage.Type.MESSAGE_EDIT:
+                ev.emit('messages.update', [
+                    {
+                        // flip the sender / fromMe properties because they're in the perspective of the sender
+                        key: { ...message.key, id: (_d = protocolMsg.key) === null || _d === void 0 ? void 0 : _d.id },
+                        update: {
+                            message: {
+                                editedMessage: {
+                                    message: protocolMsg.editedMessage
+                                }
+                            },
+                            messageTimestamp: protocolMsg.timestampMs
+                                ? Math.floor((0, generics_1.toNumber)(protocolMsg.timestampMs) / 1000)
+                                : message.messageTimestamp
+                        }
+                    }
+                ]);
+                break;
+        }
+    }
+    else if (content === null || content === void 0 ? void 0 : content.reactionMessage) {
+        const reaction = {
+            ...content.reactionMessage,
+            key: message.key,
+        };
+        ev.emit('messages.reaction', [{
+                reaction,
+                key: (_e = content.reactionMessage) === null || _e === void 0 ? void 0 : _e.key,
+            }]);
+    }
+    else if (message.messageStubType) {
+        const jid = (_f = message.key) === null || _f === void 0 ? void 0 : _f.remoteJid;
+        //let actor = whatsappID (message.participant)
+        let participants;
+        const emitParticipantsUpdate = (action) => (ev.emit('group-participants.update', { id: jid, author: message.participant, participants, action }));
+        const emitGroupUpdate = (update) => {
+            var _a;
+            ev.emit('groups.update', [{ id: jid, ...update, author: (_a = message.participant) !== null && _a !== void 0 ? _a : undefined }]);
+        };
+        const emitGroupRequestJoin = (participant, action, method) => {
+            ev.emit('group.join-request', { id: jid, author: message.participant, participant, action, method: method });
+        };
+        const participantsIncludesMe = () => participants.find(jid => (0, WABinary_1.areJidsSameUser)(meId, jid));
+        switch (message.messageStubType) {
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_CHANGE_NUMBER:
+                participants = message.messageStubParameters || [];
+                emitParticipantsUpdate('modify');
+                break;
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_LEAVE:
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_REMOVE:
+                participants = message.messageStubParameters || [];
+                emitParticipantsUpdate('remove');
+                // mark the chat read only if you left the group
+                if (participantsIncludesMe()) {
+                    chat.readOnly = true;
+                }
+                break;
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_ADD:
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_INVITE:
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_ADD_REQUEST_JOIN:
+                participants = message.messageStubParameters || [];
+                if (participantsIncludesMe()) {
+                    chat.readOnly = false;
+                }
+                emitParticipantsUpdate('add');
+                break;
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_DEMOTE:
+                participants = message.messageStubParameters || [];
+                emitParticipantsUpdate('demote');
+                break;
+            case Types_1.WAMessageStubType.GROUP_PARTICIPANT_PROMOTE:
+                participants = message.messageStubParameters || [];
+                emitParticipantsUpdate('promote');
+                break;
+            case Types_1.WAMessageStubType.GROUP_CHANGE_ANNOUNCE:
+                const announceValue = (_g = message.messageStubParameters) === null || _g === void 0 ? void 0 : _g[0];
+                emitGroupUpdate({ announce: announceValue === 'true' || announceValue === 'on' });
+                break;
+            case Types_1.WAMessageStubType.GROUP_CHANGE_RESTRICT:
+                const restrictValue = (_h = message.messageStubParameters) === null || _h === void 0 ? void 0 : _h[0];
+                emitGroupUpdate({ restrict: restrictValue === 'true' || restrictValue === 'on' });
+                break;
+            case Types_1.WAMessageStubType.GROUP_CHANGE_SUBJECT:
+                const name = (_j = message.messageStubParameters) === null || _j === void 0 ? void 0 : _j[0];
+                chat.name = name;
+                emitGroupUpdate({ subject: name });
+                break;
+            case Types_1.WAMessageStubType.GROUP_CHANGE_DESCRIPTION:
+                const description = (_k = message.messageStubParameters) === null || _k === void 0 ? void 0 : _k[0];
+                chat.description = description;
+                emitGroupUpdate({ desc: description });
+                break;
+            case Types_1.WAMessageStubType.GROUP_CHANGE_INVITE_LINK:
+                const code = (_l = message.messageStubParameters) === null || _l === void 0 ? void 0 : _l[0];
+                emitGroupUpdate({ inviteCode: code });
+                break;
+            case Types_1.WAMessageStubType.GROUP_MEMBER_ADD_MODE:
+                const memberAddValue = (_m = message.messageStubParameters) === null || _m === void 0 ? void 0 : _m[0];
+                emitGroupUpdate({ memberAddMode: memberAddValue === 'all_member_add' });
+                break;
+            case Types_1.WAMessageStubType.GROUP_MEMBERSHIP_JOIN_APPROVAL_MODE:
+                const approvalMode = (_o = message.messageStubParameters) === null || _o === void 0 ? void 0 : _o[0];
+                emitGroupUpdate({ joinApprovalMode: approvalMode === 'on' });
+                break;
+            case Types_1.WAMessageStubType.GROUP_MEMBERSHIP_JOIN_APPROVAL_REQUEST_NON_ADMIN_ADD:
+                const participant = (_p = message.messageStubParameters) === null || _p === void 0 ? void 0 : _p[0];
+                const action = (_q = message.messageStubParameters) === null || _q === void 0 ? void 0 : _q[1];
+                const method = (_r = message.messageStubParameters) === null || _r === void 0 ? void 0 : _r[2];
+                emitGroupRequestJoin(participant, action, method);
+                break;
+        }
+    }
+    else if (content === null || content === void 0 ? void 0 : content.pollUpdateMessage) {
+        const creationMsgKey = content.pollUpdateMessage.pollCreationMessageKey;
+        // we need to fetch the poll creation message to get the poll enc key
+        const pollMsg = await getMessage(creationMsgKey);
+        if (pollMsg) {
+            const meIdNormalised = (0, WABinary_1.jidNormalizedUser)(meId);
+            const pollCreatorJid = (0, generics_1.getKeyAuthor)(creationMsgKey, meIdNormalised);
+            const voterJid = (0, generics_1.getKeyAuthor)(message.key, meIdNormalised);
+            const pollEncKey = (_s = pollMsg.messageContextInfo) === null || _s === void 0 ? void 0 : _s.messageSecret;
+            try {
+                const voteMsg = decryptPollVote(content.pollUpdateMessage.vote, {
+                    pollEncKey,
+                    pollCreatorJid,
+                    pollMsgId: creationMsgKey.id,
+                    voterJid,
+                });
+                ev.emit('messages.update', [
+                    {
+                        key: creationMsgKey,
+                        update: {
+                            pollUpdates: [
+                                {
+                                    pollUpdateMessageKey: message.key,
+                                    vote: voteMsg,
+                                    senderTimestampMs: content.pollUpdateMessage.senderTimestampMs.toNumber(),
+                                }
+                            ]
+                        }
+                    }
+                ]);
+            }
+            catch (err) {
+                logger === null || logger === void 0 ? void 0 : logger.warn({ err, creationMsgKey }, 'failed to decrypt poll vote');
+            }
+        }
+        else {
+            logger === null || logger === void 0 ? void 0 : logger.warn({ creationMsgKey }, 'poll creation message not found, cannot decrypt update');
+        }
+    }
+    if (Object.keys(chat).length > 1) {
+        ev.emit('chats.update', [chat]);
+    }
+};
+exports.default = processMessage;

package/lib/Utils/retry.js

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.retryWithBackoff = void 0;
+
+async function sleep(ms, signal) {
+    return new Promise((resolve, reject) => {
+        const t = setTimeout(resolve, ms);
+        if (signal) {
+            const onAbort = () => {
+                clearTimeout(t);
+                reject(Object.assign(new Error('aborted'), { code: 'ABORT_ERR' }));
+            };
+            if (signal.aborted) {
+                onAbort();
+                return;
+            }
+            signal.addEventListener('abort', onAbort, { once: true });
+        }
+    });
+}
+
+function computeDelay(attempt, baseMs, maxMs, jitter) {
+    const exp = Math.min(maxMs, baseMs * Math.pow(2, attempt));
+    if (!jitter) return exp;
+    // full jitter strategy: random between 0 and exp
+    const rand = Math.random() * exp;
+    return Math.min(maxMs, rand);
+}
+
+async function retryWithBackoff(fn, opts = {}) {
+	const {
+		retries = 3,
+		baseMs = 300,
+		maxMs = 5000,
+		jitter = true,
+        timeoutPerAttemptMs,
+        signal,
+		onRetry,
+		shouldRetry
+	} = opts;
+	let lastErr;
+	for (let attempt = 0; attempt <= retries; attempt++) {
+		try {
+            if (timeoutPerAttemptMs && timeoutPerAttemptMs > 0) {
+                const ctrl = new AbortController();
+                const timer = setTimeout(() => ctrl.abort(), timeoutPerAttemptMs);
+                try {
+                    return await fn({ signal: ctrl.signal });
+                } finally {
+                    clearTimeout(timer);
+                }
+            }
+            return await fn({ signal });
+		} catch (err) {
+			lastErr = err;
+			if (attempt === retries) break;
+			if (shouldRetry && !shouldRetry(err)) break;
+			if (onRetry) {
+				try { onRetry(err, attempt + 1); } catch {}
+			}
+            await sleep(computeDelay(attempt, baseMs, maxMs, jitter), signal);
+		}
+	}
+	throw lastErr;
+}
+exports.retryWithBackoff = retryWithBackoff;

package/lib/Utils/signal.d.ts

@@ -0,0 +1,33 @@
+import { SignalRepository } from '../Types';
+import { AuthenticationCreds, AuthenticationState, KeyPair, SignalIdentity, SignalKeyStore, SignedKeyPair } from '../Types/Auth';
+import { BinaryNode, JidWithDevice } from '../WABinary';
+import { USyncQueryResultList } from '../WAUSync';
+export declare const createSignalIdentity: (wid: string, accountSignatureKey: Uint8Array) => SignalIdentity;
+export declare const getPreKeys: ({ get }: SignalKeyStore, min: number, limit: number) => Promise<{
+    [id: string]: KeyPair;
+}>;
+export declare const generateOrGetPreKeys: (creds: AuthenticationCreds, range: number) => {
+    newPreKeys: {
+        [id: number]: KeyPair;
+    };
+    lastPreKeyId: number;
+    preKeysRange: readonly [number, number];
+};
+export declare const xmppSignedPreKey: (key: SignedKeyPair) => BinaryNode;
+export declare const xmppPreKey: (pair: KeyPair, id: number) => BinaryNode;
+export declare const parseAndInjectE2ESessions: (node: BinaryNode, repository: SignalRepository) => Promise<void>;
+export declare const extractDeviceJids: (result: USyncQueryResultList[], myJid: string, excludeZeroDevices: boolean) => JidWithDevice[];
+/**
+ * get the next N keys for upload or processing
+ * @param count number of pre-keys to get or generate
+ */
+export declare const getNextPreKeys: ({ creds, keys }: AuthenticationState, count: number) => Promise<{
+    update: Partial<AuthenticationCreds>;
+    preKeys: {
+        [id: string]: KeyPair;
+    };
+}>;
+export declare const getNextPreKeysNode: (state: AuthenticationState, count: number) => Promise<{
+    update: Partial<AuthenticationCreds>;
+    node: BinaryNode;
+}>;

package/lib/Utils/signal.js

@@ -0,0 +1,153 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getNextPreKeysNode = exports.getNextPreKeys = exports.extractDeviceJids = exports.parseAndInjectE2ESessions = exports.xmppPreKey = exports.xmppSignedPreKey = exports.generateOrGetPreKeys = exports.getPreKeys = exports.createSignalIdentity = void 0;
+const lodash_1 = require("lodash");
+const Defaults_1 = require("../Defaults");
+const WABinary_1 = require("../WABinary");
+const crypto_1 = require("./crypto");
+const generics_1 = require("./generics");
+const createSignalIdentity = (wid, accountSignatureKey) => {
+    return {
+        identifier: { name: wid, deviceId: 0 },
+        identifierKey: (0, crypto_1.generateSignalPubKey)(accountSignatureKey)
+    };
+};
+exports.createSignalIdentity = createSignalIdentity;
+const getPreKeys = async ({ get }, min, limit) => {
+    const idList = [];
+    for (let id = min; id < limit; id++) {
+        idList.push(id.toString());
+    }
+    return get('pre-key', idList);
+};
+exports.getPreKeys = getPreKeys;
+const generateOrGetPreKeys = (creds, range) => {
+    const avaliable = creds.nextPreKeyId - creds.firstUnuploadedPreKeyId;
+    const remaining = range - avaliable;
+    const lastPreKeyId = creds.nextPreKeyId + remaining - 1;
+    const newPreKeys = {};
+    if (remaining > 0) {
+        for (let i = creds.nextPreKeyId; i <= lastPreKeyId; i++) {
+            newPreKeys[i] = crypto_1.Curve.generateKeyPair();
+        }
+    }
+    return {
+        newPreKeys,
+        lastPreKeyId,
+        preKeysRange: [creds.firstUnuploadedPreKeyId, range],
+    };
+};
+exports.generateOrGetPreKeys = generateOrGetPreKeys;
+const xmppSignedPreKey = (key) => ({
+    tag: 'skey',
+    attrs: {},
+    content: [
+        { tag: 'id', attrs: {}, content: (0, generics_1.encodeBigEndian)(key.keyId, 3) },
+        { tag: 'value', attrs: {}, content: key.keyPair.public },
+        { tag: 'signature', attrs: {}, content: key.signature }
+    ]
+});
+exports.xmppSignedPreKey = xmppSignedPreKey;
+const xmppPreKey = (pair, id) => ({
+    tag: 'key',
+    attrs: {},
+    content: [
+        { tag: 'id', attrs: {}, content: (0, generics_1.encodeBigEndian)(id, 3) },
+        { tag: 'value', attrs: {}, content: pair.public }
+    ]
+});
+exports.xmppPreKey = xmppPreKey;
+const parseAndInjectE2ESessions = async (node, repository) => {
+    const extractKey = (key) => (key ? ({
+        keyId: (0, WABinary_1.getBinaryNodeChildUInt)(key, 'id', 3),
+        publicKey: (0, crypto_1.generateSignalPubKey)((0, WABinary_1.getBinaryNodeChildBuffer)(key, 'value')),
+        signature: (0, WABinary_1.getBinaryNodeChildBuffer)(key, 'signature'),
+    }) : undefined);
+    const nodes = (0, WABinary_1.getBinaryNodeChildren)((0, WABinary_1.getBinaryNodeChild)(node, 'list'), 'user');
+    for (const node of nodes) {
+        (0, WABinary_1.assertNodeErrorFree)(node);
+    }
+    // Most of the work in repository.injectE2ESession is CPU intensive, not IO
+    // So Promise.all doesn't really help here,
+    // but blocks even loop if we're using it inside keys.transaction, and it makes it "sync" actually
+    // This way we chunk it in smaller parts and between those parts we can yield to the event loop
+    // It's rare case when you need to E2E sessions for so many users, but it's possible
+    const chunkSize = 100;
+    const chunks = (0, lodash_1.chunk)(nodes, chunkSize);
+    for (const nodesChunk of chunks) {
+        await Promise.all(nodesChunk.map(async (node) => {
+            const signedKey = (0, WABinary_1.getBinaryNodeChild)(node, 'skey');
+            const key = (0, WABinary_1.getBinaryNodeChild)(node, 'key');
+            const identity = (0, WABinary_1.getBinaryNodeChildBuffer)(node, 'identity');
+            const jid = node.attrs.jid;
+            const registrationId = (0, WABinary_1.getBinaryNodeChildUInt)(node, 'registration', 4);
+            await repository.injectE2ESession({
+                jid,
+                session: {
+                    registrationId: registrationId,
+                    identityKey: (0, crypto_1.generateSignalPubKey)(identity),
+                    signedPreKey: extractKey(signedKey),
+                    preKey: extractKey(key)
+                }
+            });
+        }));
+    }
+};
+exports.parseAndInjectE2ESessions = parseAndInjectE2ESessions;
+const extractDeviceJids = (result, myJid, excludeZeroDevices) => {
+    const { user: myUser, device: myDevice } = (0, WABinary_1.jidDecode)(myJid);
+    const extracted = [];
+    for (const userResult of result) {
+        const { devices, id } = userResult;
+        const { user } = (0, WABinary_1.jidDecode)(id);
+        const deviceList = devices === null || devices === void 0 ? void 0 : devices.deviceList;
+        if (Array.isArray(deviceList)) {
+            for (const { id: device, keyIndex } of deviceList) {
+                if ((!excludeZeroDevices || device !== 0) && // if zero devices are not-excluded, or device is non zero
+                    (myUser !== user || myDevice !== device) && // either different user or if me user, not this device
+                    (device === 0 || !!keyIndex) // ensure that "key-index" is specified for "non-zero" devices, produces a bad req otherwise
+                ) {
+                    extracted.push({ user, device });
+                }
+            }
+        }
+    }
+    return extracted;
+};
+exports.extractDeviceJids = extractDeviceJids;
+/**
+ * get the next N keys for upload or processing
+ * @param count number of pre-keys to get or generate
+ */
+const getNextPreKeys = async ({ creds, keys }, count) => {
+    const { newPreKeys, lastPreKeyId, preKeysRange } = (0, exports.generateOrGetPreKeys)(creds, count);
+    const update = {
+        nextPreKeyId: Math.max(lastPreKeyId + 1, creds.nextPreKeyId),
+        firstUnuploadedPreKeyId: Math.max(creds.firstUnuploadedPreKeyId, lastPreKeyId + 1)
+    };
+    await keys.set({ 'pre-key': newPreKeys });
+    const preKeys = await (0, exports.getPreKeys)(keys, preKeysRange[0], preKeysRange[0] + preKeysRange[1]);
+    return { update, preKeys };
+};
+exports.getNextPreKeys = getNextPreKeys;
+const getNextPreKeysNode = async (state, count) => {
+    const { creds } = state;
+    const { update, preKeys } = await (0, exports.getNextPreKeys)(state, count);
+    const node = {
+        tag: 'iq',
+        attrs: {
+            xmlns: 'encrypt',
+            type: 'set',
+            to: WABinary_1.S_WHATSAPP_NET,
+        },
+        content: [
+            { tag: 'registration', attrs: {}, content: (0, generics_1.encodeBigEndian)(creds.registrationId) },
+            { tag: 'type', attrs: {}, content: Defaults_1.KEY_BUNDLE_TYPE },
+            { tag: 'identity', attrs: {}, content: creds.signedIdentityKey.public },
+            { tag: 'list', attrs: {}, content: Object.keys(preKeys).map(k => (0, exports.xmppPreKey)(preKeys[+k], +k)) },
+            (0, exports.xmppSignedPreKey)(creds.signedPreKey)
+        ]
+    };
+    return { update, node };
+};
+exports.getNextPreKeysNode = getNextPreKeysNode;

package/lib/Utils/use-multi-file-auth-state.d.ts

@@ -0,0 +1,12 @@
+import { AuthenticationState } from '../Types';
+/**
+ * stores the full authentication state in a single folder.
+ * Far more efficient than singlefileauthstate
+ *
+ * Again, I wouldn't endorse this for any production level use other than perhaps a bot.
+ * Would recommend writing an auth state for use with a proper SQL or No-SQL DB
+ * */
+export declare const useMultiFileAuthState: (folder: string) => Promise<{
+    state: AuthenticationState;
+    saveCreds: () => Promise<void>;
+}>;