@alannxd/baileys 6.0.5 → 6.0.6

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

@@ -1,164 +1,290 @@
-import * as WAProto_1 from "../../WAProto/index.js";
-import * as Defaults_1 from "../Defaults/index.js";
-import { LabelAssociationType } from "../Types/LabelAssociation.js";
-import * as Utils_1 from "../Utils/index.js";
-import * as WABinary_1 from "../WABinary/index.js";
-import makeOrderedDictionary from "./make-ordered-dictionary.js";
-import { ObjectRepository } from "./object-repository.js";
-import KeyedDB from "./keyed-db.js";
-
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.waLabelAssociationKey = exports.waMessageID = exports.waChatKey = void 0;
+const WAProto_1 = require("../../WAProto");
+const Defaults_1 = require("../Defaults");
+const LabelAssociation_1 = require("../Types/LabelAssociation");
+const Utils_1 = require("../Utils");
+const WABinary_1 = require("../WABinary");
+const make_ordered_dictionary_1 = __importDefault(require("./make-ordered-dictionary"));
+const object_repository_1 = require("./object-repository");
 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,
+    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,
+exports.waChatKey = waChatKey;
+const waMessageID = (m) => m.key.id || '';
+exports.waMessageID = waMessageID;
+exports.waLabelAssociationKey = {
+    key: (la) => (la.type === LabelAssociation_1.LabelAssociationType.Chat ? la.chatId + la.labelId : la.chatId + la.messageId + la.labelId),
     compare: (k1, k2) => k2.localeCompare(k1)
 };
-
-const makeMessagesDictionary = () => makeOrderedDictionary(waMessageID);
-
-export default function makeInMemoryStore(config) {
+const makeMessagesDictionary = () => (0, make_ordered_dictionary_1.default)(exports.waMessageID);
+exports.default = (config) => {
     const socket = config.socket;
-    const chatKey = config.chatKey || waChatKey(true);
-    const labelAssociationKey = config.labelAssociationKey || waLabelAssociationKey;
-    const logger =
-        config.logger ||
-        Defaults_1.DEFAULT_CONNECTION_CONFIG.logger.child({
-            stream: "in-mem-store"
-        });
-
-    const chats = new KeyedDB(chatKey.compare, chatKey.key);
+    const chatKey = config.chatKey || (0, exports.waChatKey)(true);
+    const labelAssociationKey = config.labelAssociationKey || exports.waLabelAssociationKey;
+    const logger = config.logger || Defaults_1.DEFAULT_CONNECTION_CONFIG.logger.child({ stream: 'in-mem-store' });
+    const KeyedDB = require('@adiwajshing/keyed-db').default;
+    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.compare, labelAssociationKey.key);
-
+    const state = { connection: 'close' };
+    const labels = new object_repository_1.ObjectRepository();
+    const labelAssociations = new KeyedDB(labelAssociationKey, labelAssociationKey.key);
     const assertMessageList = (jid) => {
-        if (!messages[jid]) messages[jid] = makeMessagesDictionary();
+        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] = { ...(contacts[contact.id] || {}), ...contact };
+            contacts[contact.id] = Object.assign(contacts[contact.id] || {}, contact);
         }
         return oldContacts;
     };
-
     const labelsUpsert = (newLabels) => {
-        for (const label of newLabels) labels.upsertById(label.id, label);
+        for (const label of newLabels) {
+            labels.upsertById(label.id, label);
+        }
+    };
+    const getValidContacts = () => {
+        for (const contact of Object.keys(contacts)) {
+            if (contact.indexOf('@') < 0) {
+                delete contacts[contact];
+            }
+        }
+        return Object.keys(contacts);
     };
-
+    /**
+     * binds to a BaileysEventEmitter.
+     * It listens to all events and constructs a state that you can query accurate data from.
+     * Eg. can use the store to fetch chats, contacts, messages etc.
+     * @param ev typically the event emitter from the socket connection
+     */
     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_1.proto.HistorySync.HistorySyncType.ON_DEMAND) return;
+        ev.on('connection.update', update => {
+            Object.assign(state, update);
+        });
+        ev.on('messaging-history.set', ({ chats: newChats, contacts: newContacts, messages: newMessages, isLatest }) => {
             if (isLatest) {
                 chats.clear();
-                for (const id in messages) delete messages[id];
+                for (const id in messages) {
+                    delete messages[id];
+                }
             }
-            chats.insertIfAbsent(...newChats);
+            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];
+            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.remoteJid;
                 const list = assertMessageList(jid);
-                list.upsert(msg, "prepend");
+                list.upsert(msg, 'prepend');
             }
+            logger.debug({ messages: newMessages.length }, 'synced messages');
+        });
+        ev.on('contacts.upsert', contacts => {
+            contactsUpsert(contacts);
         });
-
-        ev.on("contacts.upsert", contactsUpsert);
-
-        ev.on("contacts.update", async (updates) => {
+        ev.on('contacts.update', async (updates) => {
+            var _a;
             for (const update of updates) {
-                const contact = contacts[update.id];
-                if (contact) Object.assign(contact, update);
+                let contact;
+                if (contacts[update.id]) {
+                    contact = contacts[update.id];
+                }
+                else {
+                    const validContacts = getValidContacts();
+                    const contactHashes = validContacts.map((contactId) => {
+                        const { user } = (0, WABinary_1.jidDecode)(contactId);
+                        return [contactId, ((0, Utils_1.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]) || '']; // find contact by attrs.hash, when user is not saved as a contact
+                }
+                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;
+                    }
+                    Object.assign(contacts[contact.id], contact);
+                }
+                else {
+                    logger.debug({ update }, 'got update for non-existant contact');
+                }
             }
         });
-
-        ev.on("chats.upsert", (newChats) => chats.upsert(...newChats));
-
-        ev.on("chats.update", (updates) => {
+        ev.on('chats.upsert', newChats => {
+            chats.upsert(...newChats);
+        });
+        ev.on('chats.update', updates => {
             for (let update of updates) {
-                chats.update(update.id, (chat) => Object.assign(chat, update));
+                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) labels.upsertById(label.id, label);
+        ev.on('labels.edit', (label) => {
+            if (label.deleted) {
+                return labels.deleteById(label.id);
+            }
+            // WhatsApp can store only up to 20 labels
+            if (labels.count() < 20) {
+                return labels.upsertById(label.id, label);
+            }
+            logger.error('Labels count exceed');
         });
-
-        ev.on("labels.association", ({ type, association }) => {
-            if (type === "add") labelAssociations.upsert(association);
-            if (type === "remove") labelAssociations.delete(association);
+        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 }) => {
+        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) chats.deleteById(item);
+        ev.on('chats.delete', deletions => {
+            for (const item of deletions) {
+                if (chats.get(item)) {
+                    chats.deleteById(item);
+                }
+            }
         });
-
-        ev.on("messages.upsert", ({ messages: newMessages, type }) => {
-            if (type !== "append" && type !== "notify") return;
-            for (const msg of newMessages) {
-                const jid = WABinary_1.jidNormalizedUser(msg.key.remoteJid);
-                const list = assertMessageList(jid);
-                list.upsert(msg, "append");
-                if (type === "notify" && !chats.get(jid))
-                    ev.emit("chats.upsert", [
-                        {
-                            id: jid,
-                            conversationTimestamp: Utils_1.toNumber(msg.messageTimestamp),
-                            unreadCount: 1
+        ev.on('messages.upsert', ({ messages: newMessages, type }) => {
+            switch (type) {
+                case 'append':
+                case 'notify':
+                    for (const msg of newMessages) {
+                        const jid = (0, WABinary_1.jidNormalizedUser)(msg.key.remoteJid);
+                        const list = assertMessageList(jid);
+                        list.upsert(msg, 'append');
+                        if (type === 'notify') {
+                            if (!chats.get(jid)) {
+                                ev.emit('chats.upsert', [
+                                    {
+                                        id: jid,
+                                        conversationTimestamp: (0, Utils_1.toNumber)(msg.messageTimestamp),
+                                        unreadCount: 1
+                                    }
+                                ]);
+                            }
                         }
-                    ]);
+                    }
+                    break;
             }
         });
-
-        ev.on("messages.update", (updates) => {
+        ev.on('messages.update', updates => {
+            var _a;
             for (const { update, key } of updates) {
-                const list = assertMessageList(WABinary_1.jidNormalizedUser(key.remoteJid));
-                list.updateAssign(key.id, update);
+                const list = assertMessageList((0, WABinary_1.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) messages[item.jid]?.clear();
+        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].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));
+                    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(id => ({ id, isAdmin: false, isSuperAdmin: false })));
+                        break;
+                    case 'demote':
+                    case 'promote':
+                        for (const participant of metadata.participants) {
+                            if (participants.includes(participant.id)) {
+                                participant.isAdmin = action === 'promote';
+                            }
+                        }
+                        break;
+                    case 'remove':
+                        metadata.participants = metadata.participants.filter(p => !participants.includes(p.id));
+                        break;
+                }
+            }
+        });
+        ev.on('message-receipt.update', updates => {
+            for (const { key, receipt } of updates) {
+                const obj = messages[key.remoteJid];
+                const msg = obj === null || obj === void 0 ? void 0 : obj.get(key.id);
+                if (msg) {
+                    (0, Utils_1.updateMessageWithReceipt)(msg, receipt);
+                }
+            }
+        });
+        ev.on('messages.reaction', (reactions) => {
+            for (const { key, reaction } of reactions) {
+                const obj = messages[key.remoteJid];
+                const msg = obj === null || obj === void 0 ? void 0 : obj.get(key.id);
+                if (msg) {
+                    (0, Utils_1.updateMessageWithReaction)(msg, reaction);
                 }
             }
         });
     };
-
     const toJSON = () => ({
         chats,
         contacts,
@@ -166,21 +292,18 @@ export default function makeInMemoryStore(config) {
         labels,
         labelAssociations
     });
-
     const fromJSON = (json) => {
         chats.upsert(...json.chats);
-        labelAssociations.upsert(...(json.labelAssociations || []));
+        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_1.proto.WebMessageInfo.fromObject(msg), "append");
+            for (const msg of json.messages[jid]) {
+                list.upsert(WAProto_1.proto.WebMessageInfo.fromObject(msg), 'append');
+            }
         }
     };
-    const loadMessage = async (jid, id) => {
-        return messages[jid]?.get(id)
-    }
     return {
         chats,
         contacts,
@@ -191,8 +314,114 @@ export default function makeInMemoryStore(config) {
         labels,
         labelAssociations,
         bind,
+        /** loads messages from the store, if not found -- uses the legacy connection */
+        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); // get the last X messages
+                }
+            }
+            else {
+                messages = [];
+            }
+            return messages;
+        },
+        /**
+         * 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: () => {
+            return labels;
+        },
+        /**
+         * Get labels for chat
+         *
+         * @returns Label IDs
+         **/
+        getChatLabels: (chatId) => {
+            return labelAssociations.filter((la) => la.chatId === chatId).all();
+        },
+        /**
+         * Get labels for message
+         *
+         * @returns Label IDs
+         **/
+        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];
+        },
+        // fetchBroadcastListInfo: async(jid: string, sock: WASocket | undefined) => {
+        // 	if(!groupMetadata[jid]) {
+        // 		const metadata = await sock?.getBroadcastListInfo(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,
-        loadMessage
+        writeToFile: (path) => {
+            // require fs here so that in case "fs" is not available -- the app does not crash
+            const { writeFileSync } = require('fs');
+            writeFileSync(path, JSON.stringify(toJSON()));
+        },
+        readFromFile: (path) => {
+            // require fs here so that in case "fs" is not available -- the app does not crash
+            const { readFileSync, existsSync } = require('fs');
+            if (existsSync(path)) {
+                logger.debug({ path }, 'reading from file');
+                const jsonStr = readFileSync(path, { encoding: 'utf-8' });
+                const json = JSON.parse(jsonStr);
+                fromJSON(json);
+            }
+        }
     };
-}
+};

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

@@ -1,14 +1,13 @@
 declare function makeOrderedDictionary<T>(idGetter: (item: T) => string): {
-  array: T[];
-  get: (id: string) => T | undefined;
-  upsert: (item: T, mode: 'append' | 'prepend') => void;
-  update: (item: T) => boolean;
-  remove: (item: T) => boolean;
-  updateAssign: (id: string, update: Partial<T>) => boolean;
-  clear: () => void;
-  filter: (contain: (item: T) => boolean) => void;
-  toJSON: () => T[];
-  fromJSON: (newItems: T[]) => void;
+    array: T[];
+    get: (id: string) => T | undefined;
+    upsert: (item: T, mode: 'append' | 'prepend') => void;
+    update: (item: T) => boolean;
+    remove: (item: T) => boolean;
+    updateAssign: (id: string, update: Partial<T>) => boolean;
+    clear: () => void;
+    filter: (contain: (item: T) => boolean) => void;
+    toJSON: () => T[];
+    fromJSON: (newItems: T[]) => void;
 };
 export default makeOrderedDictionary;
-export { makeOrderedDictionary };

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

@@ -1,9 +1,9 @@
-export default function makeOrderedDictionary(idGetter) {
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+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);
@@ -13,18 +13,21 @@ export default function makeOrderedDictionary(idGetter) {
         }
         return false;
     };
-
     const upsert = (item, mode) => {
         const id = idGetter(item);
         if (get(id)) {
             update(item);
-        } else {
-            if (mode === 'append') array.push(item);
-            else array.unshift(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);
@@ -35,7 +38,6 @@ export default function makeOrderedDictionary(idGetter) {
         }
         return false;
     };
-
     return {
         array,
         get,
@@ -53,8 +55,10 @@ export default function makeOrderedDictionary(idGetter) {
             return false;
         },
         clear: () => {
-            array.length = 0;
-            for (const key of Object.keys(dict)) delete dict[key];
+            array.splice(0, array.length);
+            Object.keys(dict).forEach(key => {
+                delete dict[key];
+            });
         },
         filter: (contain) => {
             let i = 0;
@@ -62,8 +66,9 @@ export default function makeOrderedDictionary(idGetter) {
                 if (!contain(array[i])) {
                     delete dict[idGetter(array[i])];
                     array.splice(i, 1);
-                } else {
-                    i++;
+                }
+                else {
+                    i += 1;
                 }
             }
         },
@@ -72,4 +77,5 @@ export default function makeOrderedDictionary(idGetter) {
             array.splice(0, array.length, ...newItems);
         }
     };
-}
+}
+exports.default = makeOrderedDictionary;

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

@@ -1,11 +1,10 @@
-declare class ObjectRepository<T> {
-  private entityMap: Map<string, T>;
-  constructor(entities?: { [id: string]: T });
-  findById(id: string): T | undefined;
-  findAll(): T[];
-  upsertById(id: string, entity: T): Map<string, T>;
-  deleteById(id: string): boolean;
-  count(): number;
-  toJSON(): T[];
+export declare class ObjectRepository<T extends object> {
+    readonly entityMap: Map<string, T>;
+    constructor(entities?: Record<string, T>);
+    findById(id: string): T | undefined;
+    findAll(): T[];
+    upsertById(id: string, entity: T): Map<string, T>;
+    deleteById(id: string): boolean;
+    count(): number;
+    toJSON(): T[];
 }
-export { ObjectRepository };