feat: Added rabbitmq to send events

DavidsonGomes · DavidsonGomes · commit e05c48979d59 · 2023-08-02T17:28:52.000-03:00
diff --git a/package.json b/package.json
@@ -47,6 +47,7 @@
     "@hapi/boom": "^10.0.1",
     "@sentry/node": "^7.59.2",
     "@whiskeysockets/baileys": "github:EvolutionAPI/Baileys",
+    "amqplib": "^0.10.3",
     "axios": "^1.3.5",
     "class-validator": "^0.13.2",
     "compression": "^1.7.4",
diff --git a/src/libs/amqp.server.ts b/src/libs/amqp.server.ts
@@ -0,0 +1,37 @@
+import * as amqp from 'amqplib/callback_api';
+
+import { Logger } from '../config/logger.config';
+
+const logger = new Logger('AMQP');
+
+let amqpChannel: amqp.Channel | null = null;
+
+export const initAMQP = () => {
+  return new Promise<void>((resolve, reject) => {
+    amqp.connect('amqp://admin:admin@localhost:5672', (error, connection) => {
+      if (error) {
+        reject(error);
+        return;
+      }
+
+      connection.createChannel((channelError, channel) => {
+        if (channelError) {
+          reject(channelError);
+          return;
+        }
+
+        const exchangeName = 'evolution_exchange';
+
+        channel.assertExchange(exchangeName, 'topic', { durable: false });
+        amqpChannel = channel;
+
+        logger.log('Serviço do RabbitMQ inicializado com sucesso.');
+        resolve();
+      });
+    });
+  });
+};
+
+export const getAMQP = (): amqp.Channel | null => {
+  return amqpChannel;
+};
diff --git a/src/libs/socket.server.ts b/src/libs/socket.server.ts
diff --git a/src/main.ts b/src/main.ts
@@ -9,7 +9,8 @@ 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 { initAMQP } from './libs/amqp.server';
+import { initIO } from './libs/socket.server';
 import { ServerUP } from './utils/server-up';
 import { HttpStatus, router } from './whatsapp/routers/index.router';
 import { waMonitor } from './whatsapp/whatsapp.module';
@@ -86,6 +87,8 @@ function bootstrap() {
 
   initIO(server);
 
+  initAMQP();
+
   onUnexpectedError();
 }
 
diff --git a/src/validate/validate.schema.ts b/src/validate/validate.schema.ts
@@ -934,3 +934,43 @@ export const websocketSchema: JSONSchema7 = {
   required: ['enabled'],
   ...isNotEmpty('enabled'),
 };
+
+export const rabbitmqSchema: JSONSchema7 = {
+  $id: v4(),
+  type: 'object',
+  properties: {
+    enabled: { type: 'boolean', enum: [true, false] },
+    events: {
+      type: 'array',
+      minItems: 0,
+      items: {
+        type: 'string',
+        enum: [
+          'APPLICATION_STARTUP',
+          'QRCODE_UPDATED',
+          'MESSAGES_SET',
+          'MESSAGES_UPSERT',
+          'MESSAGES_UPDATE',
+          'MESSAGES_DELETE',
+          'SEND_MESSAGE',
+          'CONTACTS_SET',
+          'CONTACTS_UPSERT',
+          'CONTACTS_UPDATE',
+          'PRESENCE_UPDATE',
+          'CHATS_SET',
+          'CHATS_UPSERT',
+          'CHATS_UPDATE',
+          'CHATS_DELETE',
+          'GROUPS_UPSERT',
+          'GROUP_UPDATE',
+          'GROUP_PARTICIPANTS_UPDATE',
+          'CONNECTION_UPDATE',
+          'CALL',
+          'NEW_JWT_TOKEN',
+        ],
+      },
+    },
+  },
+  required: ['enabled'],
+  ...isNotEmpty('enabled'),
+};
diff --git a/src/whatsapp/controllers/instance.controller.ts b/src/whatsapp/controllers/instance.controller.ts
@@ -11,6 +11,7 @@ import { RepositoryBroker } from '../repository/repository.manager';
 import { AuthService, OldToken } from '../services/auth.service';
 import { ChatwootService } from '../services/chatwoot.service';
 import { WAMonitoringService } from '../services/monitor.service';
+import { RabbitmqService } from '../services/rabbitmq.service';
 import { SettingsService } from '../services/settings.service';
 import { WebhookService } from '../services/webhook.service';
 import { WebsocketService } from '../services/websocket.service';
@@ -28,6 +29,7 @@ export class InstanceController {
     private readonly chatwootService: ChatwootService,
     private readonly settingsService: SettingsService,
     private readonly websocketService: WebsocketService,
+    private readonly rebbitmqService: RabbitmqService,
     private readonly cache: RedisCache,
   ) {}
 
diff --git a/src/whatsapp/controllers/rabbitmq.controller.ts b/src/whatsapp/controllers/rabbitmq.controller.ts
@@ -0,0 +1,53 @@
+import { Logger } from '../../config/logger.config';
+import { InstanceDto } from '../dto/instance.dto';
+import { RabbitmqDto } from '../dto/rabbitmq.dto';
+import { RabbitmqService } from '../services/rabbitmq.service';
+
+const logger = new Logger('RabbitmqController');
+
+export class RabbitmqController {
+  constructor(private readonly rabbitmqService: RabbitmqService) {}
+
+  public async createRabbitmq(instance: InstanceDto, data: RabbitmqDto) {
+    logger.verbose('requested createRabbitmq from ' + instance.instanceName + ' instance');
+
+    if (!data.enabled) {
+      logger.verbose('rabbitmq disabled');
+      data.events = [];
+    }
+
+    if (data.events.length === 0) {
+      logger.verbose('rabbitmq events empty');
+      data.events = [
+        'APPLICATION_STARTUP',
+        'QRCODE_UPDATED',
+        'MESSAGES_SET',
+        'MESSAGES_UPSERT',
+        'MESSAGES_UPDATE',
+        'MESSAGES_DELETE',
+        'SEND_MESSAGE',
+        'CONTACTS_SET',
+        'CONTACTS_UPSERT',
+        'CONTACTS_UPDATE',
+        'PRESENCE_UPDATE',
+        'CHATS_SET',
+        'CHATS_UPSERT',
+        'CHATS_UPDATE',
+        'CHATS_DELETE',
+        'GROUPS_UPSERT',
+        'GROUP_UPDATE',
+        'GROUP_PARTICIPANTS_UPDATE',
+        'CONNECTION_UPDATE',
+        'CALL',
+        'NEW_JWT_TOKEN',
+      ];
+    }
+
+    return this.rabbitmqService.create(instance, data);
+  }
+
+  public async findRabbitmq(instance: InstanceDto) {
+    logger.verbose('requested findRabbitmq from ' + instance.instanceName + ' instance');
+    return this.rabbitmqService.find(instance);
+  }
+}
diff --git a/src/whatsapp/dto/rabbitmq.dto.ts b/src/whatsapp/dto/rabbitmq.dto.ts
@@ -0,0 +1,4 @@
+export class RabbitmqDto {
+  enabled: boolean;
+  events?: string[];
+}
diff --git a/src/whatsapp/models/index.ts b/src/whatsapp/models/index.ts
@@ -3,6 +3,7 @@ export * from './chat.model';
 export * from './chatwoot.model';
 export * from './contact.model';
 export * from './message.model';
+export * from './rabbitmq.model';
 export * from './settings.model';
 export * from './webhook.model';
 export * from './websocket.model';
diff --git a/src/whatsapp/models/rabbitmq.model.ts b/src/whatsapp/models/rabbitmq.model.ts
@@ -0,0 +1,18 @@
+import { Schema } from 'mongoose';
+
+import { dbserver } from '../../libs/db.connect';
+
+export class RabbitmqRaw {
+  _id?: string;
+  enabled?: boolean;
+  events?: string[];
+}
+
+const rabbitmqSchema = new Schema<RabbitmqRaw>({
+  _id: { type: String, _id: true },
+  enabled: { type: Boolean, required: true },
+  events: { type: [String], required: true },
+});
+
+export const RabbitmqModel = dbserver?.model(RabbitmqRaw.name, rabbitmqSchema, 'rabbitmq');
+export type IRabbitmqModel = typeof RabbitmqModel;
diff --git a/src/whatsapp/repository/rabbitmq.repository.ts b/src/whatsapp/repository/rabbitmq.repository.ts
@@ -0,0 +1,62 @@
+import { readFileSync } from 'fs';
+import { join } from 'path';
+
+import { ConfigService } from '../../config/env.config';
+import { Logger } from '../../config/logger.config';
+import { IInsert, Repository } from '../abstract/abstract.repository';
+import { IRabbitmqModel, RabbitmqRaw } from '../models';
+
+export class RabbitmqRepository extends Repository {
+  constructor(private readonly rabbitmqModel: IRabbitmqModel, private readonly configService: ConfigService) {
+    super(configService);
+  }
+
+  private readonly logger = new Logger('RabbitmqRepository');
+
+  public async create(data: RabbitmqRaw, instance: string): Promise<IInsert> {
+    try {
+      this.logger.verbose('creating rabbitmq');
+      if (this.dbSettings.ENABLED) {
+        this.logger.verbose('saving rabbitmq to db');
+        const insert = await this.rabbitmqModel.replaceOne({ _id: instance }, { ...data }, { upsert: true });
+
+        this.logger.verbose('rabbitmq saved to db: ' + insert.modifiedCount + ' rabbitmq');
+        return { insertCount: insert.modifiedCount };
+      }
+
+      this.logger.verbose('saving rabbitmq to store');
+
+      this.writeStore<RabbitmqRaw>({
+        path: join(this.storePath, 'rabbitmq'),
+        fileName: instance,
+        data,
+      });
+
+      this.logger.verbose('rabbitmq saved to store in path: ' + join(this.storePath, 'rabbitmq') + '/' + instance);
+
+      this.logger.verbose('rabbitmq created');
+      return { insertCount: 1 };
+    } catch (error) {
+      return error;
+    }
+  }
+
+  public async find(instance: string): Promise<RabbitmqRaw> {
+    try {
+      this.logger.verbose('finding rabbitmq');
+      if (this.dbSettings.ENABLED) {
+        this.logger.verbose('finding rabbitmq in db');
+        return await this.rabbitmqModel.findOne({ _id: instance });
+      }
+
+      this.logger.verbose('finding rabbitmq in store');
+      return JSON.parse(
+        readFileSync(join(this.storePath, 'rabbitmq', instance + '.json'), {
+          encoding: 'utf-8',
+        }),
+      ) as RabbitmqRaw;
+    } catch (error) {
+      return {};
+    }
+  }
+}
diff --git a/src/whatsapp/repository/repository.manager.ts b/src/whatsapp/repository/repository.manager.ts
@@ -10,6 +10,7 @@ import { ChatwootRepository } from './chatwoot.repository';
 import { ContactRepository } from './contact.repository';
 import { MessageRepository } from './message.repository';
 import { MessageUpRepository } from './messageUp.repository';
+import { RabbitmqRepository } from './rabbitmq.repository';
 import { SettingsRepository } from './settings.repository';
 import { WebhookRepository } from './webhook.repository';
 import { WebsocketRepository } from './websocket.repository';
@@ -23,6 +24,7 @@ export class RepositoryBroker {
     public readonly chatwoot: ChatwootRepository,
     public readonly settings: SettingsRepository,
     public readonly websocket: WebsocketRepository,
+    public readonly rabbitmq: RabbitmqRepository,
     public readonly auth: AuthRepository,
     private configService: ConfigService,
     dbServer?: MongoClient,
@@ -54,6 +56,7 @@ export class RepositoryBroker {
         const chatwootDir = join(storePath, 'chatwoot');
         const settingsDir = join(storePath, 'settings');
         const websocketDir = join(storePath, 'websocket');
+        const rabbitmqDir = join(storePath, 'rabbitmq');
         const tempDir = join(storePath, 'temp');
 
         if (!fs.existsSync(authDir)) {
@@ -92,6 +95,10 @@ export class RepositoryBroker {
           this.logger.verbose('creating websocket dir: ' + websocketDir);
           fs.mkdirSync(websocketDir, { recursive: true });
         }
+        if (!fs.existsSync(rabbitmqDir)) {
+          this.logger.verbose('creating rabbitmq dir: ' + rabbitmqDir);
+          fs.mkdirSync(rabbitmqDir, { recursive: true });
+        }
         if (!fs.existsSync(tempDir)) {
           this.logger.verbose('creating temp dir: ' + tempDir);
           fs.mkdirSync(tempDir, { recursive: true });
diff --git a/src/whatsapp/routers/index.router.ts b/src/whatsapp/routers/index.router.ts
@@ -8,6 +8,7 @@ import { ChatRouter } from './chat.router';
 import { ChatwootRouter } from './chatwoot.router';
 import { GroupRouter } from './group.router';
 import { InstanceRouter } from './instance.router';
+import { RabbitmqRouter } from './rabbitmq.router';
 import { MessageRouter } from './sendMessage.router';
 import { SettingsRouter } from './settings.router';
 import { ViewsRouter } from './view.router';
@@ -46,6 +47,7 @@ router
   .use('/webhook', new WebhookRouter(...guards).router)
   .use('/chatwoot', new ChatwootRouter(...guards).router)
   .use('/settings', new SettingsRouter(...guards).router)
-  .use('/websocket', new WebsocketRouter(...guards).router);
+  .use('/websocket', new WebsocketRouter(...guards).router)
+  .use('/rabbitmq', new RabbitmqRouter(...guards).router);
 
 export { HttpStatus, router };
diff --git a/src/whatsapp/routers/rabbitmq.router.ts b/src/whatsapp/routers/rabbitmq.router.ts
@@ -0,0 +1,52 @@
+import { RequestHandler, Router } from 'express';
+
+import { Logger } from '../../config/logger.config';
+import { instanceNameSchema, rabbitmqSchema } from '../../validate/validate.schema';
+import { RouterBroker } from '../abstract/abstract.router';
+import { InstanceDto } from '../dto/instance.dto';
+import { RabbitmqDto } from '../dto/rabbitmq.dto';
+import { rabbitmqController } from '../whatsapp.module';
+import { HttpStatus } from './index.router';
+
+const logger = new Logger('RabbitmqRouter');
+
+export class RabbitmqRouter extends RouterBroker {
+  constructor(...guards: RequestHandler[]) {
+    super();
+    this.router
+      .post(this.routerPath('set'), ...guards, async (req, res) => {
+        logger.verbose('request received in setRabbitmq');
+        logger.verbose('request body: ');
+        logger.verbose(req.body);
+
+        logger.verbose('request query: ');
+        logger.verbose(req.query);
+        const response = await this.dataValidate<RabbitmqDto>({
+          request: req,
+          schema: rabbitmqSchema,
+          ClassRef: RabbitmqDto,
+          execute: (instance, data) => rabbitmqController.createRabbitmq(instance, data),
+        });
+
+        res.status(HttpStatus.CREATED).json(response);
+      })
+      .get(this.routerPath('find'), ...guards, async (req, res) => {
+        logger.verbose('request received in findRabbitmq');
+        logger.verbose('request body: ');
+        logger.verbose(req.body);
+
+        logger.verbose('request query: ');
+        logger.verbose(req.query);
+        const response = await this.dataValidate<InstanceDto>({
+          request: req,
+          schema: instanceNameSchema,
+          ClassRef: InstanceDto,
+          execute: (instance) => rabbitmqController.findRabbitmq(instance),
+        });
+
+        res.status(HttpStatus.OK).json(response);
+      });
+  }
+
+  public readonly router = Router();
+}
diff --git a/src/whatsapp/services/rabbitmq.service.ts b/src/whatsapp/services/rabbitmq.service.ts
diff --git a/src/whatsapp/services/whatsapp.service.ts b/src/whatsapp/services/whatsapp.service.ts
diff --git a/src/whatsapp/types/wa.types.ts b/src/whatsapp/types/wa.types.ts
diff --git a/src/whatsapp/whatsapp.module.ts b/src/whatsapp/whatsapp.module.ts
