96 lines
3.2 KiB
JavaScript
96 lines
3.2 KiB
JavaScript
import { createPrompt, useState, useKeypress, usePrefix, isEnterKey, } from '@inquirer/core';
|
|
import chalk from 'chalk';
|
|
const helpChoice = {
|
|
key: 'h',
|
|
name: 'Help, list all options',
|
|
value: undefined,
|
|
};
|
|
function getChoiceKey(choice, key) {
|
|
if (key === 'name') {
|
|
if ('name' in choice)
|
|
return choice.name;
|
|
return choice.value;
|
|
}
|
|
if ('value' in choice)
|
|
return choice.value;
|
|
return choice.name;
|
|
}
|
|
export default createPrompt((config, done) => {
|
|
const { choices, default: defaultKey = 'h', expanded: defaultExpandState = false, } = config;
|
|
const [status, setStatus] = useState('pending');
|
|
const [value, setValue] = useState('');
|
|
const [expanded, setExpanded] = useState(defaultExpandState);
|
|
const [errorMsg, setError] = useState(undefined);
|
|
const prefix = usePrefix();
|
|
useKeypress((event, rl) => {
|
|
if (isEnterKey(event)) {
|
|
const answer = (value || defaultKey).toLowerCase();
|
|
if (answer === 'h' && !expanded) {
|
|
setExpanded(true);
|
|
}
|
|
else {
|
|
const selectedChoice = choices.find(({ key }) => key === answer);
|
|
if (selectedChoice) {
|
|
const finalValue = getChoiceKey(selectedChoice, 'value');
|
|
setValue(finalValue);
|
|
setStatus('done');
|
|
done(finalValue);
|
|
}
|
|
else if (value === '') {
|
|
setError('Please input a value');
|
|
}
|
|
else {
|
|
setError(`"${chalk.red(value)}" isn't an available option`);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
setValue(rl.line);
|
|
setError(undefined);
|
|
}
|
|
});
|
|
const message = chalk.bold(config.message);
|
|
if (status === 'done') {
|
|
// TODO: `value` should be the display name instead of the raw value.
|
|
return `${prefix} ${message} ${chalk.cyan(value)}`;
|
|
}
|
|
const allChoices = expanded ? choices : [...choices, helpChoice];
|
|
// Collapsed display style
|
|
let longChoices = '';
|
|
let shortChoices = allChoices
|
|
.map((choice) => {
|
|
if (choice.key === defaultKey) {
|
|
return choice.key.toUpperCase();
|
|
}
|
|
return choice.key;
|
|
})
|
|
.join('');
|
|
shortChoices = chalk.dim(` (${shortChoices})`);
|
|
// Expanded display style
|
|
if (expanded) {
|
|
shortChoices = '';
|
|
longChoices = allChoices
|
|
.map((choice) => {
|
|
const line = ` ${choice.key}) ${getChoiceKey(choice, 'name')}`;
|
|
if (choice.key === value.toLowerCase()) {
|
|
return chalk.cyan(line);
|
|
}
|
|
return line;
|
|
})
|
|
.join('\n');
|
|
}
|
|
let helpTip = '';
|
|
const currentOption = allChoices.find(({ key }) => key === value.toLowerCase());
|
|
if (currentOption) {
|
|
helpTip = `${chalk.cyan('>>')} ${getChoiceKey(currentOption, 'name')}`;
|
|
}
|
|
let error = '';
|
|
if (errorMsg) {
|
|
error = chalk.red(`> ${errorMsg}`);
|
|
}
|
|
return [
|
|
`${prefix} ${message}${shortChoices} ${value}`,
|
|
[longChoices, helpTip, error].filter(Boolean).join('\n'),
|
|
];
|
|
});
|