const { RegisterGlobalCommands, RegisterGuildCommands} = require("../util/RegisterSlashCommands"); const { Collection, Client, EmbedBuilder, Routes } = require('discord.js'); const MongoClient = require('mongodb').MongoClient; const { REST } = require('@discordjs/rest'); const mongoose = require('mongoose'); const Logger = require("../util/Logger"); const path = require("path"); const fs = require('fs'); const { exit } = require("process"); class QuarksBot extends Client { constructor(options, config) { super(options) this.config = config; this.commands = new Collection(); this.logger = new Logger(path.join(__dirname, "..", "logs/Logs.log")); if (this.config.Token === "") throw new TypeError( "The config.js is not filled out. Please make sure nothing is blank, otherwise the bot will not work properly." ); this.db; this.dbo; this.connectMongo(this.config.mongoURI, this.config.dbo); this.LoadCommands(); this.LoadEvents(); this.Ready = false; this.ws.on("INTERACTION_CREATE", async (interaction) => { const start = new Date().getTime(); if (interaction.type!=3) { let GuildDB = await this.GetGuild(interaction.guild_id); const command = interaction.data.name.toLowerCase(); const args = interaction.data.options; client.log(`Interaction - ${command}`); //Easy to send respnose so ;) interaction.guild = await this.guilds.fetch(interaction.guild_id); interaction.send = async (message) => { const rest = new REST({ version: '10' }).setToken(client.config.Token); return await rest.post(Routes.interactionCallback(interaction.id, interaction.token), { body: { type: 4, data: message, } }); }; let cmd = client.commands.get(command); try { cmd.SlashCommand.run(this, interaction, args, { GuildDB }, start); // start is only used in ping / stats command } catch (err) { client.error(err) const embed = new EmbedBuilder() .setDescription(`**Internal Error:**\nUh Oh D: Its not you, its me.\nThis command has crashed\nContact the Developers\nhttps://discord.gg/YCXhvy9uZw`) .setColor(client.config.Colors.Red) return interaction.send({ embeds: [embed] }); } } }); const client = this; } async connectMongo(mongoURI, dbo) { try { // Connect to MongoDB database. this.db = await MongoClient.connect(mongoURI, {connectTimeoutMS: 5000}); this.dbo = this.db.db(dbo); mongoose.connect(`${mongoURI}/${dbo}`); this.log('Successfully connected to mongoDB'); } catch (err) { this.error('Failed to connect to mongodb'); process.exit(-1); } } exists(n) {return null != n && undefined != n && "" != n} secondsToDhms(seconds) { seconds = Number(seconds); const d = Math.floor(seconds / (3600*24)); const h = Math.floor(seconds % (3600*24) / 3600); const m = Math.floor(seconds % 3600 / 60); const s = Math.floor(seconds % 60); const dDisplay = d > 0 ? d + (d == 1 ? " day, " : " days, ") : ""; const hDisplay = h > 0 ? h + (h == 1 ? " hour, " : " hours, ") : ""; const mDisplay = m > 0 ? m + (m == 1 ? " minute, " : " minutes, ") : ""; const sDisplay = s > 0 ? s + (s == 1 ? " second" : " seconds") : ""; return dDisplay + hDisplay + mDisplay + sDisplay; } LoadCommands() { let CommandsDir = path.join(__dirname, '..', 'commands'); fs.readdir(CommandsDir, (err, files) => { if (err) this.error(err); else files.forEach((file) => { let cmd = require(CommandsDir + "/" + file); if (!this.exists(cmd.name) || !this.exists(cmd.description)) return this.error( "Unable to load Command: " + file.split(".")[0] + ", Reason: File doesn't had name/desciption" ); this.commands.set(file.split(".")[0].toLowerCase(), cmd); this.log("Command Loaded: " + file.split(".")[0]); }); }); } LoadEvents() { let EventsDir = path.join(__dirname, '..', 'events'); fs.readdir(EventsDir, (err, files) => { if (err) this.error(err); else files.forEach((file) => { const event = require(EventsDir + "/" + file);; if (file.split(".")[0] == 'interactionCreate') this.on(file.split(".")[0], i => event(this, i)); else this.on(file.split(".")[0], event.bind(null, this)); this.logger.log("Event Loaded: " + file.split(".")[0]); }); }); } sendError(Channel, Error) { let embed = new EmbedBuilder() .setColor(this.config.Red) .setDescription(Error); Channel.send(embed); } sendInternalError(Interaction, Error) { this.error(Error); const embed = new EmbedBuilder() .setDescription(`**Internal Error:**\nUh Oh D: Its not you, its me.\nThis command has crashed\nContact the Developers\nhttps://discord.gg/YCXhvy9uZw`) .setColor(client.config.Colors.Red) Interaction.send({ embeds: [embed] }); } // Calls register for guild and global commands RegisterSlashCommands() { RegisterGlobalCommands(this); this.guilds.cache.forEach((guild) => {RegisterGuildCommands(this, guild.id)}); } async GetGuild(GuildId) { let guild = await this.dbo.collection("guilds").findOne({"server.serverID":GuildId}).then(guild => guild); // If guild not found, generate guild default if (!guild) { guild = { server: { serverID: GuildId, allowedChannels: [], actionChannel: null, tweetChannel: null, advertsChannel: null, dispatchChannel: null, startingBalance: 1000.00, incomeRoles: [], botAdmin: null, commercialRole: null, officerRole: null, dispatcherRole: null, onlyOfficersSearch: false, casinoMultiplier: 2, } } this.dbo.collection("guilds").insertOne(guild, function(err, res) { if (err) throw err; }); } let guildData = { serverID: GuildId, customChannelStatus: guild.server.allowedChannels.length > 0 ? true : false, allowedChannels: guild.server.allowedChannels, actionChannel: guild.server.actionChannel, tweetChannel: guild.server.tweetChannel, advertsChannel: guild.server.advertsChannel, dispatchChannel: guild.server.dispatchChannel, startingBalance: guild.server.startingBalance, incomeRoleStatus: guild.server.incomeRoles.length > 0 ? true : false, incomeRoles: guild.server.incomeRoles, commercialRole: guild.server.commercialRole, officerRole: guild.server.officerRole, dispatcherRole: guild.server.dispatcherRole, onlyOfficersSearch: guild.server.onlyOfficersSearch, botAdmin: guild.server.botAdmin, casinoMultiplier: guild.server.casinoMultiplier, } return guildData; } log(Text) { this.logger.log(Text); } error(Text) { this.logger.error(Text); } build() { this.login(this.config.Token); } } module.exports = QuarksBot;