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) {
        this.msg = null;
        this.commandsDirectory = cmdDir;
        this.commands = [];
        this.commandName = "";
        this.argv = [];
        this.args = [];
        this.options = [];
        this.normalArgs = [];
        this.loadCommands();
        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.get('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.commandName);
    }

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

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