mrmBot-Matrix/shard.js

// shard base
const { Base } = require("eris-sharder");
// path stuff
const { readdir } = require("fs").promises;
// fancy loggings
const logger = require("./utils/logger.js");
// initialize command loader
const handler = require("./utils/handler.js");
// lavalink stuff
const sound = require("./utils/soundplayer.js");
// image processing stuff
const image = require("./utils/image.js");
// database stuff
const database = require("./utils/database.js");
// command collections
const collections = require("./utils/collections.js");
// playing messages
const { messages } = require("./messages.json");
// other stuff
const misc = require("./utils/misc.js");
// generate help page
const helpGenerator =
  process.env.OUTPUT !== "" ? require("./utils/help.js") : null;
const http = require("http");

class Shard extends Base {
  constructor(bot) {
    super(bot);
  }

  async init() {
    // register commands and their info
    const soundStatus = await sound.checkStatus();
    logger.log("info", "Attempting to load commands...");
    for await (const commandFile of this.getFiles("./commands/")) {
      logger.log("log", `Loading command from ${commandFile}...`);
      try {
        await handler.load(commandFile, soundStatus);
      } catch (e) {
        logger.error(`Failed to register command from ${commandFile}: ${e}`);
      }
    }

    // register events
    const events = await readdir("./events/");
    logger.log("info", `Attempting to load ${events.length} events...`);
    for (const file of events) {
      logger.log("log", `Loading event from ${file}...`);
      const eventName = file.split(".")[0];
      const event = require(`./events/${file}`);
      this.bot.on(eventName, event.bind(null, this.bot, this.clusterID, this.ipc));
    }

    // connect to image api if enabled
    if (process.env.API === "true") {
      for (const server of image.servers) {
        try {
          await image.connect(server);
        } catch (e) {
          logger.error(e);
        }
      }
    }

    // generate docs
    if (helpGenerator) {
      await helpGenerator.generateList();
      if (this.clusterID === 0 && helpGenerator) {
        await helpGenerator.createPage(process.env.OUTPUT);
        logger.log("info", "The help docs have been generated.");
      }
    }

    if (process.env.METRICS !== "" && process.env.METRICS !== undefined) {
      const httpServer = http.createServer(async (req, res) => {
        if (req.method !== "GET") {
          res.statusCode = 405;
          return res.end("GET only");
        }
        res.write(`# HELP connected_workers Number of workers connected
# TYPE connected_workers gauge
connected_workers ${image.connections.size}
# HELP running_jobs Number of running jobs on this worker
# TYPE running_jobs gauge
# HELP queued_jobs Number of queued jobs on this worker
# TYPE queued_jobs gauge
# HELP max_jobs Number of max allowed jobs on this worker
# TYPE max_jobs gauge
# HELP command_count Number of times a command has been run
# TYPE command_count counter
`);
        const servers = await image.getRunning();
        for (const [i, w] of servers.entries()) {
          res.write(`running_jobs{worker="${i}"} ${w.runningJobs}\n`);
          res.write(`queued_jobs{worker="${i}"} ${w.queued}\n`);
          res.write(`max_jobs{worker="${i}"} ${w.max}\n`);
        }
        const counts = await database.getCounts();
        for (const [i, w] of Object.entries(counts)) {
          res.write(`command_count{command="${i}"} ${w}\n`);
        }
        res.end();
      });
      httpServer.listen(process.env.METRICS, () => {
        logger.log("info", `Serving metrics at ${process.env.METRICS}`);
      });
    }

    // handle process stop
    process.on("SIGINT", () => {
      logger.log("warn", "SIGINT detected, shutting down...");
      this.bot.editStatus("dnd", {
        name: "Restarting/shutting down..."
      });
      for (const command in collections.commands) {
        handler.unload(command);
      }
      this.bot.disconnect();
      require("./utils/database.js").stop();
      process.exit(0);
    });
    return;
  }

  async* getFiles(dir) {
    const dirents = await readdir(dir, { withFileTypes: true });
    for (const dirent of dirents) {
      if (dirent.isDirectory()) {
        yield* this.getFiles(dir + dirent.name);
      } else {
        yield dir + (dir.charAt(dir.length - 1) !== "/" ? "/" : "") + dirent.name;
      }
    }
  }

  async launch() {
    await this.init();

    this.ipc.register("stat", (message) => {
      collections.stats = message;
    });

    this.ipc.register("restart", async () => {
      await this.bot.editStatus("dnd", {
        name: "esmBot is restarting, please stand by."
      });
      process.exit(1);
    });

    this.ipc.register("reload", async (message) => {
      const result = await handler.unload(message.cmd);
      if (result) return this.ipc.broadcast("reloadFail", { result: result });
      const result2 = await handler.load(collections.paths.get(message.cmd));
      if (result2) return this.ipc.broadcast("reloadFail", { result: result2 });
      return this.ipc.broadcast("reloadSuccess");
    });

    this.ipc.register("soundreload", async () => {
      const soundStatus = await sound.checkStatus();
      if (!soundStatus) {
        const length = await sound.connect(this.bot);
        return this.ipc.broadcast("soundReloadSuccess", { length });
      } else {
        return this.ipc.broadcast("soundReloadFail");
      }
    });
    
    // connect to lavalink
    if (!sound.status && !sound.connected) await sound.connect(this.bot);

    this.bot.privateChannels.limit = 0;

    await database.setup();

    // set activity (a.k.a. the gamer code)
    (async function activityChanger() {
      this.bot.editStatus("dnd", {
        name: `${misc.random(messages)} | @${this.bot.user.username} help`,
      });
      setTimeout(activityChanger.bind(this), 900000);
    }).bind(this)();

    logger.log("info", `Started cluster ${this.clusterID}.`);
  }

}

module.exports = Shard;