126 lines
4.0 KiB
JavaScript
126 lines
4.0 KiB
JavaScript
"use strict";
|
|
function repeat(chr, count) {
|
|
let result = "";
|
|
while (result.length < count) {
|
|
result += chr;
|
|
}
|
|
return result;
|
|
}
|
|
function _getPrompt(prompt, options, callback) {
|
|
process.stdout.write(prompt);
|
|
let stdin = process.stdin;
|
|
stdin.resume();
|
|
stdin.setRawMode(true);
|
|
stdin.resume();
|
|
stdin.setEncoding('utf8');
|
|
let message = '';
|
|
let respond = (ctrlC, message) => {
|
|
process.stdout.write('\n');
|
|
stdin.setRawMode(false);
|
|
stdin.pause();
|
|
stdin.removeListener('data', handler);
|
|
callback(ctrlC, message);
|
|
};
|
|
function handler(chr) {
|
|
chr = String(chr);
|
|
switch (chr) {
|
|
// Enter (ish)
|
|
case "\n":
|
|
case "\r":
|
|
case "\u0004":
|
|
if (options.choice) {
|
|
if (options.defaultChoice) {
|
|
respond(null, options.defaultChoice);
|
|
}
|
|
}
|
|
else {
|
|
respond(null, message);
|
|
}
|
|
break;
|
|
// Backspace
|
|
case "\u007f":
|
|
if (message.length > 0 && options.choice == null) {
|
|
message = message.substring(0, message.length - 1);
|
|
(process.stdout).clearLine();
|
|
(process.stdout).cursorTo(0);
|
|
if (options.mask) {
|
|
process.stdout.write(prompt + repeat(options.mask, message.length));
|
|
}
|
|
else {
|
|
process.stdout.write(prompt + message);
|
|
}
|
|
}
|
|
break;
|
|
// Ctrl-C
|
|
case "\u0003":
|
|
process.stdout.write('\n[ CTRL-C ]');
|
|
respond(true, null);
|
|
break;
|
|
// Any other character
|
|
default:
|
|
if (options.choice) {
|
|
if (options.choice.indexOf(chr) >= 0) {
|
|
process.stdout.write(chr);
|
|
respond(null, chr);
|
|
}
|
|
}
|
|
else {
|
|
// More passsword characters
|
|
if (options.mask) {
|
|
process.stdout.write('*');
|
|
}
|
|
else {
|
|
process.stdout.write(chr);
|
|
}
|
|
message += chr;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
stdin.on('data', handler);
|
|
}
|
|
function getPrompt(prompt, options) {
|
|
return new Promise((resolve, reject) => {
|
|
_getPrompt(prompt, options, (ctrlC, password) => {
|
|
if (ctrlC) {
|
|
return reject(new Error("cancelled"));
|
|
}
|
|
resolve(password);
|
|
});
|
|
});
|
|
}
|
|
export function getProgressBar(action) {
|
|
let lastProgress = -1;
|
|
return function (percent) {
|
|
let progress = Math.trunc(percent * 100);
|
|
if (progress == lastProgress) {
|
|
return;
|
|
}
|
|
lastProgress = progress;
|
|
process.stdin.setRawMode(false);
|
|
process.stdin.pause();
|
|
(process.stdout).clearLine();
|
|
(process.stdout).cursorTo(0);
|
|
process.stdout.write(action + "... " + progress + "%");
|
|
if (percent === 1) {
|
|
process.stdout.write('\n');
|
|
}
|
|
};
|
|
}
|
|
export function getPassword(prompt) {
|
|
return getPrompt(prompt, { mask: "*" });
|
|
}
|
|
export function getMessage(prompt) {
|
|
return getPrompt(prompt, {});
|
|
}
|
|
// @TODO: Allow choices to be an array, [ "Yes", "No", "All" ] => "(y)es/ (N)o/ (a)ll"
|
|
export function getChoice(prompt, choices, defaultChoice) {
|
|
let choice = choices.toLowerCase().split("");
|
|
if (defaultChoice) {
|
|
defaultChoice = defaultChoice.toLowerCase();
|
|
}
|
|
let options = { choice: choice, defaultChoice: defaultChoice };
|
|
let hint = choice.map((c) => ((c === defaultChoice) ? c.toUpperCase() : c)).join("/");
|
|
return getPrompt((prompt + " (" + hint + ") "), options);
|
|
}
|