management.js

import util from 'node:util'
import child_process from 'node:child_process'
const exec = util.promisify(child_process.exec)

const containerPrefix = process.env.CONTAINER_PREFIX ?? 'ss13-'
const serversRoot = process.env.SERVERS_ROOT ?? '/servers'

const restarting = new Set()

function getServerPath(server) {
  return `${serversRoot}/${server}/game`
}

function serverToContainer(server) {
  return `${containerPrefix}${server}`
}

async function getServers() {
  const { stdout: containers } = await exec(`docker container ls --format='{{.Names}}'`)
  return containers
    .trim()
    .split('\n')
    .filter((i) => i.startsWith(containerPrefix))
    .map((i) => i.replace(containerPrefix, ''))
}

async function getStateStatus(server) {
  const { stdout } = await exec(
    `docker inspect -f '{{.State.Status}}' ${serverToContainer(server)}`
  )
  return stdout.trim()
}

async function getStateHealth(server) {
  const { stdout } = await exec(
    `docker inspect -f '{{.State.Health.Status}}' ${serverToContainer(server)}`
  )
  return stdout.trim()
}

async function getStateStartedAt(server) {
  const { stdout } = await exec(
    `docker inspect -f '{{.State.StartedAt}}' ${serverToContainer(server)}`
  )
  return stdout.trim()
}

export async function getServerStatus(server, servers = null) {
  if (!servers) {
    servers = await getServers()
    if (!servers.includes(server)) throw new Error('Server not found')
  }

  return {
    restarting: restarting.has(server),
    status: await getStateStatus(server),
    health: await getStateHealth(server),
    startedAt: await getStateStartedAt(server),
  }
}

export async function getAllServersStatus() {
  const servers = await getServers()
  const ret = {}
  for (const server of servers) {
    ret[server] = await getServerStatus(server, servers)
  }
  return ret
}

export async function restartServer(server) {
  if (restarting.has(server)) throw new Error('Already restarting')
  const servers = await getServers()
  if (!servers.includes(server)) throw new Error('Server not found')
  const health = await getStateHealth(server)
  if (health === 'starting') throw new Error('Currently starting')
  restarting.add(server)
  const container = serverToContainer(server)
  const projectDir = getServerPath(server) + '/tools/server'
  child_process.exec(
    [
      `docker exec ${container} pkill -USR2 DreamDaemon`,
      `cd "${projectDir}"`,
      `./dc down`,
      `./dc up -d`,
    ].join(' && '),
    () => {
      restarting.delete(server)
    }
  )
}