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	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	rakin	46	constructor(cmdDir, loadCommands = true, loadSnippets = true) {
10	rakin	5	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	46
19			if (loadCommands)
20			this.loadCommands();
21
22			if (loadSnippets)
23			this.snippetManager = new SnippetManager();
24
25	rakin	11	this.shield = new Shield();
26	rakin	5	}
27
28			setMessage(msg) {
29			this.msg = msg;
30			this.argv = msg.content.split(/ +/g);
31			this.args = this.argv;
32	rakin	46	this.commandName = this.args.shift().trim().replace(new RegExp(`^${escapeRegex(app.config.props[msg.guild.id].prefix)}`), "");
33	rakin	5	}
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	rakin	27	return this.snippetManager.find(this.msg.guild.id, this.commandName);
55	rakin	5	}
56
57			hasValid() {
58			return this.has() && this.valid();
59			}
60
61			valid() {
62	rakin	46	return this.msg.content.startsWith(app.config.props[this.msg.guild.id].prefix);
63	rakin	5	}
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	rakin	11	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	rakin	5	async exec() {
90	rakin	7	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	rakin	5	}
99	rakin	11
100			verify() {
101			return this.shield.verify(this.msg, this);
102			}
103	rakin	5	}
104
105			module.exports = CommandManager;