doneit-web/src/app/services/chat/ws-chat.service.ts

import { Injectable } from '@angular/core';
import { v4 as uuidv4 } from 'uuid'
import { wsCallbacksParams, msgQueue, send } from 'src/app/models/rochet-chat-cliente-service'
import { deepFind } from 'src/plugin/deep'
import { environment } from 'src/environments/environment';
import { SessionStore } from 'src/app/store/session.service';
import { chatHistory, Rooms } from 'src/app/models/chatMethod';

@Injectable({
  providedIn: 'root'
})
export class WsChatService {

  isLogin = false;
  loginResponse = {}

  constructor() {}

  connect() {
    // dont connect if is already connected
    if(this.ws.connected == true) {
      return false
    }

    this.ws.connect();

    const message = {
      msg: "connect",
      version: "1",
      support: ["1"]
    }

    this.ws.send({message, loginRequired: false, requestId: 'connectMessage'})
    this.ws.send({message:{msg:"pong"}, loginRequired: false, requestId: 'connectPong'})

    this.ws.registerCallback({
      type:'Onmessage',
       funx:(message: any) => {
        if(message.msg == "ping") {
          this.ws.send({message:{msg:"pong"}, loginRequired: false})
        }
      }
    })

  }

  login() {
    // dont login if is already login
    if(this.isLogin == true) {
      return new Promise((resolve, reject)=>{
        resolve(this.loginResponse)
      })
    }

    const requestId =  uuidv4()

    const message = {
      msg: "method",
      method: "login",
      id: requestId,
      params: [
        {
          user: { username: SessionStore.user.RochetChatUser },
          password: SessionStore.user.Password
        }
      ]
    }
    this.ws.send({message, requestId: 'login', loginRequired: false})


    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId ) { // same request send

          if(message.result) {
            if(message.result.token) {

              this.isLogin = true
              this.loginResponse = message

              this.ws.wsMsgQueue()

              resolve(message)
            }
          } else {
            this.isLogin = false
            reject(message)
          }

          reject(message)

          return true
        }
      }})
    });

  }

  getRooms(roomOlder = 1480377601) {

    const requestId =  uuidv4()

    const message = {
      "msg": "method",
      "method": "rooms/get",
      "id": requestId,
      "params": [ { "$date": 1480377601 } ]
    }

    this.ws.send({message, requestId})

    return new Promise<Rooms>((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  getUserOfRoom(roomId) {

    const requestId =  uuidv4()

    const message = {
      "msg": "method",
      "method": "getUsersOfRoom",
      "id": requestId,
      "params": [
        roomId,
        false,
        {
          "limit": 100,
          "skip": 0
        } ]
    }

    this.ws.send({message, requestId})

    return new Promise<Rooms>((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  logout() {
    this.isLogin = false
    this.ws.connected = false
  }

  // send message to room
  send(roomId, msg) {

    const requestId =  uuidv4()

    var message = {
      msg: "method",
      method: "sendMessage",
      id: requestId,
      params: [{
        _id: uuidv4(),
        rid: roomId,
        msg: msg
      }]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  leaveRoom(roomId) {

    const requestId =  uuidv4()

    var message = {
      msg: "method",
      method: "leaveRoom",
      id: requestId,
      params: [
        roomId,
      ]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  addRoomOwner(roomId, userId) {

    const requestId =  uuidv4()

    var message = {
      msg: "method",
      method: "addRoomOwner",
      id: requestId,
      params: [
        roomId,
        userId
      ]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  hidingRoom(roomId) {

    const requestId =  uuidv4()

    var message = {
      msg: "method",
      method: "hideRoom",
      id: requestId,
      params: [roomId]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  joinRoom(){}

  deleteMessage(msgId) {
    const requestId = uuidv4();

    var message = {
      msg: "method",
      method: "deleteMessage",
      id: requestId,
      params:[{"_id":msgId}]
    }

    this.ws.send({message, requestId});

    return new Promise ((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message) =>{
        if(message.id == requestId || deepFind(message, 'result') == requestId){
          resolve(message)
          return true
        }
      }})
    });
  }

  createPrivateRoom(groupName, username, customFields) {
    const requestId =  uuidv4()

    var message = {
      msg: "method",
      method: "createPrivateGroup",
      id: requestId,
      params: [
        groupName,
        [username],
        false,
        customFields,
        {
          "broadcast":false,
          "encrypted":false
        }
      ]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  sendStreamNotifyRoom(roomId : string, username, event: 'typing', param: any) {

    const requestId =  uuidv4()

    let message = {
      msg: "method",
      method: "stream-notify-room",
      id: requestId,
      params: [
        `${roomId}/${event}`,
        username,
        param
      ]
    };

    this.ws.send({message, requestId})

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });

  }


  subStreamNotifyRoom(roomId : string , event: 'typing' | 'deleteMessage', param: any) {

    const requestId =  uuidv4()

    let message = {
      msg: "sub",
      id: requestId,
      name: "stream-notify-room",
      params:[
        `${roomId}/${event}`,
        param
      ]
    }


    this.ws.send({message, requestId})

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId ) { // same request send
          resolve(message)
          return true
        }
      }})
    });

  }

  receiveStreamNotifyRoom(funx: Function) {

    this.ws.registerCallback({
      type:'Onmessage',
      funx:(message)=> {
        if(message.collection == "stream-notify-room" && message.msg == 'changed') {
          funx(message)
        }
    }})
  }


  getUserStatus(funx:Function){
    this.ws.registerCallback({
      type:'Onmessage',
      funx:(message)=>{
        if(message.msg == 'changed' && message.collection == "stream-notify-logged") {
          funx(message)
        }
      }
    })
  }


  loadHistory(roomId, limit: number = 50) {

    const requestId =  uuidv4()

    const message = {
      msg: "method",
      method: "loadHistory",
      id: requestId,
      params: [
        roomId,
        null,
        limit,
        {
          "$date": 1480377601
        }
      ]
    }

    this.ws.send({message, requestId: 'loadHistory'})

    return new Promise<chatHistory>((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        // console.log(message)
        if(message.id == requestId ) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  setStatus(status: 'online' | 'busy' | 'away' | 'offline') {

    const requestId =  uuidv4()

    const message = {

      msg: "method",
      method: `UserPresence:setDefaultStatus`,
      id: requestId,
      params: [ status ]

    }

    this.ws.send({message, requestId})

  }

  subscribeNotifyRoom(roomId : string) {

    const requestId =  uuidv4()

    var message = {
      "msg": "sub",
      "id": requestId,
      "name": "stream-notify-room",
      "params":[
      `${roomId}/event`,
        false
      ]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.msg == 'ready' || deepFind(message, 'subs.0') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });

  }

updateRoomEventss(roomId, collection:string, funx: Function, ) {

    this.ws.registerCallback({
      type:'Onmessage',
      funx:(message)=>{
        //console.log(message);

        if(message.msg =='changed' && message.collection == collection) {
          if(message.fields.args[0].rid == roomId) {
            funx(message)
          }
          else if(message.fields.eventName ==`${roomId}/deleteMessage`){
            funx(message)
          }
        }
      }
    })
  }


  streamRoomMessages(roomId : string) {

    const requestId =  uuidv4()

    const message = {
      "msg": "sub",
      "id": requestId,
      "name": "stream-room-messages",
      "params":[
        `${roomId}`,
          false
      ]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  streamNotifyLogged() {

    const requestId =  uuidv4()

    const message = {
      "msg": "sub",
      "id": requestId,
      "name": "stream-notify-logged",
      "params":[
        "user-status",
          false
      ]
    }

    this.ws.send({message, requestId});

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });
  }

  streamNotifyRoom(roomId : string) {

    const requestId =  uuidv4()

    let message = {
      "msg": "method",
      "method": "stream-notify-room",
      "id": requestId,
      "params": [
        `null/typing`,
        "paulo.pinto",
        true
      ]
    };

    this.ws.send({message, requestId})

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });

  }

  streamNotifyRoomDeleteMessage(roomId:string) {

    const requestId =  uuidv4()

    let message = {
      "msg": "sub",
      "id": requestId,
      "name": "stream-notify-room",
      "params": [
        `${roomId}/deleteMessage`,
        true
      ]
    };

    this.ws.send({message, requestId})

    return new Promise((resolve, reject) => {
      this.ws.registerCallback({type:'Onmessage', funx:(message)=>{
        if(message.id == requestId || deepFind(message,'result.id') == requestId) { // same request send
          resolve(message)
          return true
        }
      }})
    });

  }

  registerCallback(data:wsCallbacksParams) {
    return this.ws.registerCallback(data)
  }

  // socket class ==================================================================
  private socket!: WebSocket;
  private wsMsgQueue : {[key: string]:  msgQueue} = {}
  private wsCallbacks: {[key: string]:  wsCallbacksParams} = {}
  private wsReconnect = 0

  private ws = {
    connected: false,
    registerCallback:(params: wsCallbacksParams) => {

      let id = params.requestId || params.key || uuidv4()
      this.wsCallbacks[id] = params

      return id
    },
    connect:()=> {
      this.socket = new WebSocket(environment.apiWsChatUrl);
      // bind function
      this.socket.onopen = this.ws.onopen;
      this.socket.onmessage = this.ws.onmessage;
      this.socket.onclose = this.ws.onclose;
      this.socket.onerror = this.ws.onerror;
    },
    onopen:()=> {
      this.ws.connected = true

      console.log('================== welcome to socket server =====================')

      this.ws.wsMsgQueue()

      if(this.wsReconnect >= 1) {
        for (const [key, value] of Object.entries(this.wsCallbacks)) {
          if(value.type== 'reConnect') {
            const dontRepeat = value.funx()

            if(dontRepeat) {
              delete this.wsCallbacks[key]
            }

          }

        }
      }

      this.wsReconnect++;
    },
    wsMsgQueue:()=> {

      for (const [key, item] of Object.entries(this.wsMsgQueue)) {

        if(item.loginRequired == true && this.isLogin == true) {
          console.log('run msgQueue ', key)
          this.ws.send(item);
          delete this.wsMsgQueue[key]
        } else if(item.loginRequired == false) {
          console.log('run msgQueue ', key)
          this.ws.send(item);
          delete this.wsMsgQueue[key]
        }
      }
    },
    send: ({message, requestId = uuidv4(), loginRequired = true}:send) => {

      if (this.ws.connected == false || loginRequired == true && this.isLogin == false) { // save data to send when back online
        // console.log('save msgQueue this.ws.connected == false || loginRequired == true && this.isLogin == false',this.ws.connected, loginRequired, this.isLogin)
        console.log('save msgQueue', requestId, message)

        this.wsMsgQueue[requestId] = {message, requestId, loginRequired}
      } else {
        let messageStr = JSON.stringify(message)
        this.socket.send(messageStr)
      }
      return requestId
    },

    onmessage:(event: any)=> {
      const data = JSON.parse(event.data)

      for (const [key, value] of Object.entries(this.wsCallbacks)) {
        if(value.type== 'Onmessage') {
          const dontRepeat = value.funx(data)

          if(dontRepeat) {
            delete this.wsCallbacks[key]
          }

        }

      }

    },

    onclose:(event: any)=> {
      this.ws.connected = false
      this.isLogin = false

      this.connect()
      this.login()

      console.log(`[close] Connection closed cleanly, code=${event.code} reason=${event.reason}`);
    },

    onerror: (event: any) =>  {
      console.log(`[error] ${event.message}`);
    }

  }}