projeto-hit/backend/src/libs/socket.ts

import { Server as SocketIO } from "socket.io";
import { Server } from "http";
import AppError from "../errors/AppError";
import { logger } from "../utils/logger";

import { v4 as uuidv4 } from 'uuid';
import ListUserParamiterService from "../services/UserServices/ListUserParamiterService";
import { addHours, addMinutes, addSeconds, intervalToDuration, add } from "date-fns";

let io: SocketIO;

//test del
import createOrUpdateOnlineUserService from "../services/UserServices/CreateOrUpdateOnlineUserService";
import { splitDateTime } from "../helpers/SplitDateTime";
import format from 'date-fns/format';
import ptBR from 'date-fns/locale/pt-BR';
import ListUserOnlineOffline from "../services/UserServices/ListUsersOnlineOfflineService";
import { handleMessage, handleMsgAck } from "../services/WbotServices/wbotMessageListener";
import { join } from "path";
import Whatsapp from "../models/Whatsapp";

let count: number = 0
let listOnline: any[] = []
let listOnlineAux: any[] = []
let countOnline: number = 0
let obj: any = { listOnline: [], uuid: null, listOnlineAux: [] }




let lstOnline: any[] = []
let lstOnlineAux: any[] = []
let lstTry: any[] = []

let dateTime = splitDateTime(new Date(format(new Date(), 'yyyy-MM-dd HH:mm:ss', { locale: ptBR })))


export const initIO = (httpServer: Server): SocketIO => {
  io = new SocketIO(httpServer, {
    cors: {
      origin: process.env.FRONTEND_URL
    }, 
    maxHttpBufferSize: 1e8
  });


  io.on("connection", socket => {
    logger.info("Client Connected");







    socket.on("joinWhatsSession", (whatsappId: string) => {
      logger.info(`A client joined a joinWhatsSession channel: ${whatsappId}`);
      socket.join(`session_${whatsappId}`);
    });

    socket.on("message_from_client", () => {
 
      socket.emit('message_from_server', 'Sent an event from the server!');
    })

    socket.on("message_create", async (data: any) => {

       handleMessage(data.msg, data);

    });


    socket.on("media_uploaded", async (data: any) => { 

      handleMessage(data.msg, data);

    });

    socket.on("message_ack", async (data: any) => {

      handleMsgAck(data.id, data.ack)

    });










    socket.on("online", (userId: any) => {

      // console.log('userId: ', userId)

      obj.uuid = uuidv4()

      if (userId.logoutUserId) {

        let index = lstOnline.findIndex((x: any) => x.id == userId.logoutUserId)

        if (index != -1) {

          lstOnline.splice(index, 1)

        }


        index = lstOnlineAux.findIndex((x: any) => x.id == userId.logoutUserId)

        if (index != -1) {

          lstOnlineAux.splice(index, 1)

        }

        return
      }

      if (lstOnline.length == 0) {

        const index = listOnlineAux.findIndex((e: any) => e.id == userId)

        if (index == -1) {
          listOnlineAux.push({ 'id': userId })
        }
        else {
          return
        }

        lstOnline.push({ 'id': userId, 'status': 'online', 'try': 0 })

        lstOnlineAux.push({ 'id': userId })
        console.log(' 1 PUSHED NEW USER ID 1: ', userId)

        obj.listOnline = lstOnline

      }
      else {

        const indexAux = lstOnlineAux.findIndex((e: any) => e.id == userId)

        if (indexAux == -1) {
          lstOnlineAux.push({ 'id': userId })
        }

        const index = lstOnline.findIndex((e: any) => e.id == userId)

        if (index == -1) {


          lstOnline.push({ 'id': userId, 'status': 'online', 'try': 0 })


          console.log(' 2 PUSHED NEW USER ID: ', userId)

          obj.listOnline = lstOnline
        }
        else {

          if (countOnline > (lstOnline.length - 1)) {

            lstOnline.forEach((x: any) => {
              if (lstOnlineAux.map((e: any) => e.id).includes(x.id)) {
                x.try = 0
                x.status = 'online'
              }
            })


            var difference = lstOnline.filter((x: any) => !lstOnlineAux.map((e: any) => e.id).includes(x.id));


            if (difference.length > 0) {

              difference.forEach((e) => {

                e.try += 1

                if (e.try > 1) {
                  e.status = 'waiting...'
                }

                if (e.try > 3) {

                  const index = lstOnline.findIndex((x: any) => x.id == e.id)

                  if (index != -1) {

                    lstOnline.splice(index, 1)

                  }
                }

              })


            }
            obj.listOnline = lstOnline
            obj.listOnlineAux = lstOnlineAux


            lstOnlineAux = []
            listOnlineAux = []

            countOnline = -1
          }


          countOnline++
        }

      }

      exports.ob = obj


    });

    socket.on("joinChatBox", (ticketId: string) => {
      logger.info("A client joined a ticket channel");
      socket.join(ticketId);
    });

    socket.on("joinNotification", () => {
      logger.info("A client joined notification channel");
      socket.join("notification");
    });

    socket.on("joinTickets", (status: string) => {
      logger.info(`A client joined to ${status} tickets channel.`);
      socket.join(status);
    });

    socket.on("disconnect", (data: any) => {
      logger.info(`Client disconnected socket: ${data}`);
    });

    socket.on("disconnecting", async () => {
      console.log('socket.rooms: ', socket.rooms); // the Set contains at least the socket ID

      let rooms = socket.rooms

      console.log('rooms: ', rooms,    ' | rooms.size: ', rooms.size)

      if(rooms && rooms.size==1) return
      if(rooms && rooms.size==2 && !([...rooms][1].startsWith('session_'))) return

      let whatsappIds: any = await Whatsapp.findAll({ attributes: ['id'], raw: true }) 

      if (whatsappIds && whatsappIds.length > 0) {

        whatsappIds = whatsappIds.map((e: any) => `${e.id}`)

        console.log('whatsappIds whatsappIds whatsappIds whatsappIds whatsappIds: ',whatsappIds) 

        if (rooms && rooms.size == 2 &&
          [...rooms][1].startsWith('session_') &&
          whatsappIds.includes([...rooms][1].replace('session_', ''))) {

          console.log([...rooms][1])

          let whatsappId = [...rooms][1].replace('session_', '')

          const whatsapp = await Whatsapp.findByPk(whatsappId, {})

          if (whatsapp) {

            await whatsapp.update({ status: 'OPENING' });

            io.emit("whatsappSession", {
              action: "update",
              session: whatsapp
            });

          }


        }

      }

    });
  });
  return io;
};

export const getIO = (): SocketIO => {
  if (!io) {
    throw new AppError("Socket IO not initialized");
  }
  return io;
};



function writeFileAsync(arg0: any, data: any, arg2: string) {
  throw new Error("Function not implemented.");
}

// exports.listOnlineUsers = listUserId
// exports.listUserId