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.
126 lines
4.5 KiB
JavaScript
126 lines
4.5 KiB
JavaScript
import { createPrompt, useMemo, useState, useKeypress, usePrefix, isEnterKey, Separator, makeTheme, ValidationError, } from '@inquirer/core';
|
|
import colors from 'yoctocolors-cjs';
|
|
const numberRegex = /\d+/;
|
|
function isSelectableChoice(choice) {
|
|
return choice != null && !Separator.isSeparator(choice);
|
|
}
|
|
function normalizeChoices(choices) {
|
|
let index = 0;
|
|
return choices.map((choice) => {
|
|
if (Separator.isSeparator(choice))
|
|
return choice;
|
|
index += 1;
|
|
if (typeof choice === 'string') {
|
|
return {
|
|
value: choice,
|
|
name: choice,
|
|
short: choice,
|
|
key: String(index),
|
|
};
|
|
}
|
|
const name = choice.name ?? String(choice.value);
|
|
return {
|
|
value: choice.value,
|
|
name,
|
|
short: choice.short ?? name,
|
|
key: choice.key ?? String(index),
|
|
};
|
|
});
|
|
}
|
|
function getSelectedChoice(input, choices) {
|
|
let selectedChoice;
|
|
const selectableChoices = choices.filter(isSelectableChoice);
|
|
if (numberRegex.test(input)) {
|
|
const answer = Number.parseInt(input, 10) - 1;
|
|
selectedChoice = selectableChoices[answer];
|
|
}
|
|
else {
|
|
selectedChoice = selectableChoices.find((choice) => choice.key === input);
|
|
}
|
|
return selectedChoice
|
|
? [selectedChoice, choices.indexOf(selectedChoice)]
|
|
: [undefined, undefined];
|
|
}
|
|
export default createPrompt((config, done) => {
|
|
const { loop = true } = config;
|
|
const choices = useMemo(() => normalizeChoices(config.choices), [config.choices]);
|
|
const [status, setStatus] = useState('idle');
|
|
const [value, setValue] = useState('');
|
|
const [errorMsg, setError] = useState();
|
|
const theme = makeTheme(config.theme);
|
|
const prefix = usePrefix({ status, theme });
|
|
const bounds = useMemo(() => {
|
|
const first = choices.findIndex(isSelectableChoice);
|
|
const last = choices.findLastIndex(isSelectableChoice);
|
|
if (first === -1) {
|
|
throw new ValidationError('[select prompt] No selectable choices. All choices are disabled.');
|
|
}
|
|
return { first, last };
|
|
}, [choices]);
|
|
useKeypress((key, rl) => {
|
|
if (isEnterKey(key)) {
|
|
const [selectedChoice] = getSelectedChoice(value, choices);
|
|
if (isSelectableChoice(selectedChoice)) {
|
|
setValue(selectedChoice.short);
|
|
setStatus('done');
|
|
done(selectedChoice.value);
|
|
}
|
|
else if (value === '') {
|
|
setError('Please input a value');
|
|
}
|
|
else {
|
|
setError(`"${colors.red(value)}" isn't an available option`);
|
|
}
|
|
}
|
|
else if (key.name === 'up' || key.name === 'down') {
|
|
rl.clearLine(0);
|
|
const [selectedChoice, active] = getSelectedChoice(value, choices);
|
|
if (!selectedChoice) {
|
|
const firstChoice = key.name === 'down'
|
|
? choices.find(isSelectableChoice)
|
|
: choices.findLast(isSelectableChoice);
|
|
setValue(firstChoice.key);
|
|
}
|
|
else if (loop ||
|
|
(key.name === 'up' && active !== bounds.first) ||
|
|
(key.name === 'down' && active !== bounds.last)) {
|
|
const offset = key.name === 'up' ? -1 : 1;
|
|
let next = active;
|
|
do {
|
|
next = (next + offset + choices.length) % choices.length;
|
|
} while (!isSelectableChoice(choices[next]));
|
|
setValue(choices[next].key);
|
|
}
|
|
}
|
|
else {
|
|
setValue(rl.line);
|
|
setError(undefined);
|
|
}
|
|
});
|
|
const message = theme.style.message(config.message, status);
|
|
if (status === 'done') {
|
|
return `${prefix} ${message} ${theme.style.answer(value)}`;
|
|
}
|
|
const choicesStr = choices
|
|
.map((choice) => {
|
|
if (Separator.isSeparator(choice)) {
|
|
return ` ${choice.separator}`;
|
|
}
|
|
const line = ` ${choice.key}) ${choice.name}`;
|
|
if (choice.key === value.toLowerCase()) {
|
|
return theme.style.highlight(line);
|
|
}
|
|
return line;
|
|
})
|
|
.join('\n');
|
|
let error = '';
|
|
if (errorMsg) {
|
|
error = theme.style.error(errorMsg);
|
|
}
|
|
return [
|
|
`${prefix} ${message} ${value}`,
|
|
[choicesStr, error].filter(Boolean).join('\n'),
|
|
];
|
|
});
|
|
export { Separator } from '@inquirer/core';
|