curate/src/commandloader.js

const fs = require('fs');
const path = require('path');
const reload = require('require-reload')(require);
const config = require('config');

module.exports = class CommandLoader {
  constructor(client, cPath) {
    this.commands = [];
    this.path = path.resolve(cPath);
    this.client = client;
  }

  /**
   * Loads commands from a folder
   * @param {String} folderPath
   */
  iterateFolder(folderPath) {
    const files = fs.readdirSync(folderPath);
    files.map(file => {
      const filePath = path.join(folderPath, file);
      const stat = fs.lstatSync(filePath);
      if (stat.isSymbolicLink()) {
        const realPath = fs.readlinkSync(filePath);
        if (stat.isFile() && file.endsWith('.js')) {
          this.load(realPath);
        } else if (stat.isDirectory()) {
          this.iterateFolder(realPath);
        }
      } else if (stat.isFile() && file.endsWith('.js'))
        this.load(filePath);
      else if (stat.isDirectory())
        this.iterateFolder(filePath);
    });
  }

  /**
   * Loads a command
   * @param {string} commandPath
   */
  load(commandPath) {
    console.fileload('Loading command', commandPath);
    const cls = reload(commandPath);
    const cmd = new cls(this.client);
    cmd.path = commandPath;
    this.commands.push(cmd);
    return cmd;
  }

  /**
   * Reloads all commands
   */
  reload() {
    this.commands = [];
    this.iterateFolder(this.path);
  }

  /**
   * Gets a command based on it's name or alias
   * @param {string} name The command's name or alias
   */
  get(name) {
    let cmd = this.commands.find(c => c.name === name);
    if (cmd) return cmd;
    this.commands.forEach(c => {
      if (c.options.aliases.includes(name)) cmd = c;
    });
    return cmd;
  }

  /**
   * Preloads a command
   * @param {string} name The command's name or alias
   */
  preload(name) {
    if (!this.get(name)) return;
    this.get(name)._preload();
  }

  /**
   * Preloads all commands
   */
  preloadAll() {
    this.commands.forEach(c => c._preload());
  }

  /**
   * Processes the cooldown of a command
   * @param {Message} message
   * @param {Command} command
   */
  async processCooldown(message, command) {
    if (config.elevated.includes(message.author.id)) return true;
    const now = Date.now() - 1;
    const cooldown = command.cooldownAbs;
    let userCD = await this.client.db.hget(`cooldowns:${message.author.id}`, command.name) || 0;
    if (userCD) userCD = parseInt(userCD);
    if (userCD + cooldown > now) return false;
    await this.client.db.hset(`cooldowns:${message.author.id}`, command.name, now);
    return true;
  }
};