mrmBot-Matrix/events/ready.js

const gm = require("gm");
const cron = require("cron");
const { promisify } = require("util");
const client = require("../utils/client.js");
const database = require("../utils/database.js");
const collections = require("../utils/collections.js");
const logger = require("../utils/logger.js");
const messages = require("../messages.json");
const misc = require("../utils/misc.js");
const soundPlayer = require("../utils/soundplayer.js");
const helpGenerator =
  process.env.OUTPUT !== "" ? require("../utils/help.js") : null;
const twitter =
  process.env.TWITTER === "true" ? require("../utils/twitter.js") : null;

// run when ready
module.exports = async () => {
  // add gm extensions
  gm.prototype.writePromise = promisify(gm.prototype.write);
  gm.prototype.streamPromise = promisify(gm.prototype.stream);
  gm.prototype.sizePromise = promisify(gm.prototype.size);
  gm.prototype.identifyPromise = promisify(gm.prototype.identify);
  gm.prototype.bufferPromise = function(format, delay, type) {
    return new Promise((resolve, reject) => {
      this.in(
        delay ? "-delay" : "",
        delay ? delay.split("/").reverse().join("x") : ""
      )
        .out(
          type !== "sonic" ? "-layers" : "",
          type !== "sonic" ? "OptimizeTransparency" : ""
        )
        .out(type !== "sonic" ? "-fuzz" : "", type !== "sonic" ? "2%" : "")
        .out("+profile", "xmp")
        .out("-limit", "memory", "64MB")
        .out("-limit", "map", "128MB")
        .stream(format, (err, stdout, stderr) => {
          if (err) return reject(err);
          const chunks = [];
          stdout.on("data", (chunk) => {
            chunks.push(chunk);
          });
          // these are 'once' because they can and do fire multiple times for multiple errors,
          // but this is a promise so you'll have to deal with them one at a time
          stdout.once("end", () => {
            resolve(Buffer.concat(chunks));
          });
          stderr.once("data", (data) => {
            reject(data.toString());
          });
        });
    });
  };

  // connect to lavalink
  if (!soundPlayer.status) await soundPlayer.connect();

  // make sure settings/tags exist
  for (const [id] of client.guilds) {
    const guildDB = (
      await database.guilds
        .findOne({
          id: id,
        })
        .exec()
    );
    if (!guildDB) {
      logger.log(`Registering guild database entry for guild ${id}...`);
      const newGuild = new database.guilds({
        id: id,
        tags: misc.tagDefaults,
        prefix: "&",
        warns: {},
        disabledChannels: []
      });
      await newGuild.save();
    } else if (guildDB) {
      if (!guildDB.warns) {
        logger.log(`Creating warn object for guild ${id}...`);
        guildDB.set("warns", {});
        await guildDB.save();
      } else if (!guildDB.disabledChannels) {
        logger.log(`Creating disabled channels object for guild ${id}...`);
        guildDB.set("disabledChannels", []);
        await guildDB.save();
      }
    }
  }

  const job = new cron.CronJob("0 0 * * 0", async () => {
    logger.log("Deleting stale guild entries in database...");
    const guildDB = (await database.guilds.find({}).exec());
    for (const { id } of guildDB) {
      if (!client.guilds.get(id)) {
        await database.guilds.deleteMany({ id: id });
        logger.log(`Deleted entry for guild ID ${id}.`);
      }
    }
    logger.log("Finished deleting stale entries.");
  });
  job.start();

  const global = (await database.global.findOne({}).exec());
  if (!global) {
    const countObject = {};
    for (const command of collections.commands.keys()) {
      countObject[command] = 0;
    }
    const newGlobal = new database.global({
      cmdCounts: countObject
    });
    await newGlobal.save();
  } else {
    for (const command of collections.commands.keys()) {
      if (!global.cmdCounts.has(command)) {
        global.cmdCounts.set(command, 0);
        await global.save();
      }
    }
  }

  // generate docs
  if (helpGenerator) await helpGenerator(process.env.OUTPUT);

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

  // tweet stuff
  if (twitter !== null && twitter.active === false) {
    const blocks = await twitter.client.blocks.ids();
    const tweet = async () => {
      const tweets = (
        await database.tweets
          .find({
            enabled: true,
          })
          .exec()
      )[0];
      const tweetContent = await misc.getTweet(tweets);
      try {
        const info = await twitter.client.statuses.update(tweetContent);
        logger.log(`Tweet with id ${info.id_str} has been posted.`);
        // with status code ${info.resp.statusCode} ${info.resp.statusMessage}
      } catch (e) {
        const error = JSON.stringify(e);
        if (error.includes("Status is a duplicate.")) {
          logger.log("Duplicate tweet, will retry in 30 minutes");
        } else {
          logger.error(e);
        }
      }
    };
    tweet();
    setInterval(tweet, 1800000);
    twitter.active = true;
    try {
      const stream = twitter.client.statuses.filter(`@${process.env.HANDLE}`);
      stream.on("data", async (tweet) => {
        if (
          tweet.user.screen_name !== "esmBot_" &&
        !blocks.ids.includes(tweet.user.id_str)
        ) {
          const tweets = (
            await database.tweets
              .find({
                enabled: true,
              })
              .exec()
          )[0];
          let tweetContent;
          if (
            tweet.text.includes("@this_vid") ||
          tweet.text.includes("@DownloaderBot") ||
          tweet.text.includes("@GetVideoBot") ||
          tweet.text.includes("@DownloaderB0t") ||
          tweet.text.includes("@thisvid_")
          ) {
            tweetContent = await misc.getTweet(tweet, true, true);
          } else {
            tweetContent = await misc.getTweet(tweets, true).replace(/{{user}}/gm, `@${tweet.user.screen_name}`);
          }
          const payload = {
            status: `@${tweet.user.screen_name} ${tweetContent}`,
            in_reply_to_status_id: tweet.id_str,
          };
          const info = await twitter.client.statuses.update(payload);
          logger.log(`Reply with id ${info.id_str} has been posted.`);
        // with status code ${info.resp.statusCode} ${info.resp.statusMessage}
        }
      });
    } catch (e) {
      logger.error(`The Twitter streaming API ran into an error: ${e}`);
    }
  }

  logger.log(
    "info",
    `Successfully started ${client.user.username}#${client.user.discriminator} with ${client.users.size} users in ${client.guilds.size} servers.`
  );
};