fileutils-cli/src/common/Favorites.js

A collection of powerful command line file utilities

const async = require('async');
const clear = require('cli-clear');
const inquirer = require('inquirer');
const os = require('os');
const term = require('terminal-kit').terminal;

const fu = require('../fu');

module.exports.Favorites = class Favorites {
  constructor(sequelize, options) {
    this.sequelize = sequelize;
    this.options = options;
  }

  async display() {
    this.allFavorites = await this.sequelize.models.Favorites.findAll();
    clear();
    let choices = await async.mapSeries(this.allFavorites, async (f) => {
      return {
        name: `${f.id}: ${f.commandString}${ f.alias ? ` (${f.alias})` : ''}`,
        value: f.id
      };
    });
    choices.push({ name: 'Exit', value: -1 });
    const selection = await inquirer.prompt({
      type: 'list',
      loop: false,
      message: 'Select a favorite to view options',
      name: 'favorite',
      default: 0,
      choices: choices,
      pageSize: 20
    });
    if (selection.favorite === -1) process.exit(0);
    await this.get(selection.favorite);
    await this.displayFavorite();
  }

  async displayFavorite() {
    clear();
    console.log(`${this.selected.id}: ${this.selected.commandString}${ this.selected.alias ? ` (${this.selected.alias})` : ''}`);
    const selection = await inquirer.prompt({
      type: 'list',
      loop: false,
      message: 'What would you like to do?',
      name: 'choice',
      default: 0,
      choices: ['Run', 'Edit Command', 'Edit Alias', 'Delete', 'Go Back', 'Exit']
    });
    switch (selection.choice) {
      case 'Run':
        await this.run();
        return;
      case 'Edit Command':
        await this.editCommand();
        await this.displayFavorite();
        return;
      case 'Edit Alias':
        await this.editAlias();
        await this.displayFavorite();
        return;
      case 'Delete':
        await this.delete();
        return;
      case 'Exit':
        process.exit(0);
        return;
      default: await this.display();
    }
  }

  async get(idOrAlias) {
    if (idOrAlias && (Number.isInteger(idOrAlias) || idOrAlias.match(/^\d+$/))) {
      const id = parseInt(idOrAlias);
      await this.getById(id);
    } else if (idOrAlias) {
      await this.getByAlias(idOrAlias);
    } else if (this.options.idOrAlias && this.options.idOrAlias.match(/^\d+$/)) {
      const id = parseInt(this.options.idOrAlias);
      await this.getById(id);
    } else if (this.options.idOrAlias) {
      await this.getByAlias(this.options.idOrAlias);
    } else {
      await this.display();
    }
  }

  async getById(id) {
    this.selected = await this.sequelize.models.Favorites.findOne({ where: {id: id } });
  }

  async getByAlias(alias) {
    this.selected = await this.sequelize.models.Favorites.findOne({ where: {alias: alias } });
  }

  async run() {
    if (!this.selected) {
      console.log('Invalid ID or Alias');
      return;
    }
    if (this.options.verbose) console.log(this.selected.commandString);
    await (fu(this.selected.command));
  }

  async delete() {
    await this.selected.destroy();
    await this.display();
  }

  async editCommand() {
    console.log('Edit the command and hit ENTER to save or ESC to cancel');
    term('Command: ');
    let keyBindingOptions = {
      ENTER: 'submit' ,
      KP_ENTER: 'submit' ,
      ESCAPE: 'cancel' ,
      BACKSPACE: 'backDelete' ,
      DELETE: 'delete' ,
      LEFT: 'backward' ,
      RIGHT: 'forward' ,
      UP: 'historyPrevious' ,
      DOWN: 'historyNext' ,
      HOME: 'startOfInput' ,
      END: 'endOfInput' ,
      TAB: 'autoComplete' ,
      CTRL_R: 'autoCompleteUsingHistory' ,
      CTRL_LEFT: 'previousWord' ,
      CTRL_RIGHT: 'nextWord' ,
      ALT_D: 'deleteNextWord' ,
      CTRL_W: 'deletePreviousWord' ,
      CTRL_U: 'deleteAllBefore' ,
      CTRL_K: 'deleteAllAfter'
    };
    if (os.platform() === 'darwin') keyBindingOptions.DELETE = 'backDelete';
    const input = await term.inputField({ cancelable: true, default: this.selected.commandString, keyBindings: keyBindingOptions }).promise;
    if (input) {
      this.selected.command = input.split(' ');
      await this.selected.save();
    }
  }

  async editAlias() {
    const input = await inquirer.prompt({
      type: 'input',
      name: 'alias',
      message: 'Enter an alias for this favorite (optional)',
    });
    if (input.alias) {
      try {
        this.selected.alias = input.alias;
        await this.selected.save();
      } catch (e) {
        console.log(`${input.alias} is already taken, please input a different alias`);
        await this.editAlias();
      }
    }
  }
};