fix: reorganization of files and folders

src/api/integrations/websocket/controllers/websocket.controller.ts

@@ -0,0 +1,58 @@
+import { Logger } from '../../../../config/logger.config';
+import { InstanceDto } from '../../../dto/instance.dto';
+import { WebsocketDto } from '../dto/websocket.dto';
+import { WebsocketService } from '../services/websocket.service';
+
+const logger = new Logger('WebsocketController');
+
+export class WebsocketController {
+  constructor(private readonly websocketService: WebsocketService) {}
+
+  public async createWebsocket(instance: InstanceDto, data: WebsocketDto) {
+    logger.verbose('requested createWebsocket from ' + instance.instanceName + ' instance');
+
+    if (!data.enabled) {
+      logger.verbose('websocket disabled');
+      data.events = [];
+    }
+
+    if (data.events.length === 0) {
+      logger.verbose('websocket 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',
+        'LABELS_EDIT',
+        'LABELS_ASSOCIATION',
+        'CALL',
+        'NEW_JWT_TOKEN',
+        'TYPEBOT_START',
+        'TYPEBOT_CHANGE_STATUS',
+        'CHAMA_AI_ACTION',
+      ];
+    }
+
+    return this.websocketService.create(instance, data);
+  }
+
+  public async findWebsocket(instance: InstanceDto) {
+    logger.verbose('requested findWebsocket from ' + instance.instanceName + ' instance');
+    return this.websocketService.find(instance);
+  }
+}

src/api/integrations/websocket/dto/websocket.dto.ts

@@ -0,0 +1,4 @@
+export class WebsocketDto {
+  enabled: boolean;
+  events?: string[];
+}

src/api/integrations/websocket/libs/socket.server.ts

@@ -0,0 +1,44 @@
+import { Server } from 'http';
+import { Server as SocketIO } from 'socket.io';
+
+import { configService, Cors, Websocket } 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) => {
+  if (configService.get<Websocket>('WEBSOCKET')?.ENABLED) {
+    io = new SocketIO(httpServer, {
+      cors: {
+        origin: cors,
+      },
+    });
+
+    io.on('connection', (socket) => {
+      logger.info('User connected');
+
+      socket.on('disconnect', () => {
+        logger.info('User disconnected');
+      });
+    });
+
+    logger.info('Socket.io initialized');
+    return io;
+  }
+  return null;
+};
+
+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/api/integrations/websocket/models/websocket.model.ts

@@ -0,0 +1,18 @@
+import { Schema } from 'mongoose';
+
+import { dbserver } from '../../../../libs/db.connect';
+
+export class WebsocketRaw {
+  _id?: string;
+  enabled?: boolean;
+  events?: string[];
+}
+
+const websocketSchema = new Schema<WebsocketRaw>({
+  _id: { type: String, _id: true },
+  enabled: { type: Boolean, required: true },
+  events: { type: [String], required: true },
+});
+
+export const WebsocketModel = dbserver?.model(WebsocketRaw.name, websocketSchema, 'websocket');
+export type IWebsocketModel = typeof WebsocketModel;

src/api/integrations/websocket/repository/websocket.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 { IWebsocketModel, WebsocketRaw } from '../../../models';
+
+export class WebsocketRepository extends Repository {
+  constructor(private readonly websocketModel: IWebsocketModel, private readonly configService: ConfigService) {
+    super(configService);
+  }
+
+  private readonly logger = new Logger('WebsocketRepository');
+
+  public async create(data: WebsocketRaw, instance: string): Promise<IInsert> {
+    try {
+      this.logger.verbose('creating websocket');
+      if (this.dbSettings.ENABLED) {
+        this.logger.verbose('saving websocket to db');
+        const insert = await this.websocketModel.replaceOne({ _id: instance }, { ...data }, { upsert: true });
+
+        this.logger.verbose('websocket saved to db: ' + insert.modifiedCount + ' websocket');
+        return { insertCount: insert.modifiedCount };
+      }
+
+      this.logger.verbose('saving websocket to store');
+
+      this.writeStore<WebsocketRaw>({
+        path: join(this.storePath, 'websocket'),
+        fileName: instance,
+        data,
+      });
+
+      this.logger.verbose('websocket saved to store in path: ' + join(this.storePath, 'websocket') + '/' + instance);
+
+      this.logger.verbose('websocket created');
+      return { insertCount: 1 };
+    } catch (error) {
+      return error;
+    }
+  }
+
+  public async find(instance: string): Promise<WebsocketRaw> {
+    try {
+      this.logger.verbose('finding websocket');
+      if (this.dbSettings.ENABLED) {
+        this.logger.verbose('finding websocket in db');
+        return await this.websocketModel.findOne({ _id: instance });
+      }
+
+      this.logger.verbose('finding websocket in store');
+      return JSON.parse(
+        readFileSync(join(this.storePath, 'websocket', instance + '.json'), {
+          encoding: 'utf-8',
+        }),
+      ) as WebsocketRaw;
+    } catch (error) {
+      return {};
+    }
+  }
+}

src/api/integrations/websocket/routes/websocket.router.ts

@@ -0,0 +1,52 @@
+import { RequestHandler, Router } from 'express';
+
+import { Logger } from '../../../../config/logger.config';
+import { instanceNameSchema, websocketSchema } from '../../../../validate/validate.schema';
+import { RouterBroker } from '../../../abstract/abstract.router';
+import { InstanceDto } from '../../../dto/instance.dto';
+import { HttpStatus } from '../../../routers/index.router';
+import { websocketController } from '../../../server.module';
+import { WebsocketDto } from '../dto/websocket.dto';
+
+const logger = new Logger('WebsocketRouter');
+
+export class WebsocketRouter extends RouterBroker {
+  constructor(...guards: RequestHandler[]) {
+    super();
+    this.router
+      .post(this.routerPath('set'), ...guards, async (req, res) => {
+        logger.verbose('request received in setWebsocket');
+        logger.verbose('request body: ');
+        logger.verbose(req.body);
+
+        logger.verbose('request query: ');
+        logger.verbose(req.query);
+        const response = await this.dataValidate<WebsocketDto>({
+          request: req,
+          schema: websocketSchema,
+          ClassRef: WebsocketDto,
+          execute: (instance, data) => websocketController.createWebsocket(instance, data),
+        });
+
+        res.status(HttpStatus.CREATED).json(response);
+      })
+      .get(this.routerPath('find'), ...guards, async (req, res) => {
+        logger.verbose('request received in findWebsocket');
+        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) => websocketController.findWebsocket(instance),
+        });
+
+        res.status(HttpStatus.OK).json(response);
+      });
+  }
+
+  public readonly router = Router();
+}

src/api/integrations/websocket/services/websocket.service.ts

@@ -0,0 +1,33 @@
+import { Logger } from '../../../../config/logger.config';
+import { InstanceDto } from '../../../dto/instance.dto';
+import { WebsocketRaw } from '../../../models';
+import { WAMonitoringService } from '../../../services/monitor.service';
+import { WebsocketDto } from '../dto/websocket.dto';
+
+export class WebsocketService {
+  constructor(private readonly waMonitor: WAMonitoringService) {}
+
+  private readonly logger = new Logger(WebsocketService.name);
+
+  public create(instance: InstanceDto, data: WebsocketDto) {
+    this.logger.verbose('create websocket: ' + instance.instanceName);
+    this.waMonitor.waInstances[instance.instanceName].setWebsocket(data);
+
+    return { websocket: { ...instance, websocket: data } };
+  }
+
+  public async find(instance: InstanceDto): Promise<WebsocketRaw> {
+    try {
+      this.logger.verbose('find websocket: ' + instance.instanceName);
+      const result = await this.waMonitor.waInstances[instance.instanceName].findWebsocket();
+
+      if (Object.keys(result).length === 0) {
+        throw new Error('Websocket not found');
+      }
+
+      return result;
+    } catch (error) {
+      return { enabled: false, events: [] };
+    }
+  }
+}

src/api/integrations/websocket/validate/websocket.schema.ts

@@ -0,0 +1,66 @@
+import { JSONSchema7 } from 'json-schema';
+import { v4 } from 'uuid';
+
+const isNotEmpty = (...propertyNames: string[]): JSONSchema7 => {
+  const properties = {};
+  propertyNames.forEach(
+    (property) =>
+      (properties[property] = {
+        minLength: 1,
+        description: `The "${property}" cannot be empty`,
+      }),
+  );
+  return {
+    if: {
+      propertyNames: {
+        enum: [...propertyNames],
+      },
+    },
+    then: { properties },
+  };
+};
+
+export const websocketSchema: 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',
+          'LABELS_EDIT',
+          'LABELS_ASSOCIATION',
+          'CALL',
+          'NEW_JWT_TOKEN',
+          'TYPEBOT_START',
+          'TYPEBOT_CHANGE_STATUS',
+          'CHAMA_AI_ACTION',
+        ],
+      },
+    },
+  },
+  required: ['enabled'],
+  ...isNotEmpty('enabled'),
+};