95 lines
3.6 KiB
JavaScript
95 lines
3.6 KiB
JavaScript
import { createPrompt, useState, useKeypress, usePrefix, isEnterKey, isBackspaceKey, makeTheme, } from '@inquirer/core';
|
|
function isStepOf(value, step, min) {
|
|
const valuePow = value * Math.pow(10, 6);
|
|
const stepPow = step * Math.pow(10, 6);
|
|
const minPow = min * Math.pow(10, 6);
|
|
return (valuePow - (Number.isFinite(min) ? minPow : 0)) % stepPow === 0;
|
|
}
|
|
function validateNumber(value, { min, max, step, }) {
|
|
if (value == null || Number.isNaN(value)) {
|
|
return false;
|
|
}
|
|
else if (value < min || value > max) {
|
|
return `Value must be between ${min} and ${max}`;
|
|
}
|
|
else if (step !== 'any' && !isStepOf(value, step, min)) {
|
|
return `Value must be a multiple of ${step}${Number.isFinite(min) ? ` starting from ${min}` : ''}`;
|
|
}
|
|
return true;
|
|
}
|
|
export default createPrompt((config, done) => {
|
|
const { validate = () => true, min = -Infinity, max = Infinity, step = 1, required = false, } = config;
|
|
const theme = makeTheme(config.theme);
|
|
const [status, setStatus] = useState('pending');
|
|
const [value, setValue] = useState(''); // store the input value as string and convert to number on "Enter"
|
|
// Ignore default if not valid.
|
|
const validDefault = validateNumber(config.default, { min, max, step }) === true
|
|
? config.default?.toString()
|
|
: undefined;
|
|
const [defaultValue = '', setDefaultValue] = useState(validDefault);
|
|
const [errorMsg, setError] = useState();
|
|
const isLoading = status === 'loading';
|
|
const prefix = usePrefix({ isLoading, theme });
|
|
useKeypress(async (key, rl) => {
|
|
// Ignore keypress while our prompt is doing other processing.
|
|
if (status !== 'pending') {
|
|
return;
|
|
}
|
|
if (isEnterKey(key)) {
|
|
const input = value || defaultValue;
|
|
const answer = input === '' ? undefined : Number(input);
|
|
setStatus('loading');
|
|
let isValid = true;
|
|
if (required || answer != null) {
|
|
isValid = validateNumber(answer, { min, max, step });
|
|
}
|
|
if (isValid === true) {
|
|
isValid = await validate(answer);
|
|
}
|
|
if (isValid === true) {
|
|
setValue(String(answer ?? ''));
|
|
setStatus('done');
|
|
done(answer);
|
|
}
|
|
else {
|
|
// Reset the readline line value to the previous value. On line event, the value
|
|
// get cleared, forcing the user to re-enter the value instead of fixing it.
|
|
rl.write(value);
|
|
setError(isValid || 'You must provide a valid numeric value');
|
|
setStatus('pending');
|
|
}
|
|
}
|
|
else if (isBackspaceKey(key) && !value) {
|
|
setDefaultValue(undefined);
|
|
}
|
|
else if (key.name === 'tab' && !value) {
|
|
setDefaultValue(undefined);
|
|
rl.clearLine(0); // Remove the tab character.
|
|
rl.write(defaultValue);
|
|
setValue(defaultValue);
|
|
}
|
|
else {
|
|
setValue(rl.line);
|
|
setError(undefined);
|
|
}
|
|
});
|
|
const message = theme.style.message(config.message);
|
|
let formattedValue = value;
|
|
if (status === 'done') {
|
|
formattedValue = theme.style.answer(value);
|
|
}
|
|
let defaultStr;
|
|
if (defaultValue && status !== 'done' && !value) {
|
|
defaultStr = theme.style.defaultAnswer(defaultValue);
|
|
}
|
|
let error = '';
|
|
if (errorMsg) {
|
|
error = theme.style.error(errorMsg);
|
|
}
|
|
return [
|
|
[prefix, message, defaultStr, formattedValue]
|
|
.filter((v) => v !== undefined)
|
|
.join(' '),
|
|
error,
|
|
];
|
|
});
|