mrmBot-Matrix/app.js

186 lines
6.3 KiB
JavaScript
Raw Normal View History

if (process.platform === "win32") console.error("\x1b[1m\x1b[31m\x1b[40m" + `WIN32 IS NOT OFFICIALLY SUPPORTED!
Although there's a (very) slim chance of it working, multiple aspects of the bot are built with UNIX-like systems in mind and could break on Win32-based systems. If you want to run the bot on Windows, using Windows Subsystem for Linux is highly recommended.
The bot will continue to run past this message, but keep in mind that it could break at any time. Continue running at your own risk; alternatively, stop the bot using Ctrl+C and install WSL.` + "\x1b[0m");
2022-01-07 17:44:18 +00:00
if (process.versions.node.split(".")[0] < 15) {
console.error(`You are currently running Node.js version ${process.version}.
esmBot requires Node.js version 15 or above.
Please refer to step 3 of the setup guide.`);
process.exit(1);
}
// load config from .env file
import { resolve, dirname } from "path";
import { fileURLToPath } from "url";
import { config } from "dotenv";
config({ path: resolve(dirname(fileURLToPath(import.meta.url)), ".env") });
2021-04-21 18:35:06 +00:00
// main sharding manager
import { Fleet } from "eris-fleet";
import { isMaster } from "cluster";
// main services
import Shard from "./shard.js";
import ImageWorker from "./utils/services/image.js";
import PrometheusWorker from "./utils/services/prometheus.js";
2021-07-05 13:14:52 +00:00
// some utils
2022-03-10 21:24:24 +00:00
import { promises, readFileSync } from "fs";
import winston from "winston";
import "winston-daily-rotate-file";
import { exec as baseExec } from "child_process";
import { promisify } from "util";
2022-03-10 21:24:24 +00:00
const exec = promisify(baseExec);
// database stuff
import database from "./utils/database.js";
2021-04-21 18:35:06 +00:00
// dbl posting
import { Api } from "@top-gg/sdk";
const dbl = process.env.NODE_ENV === "production" && process.env.DBL ? new Api(process.env.DBL) : null;
2021-07-05 13:14:52 +00:00
if (isMaster) {
const esmBotVersion = JSON.parse(readFileSync(new URL("./package.json", import.meta.url))).version;
const erisFleetVersion = JSON.parse(readFileSync(new URL("./node_modules/eris-fleet/package.json", import.meta.url))).version; // a bit of a hacky way to get the eris-fleet version
2021-07-05 13:14:52 +00:00
console.log(`
,*\`$ z\`"v
F zBw\`% A ,W "W
,\` ,EBBBWp"%. ,-=~~==-,+* 4BBE T
M BBBBBBBB* ,w=####Wpw 4BBBBB# 1
F BBBBBBBMwBBBBBBBBBBBBB#wXBBBBBH E
F BBBBBBkBBBBBBBBBBBBBBBBBBBBE4BL k
# BFBBBBBBBBBBBBF" "RBBBW F
V ' 4BBBBBBBBBBM TBBL F
F BBBBBBBBBBF JBB L
F FBBBBBBBEB BBL 4
E [BB4BBBBEBL BBL 4
I #BBBBBBBEB 4BBH *w
A 4BBBBBBBBBEW, ,BBBB W [
.A ,k 4BBBBBBBBBBBEBW####BBBBBBM BF F
k <BBBw BBBBEBBBBBBBBBBBBBBBBBQ4BM #
5, REBBB4BBBBB#BBBBBBBBBBBBP5BFF ,F
*w \`*4BBW\`"FF#F##FFFF"\` , * +"
*+, " F'"'*^~~~^"^\` V+*^
\`"""
2022-02-18 16:39:40 +00:00
esmBot ${esmBotVersion} (${(await exec("git rev-parse HEAD").then(output => output.stdout.substring(0, 7), () => "unknown commit"))}), powered by eris-fleet ${erisFleetVersion}
2021-07-05 13:14:52 +00:00
`);
}
2021-07-05 04:15:27 +00:00
const Admiral = new Fleet({
BotWorker: Shard,
2021-07-05 04:15:27 +00:00
token: `Bot ${process.env.TOKEN}`,
fetchTimeout: 900000,
maxConcurrencyOverride: 1,
2021-07-05 04:15:27 +00:00
startingStatus: {
status: "idle",
game: {
name: "Starting esmBot..."
}
},
whatToLog: {
blacklist: ["stats_update"]
},
clientOptions: {
allowedMentions: {
everyone: false,
roles: false,
users: true,
repliedUser: true
},
restMode: true,
messageLimit: 50,
intents: [
"guilds",
"guildVoiceStates",
"guildMessages",
2021-08-14 02:36:13 +00:00
"directMessages"
],
stats: {
requestTimeout: 30000
},
connectionTimeout: 30000
},
useCentralRequestHandler: process.env.DEBUG_LOG ? false : true, // workaround for eris-fleet weirdness
services: [
{ name: "prometheus", ServiceWorker: PrometheusWorker },
{ name: "image", ServiceWorker: ImageWorker }
]
2021-04-21 18:35:06 +00:00
});
2021-07-05 04:15:27 +00:00
if (isMaster) {
2021-07-05 13:14:52 +00:00
const logger = winston.createLogger({
2021-08-14 21:15:21 +00:00
levels: {
2021-07-05 13:14:52 +00:00
error: 0,
warn: 1,
info: 2,
main: 3,
debug: 4
},
transports: [
2021-08-09 15:54:56 +00:00
new winston.transports.Console({ format: winston.format.colorize({ all: true }), stderrLevels: ["error", "warn"] }),
new winston.transports.DailyRotateFile({ filename: "logs/error-%DATE%.log", level: "error", zippedArchive: true, maxSize: 4194304, maxFiles: 8 }),
new winston.transports.DailyRotateFile({ filename: "logs/main-%DATE%.log", zippedArchive: true, maxSize: 4194304, maxFiles: 8 })
2021-07-05 13:14:52 +00:00
],
level: process.env.DEBUG_LOG ? "debug" : "main",
2021-07-05 13:14:52 +00:00
format: winston.format.combine(
winston.format.timestamp({ format: "YYYY-MM-DD HH:mm:ss" }),
winston.format.printf((info) => {
const {
timestamp, level, message, ...args
} = info;
2021-08-14 21:15:21 +00:00
2021-07-05 13:14:52 +00:00
return `[${timestamp}]: [${level.toUpperCase()}] - ${message} ${Object.keys(args).length ? JSON.stringify(args, null, 2) : ""}`;
}),
)
});
2021-08-14 21:15:21 +00:00
2021-07-05 13:14:52 +00:00
winston.addColors({
info: "green",
main: "gray",
debug: "magenta",
2021-07-05 13:14:52 +00:00
warn: "yellow",
error: "red"
});
2022-02-22 15:20:52 +00:00
database.upgrade(logger).then(result => {
if (result === 1) return process.exit(1);
});
2021-07-05 13:14:52 +00:00
Admiral.on("log", (m) => logger.main(m));
Admiral.on("info", (m) => logger.info(m));
2021-07-05 13:14:52 +00:00
Admiral.on("debug", (m) => logger.debug(m));
Admiral.on("warn", (m) => logger.warn(m));
Admiral.on("error", (m) => logger.error(m));
2021-07-05 04:15:27 +00:00
2021-04-27 02:53:27 +00:00
if (dbl) {
2021-07-05 04:15:27 +00:00
Admiral.on("stats", async (m) => {
await dbl.postStats({
serverCount: m.guilds,
shardCount: m.shardCount
});
2021-04-27 02:53:27 +00:00
});
2021-04-21 18:35:06 +00:00
}
2022-03-10 21:24:24 +00:00
2022-06-11 19:30:04 +00:00
// process the threshold into bytes early
if (process.env.TEMPDIR && process.env.THRESHOLD) {
2022-03-10 21:24:24 +00:00
const matched = process.env.THRESHOLD.match(/(\d+)([KMGT])/);
const sizes = {
2022-06-11 19:30:04 +00:00
K: 1024,
M: 1048576,
G: 1073741824,
T: 1099511627776
2022-03-10 21:24:24 +00:00
};
2022-06-11 19:30:04 +00:00
if (matched && matched[1] && matched[2]) {
process.env.THRESHOLD = matched[1] * sizes[matched[2]];
2022-03-10 21:24:24 +00:00
} else {
logger.error("Invalid THRESHOLD config.");
process.env.THRESHOLD = undefined;
}
const dirstat = (await promises.readdir(process.env.TEMPDIR)).map((file) => {
return promises.stat(`${process.env.TEMPDIR}/${file}`).then((stats) => stats.size);
2022-06-11 19:30:04 +00:00
});
const size = await Promise.all(dirstat);
const reduced = size.reduce((a, b) => {
2022-06-11 19:30:04 +00:00
return a + b;
}, 0);
Admiral.centralStore.set("dirSizeCache", reduced);
2022-03-10 21:24:24 +00:00
}
}