feat: Added websocket with lib socket.io

CHANGELOG.md

@@ -6,6 +6,7 @@
 * Added extra files for chatwoot and appsmith
 * Added Get Last Message and Archive for Chat
 * Added env var QRCODE_COLOR
+* Added websocket with lib socket.io
 
 ### Fixed
 

package.json

@@ -73,6 +73,7 @@
     "qrcode-terminal": "^0.12.0",
     "redis": "^4.6.5",
     "sharp": "^0.30.7",
+    "socket.io": "^4.7.1",
     "uuid": "^9.0.0"
   },
   "devDependencies": {

src/libs/socket.ts

@@ -0,0 +1,40 @@
+import { Server } from 'http';
+import { Server as SocketIO } from 'socket.io';
+
+import { configService, Cors } from '../config/env.config';
+import { Logger } from '../config/logger.config';
+
+const logger = new Logger('Socket');
+
+let io: SocketIO;
+
+const cors = configService.get<Cors>('CORS').ORIGIN;
+
+export const initIO = (httpServer: Server) => {
+  logger.verbose('Initializing Socket.io');
+  io = new SocketIO(httpServer, {
+    cors: {
+      origin: cors,
+    },
+  });
+
+  io.on('connection', (socket) => {
+    logger.verbose('Client connected');
+    socket.on('disconnect', () => {
+      logger.verbose('Client disconnected');
+    });
+  });
+
+  return io;
+};
+
+export const getIO = (): SocketIO => {
+  logger.verbose('Getting Socket.io');
+
+  if (!io) {
+    logger.error('Socket.io not initialized');
+    throw new Error('Socket.io not initialized');
+  }
+
+  return io;
+};

src/main.ts

@@ -9,6 +9,7 @@ import { configService, Cors, HttpServer } from './config/env.config';
 import { onUnexpectedError } from './config/error.config';
 import { Logger } from './config/logger.config';
 import { ROOT_DIR } from './config/path.config';
+import { initIO } from './libs/socket';
 import { ServerUP } from './utils/server-up';
 import { HttpStatus, router } from './whatsapp/routers/index.router';
 import { waMonitor } from './whatsapp/whatsapp.module';
@@ -83,6 +84,8 @@ function bootstrap() {
 
   initWA();
 
+  initIO(server);
+
   onUnexpectedError();
 }
 

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

@@ -9,7 +9,7 @@ import {
 
 import { configService, Database } from '../config/env.config';
 import { Logger } from '../config/logger.config';
-import { dbserver } from '../db/db.connect';
+import { dbserver } from '../libs/db.connect';
 
 export async function useMultiFileAuthStateDb(
   coll: string,

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

@@ -7,7 +7,7 @@ import {
 } from '@whiskeysockets/baileys';
 
 import { Logger } from '../config/logger.config';
-import { RedisCache } from '../db/redis.client';
+import { RedisCache } from '../libs/redis.client';
 
 export async function useMultiFileAuthStateRedisDb(cache: RedisCache): Promise<{
   state: AuthenticationState;

src/whatsapp/controllers/instance.controller.ts

@@ -4,8 +4,8 @@ import EventEmitter2 from 'eventemitter2';
 
 import { ConfigService, HttpServer } from '../../config/env.config';
 import { Logger } from '../../config/logger.config';
-import { RedisCache } from '../../db/redis.client';
 import { BadRequestException, InternalServerErrorException } from '../../exceptions';
+import { RedisCache } from '../../libs/redis.client';
 import { InstanceDto } from '../dto/instance.dto';
 import { RepositoryBroker } from '../repository/repository.manager';
 import { AuthService, OldToken } from '../services/auth.service';

src/whatsapp/guards/instance.guard.ts

@@ -4,8 +4,8 @@ import { join } from 'path';
 
 import { configService, Database, Redis } from '../../config/env.config';
 import { INSTANCE_DIR } from '../../config/path.config';
-import { dbserver } from '../../db/db.connect';
 import { BadRequestException, ForbiddenException, NotFoundException } from '../../exceptions';
+import { dbserver } from '../../libs/db.connect';
 import { InstanceDto } from '../dto/instance.dto';
 import { cache, waMonitor } from '../whatsapp.module';
 

src/whatsapp/models/auth.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 
 export class AuthRaw {
   _id?: string;

src/whatsapp/models/chat.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 
 export class ChatRaw {
   _id?: string;

src/whatsapp/models/chatwoot.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 
 export class ChatwootRaw {
   _id?: string;

src/whatsapp/models/contact.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 
 export class ContactRaw {
   _id?: string;

src/whatsapp/models/message.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 import { wa } from '../types/wa.types';
 
 class Key {

src/whatsapp/models/settings.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 
 export class SettingsRaw {
   _id?: string;

src/whatsapp/models/webhook.model.ts

@@ -1,6 +1,6 @@
 import { Schema } from 'mongoose';
 
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 
 export class WebhookRaw {
   _id?: string;

src/whatsapp/routers/instance.router.ts

@@ -2,7 +2,7 @@ import { RequestHandler, Router } from 'express';
 
 import { Auth, ConfigService, Database } from '../../config/env.config';
 import { Logger } from '../../config/logger.config';
-import { dbserver } from '../../db/db.connect';
+import { dbserver } from '../../libs/db.connect';
 import { instanceNameSchema, oldTokenSchema } from '../../validate/validate.schema';
 import { RouterBroker } from '../abstract/abstract.router';
 import { InstanceDto } from '../dto/instance.dto';

src/whatsapp/services/monitor.service.ts

@@ -7,9 +7,9 @@ import { join } from 'path';
 import { Auth, ConfigService, Database, DelInstance, HttpServer, Redis } from '../../config/env.config';
 import { Logger } from '../../config/logger.config';
 import { INSTANCE_DIR, STORE_DIR } from '../../config/path.config';
-import { dbserver } from '../../db/db.connect';
-import { RedisCache } from '../../db/redis.client';
 import { NotFoundException } from '../../exceptions';
+import { dbserver } from '../../libs/db.connect';
+import { RedisCache } from '../../libs/redis.client';
 import {
   AuthModel,
   ChatwootModel,

src/whatsapp/services/whatsapp.service.ts

@@ -63,9 +63,10 @@ import {
 } from '../../config/env.config';
 import { Logger } from '../../config/logger.config';
 import { INSTANCE_DIR, ROOT_DIR } from '../../config/path.config';
-import { dbserver } from '../../db/db.connect';
-import { RedisCache } from '../../db/redis.client';
 import { BadRequestException, InternalServerErrorException, NotFoundException } from '../../exceptions';
+import { dbserver } from '../../libs/db.connect';
+import { RedisCache } from '../../libs/redis.client';
+import { getIO } from '../../libs/socket';
 import { useMultiFileAuthStateDb } from '../../utils/use-multi-file-auth-state-db';
 import { useMultiFileAuthStateRedisDb } from '../../utils/use-multi-file-auth-state-redis-db';
 import {
@@ -416,6 +417,7 @@ export class WAStartupService {
     const serverUrl = this.configService.get<HttpServer>('SERVER').URL;
     const we = event.replace(/[.-]/gm, '_').toUpperCase();
     const transformedWe = we.replace(/_/gm, '-').toLowerCase();
+    const io = getIO();
 
     const expose = this.configService.get<Auth>('AUTHENTICATION').EXPOSE_IN_FETCH_INSTANCES;
     const tokenStore = await this.repository.auth.find(this.instanceName);
@@ -553,6 +555,12 @@ export class WAStartupService {
         }
       }
     }
+
+    io.emit(event, {
+      event,
+      instance: this.instance.name,
+      data,
+    });
   }
 
   private async connectionUpdate({ qr, connection, lastDisconnect }: Partial<ConnectionState>) {

src/whatsapp/whatsapp.module.ts

@@ -1,8 +1,8 @@
 import { configService } from '../config/env.config';
 import { eventEmitter } from '../config/event.config';
 import { Logger } from '../config/logger.config';
-import { dbserver } from '../db/db.connect';
-import { RedisCache } from '../db/redis.client';
+import { dbserver } from '../libs/db.connect';
+import { RedisCache } from '../libs/redis.client';
 import { ChatController } from './controllers/chat.controller';
 import { ChatwootController } from './controllers/chatwoot.controller';
 import { GroupController } from './controllers/group.controller';

views/manager.hbs

@@ -11,7 +11,7 @@
 
 <body>
 
-   <iframe src="https://app.smith.dgcode.com.br/app/evolutionapi/page1-64c53a7a7ea84639bf879f23?embed=true" frameborder="0" style="width: 100%; height: 100vh;"></iframe>
+   <iframe src="https://app.smith.dgcode.com.br/app/evolutionapi-public/home-64ca60783615e270291978b4?embed=true" frameborder="0" style="width: 100%; height: 100vh;"></iframe>
    
 </body>