@badzz88/baileys 8.5.4 → 8.5.6

package/src/Socket/text-router.js

@@ -0,0 +1,65 @@
+'use strict';
+Object.defineProperty(exports, '__esModule', { value: true });
+exports.attachTextRouter = void 0;
+const extractText = message => {
+    if (!message)
+        return '';
+    return (message.conversation ||
+        message.extendedTextMessage?.text ||
+        message.imageMessage?.caption ||
+        message.videoMessage?.caption ||
+        message.documentMessage?.caption ||
+        '');
+};
+const escapeRegExp = str => str.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+const attachTextRouter = sock => {
+    const routes = [];
+    const addRoute = (pattern, handler) => {
+        const route = { pattern, handler };
+        routes.push(route);
+        return () => {
+            const idx = routes.indexOf(route);
+            if (idx !== -1)
+                routes.splice(idx, 1);
+        };
+    };
+    const dispatch = async ({ messages, type }) => {
+        if (type !== 'notify' && type !== 'append')
+            return;
+        for (const msg of messages) {
+            if (!msg.message || msg.key.fromMe)
+                continue;
+            const text = extractText(msg.message);
+            if (!text)
+                continue;
+            for (const route of routes.slice()) {
+                let match = null;
+                if (route.pattern instanceof RegExp) {
+                    route.pattern.lastIndex = 0;
+                    match = route.pattern.exec(text);
+                }
+                else if (typeof route.pattern === 'string') {
+                    match = text === route.pattern ? [text] : null;
+                }
+                if (!match)
+                    continue;
+                try {
+                    await route.handler(msg, match);
+                }
+                catch (error) {
+                    sock.logger?.error?.({ error, text }, 'onText/hears handler threw');
+                }
+            }
+        }
+    };
+    sock.ev.on('messages.upsert', dispatch);
+    sock.onText = addRoute;
+    sock.hears = addRoute;
+    sock.command = (name, handler) => {
+        const names = (Array.isArray(name) ? name : [name]).map(escapeRegExp);
+        const pattern = new RegExp(`^/(?:${names.join('|')})(?:@\\S+)?(?:\\s+([\\s\\S]*))?$`, 'i');
+        return addRoute(pattern, handler);
+    };
+    return sock;
+};
+exports.attachTextRouter = attachTextRouter;

package/src/Socket/username.js

@@ -0,0 +1,130 @@
+'use strict';
+Object.defineProperty(exports, '__esModule', { value: true });
+exports.makeUsernameSocket = void 0;
+const { executeWMexQuery } = require('./mex.js');
+const { USyncQuery, USyncUser } = require('../WAUSync/index.js');
+const { makeNewsletterSocket } = require('../WAUSync/Protocols/USyncNewsletterProtocol')
+const crypto = require('crypto');
+const USERNAME_QUERY_IDS = {
+    CHECK: '26124072630599518',
+    CHECK_MULTI: '27134626522840286',
+    SET: '27108705368767936',
+    GET: '32618050064506055',
+    GET_RECOMMENDATIONS: '26077456248616957',
+    PIN_SET: '25529696019976770'
+};
+const USERNAME_CHECK_RESULT = {
+    SUCCESS: 'SUCCESS',
+    INVALID: 'INVALID'
+};
+const USERNAME_SOURCE = {
+    FB: 'FB',
+    IG: 'IG',
+    USER_INPUT: 'USER_INPUT',
+    SUGGESTION: 'SUGGESTION'
+};
+const makeUsernameSocket = config => {
+    const sock = makeNewsletterSocket(config);
+    const { query, generateMessageTag, executeUSyncQuery } = sock;
+    const mexQuery = (variables, queryId, dataPath) => executeWMexQuery(variables, queryId, dataPath, query, generateMessageTag);
+    const checkUsername = async (username, includeSuggestions = true, sessionId) => {
+        const session_id = sessionId || crypto.randomUUID();
+        const data = await mexQuery({
+            username,
+            include_suggestions: includeSuggestions,
+            session_id,
+            source: USERNAME_SOURCE.USER_INPUT
+        }, USERNAME_QUERY_IDS.CHECK, 'xwa2_username_check');
+        if (data?.includes(USERNAME_CHECK_RESULT.SUCCESS) || data?.result === USERNAME_CHECK_RESULT.SUCCESS) {
+            return { available: true, username, session_id };
+        }
+        return {
+            available: false,
+            username,
+            session_id,
+            suggestions: data?.suggestions ?? [],
+            rejectionReasons: data?.rejection_reasons ?? [],
+            suggestionsEligible: data?.suggestions_eligible ?? true
+        };
+    };
+    const setUsername = async (username, options = {}) => {
+        const { source = USERNAME_SOURCE.USER_INPUT, sessionId, pin } = options;
+        const session_id = sessionId || crypto.randomUUID();
+        const variables = {
+            username,
+            reserved: true,
+            source,
+            session_id,
+            ...(pin ? { pin } : {})
+        };
+        return mexQuery(variables, USERNAME_QUERY_IDS.SET, 'xwa2_username_set');
+    };
+    const checkAndSetUsername = async (username) => {
+        const r = await checkUsername(username);
+        if (r.available) {
+            return await setUsername(username);
+        }
+        return r;
+    };
+    const deleteUsername = async () => {
+        return mexQuery({ username: null }, USERNAME_QUERY_IDS.SET, 'xwa2_username_delete');
+    };
+    const getMyUsername = async () => {
+        const data = await mexQuery({}, USERNAME_QUERY_IDS.GET, 'xwa2_username_get');
+        return data?.username ?? null;
+    };
+    const setUsernamePin = async (pin) => {
+        const variables = pin != null ? { pin } : {};
+        return mexQuery(variables, USERNAME_QUERY_IDS.PIN_SET, 'xwa2_username_pin_set');
+    };
+    const findUserByUsername = async (username, pin) => {
+        const usyncQuery = new USyncQuery().withContactProtocol();
+        const user = new USyncUser().withUsername(username);
+        if (pin)
+            user.withUsernameKey(pin);
+        usyncQuery.withUser(user);
+        const result = await executeUSyncQuery(usyncQuery);
+        if (!result?.list?.length)
+            return null;
+        const entry = result.list[0];
+        return {
+            jid: entry.id,
+            contact: entry.contact ?? false
+        };
+    };
+    const fetchContactUsernames = async (...jids) => {
+        const usyncQuery = new USyncQuery().withUsernameProtocol();
+        for (const jid of jids) {
+            usyncQuery.withUser(new USyncUser().withId(jid));
+        }
+        const result = await executeUSyncQuery(usyncQuery);
+        return result?.list ?? [];
+    };
+    const checkUsernameMulti = async (usernames) => {
+        const data = await mexQuery({ usernames }, USERNAME_QUERY_IDS.CHECK_MULTI, 'xwa2_username_check_multi');
+        return data;
+    };
+    const getUsernameRecommendations = async (source = null) => {
+        const variables = {};
+        if (source)
+            variables.source = source;
+        return mexQuery(variables, USERNAME_QUERY_IDS.GET_RECOMMENDATIONS, 'xwa2_username_get_recommendations');
+    };
+    return {
+        ...sock,
+        checkUsername,
+        checkUsernameMulti,
+        setUsername,
+        deleteUsername,
+        getMyUsername,
+        getUsernameRecommendations,
+        setUsernamePin,
+        findUserByUsername,
+        fetchContactUsernames,
+        checkAndSetUsername,
+        USERNAME_QUERY_IDS,
+        USERNAME_CHECK_RESULT,
+        USERNAME_SOURCE
+    };
+};
+exports.makeUsernameSocket = makeUsernameSocket;

package/src/Store/index.js

@@ -0,0 +1,33 @@
+'use strict';
+var __createBinding = (this && this.__createBinding) ||
+    (Object.create
+        ? function (o, m, k, k2) {
+            if (k2 === undefined)
+                k2 = k;
+            var desc = Object.getOwnPropertyDescriptor(m, k);
+            if (!desc || ('get' in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+                desc = {
+                    enumerable: true,
+                    get: function () {
+                        return m[k];
+                    }
+                };
+            }
+            Object.defineProperty(o, k2, desc);
+        }
+        : function (o, m, k, k2) {
+            if (k2 === undefined)
+                k2 = k;
+            o[k2] = m[k];
+        });
+var __exportStar = (this && this.__exportStar) ||
+    function (m, exports) {
+        for (var p in m)
+            if (p !== 'default' && !Object.prototype.hasOwnProperty.call(exports, p))
+                __createBinding(exports, m, p);
+    };
+Object.defineProperty(exports, '__esModule', { value: true });
+__exportStar(require('./make-cache-manager-store'), exports);
+__exportStar(require('./make-in-memory-store'), exports);
+__exportStar(require('./make-ordered-dictionary'), exports);
+__exportStar(require('./object-repository'), exports);

package/src/Store/keyed-db.js

@@ -0,0 +1,118 @@
+export default class KeyedDB {
+    constructor(compareFn, idGetter) {
+        if (typeof compareFn !== "function" || typeof idGetter !== "function") {
+            throw new Error("KeyedDB requires compare and idGetter functions");
+        }
+        this._compare = compareFn;
+        this._idGetter = idGetter;
+        this._array = [];
+        this._dict = {};
+    }
+    get length() {
+        return this._array.length;
+    }
+    get(id) {
+        return this._dict[id];
+    }
+    insert(entry, mode = "insert") {
+        const id = this._idGetter(entry);
+        const existing = this._dict[id];
+        if (existing && mode === "insert") {
+            return false;
+        }
+        this._dict[id] = entry;
+        let inserted = false;
+        for (let i = 0; i < this._array.length; i++) {
+            const cmp = this._compare(this._idGetter(this._array[i]), id);
+            if (cmp < 0) {
+                this._array.splice(i, 0, entry);
+                inserted = true;
+                break;
+            }
+        }
+        if (!inserted)
+            this._array.push(entry);
+        return true;
+    }
+    insertIfAbsent(...entries) {
+        const added = [];
+        for (const entry of entries) {
+            if (!this._dict[this._idGetter(entry)]) {
+                this.insert(entry);
+                added.push(entry);
+            }
+        }
+        return added;
+    }
+    upsert(...entries) {
+        const added = [];
+        for (const entry of entries) {
+            const id = this._idGetter(entry);
+            if (this._dict[id]) {
+                const idx = this._array.findIndex(e => this._idGetter(e) === id);
+                if (idx >= 0)
+                    this._array[idx] = entry;
+                this._dict[id] = entry;
+            }
+            else {
+                this.insert(entry);
+                added.push(entry);
+            }
+        }
+        return added;
+    }
+    update(id, updater) {
+        const item = this._dict[id];
+        if (!item)
+            return false;
+        updater(item);
+        const idx = this._array.findIndex(e => this._idGetter(e) === id);
+        if (idx >= 0)
+            this._array[idx] = item;
+        return true;
+    }
+    updateAssign(id, update) {
+        const item = this._dict[id];
+        if (!item)
+            return false;
+        Object.assign(item, update);
+        const idx = this._array.findIndex(e => this._idGetter(e) === id);
+        if (idx >= 0)
+            this._array[idx] = item;
+        return true;
+    }
+    deleteById(id) {
+        if (!this._dict[id])
+            return false;
+        delete this._dict[id];
+        const idx = this._array.findIndex(e => this._idGetter(e) === id);
+        if (idx >= 0)
+            this._array.splice(idx, 1);
+        return true;
+    }
+    clear() {
+        this._array = [];
+        this._dict = {};
+    }
+    all() {
+        return [...this._array];
+    }
+    filter(predicate) {
+        this._array = this._array.filter(predicate);
+        this._dict = {};
+        for (const entry of this._array) {
+            this._dict[this._idGetter(entry)] = entry;
+        }
+    }
+    count() {
+        return this._array.length;
+    }
+    toJSON() {
+        return this._array;
+    }
+    fromJSON(array) {
+        this.clear();
+        for (const entry of array)
+            this.insert(entry);
+    }
+}

package/src/Store/make-cache-manager-store.js

@@ -0,0 +1,84 @@
+'use strict';
+var __importDefault = (this && this.__importDefault) ||
+    function (mod) {
+        return mod && mod.__esModule ? mod : { default: mod };
+    };
+Object.defineProperty(exports, '__esModule', { value: true });
+const cache_manager_1 = require('cache-manager');
+const WAProto_1 = require('../../WAProto');
+const Utils_1 = require('../Utils');
+const logger_1 = __importDefault(require('../Utils/logger'));
+const makeCacheManagerAuthState = async (store, sessionKey) => {
+    const defaultKey = file => `${sessionKey}:${file}`;
+    const databaseConn = await cache_manager_1.caching(store);
+    const writeData = async (file, data) => {
+        let ttl = undefined;
+        if (file === 'creds') {
+            ttl = 63115200;
+        }
+        await databaseConn.set(defaultKey(file), JSON.stringify(data, Utils_1.BufferJSON.replacer), ttl);
+    };
+    const readData = async (file) => {
+        try {
+            const data = await databaseConn.get(defaultKey(file));
+            if (data) {
+                return JSON.parse(data, Utils_1.BufferJSON.reviver);
+            }
+            return null;
+        }
+        catch (error) {
+            logger_1.default.error(error);
+            return null;
+        }
+    };
+    const removeData = async (file) => {
+        try {
+            return await databaseConn.del(defaultKey(file));
+        }
+        catch (err) {
+            logger_1.default.error({ err }, `Error removing ${file} from session ${sessionKey}`);
+        }
+    };
+    const clearState = async () => {
+        try {
+            const result = await databaseConn.store.keys(`${sessionKey}*`);
+            await Promise.all(result.map(async (key) => await databaseConn.del(key)));
+        }
+        catch (err) {
+            logger_1.default.warn({ err }, 'clearState failed');
+        }
+    };
+    const creds = (await readData('creds')) || (0, Utils_1.initAuthCreds)();
+    return {
+        clearState,
+        saveCreds: () => writeData('creds', creds),
+        state: {
+            creds,
+            keys: {
+                get: async (type, ids) => {
+                    const data = {};
+                    await Promise.all(ids.map(async (id) => {
+                        let value = await readData(`${type}-${id}`);
+                        if (type === 'app-state-sync-key' && value) {
+                            value = WAProto_1.proto.Message.AppStateSyncKeyData.fromObject(value);
+                        }
+                        data[id] = value;
+                    }));
+                    return data;
+                },
+                set: async (data) => {
+                    const tasks = [];
+                    for (const category in data) {
+                        for (const id in data[category]) {
+                            const value = data[category][id];
+                            const key = `${category}-${id}`;
+                            tasks.push(value ? writeData(key, value) : removeData(key));
+                        }
+                    }
+                    await Promise.all(tasks);
+                }
+            }
+        }
+    };
+};
+exports.makeCacheManagerAuthState = makeCacheManagerAuthState;