feat: prisma and remove mongodb

src/utils/use-multi-file-auth-state-mongodb.ts

@@ -1,107 +0,0 @@
-import {
-  AuthenticationCreds,
-  AuthenticationState,
-  BufferJSON,
-  initAuthCreds,
-  proto,
-  SignalDataTypeMap,
-} from '@whiskeysockets/baileys';
-
-import { configService, Database } from '../config/env.config';
-import { Logger } from '../config/logger.config';
-import { mongodbServer } from '../libs/mongodb.connect';
-
-export async function useMultiFileAuthStateMongoDb(
-  coll: string,
-): Promise<{ state: AuthenticationState; saveCreds: () => Promise<void> }> {
-  const logger = new Logger(useMultiFileAuthStateMongoDb.name);
-
-  const client = mongodbServer.getClient();
-
-  const collection = client
-    .db(configService.get<Database>('DATABASE').CONNECTION.DB_PREFIX_NAME + '-instances')
-    .collection(coll);
-
-  const writeData = async (data: any, key: string): Promise<any> => {
-    try {
-      await client.connect();
-      let msgParsed = JSON.parse(JSON.stringify(data, BufferJSON.replacer));
-      if (Array.isArray(msgParsed)) {
-        msgParsed = {
-          _id: key,
-          content_array: msgParsed,
-        };
-      }
-      return await collection.replaceOne({ _id: key }, msgParsed, {
-        upsert: true,
-      });
-    } catch (error) {
-      logger.error(error);
-    }
-  };
-
-  const readData = async (key: string): Promise<any> => {
-    try {
-      await client.connect();
-      let data = (await collection.findOne({ _id: key })) as any;
-      if (data?.content_array) {
-        data = data.content_array;
-      }
-      const creds = JSON.stringify(data);
-      return JSON.parse(creds, BufferJSON.reviver);
-    } catch (error) {
-      logger.error(error);
-    }
-  };
-
-  const removeData = async (key: string) => {
-    try {
-      await client.connect();
-      return await collection.deleteOne({ _id: key });
-    } catch (error) {
-      logger.error(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 ? writeData(value, key) : removeData(key));
-            }
-          }
-
-          await Promise.all(tasks);
-        },
-      },
-    },
-    saveCreds: async () => {
-      return await writeData(creds, 'creds');
-    },
-  };
-}

src/utils/use-multi-file-auth-state-prisma.ts

@@ -1,9 +1,11 @@
-import { PrismaClient } from '@prisma/client';
 import { BufferJSON, initAuthCreds, WAProto as proto } from '@whiskeysockets/baileys';
 import fs from 'fs/promises';
 import path from 'path';
 
-const prisma = new PrismaClient();
+import { INSTANCE_DIR } from '../config/path.config';
+import { prismaServer } from '../libs/prisma.connect';
+
+const prisma = prismaServer;
 
 const fixFileName = (file) => {
   if (!file) {
@@ -28,8 +30,16 @@ export async function saveKey(sessionId, keyJson) {
   const jaExiste = await keyExists(sessionId);
   try {
     if (!jaExiste)
-      return await prisma.session.create({ data: { sessionId: sessionId, creds: JSON.stringify(keyJson) } as any });
-    await prisma.session.update({ where: { sessionId: sessionId }, data: { creds: JSON.stringify(keyJson) } });
+      return await prisma.session.create({
+        data: {
+          sessionId: sessionId,
+          creds: JSON.stringify(keyJson),
+        },
+      });
+    await prisma.session.update({
+      where: { sessionId: sessionId },
+      data: { creds: JSON.stringify(keyJson) },
+    });
   } catch (error) {
     console.log(`${error}`);
     return null;
@@ -68,7 +78,7 @@ async function fileExists(file) {
 }
 
 export default async function useMultiFileAuthStatePrisma(sessionId) {
-  const localFolder = path.join(process.cwd(), 'sessions', sessionId);
+  const localFolder = path.join(INSTANCE_DIR, sessionId);
   const localFile = (key) => path.join(localFolder, fixFileName(key) + '.json');
   await fs.mkdir(localFolder, { recursive: true });