zapo-js 0.2.0 → 1.0.0

package/dist/store/memory/device-list.store.js

@@ -0,0 +1,107 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WaDeviceListMemoryStore = void 0;
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
+const DEFAULTS = Object.freeze({
+    ttlMs: 5 * 60 * 1000,
+    maxUsers: 16384
+});
+class WaDeviceListMemoryStore {
+    constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
+        if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
+            throw new Error('device-list ttlMs must be a positive finite number');
+        }
+        this.records = new Map();
+        this.altIndex = new Map();
+        this.ttlMs = ttlMs;
+        this.maxUsers = (0, coercion_1.resolvePositive)(options.maxUsers, DEFAULTS.maxUsers, 'WaDeviceListMemoryStoreOptions.maxUsers');
+        this.logger = options.logger;
+        this.capacityWarned = false;
+        this.cleanup = (0, collections_1.createPeriodicCleanup)(ttlMs, () => {
+            void this.cleanupExpired(Date.now());
+        });
+    }
+    warnCapacity() {
+        if (this.capacityWarned || !this.logger)
+            return;
+        this.capacityWarned = true;
+        this.logger.warn('device list store at capacity, evicting oldest', {
+            max: this.maxUsers
+        });
+    }
+    async upsertUserDevicesBatch(snapshots) {
+        for (let index = 0; index < snapshots.length; index += 1) {
+            const snapshot = snapshots[index];
+            const previous = this.records.get(snapshot.userJid);
+            if (previous?.altUserJid && previous.altUserJid !== snapshot.altUserJid) {
+                this.altIndex.delete(previous.altUserJid);
+            }
+            (0, collections_1.setBoundedMapEntry)(this.records, snapshot.userJid, {
+                ...snapshot,
+                expiresAtMs: snapshot.updatedAtMs + this.ttlMs
+            }, this.maxUsers, () => this.warnCapacity());
+            if (snapshot.altUserJid) {
+                this.altIndex.set(snapshot.altUserJid, snapshot.userJid);
+            }
+        }
+    }
+    async getUserDevicesBatch(userJids, nowMs = Date.now()) {
+        const snapshots = new Array(userJids.length);
+        for (let index = 0; index < userJids.length; index += 1) {
+            snapshots[index] = this.readLive(userJids[index], nowMs);
+        }
+        return snapshots;
+    }
+    async findByAnyUserJid(jid, nowMs = Date.now()) {
+        const direct = this.readLive(jid, nowMs);
+        if (direct)
+            return direct;
+        const primary = this.altIndex.get(jid);
+        if (!primary)
+            return null;
+        return this.readLive(primary, nowMs);
+    }
+    async deleteUserDevices(userJid) {
+        const record = this.records.get(userJid);
+        if (!record)
+            return 0;
+        if (record.altUserJid)
+            this.altIndex.delete(record.altUserJid);
+        this.records.delete(userJid);
+        return 1;
+    }
+    async cleanupExpired(nowMs) {
+        let removed = 0;
+        for (const [userJid, record] of this.records) {
+            if (record.expiresAtMs > nowMs)
+                continue;
+            if (record.altUserJid)
+                this.altIndex.delete(record.altUserJid);
+            this.records.delete(userJid);
+            removed += 1;
+        }
+        return removed;
+    }
+    async clear() {
+        this.records.clear();
+        this.altIndex.clear();
+    }
+    async destroy() {
+        this.cleanup.destroy();
+        await this.clear();
+    }
+    readLive(userJid, nowMs) {
+        const record = this.records.get(userJid);
+        if (!record)
+            return null;
+        if (record.expiresAtMs <= nowMs) {
+            if (record.altUserJid)
+                this.altIndex.delete(record.altUserJid);
+            this.records.delete(userJid);
+            return null;
+        }
+        return record;
+    }
+}
+exports.WaDeviceListMemoryStore = WaDeviceListMemoryStore;

package/dist/store/{providers/memory/participants.store.js → memory/group-metadata.store.js}

@@ -1,29 +1,41 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.WaParticipantsMemoryStore = void 0;
-const coercion_1 = require("../../../util/coercion");
-const collections_1 = require("../../../util/collections");
+exports.WaGroupMetadataMemoryStore = void 0;
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
 const DEFAULTS = Object.freeze({
     ttlMs: 5 * 60 * 1000,
     maxGroups: 4096
 });
-class WaParticipantsMemoryStore {
+class WaGroupMetadataMemoryStore {
     constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
+        if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
+            throw new Error('groupMetadata ttlMs must be a positive finite number');
+        }
         this.records = new Map();
         this.ttlMs = ttlMs;
-        this.maxGroups = (0, coercion_1.resolvePositive)(options.maxGroups, DEFAULTS.maxGroups, 'WaParticipantsMemoryStoreOptions.maxGroups');
-        this.cleanupTimer = setInterval(() => {
+        this.maxGroups = (0, coercion_1.resolvePositive)(options.maxGroups, DEFAULTS.maxGroups, 'WaGroupMetadataMemoryStoreOptions.maxGroups');
+        this.logger = options.logger;
+        this.capacityWarned = false;
+        this.cleanup = (0, collections_1.createPeriodicCleanup)(ttlMs, () => {
             void this.cleanupExpired(Date.now());
-        }, (0, collections_1.resolveCleanupIntervalMs)(ttlMs));
-        this.cleanupTimer.unref();
+        });
+    }
+    warnCapacity() {
+        if (this.capacityWarned || !this.logger)
+            return;
+        this.capacityWarned = true;
+        this.logger.warn('group metadata store at capacity, evicting oldest', {
+            max: this.maxGroups
+        });
     }
-    async upsertGroupParticipants(snapshot) {
+    async upsertGroupMetadata(snapshot) {
         (0, collections_1.setBoundedMapEntry)(this.records, snapshot.groupJid, {
             ...snapshot,
             expiresAtMs: snapshot.updatedAtMs + this.ttlMs
-        }, this.maxGroups);
+        }, this.maxGroups, () => this.warnCapacity());
     }
-    async getGroupParticipants(groupJid, nowMs = Date.now()) {
+    async getGroupMetadata(groupJid, nowMs = Date.now()) {
         const record = this.records.get(groupJid);
         if (!record) {
             return null;
@@ -34,7 +46,7 @@ class WaParticipantsMemoryStore {
         }
         return record;
     }
-    async deleteGroupParticipants(groupJid) {
+    async deleteGroupMetadata(groupJid) {
         return this.records.delete(groupJid) ? 1 : 0;
     }
     async cleanupExpired(nowMs) {
@@ -51,8 +63,8 @@ class WaParticipantsMemoryStore {
         this.records.clear();
     }
     async destroy() {
-        clearInterval(this.cleanupTimer);
+        this.cleanup.destroy();
         await this.clear();
     }
 }
-exports.WaParticipantsMemoryStore = WaParticipantsMemoryStore;
+exports.WaGroupMetadataMemoryStore = WaGroupMetadataMemoryStore;

package/dist/store/memory/identity.store.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WaIdentityMemoryStore = void 0;
+const jid_1 = require("../../protocol/jid");
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
+const DEFAULT_MAX_REMOTE_IDENTITIES = 8192;
+class WaIdentityMemoryStore {
+    constructor(options = {}) {
+        this.remoteIdentities = new Map();
+        this.maxRemoteIdentities = (0, coercion_1.resolvePositive)(options.maxRemoteIdentities, DEFAULT_MAX_REMOTE_IDENTITIES, 'WaIdentityMemoryStoreOptions.maxRemoteIdentities');
+    }
+    async getRemoteIdentity(address) {
+        return this.remoteIdentities.get((0, jid_1.signalAddressKey)(address)) ?? null;
+    }
+    async getRemoteIdentities(addresses) {
+        const result = new Array(addresses.length);
+        for (let i = 0; i < addresses.length; i += 1) {
+            result[i] = this.remoteIdentities.get((0, jid_1.signalAddressKey)(addresses[i])) ?? null;
+        }
+        return result;
+    }
+    async setRemoteIdentity(address, identityKey) {
+        (0, collections_1.setBoundedMapEntry)(this.remoteIdentities, (0, jid_1.signalAddressKey)(address), identityKey, this.maxRemoteIdentities);
+    }
+    async setRemoteIdentities(entries) {
+        for (const entry of entries) {
+            (0, collections_1.setBoundedMapEntry)(this.remoteIdentities, (0, jid_1.signalAddressKey)(entry.address), entry.identityKey, this.maxRemoteIdentities);
+        }
+    }
+    async clear() {
+        this.remoteIdentities.clear();
+    }
+}
+exports.WaIdentityMemoryStore = WaIdentityMemoryStore;

package/dist/store/memory/message-secret.store.js

@@ -0,0 +1,94 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WaMessageSecretMemoryStore = void 0;
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
+const DEFAULTS = Object.freeze({
+    ttlMs: 30 * 60 * 1000,
+    maxSecrets: 10000
+});
+class WaMessageSecretMemoryStore {
+    constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
+        if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
+            throw new Error('message-secret ttlMs must be a positive finite number');
+        }
+        this.secrets = new Map();
+        this.ttlMs = ttlMs;
+        this.maxSecrets = (0, coercion_1.resolvePositive)(options.maxSecrets, DEFAULTS.maxSecrets, 'WaMessageSecretMemoryStoreOptions.maxSecrets');
+        this.logger = options.logger;
+        this.capacityWarned = false;
+        this.cleanup = (0, collections_1.createPeriodicCleanup)(ttlMs, () => {
+            void this.cleanupExpired(Date.now());
+        });
+    }
+    warnCapacity() {
+        if (this.capacityWarned || !this.logger)
+            return;
+        this.capacityWarned = true;
+        this.logger.warn('message secret store at capacity, evicting oldest', {
+            max: this.maxSecrets
+        });
+    }
+    async get(messageId, nowMs = Date.now()) {
+        const cached = this.secrets.get(messageId);
+        if (!cached)
+            return null;
+        if (cached.expiresAtMs <= nowMs) {
+            this.secrets.delete(messageId);
+            return null;
+        }
+        return { secret: cached.secret, senderJid: cached.senderJid };
+    }
+    async getBatch(messageIds, nowMs = Date.now()) {
+        const result = new Array(messageIds.length);
+        for (let i = 0; i < messageIds.length; i += 1) {
+            const cached = this.secrets.get(messageIds[i]);
+            if (!cached) {
+                result[i] = null;
+                continue;
+            }
+            if (cached.expiresAtMs <= nowMs) {
+                this.secrets.delete(messageIds[i]);
+                result[i] = null;
+                continue;
+            }
+            result[i] = { secret: cached.secret, senderJid: cached.senderJid };
+        }
+        return result;
+    }
+    async set(messageId, entry) {
+        (0, collections_1.setBoundedMapEntry)(this.secrets, messageId, {
+            secret: entry.secret,
+            senderJid: entry.senderJid,
+            expiresAtMs: Date.now() + this.ttlMs
+        }, this.maxSecrets, () => this.warnCapacity());
+    }
+    async setBatch(entries) {
+        const nowMs = Date.now();
+        for (let i = 0; i < entries.length; i += 1) {
+            (0, collections_1.setBoundedMapEntry)(this.secrets, entries[i].messageId, {
+                secret: entries[i].entry.secret,
+                senderJid: entries[i].entry.senderJid,
+                expiresAtMs: nowMs + this.ttlMs
+            }, this.maxSecrets, () => this.warnCapacity());
+        }
+    }
+    async cleanupExpired(nowMs) {
+        let removed = 0;
+        for (const [messageId, entry] of this.secrets) {
+            if (entry.expiresAtMs > nowMs)
+                continue;
+            this.secrets.delete(messageId);
+            removed += 1;
+        }
+        return removed;
+    }
+    async clear() {
+        this.secrets.clear();
+    }
+    async destroy() {
+        this.cleanup.destroy();
+        this.secrets.clear();
+    }
+}
+exports.WaMessageSecretMemoryStore = WaMessageSecretMemoryStore;

package/dist/store/{providers/memory → memory}/message.store.js

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WaMessageMemoryStore = void 0;
-const coercion_1 = require("../../../util/coercion");
-const collections_1 = require("../../../util/collections");
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
 const DEFAULT_MESSAGE_MEMORY_STORE_LIMITS = Object.freeze({
     messages: 50000
 });

package/dist/store/memory/pre-key.store.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WaPreKeyMemoryStore = void 0;
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
+const DEFAULT_MAX_PRE_KEYS = 4096;
+class WaPreKeyMemoryStore {
+    constructor(options = {}) {
+        this.preKeys = new Map();
+        this.uploadedPreKeys = new Set();
+        this.serverHasPreKeys = false;
+        this.nextPreKeyId = 1;
+        this.maxPreKeys = (0, coercion_1.resolvePositive)(options.maxPreKeys, DEFAULT_MAX_PRE_KEYS, 'WaPreKeyMemoryStoreOptions.maxPreKeys');
+    }
+    async putPreKey(record) {
+        (0, collections_1.setBoundedMapEntry)(this.preKeys, record.keyId, record, this.maxPreKeys, (keyId) => {
+            this.uploadedPreKeys.delete(keyId);
+        });
+        if (record.keyId >= this.nextPreKeyId) {
+            this.nextPreKeyId = record.keyId + 1;
+        }
+    }
+    async getOrGenPreKeys(count, generator) {
+        if (!Number.isSafeInteger(count) || count <= 0) {
+            throw new Error(`invalid prekey count: ${count}`);
+        }
+        const available = [];
+        const availableKeyIds = [];
+        for (const keyId of this.preKeys.keys()) {
+            if (!this.uploadedPreKeys.has(keyId)) {
+                availableKeyIds.push(keyId);
+            }
+        }
+        availableKeyIds.sort((left, right) => left - right);
+        for (let index = 0; index < availableKeyIds.length; index += 1) {
+            const keyId = availableKeyIds[index];
+            const record = this.preKeys.get(keyId);
+            if (!record) {
+                continue;
+            }
+            available.push(record);
+            if (available.length >= count) {
+                return available;
+            }
+        }
+        while (available.length < count) {
+            const record = await generator(this.nextPreKeyId++);
+            (0, collections_1.setBoundedMapEntry)(this.preKeys, record.keyId, record, this.maxPreKeys, (keyId) => {
+                this.uploadedPreKeys.delete(keyId);
+            });
+            available.push(record);
+        }
+        return available;
+    }
+    async getPreKeyById(keyId) {
+        return this.preKeys.get(keyId) ?? null;
+    }
+    async getPreKeysById(keyIds) {
+        const result = new Array(keyIds.length);
+        for (let i = 0; i < keyIds.length; i += 1) {
+            result[i] = this.preKeys.get(keyIds[i]) ?? null;
+        }
+        return result;
+    }
+    async consumePreKeyById(keyId) {
+        const record = this.preKeys.get(keyId) ?? null;
+        if (!record) {
+            return null;
+        }
+        this.preKeys.delete(keyId);
+        this.uploadedPreKeys.delete(keyId);
+        return record;
+    }
+    async getOrGenSinglePreKey(generator) {
+        const preKeys = await this.getOrGenPreKeys(1, generator);
+        return preKeys[0];
+    }
+    async markKeyAsUploaded(keyId) {
+        if (keyId < 0 || keyId >= this.nextPreKeyId) {
+            throw new Error(`prekey ${keyId} is out of boundary`);
+        }
+        for (const candidate of this.preKeys.keys()) {
+            if (candidate <= keyId) {
+                this.uploadedPreKeys.add(candidate);
+            }
+        }
+    }
+    async setServerHasPreKeys(value) {
+        this.serverHasPreKeys = value;
+    }
+    async getServerHasPreKeys() {
+        return this.serverHasPreKeys;
+    }
+    async clear() {
+        this.preKeys.clear();
+        this.uploadedPreKeys.clear();
+        this.serverHasPreKeys = false;
+        this.nextPreKeyId = 1;
+    }
+}
+exports.WaPreKeyMemoryStore = WaPreKeyMemoryStore;

package/dist/store/{providers/memory → memory}/privacy-token.store.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WaPrivacyTokenMemoryStore = void 0;
-const collections_1 = require("../../../util/collections");
+const collections_1 = require("../../util/collections");
 const DEFAULT_MAX_ENTRIES = 10000;
 class WaPrivacyTokenMemoryStore {
     constructor(maxEntries = DEFAULT_MAX_ENTRIES) {

package/dist/store/{providers/memory → memory}/retry.store.js

@@ -1,18 +1,46 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WaRetryMemoryStore = void 0;
-const collections_1 = require("../../../util/collections");
-const DEFAULT_RETRY_TTL_MS = 60 * 1000;
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
+const DEFAULTS = Object.freeze({
+    ttlMs: 60 * 1000,
+    maxOutboundMessages: 10000,
+    maxInboundCounters: 20000
+});
 class WaRetryMemoryStore {
-    constructor(ttlMs = DEFAULT_RETRY_TTL_MS) {
+    constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
+        if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
+            throw new Error('retry ttlMs must be a positive finite number');
+        }
         this.outboundMessages = new Map();
         this.eligibleSets = new Map();
         this.inboundCounters = new Map();
         this.ttlMs = ttlMs;
-        this.cleanupTimer = setInterval(() => {
+        this.maxOutboundMessages = (0, coercion_1.resolvePositive)(options.maxOutboundMessages, DEFAULTS.maxOutboundMessages, 'WaRetryMemoryStoreOptions.maxOutboundMessages');
+        this.maxInboundCounters = (0, coercion_1.resolvePositive)(options.maxInboundCounters, DEFAULTS.maxInboundCounters, 'WaRetryMemoryStoreOptions.maxInboundCounters');
+        this.logger = options.logger;
+        this.outboundCapacityWarned = false;
+        this.inboundCapacityWarned = false;
+        this.cleanup = (0, collections_1.createPeriodicCleanup)(this.ttlMs, () => {
             void this.cleanupExpired(Date.now());
-        }, (0, collections_1.resolveCleanupIntervalMs)(ttlMs));
-        this.cleanupTimer.unref();
+        });
+    }
+    warnOutboundCapacity() {
+        if (this.outboundCapacityWarned || !this.logger)
+            return;
+        this.outboundCapacityWarned = true;
+        this.logger.warn('retry outbound store at capacity, evicting oldest', {
+            max: this.maxOutboundMessages
+        });
+    }
+    warnInboundCapacity() {
+        if (this.inboundCapacityWarned || !this.logger)
+            return;
+        this.inboundCapacityWarned = true;
+        this.logger.warn('retry inbound counters at capacity, evicting oldest', {
+            max: this.maxInboundCounters
+        });
     }
     getTtlMs() {
         return this.ttlMs;
@@ -57,7 +85,7 @@ class WaRetryMemoryStore {
         return { eligible: true, delivered: false };
     }
     async upsertOutboundMessage(record) {
-        this.outboundMessages.set(record.messageId, {
+        const storedRecord = {
             ...record,
             eligibleRequesterDeviceJids: record.eligibleRequesterDeviceJids
                 ? [...record.eligibleRequesterDeviceJids]
@@ -65,6 +93,10 @@ class WaRetryMemoryStore {
             deliveredRequesterDeviceJids: record.deliveredRequesterDeviceJids
                 ? [...record.deliveredRequesterDeviceJids]
                 : undefined
+        };
+        (0, collections_1.setBoundedMapEntry)(this.outboundMessages, record.messageId, storedRecord, this.maxOutboundMessages, (messageId) => {
+            this.eligibleSets.delete(messageId);
+            this.warnOutboundCapacity();
         });
         if (record.eligibleRequesterDeviceJids && record.eligibleRequesterDeviceJids.length > 0) {
             this.eligibleSets.set(record.messageId, new Set(record.eligibleRequesterDeviceJids));
@@ -86,11 +118,14 @@ class WaRetryMemoryStore {
         if (!current) {
             return;
         }
-        this.outboundMessages.set(messageId, {
+        (0, collections_1.setBoundedMapEntry)(this.outboundMessages, messageId, {
             ...current,
             state,
             updatedAtMs,
             expiresAtMs
+        }, this.maxOutboundMessages, (evictedMessageId) => {
+            this.eligibleSets.delete(evictedMessageId);
+            this.warnOutboundCapacity();
         });
     }
     async markOutboundRequesterDelivered(messageId, requesterDeviceJid, updatedAtMs, expiresAtMs) {
@@ -100,21 +135,21 @@ class WaRetryMemoryStore {
         }
         const delivered = new Set(current.deliveredRequesterDeviceJids ?? []);
         delivered.add(requesterDeviceJid);
-        this.outboundMessages.set(messageId, {
+        (0, collections_1.setBoundedMapEntry)(this.outboundMessages, messageId, {
             ...current,
             deliveredRequesterDeviceJids: Array.from(delivered),
             updatedAtMs,
             expiresAtMs
+        }, this.maxOutboundMessages, (evictedMessageId) => {
+            this.eligibleSets.delete(evictedMessageId);
+            this.warnOutboundCapacity();
         });
     }
     async incrementInboundCounter(messageId, requesterJid, _updatedAtMs, expiresAtMs) {
         const key = this.counterKey(messageId, requesterJid);
         const current = this.inboundCounters.get(key);
         const count = current ? current.count + 1 : 1;
-        this.inboundCounters.set(key, {
-            count,
-            expiresAtMs
-        });
+        (0, collections_1.setBoundedMapEntry)(this.inboundCounters, key, { count, expiresAtMs }, this.maxInboundCounters, () => this.warnInboundCapacity());
         return count;
     }
     async cleanupExpired(nowMs) {
@@ -140,7 +175,7 @@ class WaRetryMemoryStore {
         this.inboundCounters.clear();
     }
     async destroy() {
-        clearInterval(this.cleanupTimer);
+        this.cleanup.destroy();
         await this.clear();
     }
     counterKey(messageId, requesterJid) {

package/dist/store/{providers/memory → memory}/sender-key.store.js

@@ -1,9 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SenderKeyMemoryStore = void 0;
-const jid_1 = require("../../../protocol/jid");
-const coercion_1 = require("../../../util/coercion");
-const collections_1 = require("../../../util/collections");
+const jid_1 = require("../../protocol/jid");
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
 const DEFAULT_SENDER_KEY_STORE_LIMITS = Object.freeze({
     senderKeys: 8192,
     senderDistributions: 8192

package/dist/store/memory/session.store.js

@@ -0,0 +1,49 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WaSessionMemoryStore = void 0;
+const jid_1 = require("../../protocol/jid");
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
+const DEFAULT_MAX_SESSIONS = 8192;
+class WaSessionMemoryStore {
+    constructor(options = {}) {
+        this.signalSessions = new Map();
+        this.maxSessions = (0, coercion_1.resolvePositive)(options.maxSessions, DEFAULT_MAX_SESSIONS, 'WaSessionMemoryStoreOptions.maxSessions');
+    }
+    async hasSession(address) {
+        return this.signalSessions.has((0, jid_1.signalAddressKey)(address));
+    }
+    async hasSessions(addresses) {
+        const result = new Array(addresses.length);
+        for (let i = 0; i < addresses.length; i += 1) {
+            result[i] = this.signalSessions.has((0, jid_1.signalAddressKey)(addresses[i]));
+        }
+        return result;
+    }
+    async getSession(address) {
+        return this.signalSessions.get((0, jid_1.signalAddressKey)(address)) ?? null;
+    }
+    async getSessionsBatch(addresses) {
+        const result = new Array(addresses.length);
+        for (let i = 0; i < addresses.length; i += 1) {
+            result[i] = this.signalSessions.get((0, jid_1.signalAddressKey)(addresses[i])) ?? null;
+        }
+        return result;
+    }
+    async setSession(address, session) {
+        (0, collections_1.setBoundedMapEntry)(this.signalSessions, (0, jid_1.signalAddressKey)(address), session, this.maxSessions);
+    }
+    async setSessionsBatch(entries) {
+        for (let index = 0; index < entries.length; index += 1) {
+            const entry = entries[index];
+            (0, collections_1.setBoundedMapEntry)(this.signalSessions, (0, jid_1.signalAddressKey)(entry.address), entry.session, this.maxSessions);
+        }
+    }
+    async deleteSession(address) {
+        this.signalSessions.delete((0, jid_1.signalAddressKey)(address));
+    }
+    async clear() {
+        this.signalSessions.clear();
+    }
+}
+exports.WaSessionMemoryStore = WaSessionMemoryStore;

package/dist/store/memory/signal.store.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WaSignalMemoryStore = void 0;
+class WaSignalMemoryStore {
+    constructor() {
+        this.registrationInfo = null;
+        this.signedPreKey = null;
+        this.signedPreKeyRotationTs = null;
+    }
+    async getRegistrationInfo() {
+        return this.registrationInfo;
+    }
+    async setRegistrationInfo(info) {
+        this.registrationInfo = info;
+    }
+    async getSignedPreKey() {
+        return this.signedPreKey;
+    }
+    async setSignedPreKey(record) {
+        this.signedPreKey = record;
+    }
+    async getSignedPreKeyById(keyId) {
+        if (!this.signedPreKey) {
+            return null;
+        }
+        return this.signedPreKey.keyId === keyId ? this.signedPreKey : null;
+    }
+    async setSignedPreKeyRotationTs(value) {
+        this.signedPreKeyRotationTs = value;
+    }
+    async getSignedPreKeyRotationTs() {
+        return this.signedPreKeyRotationTs;
+    }
+    async clear() {
+        this.registrationInfo = null;
+        this.signedPreKey = null;
+        this.signedPreKeyRotationTs = null;
+    }
+}
+exports.WaSignalMemoryStore = WaSignalMemoryStore;

package/dist/store/{providers/memory → memory}/thread.store.js

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WaThreadMemoryStore = void 0;
-const coercion_1 = require("../../../util/coercion");
-const collections_1 = require("../../../util/collections");
+const coercion_1 = require("../../util/coercion");
+const collections_1 = require("../../util/collections");
 const DEFAULT_THREAD_MEMORY_STORE_LIMITS = Object.freeze({
     threads: 10000
 });