1
0
Fork 0

Added support for subcommands

refactor/models
Bsian 2019-10-29 21:44:12 +00:00
parent e7437d4b59
commit 3f001eebd5
No known key found for this signature in database
GPG Key ID: 097FB9A291026091
3 changed files with 51 additions and 19 deletions

View File

@ -18,7 +18,7 @@ export default class Command {
guildOnly?: boolean guildOnly?: boolean
subcommands: Command[] subcommands: Map<string, Command>
public run(message: Message, args: string[]) {} // eslint-disable-line public run(message: Message, args: string[]) {} // eslint-disable-line
@ -30,7 +30,7 @@ export default class Command {
this.aliases = []; this.aliases = [];
this.guildOnly = true; this.guildOnly = true;
this.client = client; this.client = client;
this.subcommands = []; this.subcommands = new Map();
this.permissions = {}; this.permissions = {};
} }
} }

View File

@ -36,15 +36,47 @@ export default class Util {
return result; return result;
} }
public resolveCommand(command: string): Command { public resolveCommand(command: string, args: string[]): { command: Command, args: string[]} {
if (this.client.commands.has(command)) return this.client.commands.get(command); let Cmd: Command;
for (const cmd of this.client.commands.values()) { if (this.client.commands.has(command)) Cmd = this.client.commands.get(command);
if (!cmd.aliases) continue;// eslint-disable-line no-continue else {
for (const alias of cmd.aliases) { for (const cmd of this.client.commands.values()) {
if (command === alias.toLowerCase()) return cmd; if (!cmd.aliases) continue;// eslint-disable-line no-continue
for (const alias of cmd.aliases) {
if (command.toLowerCase() === alias.toLowerCase()) {
Cmd = cmd;
} else {
return undefined;
}
}
} }
} }
return undefined; if (!Cmd) return undefined;
let hasSubcmd = true;
while (hasSubcmd) {
if (!Cmd.subcommands.size) {
hasSubcmd = false; break;
}
if (Cmd.subcommands.has(args[0])) {
Cmd = Cmd.subcommands.get(args[0]);
args.shift();
} else {
for (const cmd of Cmd.subcommands.values()) {
if (!cmd.aliases) continue; // eslint-disable-line
for (const alias of cmd.aliases) {
if (args[0].toLowerCase() === alias.toLowerCase()) {
Cmd = cmd; args.shift(); break;
}
}
break;
}
}
}
return { command: Cmd, args };
}
public resolveSubcommand(parentCmd: Command, args?: string[]) {
} }
public async handleError(error: Error, message?: Message, command?: Command): Promise<void> { public async handleError(error: Error, message?: Message, command?: Command): Promise<void> {

View File

@ -15,30 +15,30 @@ export default class {
if (message.content.indexOf(this.client.config.prefix) !== 0) return; if (message.content.indexOf(this.client.config.prefix) !== 0) return;
const noPrefix: string[] = message.content.slice(this.client.config.prefix.length).trim().split(/ +/g); const noPrefix: string[] = message.content.slice(this.client.config.prefix.length).trim().split(/ +/g);
const command: string = noPrefix[0].toLowerCase(); const command: string = noPrefix[0].toLowerCase();
const resolved: Command = this.client.util.resolveCommand(command); const Args = noPrefix.slice(1);
const resolved: { command: Command, args: string[] } = this.client.util.resolveCommand(command, Args);
if (!resolved) return; if (!resolved) return;
if (resolved.guildOnly && !(message.channel instanceof TextChannel)) return; if (resolved.command.guildOnly && !(message.channel instanceof TextChannel)) return;
let hasUserPerms: boolean; let hasUserPerms: boolean;
if (resolved.permissions.users) { if (resolved.command.permissions.users) {
hasUserPerms = resolved.permissions.users.includes(message.author.id); hasUserPerms = resolved.command.permissions.users.includes(message.author.id);
} }
let hasRolePerms: boolean = false; let hasRolePerms: boolean = false;
if (resolved.permissions.roles) { if (resolved.command.permissions.roles) {
for (const role of resolved.permissions.roles) { for (const role of resolved.command.permissions.roles) {
if (message.member && message.member.roles.includes(role)) { if (message.member && message.member.roles.includes(role)) {
// this.client.signale.debug(message.member.roles.includes(role)); // this.client.signale.debug(message.member.roles.includes(role));
hasRolePerms = true; break; hasRolePerms = true; break;
} }
} }
} }
if (!resolved.permissions.users && !resolved.permissions.roles) { if (!resolved.command.permissions.users && !resolved.command.permissions.roles) {
hasUserPerms = true; hasUserPerms = true;
hasRolePerms = true; hasRolePerms = true;
} }
if (!hasRolePerms && !hasUserPerms) return; if (!hasRolePerms && !hasUserPerms) return;
if (!resolved.enabled) { message.channel.createMessage(`***${this.client.stores.emojis.error} This command has been disabled***`); return; } if (!resolved.command.enabled) { message.channel.createMessage(`***${this.client.stores.emojis.error} This command has been disabled***`); return; }
const args: string[] = noPrefix.slice(1); resolved.command.run(message, resolved.args);
resolved.run(message, args);
} catch (error) { } catch (error) {
this.client.util.handleError(error, message); this.client.util.handleError(error, message);
} }