658 lines
17 KiB
JavaScript
658 lines
17 KiB
JavaScript
'use strict';
|
|
|
|
const colors = require('ansi-colors');
|
|
const Prompt = require('../prompt');
|
|
const roles = require('../roles');
|
|
const utils = require('../utils');
|
|
const { reorder, scrollUp, scrollDown, isObject, swap } = utils;
|
|
|
|
class ArrayPrompt extends Prompt {
|
|
constructor(options) {
|
|
super(options);
|
|
this.cursorHide();
|
|
this.maxSelected = options.maxSelected || Infinity;
|
|
this.multiple = options.multiple || false;
|
|
this.initial = options.initial || 0;
|
|
this.delay = options.delay || 0;
|
|
this.longest = 0;
|
|
this.num = '';
|
|
}
|
|
|
|
async initialize() {
|
|
if (typeof this.options.initial === 'function') {
|
|
this.initial = await this.options.initial.call(this);
|
|
}
|
|
await this.reset(true);
|
|
await super.initialize();
|
|
}
|
|
|
|
async reset() {
|
|
let { choices, initial, autofocus, suggest } = this.options;
|
|
this.state._choices = [];
|
|
this.state.choices = [];
|
|
|
|
this.choices = await Promise.all(await this.toChoices(choices));
|
|
this.choices.forEach(ch => (ch.enabled = false));
|
|
|
|
if (typeof suggest !== 'function' && this.selectable.length === 0) {
|
|
throw new Error('At least one choice must be selectable');
|
|
}
|
|
|
|
if (isObject(initial)) initial = Object.keys(initial);
|
|
if (Array.isArray(initial)) {
|
|
if (autofocus != null) this.index = this.findIndex(autofocus);
|
|
initial.forEach(v => this.enable(this.find(v)));
|
|
await this.render();
|
|
} else {
|
|
if (autofocus != null) initial = autofocus;
|
|
if (typeof initial === 'string') initial = this.findIndex(initial);
|
|
if (typeof initial === 'number' && initial > -1) {
|
|
this.index = Math.max(0, Math.min(initial, this.choices.length));
|
|
this.enable(this.find(this.index));
|
|
}
|
|
}
|
|
|
|
if (this.isDisabled(this.focused)) {
|
|
await this.down();
|
|
}
|
|
}
|
|
|
|
async toChoices(value, parent) {
|
|
this.state.loadingChoices = true;
|
|
let choices = [];
|
|
let index = 0;
|
|
|
|
let toChoices = async(items, parent) => {
|
|
if (typeof items === 'function') items = await items.call(this);
|
|
if (items instanceof Promise) items = await items;
|
|
|
|
for (let i = 0; i < items.length; i++) {
|
|
let choice = items[i] = await this.toChoice(items[i], index++, parent);
|
|
choices.push(choice);
|
|
|
|
if (choice.choices) {
|
|
await toChoices(choice.choices, choice);
|
|
}
|
|
}
|
|
return choices;
|
|
};
|
|
|
|
return toChoices(value, parent)
|
|
.then(choices => {
|
|
this.state.loadingChoices = false;
|
|
return choices;
|
|
});
|
|
}
|
|
|
|
async toChoice(ele, i, parent) {
|
|
if (typeof ele === 'function') ele = await ele.call(this, this);
|
|
if (ele instanceof Promise) ele = await ele;
|
|
if (typeof ele === 'string') ele = { name: ele };
|
|
|
|
if (ele.normalized) return ele;
|
|
ele.normalized = true;
|
|
|
|
let origVal = ele.value;
|
|
let role = roles(ele.role, this.options);
|
|
ele = role(this, ele);
|
|
|
|
if (typeof ele.disabled === 'string' && !ele.hint) {
|
|
ele.hint = ele.disabled;
|
|
ele.disabled = true;
|
|
}
|
|
|
|
if (ele.disabled === true && ele.hint == null) {
|
|
ele.hint = '(disabled)';
|
|
}
|
|
|
|
// if the choice was already normalized, return it
|
|
if (ele.index != null) return ele;
|
|
ele.name = ele.name || ele.key || ele.title || ele.value || ele.message;
|
|
ele.message = ele.message || ele.name || '';
|
|
ele.value = [ele.value, ele.name].find(this.isValue.bind(this));
|
|
|
|
ele.input = '';
|
|
ele.index = i;
|
|
ele.cursor = 0;
|
|
|
|
utils.define(ele, 'parent', parent);
|
|
ele.level = parent ? parent.level + 1 : 1;
|
|
if (ele.indent == null) {
|
|
ele.indent = parent ? parent.indent + ' ' : (ele.indent || '');
|
|
}
|
|
|
|
ele.path = parent ? parent.path + '.' + ele.name : ele.name;
|
|
ele.enabled = !!(this.multiple && !this.isDisabled(ele) && (ele.enabled || this.isSelected(ele)));
|
|
|
|
if (!this.isDisabled(ele)) {
|
|
this.longest = Math.max(this.longest, colors.unstyle(ele.message).length);
|
|
}
|
|
|
|
// shallow clone the choice first
|
|
let choice = { ...ele };
|
|
|
|
// then allow the choice to be reset using the "original" values
|
|
ele.reset = (input = choice.input, value = choice.value) => {
|
|
for (let key of Object.keys(choice)) ele[key] = choice[key];
|
|
ele.input = input;
|
|
ele.value = value;
|
|
};
|
|
|
|
if (origVal == null && typeof ele.initial === 'function') {
|
|
ele.input = await ele.initial.call(this, this.state, ele, i);
|
|
}
|
|
|
|
return ele;
|
|
}
|
|
|
|
async onChoice(choice, i) {
|
|
this.emit('choice', choice, i, this);
|
|
|
|
if (typeof choice.onChoice === 'function') {
|
|
await choice.onChoice.call(this, this.state, choice, i);
|
|
}
|
|
}
|
|
|
|
async addChoice(ele, i, parent) {
|
|
let choice = await this.toChoice(ele, i, parent);
|
|
this.choices.push(choice);
|
|
this.index = this.choices.length - 1;
|
|
this.limit = this.choices.length;
|
|
return choice;
|
|
}
|
|
|
|
async newItem(item, i, parent) {
|
|
let ele = { name: 'New choice name?', editable: true, newChoice: true, ...item };
|
|
let choice = await this.addChoice(ele, i, parent);
|
|
|
|
choice.updateChoice = () => {
|
|
delete choice.newChoice;
|
|
choice.name = choice.message = choice.input;
|
|
choice.input = '';
|
|
choice.cursor = 0;
|
|
};
|
|
|
|
return this.render();
|
|
}
|
|
|
|
indent(choice) {
|
|
if (choice.indent == null) {
|
|
return choice.level > 1 ? ' '.repeat(choice.level - 1) : '';
|
|
}
|
|
return choice.indent;
|
|
}
|
|
|
|
dispatch(s, key) {
|
|
if (this.multiple && this[key.name]) return this[key.name]();
|
|
this.alert();
|
|
}
|
|
|
|
focus(choice, enabled) {
|
|
if (typeof enabled !== 'boolean') enabled = choice.enabled;
|
|
if (enabled && !choice.enabled && this.selected.length >= this.maxSelected) {
|
|
return this.alert();
|
|
}
|
|
this.index = choice.index;
|
|
choice.enabled = enabled && !this.isDisabled(choice);
|
|
return choice;
|
|
}
|
|
|
|
space() {
|
|
if (!this.multiple) return this.alert();
|
|
this.toggle(this.focused);
|
|
return this.render();
|
|
}
|
|
|
|
a() {
|
|
if (this.maxSelected < this.choices.length) return this.alert();
|
|
let enabled = this.selectable.every(ch => ch.enabled);
|
|
this.choices.forEach(ch => (ch.enabled = !enabled));
|
|
return this.render();
|
|
}
|
|
|
|
i() {
|
|
// don't allow choices to be inverted if it will result in
|
|
// more than the maximum number of allowed selected items.
|
|
if (this.choices.length - this.selected.length > this.maxSelected) {
|
|
return this.alert();
|
|
}
|
|
this.choices.forEach(ch => (ch.enabled = !ch.enabled));
|
|
return this.render();
|
|
}
|
|
|
|
g(choice = this.focused) {
|
|
if (!this.choices.some(ch => !!ch.parent)) return this.a();
|
|
this.toggle((choice.parent && !choice.choices) ? choice.parent : choice);
|
|
return this.render();
|
|
}
|
|
|
|
toggle(choice, enabled) {
|
|
if (!choice.enabled && this.selected.length >= this.maxSelected) {
|
|
return this.alert();
|
|
}
|
|
|
|
if (typeof enabled !== 'boolean') enabled = !choice.enabled;
|
|
choice.enabled = enabled;
|
|
|
|
if (choice.choices) {
|
|
choice.choices.forEach(ch => this.toggle(ch, enabled));
|
|
}
|
|
|
|
let parent = choice.parent;
|
|
while (parent) {
|
|
let choices = parent.choices.filter(ch => this.isDisabled(ch));
|
|
parent.enabled = choices.every(ch => ch.enabled === true);
|
|
parent = parent.parent;
|
|
}
|
|
|
|
reset(this, this.choices);
|
|
this.emit('toggle', choice, this);
|
|
return choice;
|
|
}
|
|
|
|
enable(choice) {
|
|
if (this.selected.length >= this.maxSelected) return this.alert();
|
|
choice.enabled = !this.isDisabled(choice);
|
|
choice.choices && choice.choices.forEach(this.enable.bind(this));
|
|
return choice;
|
|
}
|
|
|
|
disable(choice) {
|
|
choice.enabled = false;
|
|
choice.choices && choice.choices.forEach(this.disable.bind(this));
|
|
return choice;
|
|
}
|
|
|
|
number(n) {
|
|
this.num += n;
|
|
|
|
let number = num => {
|
|
let i = Number(num);
|
|
if (i > this.choices.length - 1) return this.alert();
|
|
|
|
let focused = this.focused;
|
|
let choice = this.choices.find(ch => i === ch.index);
|
|
|
|
if (!choice.enabled && this.selected.length >= this.maxSelected) {
|
|
return this.alert();
|
|
}
|
|
|
|
if (this.visible.indexOf(choice) === -1) {
|
|
let choices = reorder(this.choices);
|
|
let actualIdx = choices.indexOf(choice);
|
|
|
|
if (focused.index > actualIdx) {
|
|
let start = choices.slice(actualIdx, actualIdx + this.limit);
|
|
let end = choices.filter(ch => !start.includes(ch));
|
|
this.choices = start.concat(end);
|
|
} else {
|
|
let pos = actualIdx - this.limit + 1;
|
|
this.choices = choices.slice(pos).concat(choices.slice(0, pos));
|
|
}
|
|
}
|
|
|
|
this.index = this.choices.indexOf(choice);
|
|
this.toggle(this.focused);
|
|
return this.render();
|
|
};
|
|
|
|
clearTimeout(this.numberTimeout);
|
|
|
|
return new Promise(resolve => {
|
|
let len = this.choices.length;
|
|
let num = this.num;
|
|
|
|
let handle = (val = false, res) => {
|
|
clearTimeout(this.numberTimeout);
|
|
if (val) res = number(num);
|
|
this.num = '';
|
|
resolve(res);
|
|
};
|
|
|
|
if (num === '0' || (num.length === 1 && Number(num + '0') > len)) {
|
|
return handle(true);
|
|
}
|
|
|
|
if (Number(num) > len) {
|
|
return handle(false, this.alert());
|
|
}
|
|
|
|
this.numberTimeout = setTimeout(() => handle(true), this.delay);
|
|
});
|
|
}
|
|
|
|
home() {
|
|
this.choices = reorder(this.choices);
|
|
this.index = 0;
|
|
return this.render();
|
|
}
|
|
|
|
end() {
|
|
let pos = this.choices.length - this.limit;
|
|
let choices = reorder(this.choices);
|
|
this.choices = choices.slice(pos).concat(choices.slice(0, pos));
|
|
this.index = this.limit - 1;
|
|
return this.render();
|
|
}
|
|
|
|
first() {
|
|
this.index = 0;
|
|
return this.render();
|
|
}
|
|
|
|
last() {
|
|
this.index = this.visible.length - 1;
|
|
return this.render();
|
|
}
|
|
|
|
prev() {
|
|
if (this.visible.length <= 1) return this.alert();
|
|
return this.up();
|
|
}
|
|
|
|
next() {
|
|
if (this.visible.length <= 1) return this.alert();
|
|
return this.down();
|
|
}
|
|
|
|
right() {
|
|
if (this.cursor >= this.input.length) return this.alert();
|
|
this.cursor++;
|
|
return this.render();
|
|
}
|
|
|
|
left() {
|
|
if (this.cursor <= 0) return this.alert();
|
|
this.cursor--;
|
|
return this.render();
|
|
}
|
|
|
|
up() {
|
|
let len = this.choices.length;
|
|
let vis = this.visible.length;
|
|
let idx = this.index;
|
|
if (this.options.scroll === false && idx === 0) {
|
|
return this.alert();
|
|
}
|
|
if (len > vis && idx === 0) {
|
|
return this.scrollUp();
|
|
}
|
|
this.index = ((idx - 1 % len) + len) % len;
|
|
if (this.isDisabled()) {
|
|
return this.up();
|
|
}
|
|
return this.render();
|
|
}
|
|
|
|
down() {
|
|
let len = this.choices.length;
|
|
let vis = this.visible.length;
|
|
let idx = this.index;
|
|
if (this.options.scroll === false && idx === vis - 1) {
|
|
return this.alert();
|
|
}
|
|
if (len > vis && idx === vis - 1) {
|
|
return this.scrollDown();
|
|
}
|
|
this.index = (idx + 1) % len;
|
|
if (this.isDisabled()) {
|
|
return this.down();
|
|
}
|
|
return this.render();
|
|
}
|
|
|
|
scrollUp(i = 0) {
|
|
this.choices = scrollUp(this.choices);
|
|
this.index = i;
|
|
if (this.isDisabled()) {
|
|
return this.up();
|
|
}
|
|
return this.render();
|
|
}
|
|
|
|
scrollDown(i = this.visible.length - 1) {
|
|
this.choices = scrollDown(this.choices);
|
|
this.index = i;
|
|
if (this.isDisabled()) {
|
|
return this.down();
|
|
}
|
|
return this.render();
|
|
}
|
|
|
|
async shiftUp() {
|
|
if (this.options.sort === true) {
|
|
this.sorting = true;
|
|
this.swap(this.index - 1);
|
|
await this.up();
|
|
this.sorting = false;
|
|
return;
|
|
}
|
|
return this.scrollUp(this.index);
|
|
}
|
|
|
|
async shiftDown() {
|
|
if (this.options.sort === true) {
|
|
this.sorting = true;
|
|
this.swap(this.index + 1);
|
|
await this.down();
|
|
this.sorting = false;
|
|
return;
|
|
}
|
|
return this.scrollDown(this.index);
|
|
}
|
|
|
|
pageUp() {
|
|
if (this.visible.length <= 1) return this.alert();
|
|
this.limit = Math.max(this.limit - 1, 0);
|
|
this.index = Math.min(this.limit - 1, this.index);
|
|
this._limit = this.limit;
|
|
if (this.isDisabled()) {
|
|
return this.up();
|
|
}
|
|
return this.render();
|
|
}
|
|
|
|
pageDown() {
|
|
if (this.visible.length >= this.choices.length) return this.alert();
|
|
this.index = Math.max(0, this.index);
|
|
this.limit = Math.min(this.limit + 1, this.choices.length);
|
|
this._limit = this.limit;
|
|
if (this.isDisabled()) {
|
|
return this.down();
|
|
}
|
|
return this.render();
|
|
}
|
|
|
|
swap(pos) {
|
|
swap(this.choices, this.index, pos);
|
|
}
|
|
|
|
isDisabled(choice = this.focused) {
|
|
let keys = ['disabled', 'collapsed', 'hidden', 'completing', 'readonly'];
|
|
if (choice && keys.some(key => choice[key] === true)) {
|
|
return true;
|
|
}
|
|
return choice && choice.role === 'heading';
|
|
}
|
|
|
|
isEnabled(choice = this.focused) {
|
|
if (Array.isArray(choice)) return choice.every(ch => this.isEnabled(ch));
|
|
if (choice.choices) {
|
|
let choices = choice.choices.filter(ch => !this.isDisabled(ch));
|
|
return choice.enabled && choices.every(ch => this.isEnabled(ch));
|
|
}
|
|
return choice.enabled && !this.isDisabled(choice);
|
|
}
|
|
|
|
isChoice(choice, value) {
|
|
return choice.name === value || choice.index === Number(value);
|
|
}
|
|
|
|
isSelected(choice) {
|
|
if (Array.isArray(this.initial)) {
|
|
return this.initial.some(value => this.isChoice(choice, value));
|
|
}
|
|
return this.isChoice(choice, this.initial);
|
|
}
|
|
|
|
map(names = [], prop = 'value') {
|
|
return [].concat(names || []).reduce((acc, name) => {
|
|
acc[name] = this.find(name, prop);
|
|
return acc;
|
|
}, {});
|
|
}
|
|
|
|
filter(value, prop) {
|
|
let isChoice = (ele, i) => [ele.name, i].includes(value);
|
|
let fn = typeof value === 'function' ? value : isChoice;
|
|
let choices = this.options.multiple ? this.state._choices : this.choices;
|
|
let result = choices.filter(fn);
|
|
if (prop) {
|
|
return result.map(ch => ch[prop]);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
find(value, prop) {
|
|
if (isObject(value)) return prop ? value[prop] : value;
|
|
let isChoice = (ele, i) => [ele.name, i].includes(value);
|
|
let fn = typeof value === 'function' ? value : isChoice;
|
|
let choice = this.choices.find(fn);
|
|
if (choice) {
|
|
return prop ? choice[prop] : choice;
|
|
}
|
|
}
|
|
|
|
findIndex(value) {
|
|
return this.choices.indexOf(this.find(value));
|
|
}
|
|
|
|
async submit() {
|
|
let choice = this.focused;
|
|
if (!choice) return this.alert();
|
|
|
|
if (choice.newChoice) {
|
|
if (!choice.input) return this.alert();
|
|
choice.updateChoice();
|
|
return this.render();
|
|
}
|
|
|
|
if (this.choices.some(ch => ch.newChoice)) {
|
|
return this.alert();
|
|
}
|
|
|
|
let { reorder, sort } = this.options;
|
|
let multi = this.multiple === true;
|
|
let value = this.selected;
|
|
if (value === void 0) {
|
|
return this.alert();
|
|
}
|
|
|
|
// re-sort choices to original order
|
|
if (Array.isArray(value) && reorder !== false && sort !== true) {
|
|
value = utils.reorder(value);
|
|
}
|
|
|
|
this.value = multi ? value.map(ch => ch.name) : value.name;
|
|
return super.submit();
|
|
}
|
|
|
|
set choices(choices = []) {
|
|
this.state._choices = this.state._choices || [];
|
|
this.state.choices = choices;
|
|
|
|
for (let choice of choices) {
|
|
if (!this.state._choices.some(ch => ch.name === choice.name)) {
|
|
this.state._choices.push(choice);
|
|
}
|
|
}
|
|
|
|
if (!this._initial && this.options.initial) {
|
|
this._initial = true;
|
|
let init = this.initial;
|
|
if (typeof init === 'string' || typeof init === 'number') {
|
|
let choice = this.find(init);
|
|
if (choice) {
|
|
this.initial = choice.index;
|
|
this.focus(choice, true);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
get choices() {
|
|
return reset(this, this.state.choices || []);
|
|
}
|
|
|
|
set visible(visible) {
|
|
this.state.visible = visible;
|
|
}
|
|
get visible() {
|
|
return (this.state.visible || this.choices).slice(0, this.limit);
|
|
}
|
|
|
|
set limit(num) {
|
|
this.state.limit = num;
|
|
}
|
|
get limit() {
|
|
let { state, options, choices } = this;
|
|
let limit = state.limit || this._limit || options.limit || choices.length;
|
|
return Math.min(limit, this.height);
|
|
}
|
|
|
|
set value(value) {
|
|
super.value = value;
|
|
}
|
|
get value() {
|
|
if (typeof super.value !== 'string' && super.value === this.initial) {
|
|
return this.input;
|
|
}
|
|
return super.value;
|
|
}
|
|
|
|
set index(i) {
|
|
this.state.index = i;
|
|
}
|
|
get index() {
|
|
return Math.max(0, this.state ? this.state.index : 0);
|
|
}
|
|
|
|
get enabled() {
|
|
return this.filter(this.isEnabled.bind(this));
|
|
}
|
|
|
|
get focused() {
|
|
let choice = this.choices[this.index];
|
|
if (choice && this.state.submitted && this.multiple !== true) {
|
|
choice.enabled = true;
|
|
}
|
|
return choice;
|
|
}
|
|
|
|
get selectable() {
|
|
return this.choices.filter(choice => !this.isDisabled(choice));
|
|
}
|
|
|
|
get selected() {
|
|
return this.multiple ? this.enabled : this.focused;
|
|
}
|
|
}
|
|
|
|
function reset(prompt, choices) {
|
|
if (choices instanceof Promise) return choices;
|
|
if (typeof choices === 'function') {
|
|
if (utils.isAsyncFn(choices)) return choices;
|
|
choices = choices.call(prompt, prompt);
|
|
}
|
|
for (let choice of choices) {
|
|
if (Array.isArray(choice.choices)) {
|
|
let items = choice.choices.filter(ch => !prompt.isDisabled(ch));
|
|
choice.enabled = items.every(ch => ch.enabled === true);
|
|
}
|
|
if (prompt.isDisabled(choice) === true) {
|
|
delete choice.enabled;
|
|
}
|
|
}
|
|
return choices;
|
|
}
|
|
|
|
module.exports = ArrayPrompt;
|