feat: add support for remote session storage using mongoDB

.env.example

@@ -1,4 +1,5 @@
 OPENAI_API_KEY=ADD_YOUR_KEY
 DREAMSTUDIO_API_KEY=ADD_YOUR_KEY
 GEMINI_API_KEY=ADD_YOUR_KEY
-HF_TOKEN=HUGGING_FACE_TOKEN
+HF_TOKEN=HUGGING_FACE_TOKEN
+MONGO_URL=YOUR_MONGO_DB_URL

package.json

@@ -23,6 +23,8 @@
         "@types/qrcode-terminal": "^0.12.0",
         "@whiskeysockets/baileys": "^6.7.7",
         "dotenv": "^16.0.3",
+        "mongo-baileys": "^1.0.1",
+        "mongodb": "^6.8.0",
         "openai": "^4.56.0",
         "ora": "3.2.0",
         "qrcode-terminal": "^0.12.0",

src/lib/baileys/index.ts

@@ -3,9 +3,15 @@ import { Boom } from '@hapi/boom';
 import * as fs from 'fs';
 import path from 'path';
 import { messagesHandler } from './handlers/messages';
+import { useMongoDBAuthState } from 'mongo-baileys';
+import { ENV } from '../env';
+import { connectToMongoDB } from '../mongo-db';
 
 export async function connectToWhatsApp() {
-  const { state, saveCreds } = await useMultiFileAuthState('auth_info_baileys');
+  const { state, saveCreds } =
+    ENV.MONGO_URL !== undefined
+      ? await useMongoDBAuthState((await connectToMongoDB()).collection as any)
+      : await useMultiFileAuthState('auth_info_baileys');
   const sock = makeWASocket({ printQRInTerminal: true, auth: state });
 
   sock.ev.on('connection.update', async (update) => {

src/lib/env.ts

@@ -7,5 +7,6 @@ export const ENV = {
   openAIKey: process.env.OPENAI_API_KEY,
   dreamStudioKey: process.env.DREAMSTUDIO_API_KEY,
   geminiKey: process.env.GEMINI_API_KEY,
-  HFKey: process.env.HF_TOKEN
+  HFKey: process.env.HF_TOKEN,
+  MONGO_URL: process.env.MONGO_URL
 };

src/lib/mongo-db/index.ts

@@ -0,0 +1,16 @@
+import { MongoClient } from 'mongodb';
+import { ENV } from '../env';
+import { AuthenticationCreds } from '@whiskeysockets/baileys';
+
+interface AuthDocument extends Document {
+  _id: string;
+  creds?: AuthenticationCreds;
+}
+
+export async function connectToMongoDB() {
+  const client = new MongoClient(ENV.MONGO_URL || '');
+  await client.connect();
+  const db = client.db('whatsappAIBot');
+  const collection = db.collection<AuthDocument>('authState');
+  return { client, collection };
+}