zapo-js 0.1.2 → 0.2.0

package/dist/client/coordinators/WaPassiveTasksCoordinator.js

@@ -34,6 +34,16 @@ class WaPassiveTasksCoordinator {
             this.passiveTasksPromise = null;
         });
     }
+    async handlePreKeyLowNotification() {
+        await Promise.all([
+            this.signalStore.setServerHasPreKeys(false),
+            this.runtime.persistServerHasPreKeys(false)
+        ]);
+        await this.uploadPreKeysIfMissing(false);
+    }
+    async handleDigestNotification() {
+        await this.validateDigestAndRecoverPreKeys();
+    }
     resetInFlightState() {
         if (!this.passiveTasksPromise) {
             return;
@@ -47,14 +57,18 @@ class WaPassiveTasksCoordinator {
             this.logger.trace('passive connect tasks skipped: session is not registered');
             return;
         }
-        await this.uploadPreKeysIfMissing();
-        await this.validateDigestAndRecoverPreKeys();
-        await this.rotateSignedPreKeyIfDue();
+        const signalMeta = await this.signalStore.getSignalMeta();
+        const prefetchedLocalKeyBundle = this.resolveLocalKeyBundle(signalMeta);
+        await this.uploadPreKeysIfMissing(signalMeta.serverHasPreKeys, prefetchedLocalKeyBundle);
+        await this.validateDigestAndRecoverPreKeys(prefetchedLocalKeyBundle);
+        await this.rotateSignedPreKeyIfDue(signalMeta.signedPreKeyRotationTs);
         await this.flushDanglingReceipts();
     }
-    async validateDigestAndRecoverPreKeys() {
+    async validateDigestAndRecoverPreKeys(prefetchedLocalKeyBundle) {
         try {
-            const validation = await this.signalDigestSync.validateLocalKeyBundle();
+            const validation = prefetchedLocalKeyBundle
+                ? await this.signalDigestSync.validateLocalKeyBundle(prefetchedLocalKeyBundle)
+                : await this.signalDigestSync.validateLocalKeyBundle();
             if (validation.valid) {
                 this.logger.debug('signal digest validated', {
                     preKeyCount: validation.preKeyCount
@@ -73,7 +87,7 @@ class WaPassiveTasksCoordinator {
                 this.signalStore.setServerHasPreKeys(false),
                 this.runtime.persistServerHasPreKeys(false)
             ]);
-            await this.uploadPreKeysIfMissing();
+            await this.uploadPreKeysIfMissing(false, prefetchedLocalKeyBundle);
         }
         catch (error) {
             this.logger.warn('signal digest validation failed with exception', {
@@ -81,20 +95,18 @@ class WaPassiveTasksCoordinator {
             });
         }
     }
-    async uploadPreKeysIfMissing() {
-        const serverHasPreKeys = await this.signalStore.getServerHasPreKeys();
+    async uploadPreKeysIfMissing(serverHasPreKeysHint, prefetchedLocalKeyBundle) {
+        const serverHasPreKeys = serverHasPreKeysHint ?? (await this.signalStore.getServerHasPreKeys());
         if (serverHasPreKeys) {
             this.logger.trace('prekey upload skipped: server already has prekeys');
             return;
         }
-        const [registrationInfo, signedPreKey] = await Promise.all([
-            this.signalStore.getRegistrationInfo(),
-            this.signalStore.getSignedPreKey()
-        ]);
-        if (!registrationInfo || !signedPreKey) {
+        const resolvedLocalKeyBundle = prefetchedLocalKeyBundle ?? (await this.resolveLocalKeyBundleFromStore());
+        if (!resolvedLocalKeyBundle) {
             this.logger.warn('prekey upload skipped: registration info is missing');
             return;
         }
+        const { registrationInfo, signedPreKey } = resolvedLocalKeyBundle;
         const preKeys = await this.signalStore.getOrGenPreKeys(constants_2.SIGNAL_UPLOAD_PREKEYS_COUNT, keygen_1.generatePreKeyPair);
         if (preKeys.length === 0) {
             throw new Error('no prekey available for upload');
@@ -106,6 +118,7 @@ class WaPassiveTasksCoordinator {
             lastPreKeyId
         });
         if (response.attrs.type === 'result') {
+            // Mark uploaded key first so the serverHasPreKeys flag never commits ahead of local key progress.
             await this.signalStore.markKeyAsUploaded(lastPreKeyId);
             await Promise.all([
                 this.signalStore.setServerHasPreKeys(true),
@@ -124,10 +137,12 @@ class WaPassiveTasksCoordinator {
             errorText: failure.errorText
         });
     }
-    async rotateSignedPreKeyIfDue() {
+    async rotateSignedPreKeyIfDue(signedPreKeyRotationTsHint) {
         try {
             const nowMs = Date.now();
-            const lastRotationTs = await this.signalStore.getSignedPreKeyRotationTs();
+            const lastRotationTs = signedPreKeyRotationTsHint === undefined
+                ? await this.signalStore.getSignedPreKeyRotationTs()
+                : signedPreKeyRotationTsHint;
             if (lastRotationTs === null) {
                 await this.signalStore.setSignedPreKeyRotationTs(nowMs);
                 this.logger.trace('signal rotate key skipped on first run');
@@ -153,6 +168,19 @@ class WaPassiveTasksCoordinator {
             });
         }
     }
+    resolveLocalKeyBundle(signalMeta) {
+        const { registrationInfo, signedPreKey } = signalMeta;
+        if (!registrationInfo || !signedPreKey) {
+            return null;
+        }
+        return {
+            registrationInfo,
+            signedPreKey
+        };
+    }
+    async resolveLocalKeyBundleFromStore() {
+        return this.resolveLocalKeyBundle(await this.signalStore.getSignalMeta());
+    }
     resolveRotationTimestamp(nowMs, errorCode) {
         if (errorCode !== undefined && errorCode >= 500) {
             const retryInMs = Math.min(this.signedPreKeyServerErrorBackoffMs, this.signedPreKeyRotationIntervalMs);
@@ -170,29 +198,44 @@ class WaPassiveTasksCoordinator {
             return;
         }
         this.logger.info('flushing dangling receipts', { count: pending.length });
-        for (let index = 0; index < pending.length; index += 1) {
-            const node = pending[index];
-            try {
-                await this.runtime.sendNodeDirect(node);
+        let cursor = 0;
+        while (cursor < pending.length) {
+            const batchEnd = Math.min(cursor + 4, pending.length);
+            const promises = [];
+            for (let i = cursor; i < batchEnd; i += 1) {
+                promises.push(this.runtime.sendNodeDirect(pending[i]));
             }
-            catch (error) {
-                const normalized = (0, primitives_1.toError)(error);
+            const results = await Promise.allSettled(promises);
+            let transientCount = 0;
+            for (let i = 0; i < results.length; i += 1) {
+                const result = results[i];
+                if (result.status === 'fulfilled') {
+                    continue;
+                }
+                const node = pending[cursor + i];
+                const normalized = (0, primitives_1.toError)(result.reason);
                 if (this.runtime.shouldQueueDanglingReceipt(node, normalized)) {
-                    for (let restoreIndex = index; restoreIndex < pending.length; restoreIndex += 1) {
-                        this.runtime.requeueDanglingReceipt(pending[restoreIndex]);
-                    }
-                    this.logger.warn('stopped dangling receipt flush due transient send error', {
-                        remaining: pending.length - index,
+                    this.runtime.requeueDanglingReceipt(node);
+                    transientCount += 1;
+                }
+                else {
+                    this.logger.warn('dropping dangling receipt due non-retryable send error', {
+                        id: node.attrs.id,
+                        to: node.attrs.to,
                         message: normalized.message
                     });
-                    return;
                 }
-                this.logger.warn('dropping dangling receipt due non-retryable send error', {
-                    id: node.attrs.id,
-                    to: node.attrs.to,
-                    message: normalized.message
+            }
+            if (transientCount > 0) {
+                for (let i = batchEnd; i < pending.length; i += 1) {
+                    this.runtime.requeueDanglingReceipt(pending[i]);
+                }
+                this.logger.warn('stopped dangling receipt flush due transient send error', {
+                    remaining: transientCount + (pending.length - batchEnd)
                 });
+                return;
             }
+            cursor = batchEnd;
         }
         this.logger.info('dangling receipts flushed');
     }

package/dist/client/coordinators/WaPrivacyCoordinator.js

@@ -0,0 +1,134 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createPrivacyCoordinator = createPrivacyCoordinator;
+const privacy_1 = require("../../protocol/privacy");
+const privacy_2 = require("../../transport/node/builders/privacy");
+const helpers_1 = require("../../transport/node/helpers");
+const query_1 = require("../../transport/node/query");
+const IGNORED_SERVER_CATEGORIES = new Set([
+    'pix',
+    'linked_profiles',
+    'stickers',
+    'dependentaccountmessages',
+    'cover_photo',
+    'dependent_account_calling',
+    'groupcreation'
+]);
+function isValidPrivacyValue(value) {
+    return (value !== privacy_1.WA_PRIVACY_VALUES.ERROR &&
+        Object.values(privacy_1.WA_PRIVACY_VALUES).includes(value));
+}
+function parsePrivacySettings(result) {
+    const privacyNode = (0, helpers_1.findNodeChild)(result, 'privacy');
+    if (!privacyNode) {
+        return {};
+    }
+    const settings = {};
+    const categories = (0, helpers_1.getNodeChildrenByTag)(privacyNode, privacy_1.WA_PRIVACY_TAGS.CATEGORY);
+    for (let i = 0; i < categories.length; i += 1) {
+        const node = categories[i];
+        const name = node.attrs.name;
+        const value = node.attrs.value;
+        if (!name || !value) {
+            continue;
+        }
+        if (IGNORED_SERVER_CATEGORIES.has(name)) {
+            continue;
+        }
+        if (!isValidPrivacyValue(value)) {
+            continue;
+        }
+        const settingName = privacy_1.WA_PRIVACY_CATEGORY_TO_SETTING[name];
+        if (settingName) {
+            settings[settingName] = value;
+        }
+    }
+    return settings;
+}
+function parseDisallowedList(result) {
+    const privacyNode = (0, helpers_1.findNodeChild)(result, 'privacy');
+    if (!privacyNode) {
+        return { jids: [] };
+    }
+    const listNode = (0, helpers_1.findNodeChild)(privacyNode, privacy_1.WA_PRIVACY_TAGS.LIST);
+    if (!listNode) {
+        return { jids: [] };
+    }
+    const dhash = listNode.attrs.dhash;
+    const userNodes = (0, helpers_1.getNodeChildrenByTag)(listNode, privacy_1.WA_PRIVACY_TAGS.USER);
+    const jids = new Array(userNodes.length);
+    let jidsCount = 0;
+    for (let i = 0; i < userNodes.length; i += 1) {
+        const jid = userNodes[i].attrs.jid;
+        if (jid) {
+            jids[jidsCount] = jid;
+            jidsCount += 1;
+        }
+    }
+    jids.length = jidsCount;
+    return { jids, dhash };
+}
+function parseBlocklist(result) {
+    const listNode = (0, helpers_1.findNodeChild)(result, 'list');
+    if (!listNode) {
+        return { jids: [] };
+    }
+    const dhash = listNode.attrs.dhash;
+    const itemNodes = (0, helpers_1.getNodeChildren)(listNode);
+    const jids = new Array(itemNodes.length);
+    let jidsCount = 0;
+    for (let i = 0; i < itemNodes.length; i += 1) {
+        const jid = itemNodes[i].attrs.jid;
+        if (jid) {
+            jids[jidsCount] = jid;
+            jidsCount += 1;
+        }
+    }
+    jids.length = jidsCount;
+    return { jids, dhash };
+}
+function createPrivacyCoordinator(options) {
+    const { queryWithContext } = options;
+    return {
+        getPrivacySettings: async () => {
+            const node = (0, privacy_2.buildGetPrivacySettingsIq)();
+            const result = await queryWithContext('privacy.getSettings', node);
+            (0, query_1.assertIqResult)(result, 'privacy.getSettings');
+            return parsePrivacySettings(result);
+        },
+        setPrivacySetting: async (setting, value) => {
+            const category = privacy_1.WA_PRIVACY_SETTING_TO_CATEGORY[setting];
+            const node = (0, privacy_2.buildSetPrivacyCategoryIq)(category, value);
+            const result = await queryWithContext('privacy.setSetting', node, undefined, {
+                category,
+                value
+            });
+            (0, query_1.assertIqResult)(result, 'privacy.setSetting');
+        },
+        getDisallowedList: async (setting) => {
+            const category = privacy_1.WA_PRIVACY_SETTING_TO_CATEGORY[setting];
+            const node = (0, privacy_2.buildGetPrivacyDisallowedListIq)(category);
+            const result = await queryWithContext('privacy.getDisallowedList', node, undefined, {
+                category
+            });
+            (0, query_1.assertIqResult)(result, 'privacy.getDisallowedList');
+            return parseDisallowedList(result);
+        },
+        getBlocklist: async () => {
+            const node = (0, privacy_2.buildGetBlocklistIq)();
+            const result = await queryWithContext('privacy.getBlocklist', node);
+            (0, query_1.assertIqResult)(result, 'privacy.getBlocklist');
+            return parseBlocklist(result);
+        },
+        blockUser: async (jid) => {
+            const node = (0, privacy_2.buildBlocklistChangeIq)(jid, 'block');
+            const result = await queryWithContext('privacy.blockUser', node, undefined, { jid });
+            (0, query_1.assertIqResult)(result, 'privacy.blockUser');
+        },
+        unblockUser: async (jid) => {
+            const node = (0, privacy_2.buildBlocklistChangeIq)(jid, 'unblock');
+            const result = await queryWithContext('privacy.unblockUser', node, undefined, { jid });
+            (0, query_1.assertIqResult)(result, 'privacy.unblockUser');
+        }
+    };
+}

package/dist/client/coordinators/WaProfileCoordinator.js

@@ -0,0 +1,212 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createProfileCoordinator = createProfileCoordinator;
+const profile_1 = require("../../transport/node/builders/profile");
+const usync_1 = require("../../transport/node/builders/usync");
+const helpers_1 = require("../../transport/node/helpers");
+const query_1 = require("../../transport/node/query");
+const bytes_1 = require("../../util/bytes");
+function parseProfilePicture(result) {
+    const pictureNode = (0, helpers_1.findNodeChild)(result, 'picture');
+    if (!pictureNode) {
+        return {};
+    }
+    return {
+        url: pictureNode.attrs.url,
+        directPath: pictureNode.attrs.direct_path,
+        id: pictureNode.attrs.id,
+        type: pictureNode.attrs.type
+    };
+}
+function parseSetPictureResult(result) {
+    const pictureNode = (0, helpers_1.findNodeChild)(result, 'picture');
+    return pictureNode?.attrs.id ?? null;
+}
+function parseUsyncProfiles(result) {
+    const usyncNode = (0, helpers_1.findNodeChild)(result, 'usync');
+    if (!usyncNode) {
+        return [];
+    }
+    const listNode = (0, helpers_1.findNodeChild)(usyncNode, 'list');
+    if (!listNode) {
+        return [];
+    }
+    const userNodes = (0, helpers_1.getNodeChildren)(listNode);
+    const profiles = new Array(userNodes.length);
+    let count = 0;
+    for (let i = 0; i < userNodes.length; i += 1) {
+        const userNode = userNodes[i];
+        if (userNode.tag !== 'user') {
+            continue;
+        }
+        const jid = userNode.attrs.jid;
+        if (!jid) {
+            continue;
+        }
+        const info = { jid };
+        const userContent = userNode.content;
+        if (!Array.isArray(userContent)) {
+            profiles[count] = info;
+            count += 1;
+            continue;
+        }
+        for (let j = 0; j < userContent.length; j += 1) {
+            const child = userContent[j];
+            if (child.tag === 'picture') {
+                const idAttr = child.attrs.id;
+                if (idAttr) {
+                    const parsed = Number.parseInt(idAttr, 10);
+                    if (Number.isSafeInteger(parsed)) {
+                        info.pictureId = parsed;
+                    }
+                }
+            }
+            else if (child.tag === 'status') {
+                const code = child.attrs.code;
+                if (code !== undefined && Number.parseInt(code, 10) === 401) {
+                    info.status = '';
+                }
+                else {
+                    const content = child.content;
+                    if (content instanceof Uint8Array) {
+                        const decoded = bytes_1.TEXT_DECODER.decode(content);
+                        info.status = decoded.length > 0 ? decoded : null;
+                    }
+                    else if (typeof content === 'string') {
+                        info.status = content.length > 0 ? content : null;
+                    }
+                    else {
+                        info.status = null;
+                    }
+                }
+            }
+        }
+        profiles[count] = info;
+        count += 1;
+    }
+    profiles.length = count;
+    return profiles;
+}
+function parseUsyncDisappearingModes(result) {
+    const usyncNode = (0, helpers_1.findNodeChild)(result, 'usync');
+    if (!usyncNode)
+        return [];
+    const listNode = (0, helpers_1.findNodeChild)(usyncNode, 'list');
+    if (!listNode)
+        return [];
+    const userNodes = (0, helpers_1.getNodeChildren)(listNode);
+    const results = new Array(userNodes.length);
+    let count = 0;
+    for (let i = 0; i < userNodes.length; i += 1) {
+        const userNode = userNodes[i];
+        if (userNode.tag !== 'user')
+            continue;
+        const userContent = userNode.content;
+        if (!Array.isArray(userContent))
+            continue;
+        for (let j = 0; j < userContent.length; j += 1) {
+            const child = userContent[j];
+            if (child.tag !== 'disappearing_mode')
+                continue;
+            const errorNode = (0, helpers_1.findNodeChild)(child, 'error');
+            if (errorNode)
+                continue;
+            const duration = Number.parseInt(child.attrs.duration ?? '0', 10);
+            const timestamp = Number.parseInt(child.attrs.t ?? '0', 10);
+            const entry = { duration, timestamp };
+            if (child.attrs.ephemerality_disabled === 'true') {
+                entry.ephemeralityDisabled = true;
+            }
+            results[count] = entry;
+            count += 1;
+        }
+    }
+    results.length = count;
+    return results;
+}
+function parseUsyncStatus(result) {
+    const profiles = parseUsyncProfiles(result);
+    if (profiles.length === 0) {
+        return { status: null };
+    }
+    return { status: profiles[0].status ?? null };
+}
+function createProfileCoordinator(options) {
+    const { queryWithContext, generateSid } = options;
+    return {
+        getProfilePicture: async (jid, type, existingId) => {
+            const node = (0, profile_1.buildGetProfilePictureIq)(jid, type, existingId);
+            const result = await queryWithContext('profile.getPicture', node, undefined, {
+                jid,
+                type: type ?? 'preview'
+            });
+            (0, query_1.assertIqResult)(result, 'profile.getPicture');
+            return parseProfilePicture(result);
+        },
+        setProfilePicture: async (imageBytes, targetJid) => {
+            const node = (0, profile_1.buildSetProfilePictureIq)(imageBytes, targetJid);
+            const result = await queryWithContext('profile.setPicture', node, undefined, {
+                targetJid,
+                size: imageBytes.length
+            });
+            (0, query_1.assertIqResult)(result, 'profile.setPicture');
+            return parseSetPictureResult(result);
+        },
+        deleteProfilePicture: async (targetJid) => {
+            const node = (0, profile_1.buildDeleteProfilePictureIq)(targetJid);
+            const result = await queryWithContext('profile.deletePicture', node, undefined, {
+                targetJid
+            });
+            (0, query_1.assertIqResult)(result, 'profile.deletePicture');
+        },
+        getStatus: async (jid) => {
+            const sid = await generateSid();
+            const queryNodes = (0, profile_1.buildGetStatusUsyncQueryNodes)();
+            const usyncNode = (0, usync_1.buildUsyncIq)({
+                sid,
+                queryProtocolNodes: [queryNodes[1]],
+                users: [{ jid }]
+            });
+            const result = await queryWithContext('profile.getStatus', usyncNode, undefined, {
+                jid
+            });
+            (0, query_1.assertIqResult)(result, 'profile.getStatus');
+            return parseUsyncStatus(result);
+        },
+        setStatus: async (text) => {
+            const node = (0, profile_1.buildSetStatusIq)(text);
+            const result = await queryWithContext('profile.setStatus', node);
+            (0, query_1.assertIqResult)(result, 'profile.setStatus');
+        },
+        getProfiles: async (jids) => {
+            if (jids.length === 0) {
+                return [];
+            }
+            const sid = await generateSid();
+            const queryProtocolNodes = (0, profile_1.buildGetStatusUsyncQueryNodes)();
+            const usyncNode = (0, usync_1.buildUsyncIq)({
+                sid,
+                queryProtocolNodes,
+                users: jids.map((jid) => ({ jid }))
+            });
+            const result = await queryWithContext('profile.getProfiles', usyncNode, undefined, {
+                count: jids.length
+            });
+            (0, query_1.assertIqResult)(result, 'profile.getProfiles');
+            return parseUsyncProfiles(result);
+        },
+        getDisappearingMode: async (jids) => {
+            if (jids.length === 0)
+                return [];
+            const sid = await generateSid();
+            const usyncNode = (0, usync_1.buildUsyncIq)({
+                sid,
+                queryProtocolNodes: [(0, profile_1.buildGetDisappearingModeUsyncQueryNode)()],
+                users: jids.map((jid) => ({ jid }))
+            });
+            const result = await queryWithContext('profile.getDisappearingMode', usyncNode, undefined, { count: jids.length });
+            (0, query_1.assertIqResult)(result, 'profile.getDisappearingMode');
+            return parseUsyncDisappearingModes(result);
+        }
+    };
+}