You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
discordbot.js/lib/cmd.js

180 lines
5.7 KiB
JavaScript

/* Module definition */
/* Variable Definition */
let logger = require('winston'),
globCommands = {},
ownerCommands = {},
args = require('args-parser')(process.argv);
/* Function Definition */
/**
* @type {Servant}
*/
exports.Servant = class {
constructor(prefix) {
this.commands = {};
6 years ago
this.createCommand(((prefix || '~') + 'help') || "~help", () => {
let helpstr = "```markdown\n";
helpstr += "Commands\n===\n";
helpstr += 'Global Commands\n---\n';
Object.entries(globCommands).sort().forEach(([key, value]) => {
if (value.role !== 'owner' || msg.author.tag === args.owner) {
let cmdhelp = `${key} [${value.args.join('] [')}]`.replace('[]', '').padEnd(25, ' ');
cmdhelp += value.description || '';
helpstr += `\n${cmdhelp}\n`;
}
});
helpstr += '\nServer Commands\n---\n';
Object.entries(this.commands).sort().forEach(([key, value]) => {
if (value.role !== 'owner' || msg.author.tag === args.owner) {
let cmdhelp = `${key} [${value.args.join('] [')}]`.replace('[]', '').padEnd(25, ' ');
cmdhelp += value.description || '';
helpstr += `\n${cmdhelp}\n`;
}
});
helpstr += "```";
return helpstr;
}, [], "Shows this help.");
}
/**
* Creates a command entry in the private commands dict
* @param command
* @param call
* @param args
* @param description
*/
createCommand(command, call, args, description, role) {
6 years ago
this.commands[command] = {
'args': args,
'description': description,
'callback': call,
'role': role
};
}
/**
* Removes a command
* @param command
*/
removeCommand(command) {
delete this.commands[command];
}
/**
* Parses the message and executes the command callback for the found command entry in the commands dict
* @param msg
* @returns {*}
*/
parseCommand(msg) {
let globResult = parseGlobalCommand(msg);
logger.debug(`Global command result is ${globResult}`);
let content = msg.content;
let command = (content.match(/^.\w+/) || [])[0];
if (!command || !this.commands[command]) return globResult;
let cmd = this.commands[command];
if (!checkPermission(msg, cmd.role)) return 'No Permission';
logger.debug(`Permission <${cmd.role || 'all'}> granted for command ${command} for user <${msg.author.tag}>`);
let argvars = content.match(/(?<= )\S+/g) || [];
let kwargs = {};
let nLength = Math.min(cmd.args.length, argvars.length);
for (let i = 0; i < nLength; i++) {
kwargs[cmd.args[i]] = argvars[i];
}
let argv = argvars.slice(nLength);
logger.debug(`Executing callback for command: ${command}, kwargs: ${kwargs}, argv: ${argv}`);
return cmd.callback(msg, kwargs, argv) || globResult;
}
};
/**
* Getting the logger
* @param {Object} newLogger
*/
6 years ago
exports.setLogger = function (newLogger) {
logger = newLogger;
};
/**
* Creates a global command that can be executed in every channel.
* @param command
* @param call
* @param args
* @param description
* @param role
*/
exports.createGlobalCommand = function (command, call, args, description, role) {
6 years ago
globCommands[command] = {
'args': args || [],
'description': description,
'callback': call,
'role': role
};
logger.debug(`Created command: ${command}, args: ${args}`);
};
/**
* Parses a message for a global command
* @param msg
* @returns {boolean|*}
*/
6 years ago
exports.parseMessage = function (msg) {
return parseGlobalCommand(msg);
6 years ago
};
/**
* Initializes the module by creating a help command
*/
6 years ago
exports.init = function (prefix) {
logger.verbose("Created help command");
this.createGlobalCommand((prefix + 'help') || "~help", (msg) => {
let helpstr = "```markdown\n";
helpstr += "Commands\n---\n";
Object.entries(globCommands).sort().forEach(([key, value]) => {
if (value.role !== 'owner' || msg.author.tag === args.owner) {
let cmdhelp = `${key} [${value.args.join('] [')}]`.replace('[]', '').padEnd(25, ' ');
cmdhelp += value.description || '';
helpstr += `\n${cmdhelp}\n`;
}
});
helpstr += "```";
return helpstr;
}, [], "Shows this help.");
};
/**
* Parses the message by calling the assigned function for the command with arguments
* @param msg
*/
function parseGlobalCommand(msg) {
let content = msg.content;
let command = (content.match(/^.\w+/) || [])[0];
if (!command || !globCommands[command]) return false;
let cmd = globCommands[command];
if (!checkPermission(msg, cmd.role)) return false;
logger.debug(`Permission <${cmd.role}> granted for command ${command} for user <${msg.author.tag}>`);
let argvars = content.match(/(?<= )\S+/g) || [];
let kwargs = {};
let nLength = Math.min(cmd.args.length, argvars.length);
for (let i = 0; i < nLength; i++) {
kwargs[cmd.args[i]] = argvars[i];
}
let argv = argvars.slice(nLength);
logger.debug(`Executing callback for command: ${command}, kwargs: ${JSON.stringify(kwargs)}, argv: ${argv}`);
return cmd.callback(msg, kwargs, argv);
}
function checkPermission(msg, role) {
if (!role || ['all', 'any', 'everyone'].includes(role))
return true;
if (msg.author.tag === args.owner) {
return true;
} else {
if (msg.member && role && msg.member.roles.find("id", role.id))
return true
}
return false
}