@nerxa/socketon 0.0.0-stage → 1.21.0

package/src/Store/index.d.ts

@@ -0,0 +1,3 @@
+export * from "./make-in-memory-store.js";
+export * from "./make-ordered-dictionary.js";
+export * from "./object-repository.js";

package/src/Store/index.js

@@ -0,0 +1,3 @@
+export * from './make-in-memory-store.js';
+export * from './make-ordered-dictionary.js';
+export * from './object-repository.js';

package/src/Store/make-in-memory-store.d.ts

@@ -0,0 +1,62 @@
+export function waChatKey(pin: any): {
+    key: (c: any) => string;
+    compare: (k1: any, k2: any) => any;
+};
+export function waMessageID(m: any): any;
+export namespace waLabelAssociationKey {
+    function key(la: any): any;
+    function compare(k1: any, k2: any): any;
+}
+export function makeInMemoryStore(config?: {}): {
+    chats: any;
+    contacts: {};
+    messages: {};
+    groupMetadata: {};
+    state: {
+        connection: string;
+    };
+    presences: {};
+    labels: ObjectRepository;
+    labelAssociations: any;
+    bind: (ev: any) => void;
+    /** loads messages from the store, if not found -- uses the legacy connection */
+    loadMessages: (jid: any, count: any, cursor: any) => Promise<any>;
+    /**
+     * Get all available labels for profile
+     *
+     * Keep in mind that the list is formed from predefined tags and tags
+     * that were "caught" during their editing.
+     */
+    getLabels: () => ObjectRepository;
+    /**
+     * Get labels for chat
+     *
+     * @returns Label IDs
+     **/
+    getChatLabels: (chatId: any) => any;
+    /**
+     * Get labels for message
+     *
+     * @returns Label IDs
+     **/
+    getMessageLabels: (messageId: any) => any;
+    loadMessage: (jid: any, id: any) => Promise<any>;
+    mostRecentMessage: (jid: any) => Promise<any>;
+    fetchImageUrl: (jid: any, sock: any) => Promise<any>;
+    fetchGroupMetadata: (jid: any, sock: any) => Promise<any>;
+    fetchMessageReceipts: ({ remoteJid, id }: {
+        remoteJid: any;
+        id: any;
+    }) => Promise<any>;
+    toJSON: () => {
+        chats: any;
+        contacts: {};
+        messages: {};
+        labels: ObjectRepository;
+        labelAssociations: any;
+    };
+    fromJSON: (json: any) => void;
+    writeToFile: (path: any) => void;
+    readFromFile: (path: any) => void;
+};
+import { ObjectRepository } from './object-repository.js';

package/src/Store/make-in-memory-store.js

@@ -0,0 +1,396 @@
+import keyedDB from '@adiwajshing/keyed-db';
+import { writeFileSync, readFileSync, existsSync } from 'fs';
+import { WAProto } from '../Types/index.js';
+import { LabelAssociationType } from '../Types/LabelAssociation.js';
+import { DEFAULT_CONNECTION_CONFIG } from '../Defaults/index.js';
+import { md5, toNumber, updateMessageWithReceipt, updateMessageWithReaction } from '../Utils/index.js';
+import { jidDecode, jidNormalizedUser } from '../WABinary/index.js';
+import { makeOrderedDictionary } from './make-ordered-dictionary.js';
+import { ObjectRepository } from './object-repository.js';
+export const waChatKey = (pin) => ({
+    key: (c) => (pin ? (c.pinned ? '1' : '0') : '') + (c.archived ? '0' : '1') + (c.conversationTimestamp ? c.conversationTimestamp.toString(16).padStart(8, '0') : '') + c.id,
+    compare: (k1, k2) => k2.localeCompare(k1)
+});
+export const waMessageID = (m) => m.key.id || '';
+export const waLabelAssociationKey = {
+    key: (la) => (la.type === LabelAssociationType.Chat ? la.chatId + la.labelId : la.chatId + la.messageId + la.labelId),
+    compare: (k1, k2) => k2.localeCompare(k1)
+};
+const makeMessagesDictionary = () => makeOrderedDictionary(waMessageID);
+export const makeInMemoryStore = (config = {}) => {
+    const socket = config.socket;
+    const chatKey = config.chatKey || waChatKey(true);
+    const labelAssociationKey = config.labelAssociationKey || waLabelAssociationKey;
+    const logger = config.logger || DEFAULT_CONNECTION_CONFIG.logger.child({ stream: 'in-mem-store' });
+    const KeyedDB = keyedDB?.default ?? keyedDB;
+    const chats = new KeyedDB(chatKey, c => c.id);
+    const messages = {};
+    const contacts = {};
+    const groupMetadata = {};
+    const presences = {};
+    const state = { connection: 'close' };
+    const labels = new ObjectRepository();
+    const labelAssociations = new KeyedDB(labelAssociationKey, labelAssociationKey.key);
+    const assertMessageList = (jid) => {
+        if (!messages[jid]) {
+            messages[jid] = makeMessagesDictionary();
+        }
+        return messages[jid];
+    };
+    const contactsUpsert = (newContacts) => {
+        const oldContacts = new Set(Object.keys(contacts));
+        for (const contact of newContacts) {
+            oldContacts.delete(contact.id);
+            contacts[contact.id] = Object.assign(contacts[contact.id] || {}, contact);
+        }
+        return oldContacts;
+    };
+    const labelsUpsert = (newLabels) => {
+        for (const label of newLabels) {
+            labels.upsertById(label.id, label);
+        }
+    };
+
+    const bind = (ev) => {
+        ev.on('connection.update', update => {
+            Object.assign(state, update);
+        });
+        ev.on('messaging-history.set', ({ chats: newChats, contacts: newContacts, messages: newMessages, isLatest, syncType }) => {
+            if (syncType === WAProto.HistorySync.HistorySyncType.ON_DEMAND) {
+                return;
+
+            }
+            if (isLatest) {
+                chats.clear();
+                for (const id in messages) {
+                    delete messages[id];
+                }
+            }
+            const chatsAdded = chats.insertIfAbsent(...newChats).length;
+            logger.debug({ chatsAdded }, 'synced chats');
+            const oldContacts = contactsUpsert(newContacts);
+            if (isLatest) {
+                for (const jid of oldContacts) {
+                    delete contacts[jid];
+                }
+            }
+            logger.debug({ deletedContacts: isLatest ? oldContacts.size : 0, newContacts }, 'synced contacts');
+            for (const msg of newMessages) {
+                const jid = msg.key.remoteJidAlt || msg.key.remoteJid;
+                const list = assertMessageList(jid);
+                list.upsert(msg, 'prepend');
+            }
+            logger.debug({ messages: newMessages.length }, 'synced messages');
+        });
+        ev.on('contacts.upsert', contacts => {
+            contactsUpsert(contacts);
+        });
+        ev.on('contacts.update', async (updates) => {
+            var _a;
+            for (const update of updates) {
+                let contact;
+                if (contacts[update.id]) {
+                    contact = contacts[update.id];
+                }
+                else {
+                    const contactHashes = await Promise.all(Object.keys(contacts).map(async (contactId) => {
+                        const { user } = jidDecode(contactId);
+                        return [contactId, (await md5(Buffer.from(user + 'WA_ADD_NOTIF', 'utf8'))).toString('base64').slice(0, 3)];
+                    }));
+                    contact = contacts[((_a = contactHashes.find(([, b]) => b === update.id)) === null || _a === void 0 ? void 0 : _a[0]) || ''];
+                }
+                if (contact) {
+                    if (update.imgUrl === 'changed') {
+                        contact.imgUrl = socket ? await (socket === null || socket === void 0 ? void 0 : socket.profilePictureUrl(contact.id)) : undefined;
+                    }
+                    else if (update.imgUrl === 'removed') {
+                        delete contact.imgUrl;
+                    }
+                }
+                else {
+                    return logger.debug({ update }, 'got update for non-existant contact');
+                }
+                Object.assign(contacts[contact.id], contact);
+            }
+        });
+        ev.on('chats.upsert', newChats => {
+            chats.upsert(...newChats);
+        });
+        ev.on('chats.update', updates => {
+            for (let update of updates) {
+                const result = chats.update(update.id, chat => {
+                    if (update.unreadCount > 0) {
+                        update = { ...update };
+                        update.unreadCount = (chat.unreadCount || 0) + update.unreadCount;
+                    }
+                    Object.assign(chat, update);
+                });
+                if (!result) {
+                    logger.debug({ update }, 'got update for non-existant chat');
+                }
+            }
+        });
+        ev.on('labels.edit', (label) => {
+            if (label.deleted) {
+                return labels.deleteById(label.id);
+            }
+
+            if (labels.count() < 20) {
+                return labels.upsertById(label.id, label);
+            }
+            logger.error('Labels count exceed');
+        });
+        ev.on('labels.association', ({ type, association }) => {
+            switch (type) {
+                case 'add':
+                    labelAssociations.upsert(association);
+                    break;
+                case 'remove':
+                    labelAssociations.delete(association);
+                    break;
+                default:
+                    console.error(`unknown operation type [${type}]`);
+            }
+        });
+        ev.on('presence.update', ({ id, presences: update }) => {
+            presences[id] = presences[id] || {};
+            Object.assign(presences[id], update);
+        });
+        ev.on('chats.delete', deletions => {
+            for (const item of deletions) {
+                if (chats.get(item)) {
+                    chats.deleteById(item);
+                }
+            }
+        });
+        ev.on('messages.upsert', ({ messages: newMessages, type }) => {
+            switch (type) {
+                case 'append':
+                case 'notify':
+                    for (const msg of newMessages) {
+                        const jid = jidNormalizedUser(msg.key.remoteJidAlt || msg.key.remoteJid);
+                        const list = assertMessageList(jid);
+                        list.upsert(msg, 'append');
+                        if (type === 'notify' && !chats.get(jid)) {
+                            ev.emit('chats.upsert', [
+                                {
+                                    id: jid,
+                                    conversationTimestamp: toNumber(msg.messageTimestamp),
+                                    unreadCount: 1
+                                }
+                            ]);
+                        }
+                    }
+                    break;
+            }
+        });
+        ev.on('messages.update', updates => {
+            var _a;
+            for (const { update, key } of updates) {
+                const list = assertMessageList(jidNormalizedUser(key.remoteJid));
+                if (update === null || update === void 0 ? void 0 : update.status) {
+                    const listStatus = (_a = list.get(key.id)) === null || _a === void 0 ? void 0 : _a.status;
+                    if (listStatus && (update === null || update === void 0 ? void 0 : update.status) <= listStatus) {
+                        logger.debug({ update, storedStatus: listStatus }, 'status stored newer then update');
+                        delete update.status;
+                        logger.debug({ update }, 'new update object');
+                    }
+                }
+                const result = list.updateAssign(key.id, update);
+                if (!result) {
+                    logger.debug({ update }, 'got update for non-existent message');
+                }
+            }
+        });
+        ev.on('messages.delete', item => {
+            if ('all' in item) {
+                const list = messages[item.jid];
+                list === null || list === void 0 ? void 0 : list.clear();
+            }
+            else {
+                const jid = item.keys[0].remoteJidAlt || item.keys[0].remoteJid;
+                const list = messages[jid];
+                if (list) {
+                    const idSet = new Set(item.keys.map(k => k.id));
+                    list.filter(m => !idSet.has(m.key.id));
+                }
+            }
+        });
+        ev.on('groups.update', updates => {
+            for (const update of updates) {
+                const id = update.id;
+                if (groupMetadata[id]) {
+                    Object.assign(groupMetadata[id], update);
+                }
+                else {
+                    logger.debug({ update }, 'got update for non-existant group metadata');
+                }
+            }
+        });
+        ev.on('group-participants.update', ({ id, participants, action }) => {
+            const metadata = groupMetadata[id];
+            if (metadata) {
+                switch (action) {
+                    case 'add':
+                        metadata.participants.push(...participants.map(participant => ({ id: participant.id, phoneNumber: participant.phoneNumber, admin: participant.admin })));
+                        break;
+                    case 'demote':
+                    case 'promote':
+                        for (const participant of metadata.participants) {
+                            for (const participantData of participants) {
+                                if (participantData.id === participant.id || participantData.phoneNumber === participant.phoneNumber) {
+                                    participant.admin = action === 'promote' && 'admin';
+                                }
+                            }
+                        }
+                        break;
+                    case 'remove':
+                        const removeSet = new Set();
+                        for (const p of participants) {
+                            if (p.id)
+                                removeSet.add(p.id);
+                            if (p.phoneNumber)
+                                removeSet.add(p.phoneNumber);
+                        }
+                        metadata.participants = metadata.participants.reduce((acc, p) => {
+                            if (!removeSet.has(p.id) && !removeSet.has(p.phoneNumber))
+                                acc.push(p);
+                            return acc;
+                        }, []);
+                        break;
+                }
+            }
+        });
+        ev.on('message-receipt.update', updates => {
+            for (const { key, receipt } of updates) {
+                const obj = messages[key.remoteJidAlt || key.remoteJid];
+                const msg = obj === null || obj === void 0 ? void 0 : obj.get(key.id);
+                if (msg) {
+                    updateMessageWithReceipt(msg, receipt);
+                }
+            }
+        });
+        ev.on('messages.reaction', (reactions) => {
+            for (const { key, reaction } of reactions) {
+                const obj = messages[key.remoteJidAlt || key.remoteJid];
+                const msg = obj === null || obj === void 0 ? void 0 : obj.get(key.id);
+                if (msg) {
+                    updateMessageWithReaction(msg, reaction);
+                }
+            }
+        });
+    };
+    const toJSON = () => ({
+        chats,
+        contacts,
+        messages,
+        labels,
+        labelAssociations
+    });
+    const fromJSON = (json) => {
+        chats.upsert(...json.chats);
+        labelAssociations.upsert(...json.labelAssociations || []);
+        contactsUpsert(Object.values(json.contacts));
+        labelsUpsert(Object.values(json.labels || {}));
+        for (const jid in json.messages) {
+            const list = assertMessageList(jid);
+            for (const msg of json.messages[jid]) {
+                list.upsert(WAProto.WebMessageInfo.fromObject(msg), 'append');
+            }
+        }
+    };
+    return {
+        chats,
+        contacts,
+        messages,
+        groupMetadata,
+        state,
+        presences,
+        labels,
+        labelAssociations,
+        bind,
+
+        loadMessages: async (jid, count, cursor) => {
+            const list = assertMessageList(jid);
+            const mode = !cursor || 'before' in cursor ? 'before' : 'after';
+            const cursorKey = !!cursor ? ('before' in cursor ? cursor.before : cursor.after) : undefined;
+            const cursorValue = cursorKey ? list.get(cursorKey.id) : undefined;
+            let messages;
+            if (list && mode === 'before' && (!cursorKey || cursorValue)) {
+                if (cursorValue) {
+                    const msgIdx = list.array.findIndex(m => m.key.id === (cursorKey === null || cursorKey === void 0 ? void 0 : cursorKey.id));
+                    messages = list.array.slice(0, msgIdx);
+                }
+                else {
+                    messages = list.array;
+                }
+                const diff = count - messages.length;
+                if (diff < 0) {
+                    messages = messages.slice(-count);
+                }
+            }
+            else {
+                messages = [];
+            }
+            return messages;
+        },
+
+        getLabels: () => {
+            return labels;
+        },
+
+        getChatLabels: (chatId) => {
+            return labelAssociations.filter((la) => la.chatId === chatId).all();
+        },
+
+        getMessageLabels: (messageId) => {
+            const associations = labelAssociations
+                .filter((la) => la.messageId === messageId)
+                .all();
+            return associations.map(({ labelId }) => labelId);
+        },
+        loadMessage: async (jid, id) => { var _a; return (_a = messages[jid]) === null || _a === void 0 ? void 0 : _a.get(id); },
+        mostRecentMessage: async (jid) => {
+            var _a;
+            const message = (_a = messages[jid]) === null || _a === void 0 ? void 0 : _a.array.slice(-1)[0];
+            return message;
+        },
+        fetchImageUrl: async (jid, sock) => {
+            const contact = contacts[jid];
+            if (!contact) {
+                return sock === null || sock === void 0 ? void 0 : sock.profilePictureUrl(jid);
+            }
+            if (typeof contact.imgUrl === 'undefined') {
+                contact.imgUrl = await (sock === null || sock === void 0 ? void 0 : sock.profilePictureUrl(jid));
+            }
+            return contact.imgUrl;
+        },
+        fetchGroupMetadata: async (jid, sock) => {
+            if (!groupMetadata[jid]) {
+                const metadata = await (sock === null || sock === void 0 ? void 0 : sock.groupMetadata(jid));
+                if (metadata) {
+                    groupMetadata[jid] = metadata;
+                }
+            }
+            return groupMetadata[jid];
+        },
+        fetchMessageReceipts: async ({ remoteJid, id }) => {
+            const list = messages[remoteJid];
+            const msg = list === null || list === void 0 ? void 0 : list.get(id);
+            return msg === null || msg === void 0 ? void 0 : msg.userReceipt;
+        },
+        toJSON,
+        fromJSON,
+        writeToFile: (path) => {
+            writeFileSync(path, JSON.stringify(toJSON()));
+        },
+        readFromFile: (path) => {
+            if (existsSync(path)) {
+                logger.debug({ path }, 'reading from file');
+                const jsonStr = readFileSync(path, { encoding: 'utf-8' });
+                const json = JSON.parse(jsonStr);
+                fromJSON(json);
+            }
+        }
+    };
+};

package/src/Store/make-ordered-dictionary.d.ts

@@ -0,0 +1,12 @@
+export function makeOrderedDictionary(idGetter: any): {
+    array: any[];
+    get: (id: any) => any;
+    upsert: (item: any, mode: any) => void;
+    update: (item: any) => boolean;
+    remove: (item: any) => boolean;
+    updateAssign: (id: any, update: any) => boolean;
+    clear: () => void;
+    filter: (contain: any) => void;
+    toJSON: () => any[];
+    fromJSON: (newItems: any) => void;
+};

package/src/Store/make-ordered-dictionary.js

@@ -0,0 +1,78 @@
+export function makeOrderedDictionary(idGetter) {
+    const array = [];
+    const dict = {};
+    const get = (id) => dict[id];
+    const update = (item) => {
+        const id = idGetter(item);
+        const idx = array.findIndex(i => idGetter(i) === id);
+        if (idx >= 0) {
+            array[idx] = item;
+            dict[id] = item;
+        }
+        return false;
+    };
+    const upsert = (item, mode) => {
+        const id = idGetter(item);
+        if (get(id)) {
+            update(item);
+        }
+        else {
+            if (mode === 'append') {
+                array.push(item);
+            }
+            else {
+                array.splice(0, 0, item);
+            }
+            dict[id] = item;
+        }
+    };
+    const remove = (item) => {
+        const id = idGetter(item);
+        const idx = array.findIndex(i => idGetter(i) === id);
+        if (idx >= 0) {
+            array.splice(idx, 1);
+            delete dict[id];
+            return true;
+        }
+        return false;
+    };
+    return {
+        array,
+        get,
+        upsert,
+        update,
+        remove,
+        updateAssign: (id, update) => {
+            const item = get(id);
+            if (item) {
+                Object.assign(item, update);
+                delete dict[id];
+                dict[idGetter(item)] = item;
+                return true;
+            }
+            return false;
+        },
+        clear: () => {
+            array.splice(0, array.length);
+            for (const key of Object.keys(dict)) {
+                delete dict[key];
+            }
+        },
+        filter: (contain) => {
+            let i = 0;
+            while (i < array.length) {
+                if (!contain(array[i])) {
+                    delete dict[idGetter(array[i])];
+                    array.splice(i, 1);
+                }
+                else {
+                    i += 1;
+                }
+            }
+        },
+        toJSON: () => array,
+        fromJSON: (newItems) => {
+            array.splice(0, array.length, ...newItems);
+        }
+    };
+}

package/src/Store/object-repository.d.ts

@@ -0,0 +1,10 @@
+export class ObjectRepository {
+    constructor(entities?: {});
+    entityMap: Map<string, any>;
+    findById(id: any): any;
+    findAll(): any[];
+    upsertById(id: any, entity: any): Map<string, any>;
+    deleteById(id: any): boolean;
+    count(): number;
+    toJSON(): any[];
+}

package/src/Store/object-repository.js

@@ -0,0 +1,23 @@
+export class ObjectRepository {
+    constructor(entities = {}) {
+        this.entityMap = new Map(Object.entries(entities));
+    }
+    findById(id) {
+        return this.entityMap.get(id);
+    }
+    findAll() {
+        return Array.from(this.entityMap.values());
+    }
+    upsertById(id, entity) {
+        return this.entityMap.set(id, { ...entity });
+    }
+    deleteById(id) {
+        return this.entityMap.delete(id);
+    }
+    count() {
+        return this.entityMap.size;
+    }
+    toJSON() {
+        return this.findAll();
+    }
+}

package/src/Types/Auth.d.ts

@@ -0,0 +1 @@
+export {};

package/src/Types/Auth.js

@@ -0,0 +1 @@
+export {};

package/src/Types/Bussines.d.ts

@@ -0,0 +1 @@
+export {};

package/src/Types/Bussines.js

@@ -0,0 +1 @@
+export {};

package/src/Types/Call.d.ts

@@ -0,0 +1 @@
+export {};

package/src/Types/Call.js

@@ -0,0 +1 @@
+export {};

package/src/Types/Chat.d.ts

@@ -0,0 +1,2 @@
+export const ALL_WA_PATCH_NAMES: string[];
+export const SPAM_FLOWS: Readonly<Record<string, string>>;

package/src/Types/Chat.js

@@ -0,0 +1,24 @@
+export const ALL_WA_PATCH_NAMES = [
+    'critical_block',
+    'critical_unblock_low',
+    'regular_high',
+    'regular_low',
+    'regular'
+];
+export const SPAM_FLOWS = Object.freeze({
+    Block: 'block_dialog',
+    OverflowMenuBlock: 'overflow_menu_block',
+    OverflowMenuReport: 'overflow_menu_report',
+    MediaViewer: 'media_viewer',
+    MessageMenu: 'message_menu',
+    CommunityHome: 'community_home',
+    AccountInfoReport: 'account_info_report',
+    GroupInfoReport: 'group_info_report',
+    GroupSpamBannerReport: 'group_spam_banner_report',
+    OneToOneChatSpamBannerReport: '1_1_spam_banner_report',
+    BizSpamBannerBlock: 'biz_spam_banner_block',
+    ChatListBlock: 'chat_list_block',
+    NotificationBlock: 'notification_block',
+    StatusPostReport: 'status_post_report',
+    NewsletterInfoReport: 'newsletter_info_report'
+});

package/src/Types/Contact.d.ts

@@ -0,0 +1 @@
+export {};

package/src/Types/Contact.js

@@ -0,0 +1 @@
+export {};

package/src/Types/Events.d.ts

@@ -0,0 +1 @@
+export {};

package/src/Types/GroupMetadata.d.ts

@@ -0,0 +1,5 @@
+export declare enum GroupHistoryToggleMode {
+    DEFAULT = 0,
+    ON = 1,
+    OFF = 2
+}

package/src/Types/GroupMetadata.js

@@ -0,0 +1,6 @@
+export var GroupHistoryToggleMode;
+(function (GroupHistoryToggleMode) {
+    GroupHistoryToggleMode[GroupHistoryToggleMode["DEFAULT"] = 0] = "DEFAULT";
+    GroupHistoryToggleMode[GroupHistoryToggleMode["ON"] = 1] = "ON";
+    GroupHistoryToggleMode[GroupHistoryToggleMode["OFF"] = 2] = "OFF";
+})(GroupHistoryToggleMode || (GroupHistoryToggleMode = {}));

package/src/Types/Label.d.ts

@@ -0,0 +1,2 @@
+/** WhatsApp has 20 predefined colors */
+export const LabelColor: any;