app.ts

import { Server } from 'socket.io'
import { createServer } from "http"
import { GStatus } from './lib/types'

const port = parseInt(process.env.PORT || '3001')
const httpServer = createServer()
const io = new Server(httpServer, {
  cors: {
    origin: process.env.CLIENT_URL || '*',
  }
});

httpServer.listen(port)

console.log("Server established at port", port)

interface Bot {
  serverUrl: string;
  username: string;
  roomId: string;
  status: GStatus;
}

const maxWorkerNum = parseInt(process.env.MAX_WORKER_NUM || '0');
var workers = new Array<Worker>()
var bots = new Array<Bot>()

io.on('connection', (socket) => {
  socket.on('query', () => {
    socket.emit('botlist', workers)
  })

  socket.on('add_bot', (arg) => {
    if (maxWorkerNum && workers.length >= maxWorkerNum) {
      socket.emit('error', 'Number of players exceeded.')
      return
    }

    bots.push({...arg, status: 0})

    const worker = new Worker("./worker.ts")

    worker.onmessage = (ev) => {
      console.log("Received from bot", ev.data.id, "STATUS:", ev.data.status)
      let id = parseInt(ev.data.id)
      bots[id].status = ev.data.status
      socket.emit('botlist', bots)
    }

    worker.postMessage({ id: workers.length, ...arg })

    workers.push(worker)

    console.log("Added bot")

    socket.emit('add_success')
    socket.emit('botlist', bots)
  })

  socket.on('remove_bot', (arg) => {
    let id = parseInt(arg)
    console.log("Removing bot", id)

    if (Number.isNaN(id) || id < 0 || id >= workers.length) {
      socket.emit('error', 'Bot ID is invalid.')
      return
    }

    workers[id].terminate()
    workers = [...workers.filter((_, index) => index !== id)]
    bots = [...bots.filter((_, index) => index !== id)]

    socket.emit('botlist', bots)
  })
})