diff options
Diffstat (limited to 'src/lib/extensions/BushClient.ts')
| -rw-r--r-- | src/lib/extensions/BushClient.ts | 27 |
1 files changed, 21 insertions, 6 deletions
diff --git a/src/lib/extensions/BushClient.ts b/src/lib/extensions/BushClient.ts index 5c7729c..8592281 100644 --- a/src/lib/extensions/BushClient.ts +++ b/src/lib/extensions/BushClient.ts @@ -3,6 +3,7 @@ import { AkairoClient, TaskHandler } from 'discord-akairo'; import { APIMessage, Guild, Intents, Message, MessageOptions, Snowflake, UserResolvable } from 'discord.js'; import * as path from 'path'; import { exit } from 'process'; +import readline from 'readline'; import { Sequelize } from 'sequelize'; import * as config from '../../config/options'; import * as Models from '../models'; @@ -17,6 +18,12 @@ import { BushListenerHandler } from './BushListenerHandler'; export type BotConfig = typeof config; export type BushMessageType = string | APIMessage | (MessageOptions & { split?: false }); +const rl = readline.createInterface({ + input: process.stdin, + output: process.stdout, + terminal: false +}); + export class BushClient extends AkairoClient { public config: BotConfig; public listenerHandler: BushListenerHandler; @@ -66,9 +73,9 @@ export class BushClient extends AkairoClient { this.commandHandler = new BushCommandHandler(this, { directory: path.join(__dirname, '..', '..', 'commands'), prefix: async ({ guild }: { guild: Guild }) => { + if (this.config.dev) return 'dev'; const row = await Models.Guild.findByPk(guild.id); - if (!row) return this.config.prefix; - return row.prefix as string; + return (row?.prefix || this.config.prefix) as string; }, allowMention: true, handleEdits: true, @@ -114,9 +121,14 @@ export class BushClient extends AkairoClient { this.commandHandler.useListenerHandler(this.listenerHandler); this.commandHandler.useInhibitorHandler(this.inhibitorHandler); this.listenerHandler.setEmitters({ + client: this, commandHandler: this.commandHandler, listenerHandler: this.listenerHandler, - process + inhibitorHandler: this.inhibitorHandler, + taskHandler: this.taskHandler, + process, + stdin: rl, + gateway: this.ws }); // loads all the handlers const loaders = { @@ -128,7 +140,7 @@ export class BushClient extends AkairoClient { for (const loader of Object.keys(loaders)) { try { loaders[loader].loadAll(); - this.logger.success('Startup', `Successfully loaded <<${loader}>>.` + chalk.cyan() + '.', false); + this.logger.success('Startup', `Successfully loaded <<${loader}>>.`, false); } catch (e) { this.logger.error('Startup', `Unable to load loader <<${loader}>> with error:\n${e?.stack}`, false); } @@ -145,7 +157,10 @@ export class BushClient extends AkairoClient { Models.Ban.initModel(this.db); Models.Level.initModel(this.db); await this.db.sync(); // Sync all tables to fix everything if updated - } catch (error) {} + this.console.success('Startup', `Successfully connected to <<database>>.`, false); + } catch (error) { + this.console.error('Startup', `Failed to connect to <<database>> with error:\n` + error, false); + } } public async start(): Promise<void> { @@ -158,7 +173,7 @@ export class BushClient extends AkairoClient { } } - public destroy(relogin = true): void | Promise<string> { + public destroy(relogin = false): void | Promise<string> { super.destroy(); if (relogin) { return this.login(this.token); |