trunk/src/CommandManager.js

const fs = require("fs");
const path = require("path");
const MessageEmbed = require("./MessageEmbed");
const SnippetManager = require("./SnippetManager");
const Shield = require("./Shield");
const { escapeRegex } = require("./util");

class CommandManager {
    constructor(cmdDir, loadCommands = true, loadSnippets = true) {
        this.msg = null;
        this.commandsDirectory = cmdDir;
        this.commands = [];
        this.commandName = "";
        this.argv = [];
        this.args = [];
        this.options = [];
        this.normalArgs = [];

        if (loadCommands)
            this.loadCommands();
        
        if (loadSnippets) 
            this.snippetManager = new SnippetManager();
        
        this.shield = new Shield();
    }

    setMessage(msg) {
        this.msg = msg;
        this.argv = msg.content.split(/ +/g);
        this.args = this.argv;
        this.commandName = this.args.shift().trim().replace(new RegExp(`^${escapeRegex(app.config.props[msg.guild.id].prefix)}`), "");
    }

    loadCommands() {
        fs.readdir(this.commandsDirectory, (err, files) => {
            if (err) {
                return console.log('Unable to scan commands directory: ' + err);
            } 

            this.commandNames = files.map(file => file.replace('\.js', ''));
            
            for (let index in files) {
                this.commands[this.commandNames[index]] = require(path.resolve(this.commandsDirectory, files[index]));
            }
        });
    }

    has() {
        return typeof this.commands[this.commandName] !== 'undefined';
    }

    snippet() {
        return this.snippetManager.find(this.msg.guild.id, this.commandName);
    }

    hasValid() {
        return this.has() && this.valid();
    }

    valid() {
        return this.msg.content.startsWith(app.config.props[this.msg.guild.id].prefix);
    }

    async notFound() {
        if (app.config.get('debug')) {
            await app.msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`${escapeRegex(this.commandName)}: command not found`)
                ]
            });
        }
    }

    async notAllowed() {
        if (app.config.get('warn_notallowed')) {
            await app.msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`:x: You don't have permission to run this command.`)
                ]
            });
        }
    }

    async exec() {
        let cmd = this.commands[this.commandName];

        if (cmd.needsOptionParse) {
            this.normalArgs = await this.args.filter(arg => arg[0] !== '-');
            this.options = await this.args.filter(arg => arg[0] === '-');
        }

        return await cmd.handle(this.msg, this);
    }

    verify() {
        return this.shield.verify(this.msg, this);
    }
}

module.exports = CommandManager;
1	const fs = require("fs");
2	const path = require("path");
3	const MessageEmbed = require("./MessageEmbed");
4	const SnippetManager = require("./SnippetManager");
5	const Shield = require("./Shield");
6	const { escapeRegex } = require("./util");
7
8	class CommandManager {
9	constructor(cmdDir, loadCommands = true, loadSnippets = true) {
10	this.msg = null;
11	this.commandsDirectory = cmdDir;
12	this.commands = [];
13	this.commandName = "";
14	this.argv = [];
15	this.args = [];
16	this.options = [];
17	this.normalArgs = [];
18
19	if (loadCommands)
20	this.loadCommands();
21
22	if (loadSnippets)
23	this.snippetManager = new SnippetManager();
24
25	this.shield = new Shield();
26	}
27
28	setMessage(msg) {
29	this.msg = msg;
30	this.argv = msg.content.split(/ +/g);
31	this.args = this.argv;
32	this.commandName = this.args.shift().trim().replace(new RegExp(`^${escapeRegex(app.config.props[msg.guild.id].prefix)}`), "");
33	}
34
35	loadCommands() {
36	fs.readdir(this.commandsDirectory, (err, files) => {
37	if (err) {
38	return console.log('Unable to scan commands directory: ' + err);
39	}
40
41	this.commandNames = files.map(file => file.replace('\.js', ''));
42
43	for (let index in files) {
44	this.commands[this.commandNames[index]] = require(path.resolve(this.commandsDirectory, files[index]));
45	}
46	});
47	}
48
49	has() {
50	return typeof this.commands[this.commandName] !== 'undefined';
51	}
52
53	snippet() {
54	return this.snippetManager.find(this.msg.guild.id, this.commandName);
55	}
56
57	hasValid() {
58	return this.has() && this.valid();
59	}
60
61	valid() {
62	return this.msg.content.startsWith(app.config.props[this.msg.guild.id].prefix);
63	}
64
65	async notFound() {
66	if (app.config.get('debug')) {
67	await app.msg.reply({
68	embeds: [
69	new MessageEmbed()
70	.setColor('#f14a60')
71	.setDescription(`${escapeRegex(this.commandName)}: command not found`)
72	]
73	});
74	}
75	}
76
77	async notAllowed() {
78	if (app.config.get('warn_notallowed')) {
79	await app.msg.reply({
80	embeds: [
81	new MessageEmbed()
82	.setColor('#f14a60')
83	.setDescription(`:x: You don't have permission to run this command.`)
84	]
85	});
86	}
87	}
88
89	async exec() {
90	let cmd = this.commands[this.commandName];
91
92	if (cmd.needsOptionParse) {
93	this.normalArgs = await this.args.filter(arg => arg[0] !== '-');
94	this.options = await this.args.filter(arg => arg[0] === '-');
95	}
96
97	return await cmd.handle(this.msg, this);
98	}
99
100	verify() {
101	return this.shield.verify(this.msg, this);
102	}
103	}
104
105	module.exports = CommandManager;