@stadium-ws/core/src/Connection.ts

Build scalable realtime applications.

import WebSocket from 'isomorphic-ws'

interface ConnectionConfig {
  baseUrl: string
}

interface Connect {
  onEvent: (event: SocketMessage) => void
  token: string
}

export enum MessageType {
  EVENT_CREATE = 0,
  EVENT_DELETE = 1,
  EVENT_UPDATE = 2,
  EVENT_REACTION_CREATE = 3,
  EVENT_REACTION_DELETE = 4,
  CHANNEL_UPDATE = 5,
  CHANNEL_USER_ADDED = 6
}

export interface SocketMessage {
  type: MessageType
  data: any
}

class Connection {
  private baseUrl: string
  private webSocket: WebSocket
  private isConnected: boolean = false

  constructor (config: ConnectionConfig) {
    this.baseUrl = config.baseUrl
  }

  public connect = ({ onEvent, token }: Connect): Promise<void> => {
    return new Promise((resolve, reject) => {
      this.webSocket = new WebSocket(`${this.baseUrl}?token=${token}`)

      this.webSocket.addEventListener('error', () => {
        reject(new Error('nope'))

        // TODO: make me
      })

      this.webSocket.addEventListener('message', (e: any) => {
        const data = JSON.parse(e.data) as SocketMessage
        onEvent(data)
      })

      this.webSocket.addEventListener('open', () => {
        this.isConnected = true
        resolve()
      })

      this.webSocket.addEventListener('close', () => {
        this.isConnected = false

        // TODO: make me
      })
    })
  }

  public disconnect = async (): Promise<void> => {
    if (!this.isConnected) {
      return
    }

    // TODO: perhaps add more params, as they are supported...
    this.webSocket.close()
  }
}

export default Connection