DiscordTickets/src/index.js

/**
 * Discord Tickets
 * Copyright (C) 2021 Isaac Saunders
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 * 
 * @name @eartharoid/discord-tickets
 * @description An open-source Discord bot for ticket management
 * @copyright 2021 Isaac Saunders
 * @license GNU-GPLv3
 */

process.title = 'Discord Tickets';

const node_version = Number(process.versions.node.split('.')[0]);
if (node_version < 14)
	return console.log(`\x07Error: Discord Tickets does not work on Node v${node_version}. Please upgrade to v14 or above.`);

const leeks = require('leeks.js');
const fs = require('fs');
const { path } = require('./utils/fs');

const checkFile = (file, example) => {
	if (fs.existsSync(path(file))) return true;
	if (!fs.existsSync(path(example))) {
		console.log(`\x07Error: "${file}" not found, and unable to create it due to "${example}" being missing.`);
		return process.exit();
	}
	console.log(`Copying "${example}" to "${file}"...`);
	fs.copyFileSync(path(example), path(file));
	return false;
};

checkFile('./user/config.js', './user/example.config.js');

if (!checkFile('./.env', './example.env')) {
	console.log('Generating database encryption key...');

	const file = path('./.env');
	const crypto = require('crypto');

	let key = 'DB_ENCRYPTION_KEY=';
	let value = crypto
		.randomBytes(24)
		.toString('hex');

	let data = fs.readFileSync(file, {
		encoding: 'utf-8'
	});
	data = data.replace(key, key + value);

	fs.writeFileSync(file, data);

	console.log('Saved.');
	console.log(leeks.colours.yellow('Warning: do not lose your ENV file or encryption key; you will lose access to data in the database.'));
	console.log('\x07Please set your bot\'s "DISCORD_TOKEN" in "./.env".');

	process.exit();
}

require('dotenv').config({
	path: path('./.env')
});

require('./banner')();

const log = require('./logger');

const { version } = require('../package.json');
process.on('unhandledRejection', error => {
	log.notice('PLEASE INCLUDE THIS INFORMATION IF YOU ASK FOR HELP ABOUT THE FOLLOWING ERROR:');
	log.notice(`Discord Tickets v${version}, Node v${process.versions.node} on ${process.platform}`);
	log.warn('An error was not caught');
	if (error instanceof Error) log.warn(`Uncaught ${error.name}`);
	log.error(error);
});

const { selectPresence } = require('./utils/discord');
const Cryptr = require('cryptr');
const I18n = require('@eartharoid/i18n');
const CommandManager = require('./modules/commands/manager');
const PluginManager = require('./modules/plugins/manager');
const TicketManager = require('./modules/tickets/manager');

const fetch = require('node-fetch');

require('./modules/structures')(); // load extended structures before creating the client

const {
	Client,
	Intents
} = require('discord.js');

/**
 * The bot client
 * @extends {Client}
 */
class Bot extends Client {
	constructor() {
		super({
			partials: [
				'MESSAGE',
				'CHANNEL',
				'REACTION'
			],
			presence: selectPresence(),
			ws: {
				intents: Intents.NON_PRIVILEGED,
			}
		});
		
		(async () => {
			/** The global bot configuration */
			this.config = require('../user/config');

			/** A [leekslazylogger](https://logger.eartharoid.me) instance */
			this.log = log;

			/** A [Cryptr](https://www.npmjs.com/package/cryptr) instance */
			this.cryptr = new Cryptr(process.env.DB_ENCRYPTION_KEY);

			let locales = {};
			fs.readdirSync(path('./src/locales'))
				.filter(file => file.endsWith('.json'))
				.forEach(file => {
					let data = fs.readFileSync(path(`./src/locales/${file}`), {
						encoding: 'utf8'
					});
					let name = file.slice(0, file.length - 5);
					locales[name] = JSON.parse(data);
				});

			/** An [@eartharoid/i18n](https://github.com/eartharoid/i18n) instance */
			this.i18n = new I18n('en-GB', locales);

			/** A sequelize instance */
			this.db = await require('./database')(this), // this.db.models.Ticket...

			this.setMaxListeners(this.config.max_listeners); // set the max listeners for each event
	
			require('./updater')(this); // check for updates
			require('./modules/listeners')(this); // load internal listeners

			/** The ticket manager */
			this.tickets = new TicketManager(this);

			/** The command manager, used by internal and plugin commands */
			this.commands = new CommandManager(this);

			/** The plugin manager */
			this.plugins = new PluginManager(this);
			this.plugins.load(); // load plugins

			this.log.info('Connecting to Discord API...');

			this.login();
		})();
	}

	async postStats() {
		/**
		 * OH NO, TELEMETRY!?
		 * Relax, it just counts how many people are using Discord Tickets.
		 * You can see the source here: https://github.com/discord-tickets/stats
		 */
		if (this.config.super_secret_setting) { // you can disable it if you really want
			let tickets = await this.db.models.Ticket.count();
			await fetch(`https://stats.discordtickets.app/client?id=${this.user.id}&tickets=${tickets}`, {
				method: 'post',
			}).catch(e => {
				this.log.warn('Failed to post tickets count to stats server (you can disable sending stats by setting "super_secret_setting" to false)');
				this.log.debug(e);
			});
			this.guilds.cache.forEach(async g => {
				let members = (await g.fetch()).approximateMemberCount;
				await fetch(`https://stats.discordtickets.app/guild?id=${g.id}&members=${members}`, {
					method: 'post',
				}).catch(e => {
					// don't spam a warning for each server
					this.log.debug(e);
				});
			});
		}
	}

}

new Bot();