@itsliaaa/baileys 0.1.29 → 0.1.31

package/lib/Utils/use-single-file-auth-state.js

@@ -3,27 +3,38 @@ import { DEFAULT_CACHE_TTLS } from '../Defaults/index.js';
 import { proto } from '../../WAProto/index.js';
 import { initAuthCreds } from './auth-utils.js';
 import { BufferJSON } from './generics.js';
+import { LRUCache } from 'lru-cache';
 // Lia@Changes 25-03-26 --- Add useSingleFileAuthState with integrated cache
 const FLUSH_TIMEOUT_MS = 3000;
-// Lia@Note 21-04-26 --- Unstable need enhancement soon!
+// Lia@Changes 22-04-26 --- Enhanced useSingleFileAuthState with LRUCache
 export const useSingleFileAuthState = async (fileName) => {
-    const cache = new Map();
-    let isLoaded,
-        isWriting,
-        isNeedWrite,
-        flushTimeout,
-        loadPromise;
+    const cache = new LRUCache({
+        max: 10000,
+        ttl: 1000 * DEFAULT_CACHE_TTLS.SIGNAL_STORE,
+        updateAgeOnGet: false,
+        updateAgeOnHas: false,
+        ttlAutopurge: true
+    });
+    let fileData = {},
+        isLoaded = false,
+        isWriting = false,
+        isNeedWrite = false,
+        flushTimeout = null,
+        loadPromise = null;
     const loadKey = () => {
         if (isLoaded) return;
         if (loadPromise) return loadPromise;
         loadPromise = (async () => {
             try {
                 const data = JSON.parse(await readFile(fileName, 'utf-8'), BufferJSON.reviver);
-                for (const [keyName, value] of Object.entries(data)) {
+                fileData = data || {};
+                for (const [keyName, value] of Object.entries(fileData)) {
                     cache.set(keyName, value);
                 }
             }
-            catch { }
+            catch {
+                fileData = {};
+            }
             isLoaded = true;
             loadPromise = null;
         })();
@@ -41,20 +52,20 @@ export const useSingleFileAuthState = async (fileName) => {
             do {
                 isNeedWrite = false;
                 const tempFile = fileName + '.temp';
-                const value = Object.fromEntries(cache);
-                await writeFile(tempFile, JSON.stringify(value, BufferJSON.replacer));
+                await writeFile(tempFile, JSON.stringify(fileData, BufferJSON.replacer));
                 await rename(tempFile, fileName);
-            }
-            while (isNeedWrite);
+            } while (isNeedWrite);
             isWriting = false;
         }, FLUSH_TIMEOUT_MS);
     };
     const writeKey = (keyName, value) => {
         cache.set(keyName, value);
+        fileData[keyName] = value;
         flushKey();
     };
     const removeKey = (keyName) => {
         cache.delete(keyName);
+        delete fileData[keyName];
         flushKey();
     };
     const fileInfo = await stat(fileName).catch(() => null);
@@ -65,7 +76,7 @@ export const useSingleFileAuthState = async (fileName) => {
         throw new Error(`found something that is not a file at ${fileName}, either delete it or specify a different location`);
     }
     await loadKey();
-    const creds = cache.get('creds') || initAuthCreds();
+    const creds = fileData['creds'] || initAuthCreds();
     return {
         state: {
             creds,
@@ -73,7 +84,12 @@ export const useSingleFileAuthState = async (fileName) => {
                 get: (type, ids) => {
                     const data = {};
                     for (const id of ids) {
-                        let value = cache.get(type + id);
+                        const keyName = type + id;
+                        let value = cache.get(keyName);
+                        if (value === undefined && fileData[keyName] !== undefined) {
+                            value = fileData[keyName];
+                            cache.set(keyName, value);
+                        }
                         if (type === 'app-state-sync-key' && value) {
                             value = proto.Message.AppStateSyncKeyData.fromObject(value);
                         }

package/package.json

@@ -1,6 +1,6 @@
 {
    "name": "@itsliaaa/baileys",
-   "version": "0.1.29",
+   "version": "0.1.31",
    "description": "Enhanced Baileys v7 with fixed newsletter media upload, plus support for interactive messages, albums, and more message types.",
    "main": "lib/index.js",
    "type": "module",