init project evolution api

src/utils/use-multi-file-auth-state-redis-db.ts

@@ -0,0 +1,89 @@
+import {
+  AuthenticationCreds,
+  AuthenticationState,
+  initAuthCreds,
+  proto,
+  SignalDataTypeMap,
+} from '@evolution/base';
+import { RedisCache } from '../db/redis.client';
+import { Logger } from '../config/logger.config';
+import { Redis } from '../config/env.config';
+
+export async function useMultiFileAuthStateRedisDb(
+  redisEnv: Partial<Redis>,
+  instanceName: string,
+): Promise<{
+  state: AuthenticationState;
+  saveCreds: () => Promise<void>;
+}> {
+  const logger = new Logger(useMultiFileAuthStateRedisDb.name);
+
+  const cache = new RedisCache(redisEnv, instanceName);
+
+  const writeData = async (data: any, key: string): Promise<any> => {
+    try {
+      return await cache.writeData(key, data);
+    } catch (error) {
+      return logger.error({ localError: 'writeData', error });
+    }
+  };
+
+  const readData = async (key: string): Promise<any> => {
+    try {
+      return await cache.readData(key);
+    } catch (error) {
+      logger.error({ readData: 'writeData', error });
+      return;
+    }
+  };
+
+  const removeData = async (key: string) => {
+    try {
+      return await cache.removeData(key);
+    } catch (error) {
+      logger.error({ readData: 'removeData', error });
+    }
+  };
+
+  const creds: AuthenticationCreds = (await readData('creds')) || initAuthCreds();
+
+  return {
+    state: {
+      creds,
+      keys: {
+        get: async (type, ids: string[]) => {
+          // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+          // @ts-ignore
+          const data: { [_: string]: SignalDataTypeMap[type] } = {};
+          await Promise.all(
+            ids.map(async (id) => {
+              let value = await readData(`${type}-${id}`);
+              if (type === 'app-state-sync-key' && value) {
+                value = proto.Message.AppStateSyncKeyData.fromObject(value);
+              }
+
+              data[id] = value;
+            }),
+          );
+
+          return data;
+        },
+        set: async (data: any) => {
+          const tasks: Promise<void>[] = [];
+          for (const category in data) {
+            for (const id in data[category]) {
+              const value = data[category][id];
+              const key = `${category}-${id}`;
+              tasks.push(value ? await writeData(value, key) : await removeData(key));
+            }
+          }
+
+          await Promise.all(tasks);
+        },
+      },
+    },
+    saveCreds: async () => {
+      return await writeData(creds, 'creds');
+    },
+  };
+}