Function createManageServer

src/cluster/common.ts:104–229

(
  port: string,
  metrics: () => MetricsInfo,
  restart: (id: number) => boolean,
  restartAll: () => void,
  recalcShards?: (shards: number[][]) => void,
)

}

export async function createManageServer(
  port: string,
  metrics: () => MetricsInfo,
  restart: (id: number) => boolean,
  restartAll: () => void,
  recalcShards?: (shards: number[][]) => void,
) {
  const database = await dbInit();
  const httpServer = createServer(async (req, res) => {
    if (req.method !== "GET") {
      res.statusCode = 405;
      return res.end("GET only");
    }
    if (!req.url) throw Error("URL not found");

    const reqUrl = new URL(req.url, `http://${req.headers.host}`);
    if (reqUrl.pathname === "/" || reqUrl.pathname === "/metrics") {
      res.setHeader("Content-Type", "text/plain; version=0.0.4; charset=utf-8");
      res.write(`# HELP esmbot_command_count Number of times a command has been run
# TYPE esmbot_command_count counter
`);
      if (database) {
        const counts = await database.getCounts(true);
        for (const [i, w] of counts.entries()) {
          res.write(`esmbot_command_count{command="${i}"} ${w}\n`);
        }
      }

      const info = metrics();

      res.write(`# HELP esmbot_servers Number of servers/guilds the bot is in
# TYPE esmbot_servers gauge
esmbot_servers ${info.servers}
# HELP esmbot_shards Number of shards the bot has
# TYPE esmbot_shards gauge
esmbot_shards ${info.shards.length}
# HELP esmbot_shard_ping Latency of each of the bot's shards
# TYPE esmbot_shard_ping gauge
`);

      for (const shard of info.shards) {
        if (shard.latency)
          res.write(`esmbot_shard_ping{shard="${shard.id}", status="${shard.status}"} ${shard.latency}\n`);
      }

      res.write(`# HELP esmbot_total_mem Total memory usage of the bot
# TYPE esmbot_total_mem gauge
esmbot_total_mem ${info.totalMem}
`);

      return res.end();
    }

    if (reqUrl.pathname === "/shard") {
      if (!reqUrl.searchParams.has("id")) {
        res.statusCode = 400;
        return res.end("400 Bad Request");
      }