gittoken-socket/src/server/index.js

GitToken Socket Server and Client Libraries

import Promise from 'bluebird'
import WebSocket from 'ws'
import mysql from 'mysql'
import split from 'split'

import GitTokenEventWatcherClient  from 'gittoken-event-listener/dist/client/index'

import {
  handleMsg
} from './utils/index'

import {
  proxyQuery,
  queryString,
  query
} from './sql/index'

import {
  store
} from './redux/index'

export default class GitTokenSocketServer extends GitTokenEventWatcherClient {
  constructor({
    socketPort,
    mysqlHost,
    mysqlUser,
    mysqlRootPassword,
    mysqlDatabase,
    watcherIpcPath
  }) {
    super({ watcherIpcPath })
    this.query       = query.bind(this)
    this.handleMsg   = handleMsg.bind(this)
    this.proxyQuery  = proxyQuery.bind(this)

    this.queryString = queryString
    this.store = store

    // this.unsubscribe = this.store.subscribe(() => {
    //   console.log('store', JSON.stringify(this.store.getState(), null, 2))
    // })

    // Instantiate MySql Connection
    this.mysql = mysql.createConnection({
      host: mysqlHost,
      user: mysqlUser,
      password: mysqlRootPassword,
      database: mysqlDatabase,
    })

    this.server = new WebSocket.Server({ port: socketPort }, () => {
      console.log('GitToken Socket Server Started on Port: ', socketPort)

      // Listen for contract event listener messages;
      this.eventListener.pipe(split(JSON.parse)).on('data', (msg) => {
        try {
          this.store.dispatch({
            type: 'WATCH_TOKEN',
            event: msg['event'],
            org: msg['data']['organization'],
            id: msg['data']['transactionHash'],
            data: msg
          })
        } catch(error) {
          console.error(error)
          console.log('msg', msg)
        }

      })

      const unsubscribe = this.store.subscribe(() => {
        // Subscribe to state updates
        console.log('State: ', JSON.stringify(this.store.getState(), null, 2))
        // send updates back to
      })
    })

    this.server.on('connection', (socket, req) => {
      socket.on('message', (message) => {
        this.handleMsg({ socket, message })
      })
    })
  }

}