@budibase/server/src/websockets/index.ts

Budibase Web Server

import http from "http"
import Koa from "koa"
import ClientAppSocket from "./client"
import GridSocket from "./grid"
import BuilderSocket from "./builder"
import env from "../environment"

let clientAppSocket: ClientAppSocket | undefined
let gridSocket: GridSocket | undefined
let builderSocket: BuilderSocket | undefined

export const initialise = (app: Koa, server: http.Server) => {
  // have to remove these for testing until ioredis-mock can be fully removed
  if (!env.isTest()) {
    clientAppSocket = new ClientAppSocket(app, server)
    gridSocket = new GridSocket(app, server)
    builderSocket = new BuilderSocket(app, server)
  }
}

export { clientAppSocket, gridSocket, builderSocket }