@badzz88/baileys 8.4.7 → 8.5.0

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

@@ -1,23 +1,26 @@
-import { Mutex } from 'async-mutex';
-import { mkdir, readFile, stat, unlink, writeFile } from 'fs/promises';
-import { join } from 'path';
-import { proto } from '../../WAProto/index.js';
-import { initAuthCreds } from './auth-utils.js';
-import { BufferJSON } from './generics.js';
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.useMultiFileAuthState = void 0
+const async_mutex_1 = require('async-mutex')
+const promises_1 = require('fs/promises')
+const path_1 = require('path')
+const index_js_1 = require('../../WAProto/index.js')
+const auth_utils_1 = require('./auth-utils')
+const generics_1 = require('./generics')
 // We need to lock files due to the fact that we are using async functions to read and write files
 // https://github.com/WhiskeySockets/Baileys/issues/794
 // https://github.com/nodejs/node/issues/26338
 // Use a Map to store mutexes for each file path
-const fileLocks = new Map();
+const fileLocks = new Map()
 // Get or create a mutex for a specific file path
-const getFileLock = (path) => {
-    let mutex = fileLocks.get(path);
-    if (!mutex) {
-        mutex = new Mutex();
-        fileLocks.set(path, mutex);
-    }
-    return mutex;
-};
+const getFileLock = path => {
+	let mutex = fileLocks.get(path)
+	if (!mutex) {
+		mutex = new async_mutex_1.Mutex()
+		fileLocks.set(path, mutex)
+	}
+	return mutex
+}
 /**
  * stores the full authentication state in a single folder.
  * Far more efficient than singlefileauthstate
@@ -25,97 +28,94 @@ const getFileLock = (path) => {
  * Again, I wouldn't endorse this for any production level use other than perhaps a bot.
  * Would recommend writing an auth state for use with a proper SQL or No-SQL DB
  * */
-export const useMultiFileAuthState = async (folder) => {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const writeData = async (data, file) => {
-        const filePath = join(folder, fixFileName(file));
-        const mutex = getFileLock(filePath);
-        return mutex.acquire().then(async (release) => {
-            try {
-                await writeFile(filePath, JSON.stringify(data, BufferJSON.replacer));
-            }
-            finally {
-                release();
-            }
-        });
-    };
-    const readData = async (file) => {
-        try {
-            const filePath = join(folder, fixFileName(file));
-            const mutex = getFileLock(filePath);
-            return await mutex.acquire().then(async (release) => {
-                try {
-                    const data = await readFile(filePath, { encoding: 'utf-8' });
-                    return JSON.parse(data, BufferJSON.reviver);
-                }
-                finally {
-                    release();
-                }
-            });
-        }
-        catch (error) {
-            return null;
-        }
-    };
-    const removeData = async (file) => {
-        try {
-            const filePath = join(folder, fixFileName(file));
-            const mutex = getFileLock(filePath);
-            return mutex.acquire().then(async (release) => {
-                try {
-                    await unlink(filePath);
-                }
-                catch {
-                }
-                finally {
-                    release();
-                }
-            });
-        }
-        catch { }
-    };
-    const folderInfo = await stat(folder).catch(() => { });
-    if (folderInfo) {
-        if (!folderInfo.isDirectory()) {
-            throw new Error(`found something that is not a directory at ${folder}, either delete it or specify a different location`);
-        }
-    }
-    else {
-        await mkdir(folder, { recursive: true });
-    }
-    const fixFileName = (file) => file?.replace(/\//g, '__')?.replace(/:/g, '-');
-    const creds = (await readData('creds.json')) || initAuthCreds();
-    return {
-        state: {
-            creds,
-            keys: {
-                get: async (type, ids) => {
-                    const data = {};
-                    await Promise.all(ids.map(async (id) => {
-                        let value = await readData(`${type}-${id}.json`);
-                        if (type === 'app-state-sync-key' && value) {
-                            value = 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 file = `${category}-${id}.json`;
-                            tasks.push(value ? writeData(value, file) : removeData(file));
-                        }
-                    }
-                    await Promise.all(tasks);
-                }
-            }
-        },
-        saveCreds: async () => {
-            return writeData(creds, 'creds.json');
-        }
-    };
-};
-//# sourceMappingURL=use-multi-file-auth-state.js.map
+const useMultiFileAuthState = async folder => {
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	const writeData = async (data, file) => {
+		const filePath = (0, path_1.join)(folder, fixFileName(file))
+		const mutex = getFileLock(filePath)
+		return mutex.acquire().then(async release => {
+			try {
+				await (0, promises_1.writeFile)(filePath, JSON.stringify(data, generics_1.BufferJSON.replacer))
+			} finally {
+				release()
+			}
+		})
+	}
+	const readData = async file => {
+		try {
+			const filePath = (0, path_1.join)(folder, fixFileName(file))
+			const mutex = getFileLock(filePath)
+			return await mutex.acquire().then(async release => {
+				try {
+					const data = await (0, promises_1.readFile)(filePath, { encoding: 'utf-8' })
+					return JSON.parse(data, generics_1.BufferJSON.reviver)
+				} finally {
+					release()
+				}
+			})
+		} catch (error) {
+			return null
+		}
+	}
+	const removeData = async file => {
+		try {
+			const filePath = (0, path_1.join)(folder, fixFileName(file))
+			const mutex = getFileLock(filePath)
+			return mutex.acquire().then(async release => {
+				try {
+					await (0, promises_1.unlink)(filePath)
+				} catch {
+				} finally {
+					release()
+				}
+			})
+		} catch {}
+	}
+	const folderInfo = await (0, promises_1.stat)(folder).catch(() => {})
+	if (folderInfo) {
+		if (!folderInfo.isDirectory()) {
+			throw new Error(
+				`found something that is not a directory at ${folder}, either delete it or specify a different location`
+			)
+		}
+	} else {
+		await (0, promises_1.mkdir)(folder, { recursive: true })
+	}
+	const fixFileName = file => file?.replace(/\//g, '__')?.replace(/:/g, '-')
+	const creds = (await readData('creds.json')) || (0, auth_utils_1.initAuthCreds)()
+	return {
+		state: {
+			creds,
+			keys: {
+				get: async (type, ids) => {
+					const data = {}
+					await Promise.all(
+						ids.map(async id => {
+							let value = await readData(`${type}-${id}.json`)
+							if (type === 'app-state-sync-key' && value) {
+								value = index_js_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 file = `${category}-${id}.json`
+							tasks.push(value ? writeData(value, file) : removeData(file))
+						}
+					}
+					await Promise.all(tasks)
+				}
+			}
+		},
+		saveCreds: async () => {
+			return writeData(creds, 'creds.json')
+		}
+	}
+}
+exports.useMultiFileAuthState = useMultiFileAuthState