budibase/packages/server/src/websockets/grid.ts

import authorized from "../middleware/authorized"
import { BaseSocket } from "./websocket"
import { permissions } from "@budibase/backend-core"
import http from "http"
import Koa from "koa"
import { getTableId } from "../api/controllers/row/utils"
import { Row, Table } from "@budibase/types"
import { Socket } from "socket.io"
import { GridSocketEvents } from "@budibase/shared-core"

export default class GridSocket extends BaseSocket {
  constructor(app: Koa, server: http.Server) {
    super(app, server, "/socket/grid", [authorized(permissions.BUILDER)])
  }

  async onConnect(socket: Socket) {
    // Initial identification of connected spreadsheet
    socket.on(GridSocketEvents.SelectTable, async (tableId, callback) => {
      await this.joinRoom(socket, tableId)

      // Reply with all users in current roome
      const users = await this.getSocketUsers(tableId)
      callback({ users })
    })

    // Handle users selecting a new cell
    socket.on(GridSocketEvents.SelectCell, cellId => {
      this.updateUser(socket, { focusedCellId: cellId })
    })
  }

  emitRowUpdate(ctx: any, row: Row) {
    const tableId = getTableId(ctx)
    this.io.in(tableId).emit(GridSocketEvents.RowChange, { id: row._id, row })
  }

  emitRowDeletion(ctx: any, id: string) {
    const tableId = getTableId(ctx)
    this.io.in(tableId).emit(GridSocketEvents.RowChange, { id, row: null })
  }

  emitTableUpdate(table: Table) {
    this.io
      .in(table._id!)
      .emit(GridSocketEvents.TableChange, { id: table._id, table })
  }

  emitTableDeletion(id: string) {
    this.io.in(id).emit(GridSocketEvents.TableChange, { id, table: null })
  }
}