gittoken-socket/src/server/utils/handleMsg.js

GitToken Socket Server and Client Libraries

import WebSocket from 'ws'
import split from 'split'

export default function handleMsg({ socket, message }) {
  const { type, data } = JSON.parse(message)
  switch(type.toUpperCase()) {
    case 'WATCH_TOKEN':
      const { organization } = data
      const orgData = this.store.getState()['organizations'][organization]

      // Attempt to send cached data before watching the token
      if(orgData != null) {
        socket.send(JSON.stringify({
          type: 'ORGANIZATION_DATA',
          org: organization,
          data: orgData
        }))
      } else {
        this.eventListener.write(message)
      }

      this.eventListener.pipe(split(JSON.parse)).on('data', (msg) => {
        try {
          if (
            organization == msg['data']['organization'] &&
            socket.readyState === WebSocket.OPEN
          ) {
            socket.send(JSON.stringify({
              type: 'WATCH_TOKEN',
              event: msg['event'],
              org: msg['data']['organization'],
              id: msg['data']['transactionHash'],
              data: msg
            }))
          }
        } catch(error) {
          console.error(error)
        }
      })

      break;
    case 'GET_REGISTERED':
      this.proxyQuery({
        queryString: this.queryString[type],
        type,
        socket
      })
      break;
    default:
      socket.send(JSON.stringify({
        type: 'error',
        result: `${event} is an invalid event.`
      }))
      return null
  }
}