blitzkit/packages/bot/src/main.ts

import { assertSecret } from "@blitzkit/core";
import { ActivityType, ShardingManager } from "discord.js";
import {
  discoveredIdsDefinitions,
  tankDefinitions,
} from "./core/blitzkit/nonBlockingPromises";

let manager: ShardingManager;
let iteration = 0;
const interval = setInterval(async () => {
  console.log(`Attempting launch on iteration ${++iteration}`);

  manager?.respawnAll();
  manager = new ShardingManager("dist/bot/workers/bot.js", {
    token: assertSecret(import.meta.env.DISCORD_TOKEN),
  });

  const shards = await manager
    .on("shardCreate", (shard) => {
      clearInterval(interval);
      console.log(`🟡 Launching shard ${shard.id}`);
      shard.on("ready", () => console.log(`🟢 Launched shard ${shard.id}`));
    })
    .spawn({ timeout: 60000 });

  try {
    let servers = 0;
    let channels = 0;
    let users = 0;

    for (const [, shard] of shards) {
      const guilds = (await shard.fetchClientValue("guilds.cache")) as any;

      servers += guilds.length;

      for (const guild of guilds) {
        channels += guild.channels.length;
        users += guild.memberCount;
      }
    }

    const awaitedTankDefinitions = await tankDefinitions;
    const tanks = Object.values(awaitedTankDefinitions.tanks).length;
    const awaitedDiscoveredIdsDefinitions = await discoveredIdsDefinitions;
    const usernames = awaitedDiscoveredIdsDefinitions.count;

    manager.broadcastEval(
      async (client, { servers, users, channels, tanks, usernames }) => {
        const bios = [
          `Living in ${servers.toLocaleString()} servers`,
          `Serving ${users.toLocaleString()} users`,
          `Observing ${channels.toLocaleString()} channels`,
          `Tinkering with ${tanks.toLocaleString()} tanks`,
          `Memorizing ${usernames.toLocaleString()} usernames`,
        ];

        client.user?.setPresence({
          activities: [
            {
              type: ActivityType.Custom,
              name: bios[Math.floor(Math.random() * bios.length)],
            },
          ],
        });

        for (const guild of client.guilds.cache.values()) {
          try {
            const botMember = await guild.members.fetch(client.user.id);
            await botMember.setNickname("BlitzKit");
          } catch {}
        }
      },
      { context: { servers, channels, users, tanks, usernames } },
    );
  } catch (error) {
    console.warn(
      "An error occurred when setting up presence, ignoring to keep the process alive",
      error,
    );
  }
}, 15000);