318 lines
9.1 KiB
JavaScript
318 lines
9.1 KiB
JavaScript
/*
|
|
* readlineSync
|
|
* https://github.com/anseki/readline-sync
|
|
*
|
|
* Copyright (c) 2015 anseki
|
|
* Licensed under the MIT license.
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
var
|
|
IS_WIN = process.platform === 'win32',
|
|
SHELL_PATH = IS_WIN ? 'cmd.exe' : '/bin/sh',
|
|
SHELL_CMD = __dirname + (IS_WIN ? '\\read.bat' : '/read.sh'),
|
|
ALGORITHM_CIPHER = 'aes-256-cbc',
|
|
ALGORITHM_HASH = 'sha256',
|
|
DEFAULT_ERR_MSG = 'The platform doesn\'t support interactive reading',
|
|
|
|
fs = require('fs'),
|
|
childProc = require('child_process'),
|
|
|
|
promptText = '> ',
|
|
encoding = 'utf8',
|
|
bufSize = 1024,
|
|
print,
|
|
tempdir, salt = 0, useShell = false;
|
|
|
|
function _readlineSync(options) { // options.display is string
|
|
var input = '', fd, buffer, rsize, res, isOpened, fsBind, constBind;
|
|
|
|
if (options.display !== '' && typeof print === 'function')
|
|
{ print(options.display, encoding); }
|
|
|
|
if (useShell || options.noEchoBack) {
|
|
res = _readlineShell(options);
|
|
if (res.error) { throw res.error; }
|
|
input = res.input;
|
|
} else {
|
|
|
|
if (IS_WIN) { // r/w mode not supported
|
|
if (process.stdin.isTTY && process.stdout.isTTY) {
|
|
|
|
console.warn('STD TRY');
|
|
if (options.display !== '') {
|
|
// process.stdout.write(options.display, encoding);
|
|
fs.writeSync(process.stdout.fd, options.display);
|
|
options.display = '';
|
|
}
|
|
fd = process.stdin.fd;
|
|
isOpened = true;
|
|
console.warn('STD OK');
|
|
|
|
} else {
|
|
|
|
try {
|
|
console.warn('CON TRY');
|
|
if (options.display !== '') {
|
|
fd = fs.openSync('\\\\.\\CON', 'w');
|
|
fs.writeSync(fd, options.display);
|
|
fs.closeSync(fd);
|
|
options.display = '';
|
|
}
|
|
fd = fs.openSync('\\\\.\\CON', 'rs');
|
|
isOpened = true;
|
|
console.warn('CON OK');
|
|
} catch (e) {}
|
|
|
|
if (!isOpened || options.display !== '') { // Retry
|
|
|
|
try {
|
|
console.warn('CONINOUT TRY');
|
|
// For raw device path
|
|
// On XP, 2000, 7 (x86), it might fail.
|
|
// And, process.binding('fs') might be no good.
|
|
fsBind = process.binding('fs');
|
|
constBind = process.binding('constants');
|
|
if (options.display !== '') {
|
|
fd = fsBind.open('CONOUT$',
|
|
constBind.O_RDWR | constBind.O_SYNC, parseInt('0666', 8));
|
|
fs.writeSync(fd, options.display);
|
|
fs.closeSync(fd);
|
|
options.display = '';
|
|
}
|
|
fd = fsBind.open('CONIN$',
|
|
constBind.O_RDWR | constBind.O_SYNC, parseInt('0666', 8));
|
|
isOpened = true;
|
|
console.warn('CONINOUT OK');
|
|
} catch (e) {}
|
|
|
|
}
|
|
}
|
|
} else {
|
|
|
|
try {
|
|
console.warn('/dev/tty TRY');
|
|
fd = fs.openSync('/dev/tty', 'rs+');
|
|
isOpened = true;
|
|
if (options.display !== '') {
|
|
fs.writeSync(fd, options.display);
|
|
options.display = '';
|
|
}
|
|
console.warn('/dev/tty OK');
|
|
} catch (e) {}
|
|
|
|
}
|
|
|
|
if (isOpened && options.display === '') {
|
|
|
|
buffer = new Buffer(bufSize);
|
|
while (true) {
|
|
rsize = 0;
|
|
|
|
try {
|
|
rsize = fs.readSync(fd, buffer, 0, bufSize);
|
|
} catch (e) {
|
|
if (e.code === 'EOF') { break; }
|
|
|
|
res = _readlineShell(options);
|
|
if (res.error) { throw res.error; }
|
|
input += res.input;
|
|
break;
|
|
}
|
|
|
|
if (rsize === 0) { break; }
|
|
input += buffer.toString(encoding, 0, rsize);
|
|
if (/[\r\n]$/.test(input)) { break; }
|
|
}
|
|
|
|
} else {
|
|
res = _readlineShell(options);
|
|
if (res.error) { throw res.error; }
|
|
input = res.input;
|
|
}
|
|
|
|
if (isOpened && !process.stdin.isTTY) { fs.closeSync(fd); }
|
|
}
|
|
|
|
return options.noTrim ? input.replace(/[\r\n]+$/, '') : input.trim();
|
|
}
|
|
|
|
function _readlineShell(options) {
|
|
var cmdArgs = [], execArgs, res = {},
|
|
execOptions = {
|
|
env: process.env,
|
|
// ScriptPW (Win XP and Server2003) for `noEchoBack` needs TTY stream.
|
|
// In this case, If STDIN isn't TTY, an error is thrown.
|
|
stdio: [process.stdin],
|
|
encoding: encoding
|
|
};
|
|
|
|
if (options.noEchoBack) {
|
|
cmdArgs.push('noechoback');
|
|
}
|
|
|
|
if (childProc.execFileSync) {
|
|
console.warn('execFileSync');
|
|
if (IS_WIN) {
|
|
process.env.Q = '"'; // The quote (") that isn't escaped.
|
|
execArgs = ['/V:ON', '/S', '/C',
|
|
'%Q%' + SHELL_CMD + '%Q%' +
|
|
cmdArgs.map(function(arg) { return ' %Q%' + arg + '%Q%'; }).join('') +
|
|
' & exit !ERRORLEVEL!'];
|
|
} else {
|
|
execArgs = [SHELL_CMD].concat(cmdArgs);
|
|
}
|
|
try {
|
|
res.input = childProc.execFileSync(SHELL_PATH, execArgs, execOptions);
|
|
} catch (e) { // non-zero exit code
|
|
res.error = new Error(DEFAULT_ERR_MSG);
|
|
res.error.method = 'execFileSync';
|
|
res.error.command = SHELL_CMD;
|
|
res.error.args = cmdArgs;
|
|
res.error.shellMessage = e.stderr.trim();
|
|
}
|
|
} else {
|
|
console.warn('_execSyncByFile');
|
|
res = _execSyncByFile(cmdArgs, execOptions);
|
|
}
|
|
if (!res.error) { res.input = res.input.replace(/^'|'$/g, ''); }
|
|
|
|
return res;
|
|
}
|
|
|
|
// piping via files (node v0.10-)
|
|
function _execSyncByFile(cmdArgs, execOptions) {
|
|
|
|
function getTempfile(name) {
|
|
var path = require('path'), filepath, suffix = '', fd;
|
|
tempdir = tempdir || require('os').tmpdir();
|
|
|
|
while (true) {
|
|
filepath = path.join(tempdir, name + suffix);
|
|
try {
|
|
fd = fs.openSync(filepath, 'wx');
|
|
} catch (e) {
|
|
if (e.code === 'EEXIST') {
|
|
suffix++;
|
|
continue;
|
|
} else {
|
|
throw e;
|
|
}
|
|
}
|
|
fs.closeSync(fd);
|
|
break;
|
|
}
|
|
return filepath;
|
|
}
|
|
|
|
var execArgs, res = {},
|
|
pathStdout = getTempfile('readline-sync.stdout'),
|
|
pathStderr = getTempfile('readline-sync.stderr'),
|
|
pathStatus = getTempfile('readline-sync.status'),
|
|
pathDone = getTempfile('readline-sync.done'),
|
|
crypto = require('crypto'), shasum, decipher, password;
|
|
|
|
shasum = crypto.createHash(ALGORITHM_HASH);
|
|
shasum.update('' + process.pid + (salt++) + Math.random());
|
|
password = shasum.digest('hex');
|
|
decipher = crypto.createDecipher(ALGORITHM_CIPHER, password);
|
|
|
|
if (IS_WIN) {
|
|
process.env.Q = '"'; // The quote (") that isn't escaped.
|
|
// `()` for ignore space by echo
|
|
execArgs = ['/V:ON', '/S', '/C',
|
|
'(' + SHELL_PATH + ' /V:ON /S /C %Q%%Q%' + SHELL_CMD + '%Q%' +
|
|
cmdArgs.map(function(arg) { return ' %Q%' + arg + '%Q%'; }).join('') +
|
|
' & (echo !ERRORLEVEL!)>%Q%' + pathStatus + '%Q%%Q%) 2>%Q%' + pathStderr + '%Q%' +
|
|
' |%Q%' + process.execPath + '%Q% %Q%' + __dirname + '\\encrypt.js%Q%' +
|
|
' %Q%' + ALGORITHM_CIPHER + '%Q% %Q%' + password + '%Q%' +
|
|
' >%Q%' + pathStdout + '%Q%' +
|
|
' & (echo 1)>%Q%' + pathDone + '%Q%'];
|
|
} else {
|
|
execArgs = ['-c',
|
|
// Use `()`, not `{}` for `-c` (text param)
|
|
'(' + SHELL_PATH + ' "' + SHELL_CMD + '"' +
|
|
cmdArgs.map(function(arg) { return ' "' + arg + '"'; }).join('') +
|
|
'; echo $?>"' + pathStatus + '") 2>"' + pathStderr + '"' +
|
|
' |"' + process.execPath + '" "' + __dirname + '/encrypt.js"' +
|
|
' "' + ALGORITHM_CIPHER + '" "' + password + '"' +
|
|
' >"' + pathStdout + '"' +
|
|
'; echo 1 >"' + pathDone + '"'];
|
|
}
|
|
childProc.spawn(SHELL_PATH, execArgs, execOptions);
|
|
|
|
while (fs.readFileSync(pathDone, {encoding: encoding}).trim() !== '1') {}
|
|
if (fs.readFileSync(pathStatus, {encoding: encoding}).trim() === '0') {
|
|
res.input =
|
|
decipher.update(fs.readFileSync(pathStdout, {encoding: 'binary'}), 'hex', encoding) +
|
|
decipher.final(encoding);
|
|
} else {
|
|
res.error = new Error(DEFAULT_ERR_MSG);
|
|
res.error.method = '_execSyncByFile';
|
|
res.error.command = SHELL_CMD;
|
|
res.error.args = cmdArgs;
|
|
res.error.shellMessage = fs.readFileSync(pathStderr, {encoding: encoding}).trim();
|
|
}
|
|
|
|
fs.unlinkSync(pathStdout);
|
|
fs.unlinkSync(pathStderr);
|
|
fs.unlinkSync(pathStatus);
|
|
fs.unlinkSync(pathDone);
|
|
|
|
return res;
|
|
}
|
|
|
|
// for dev
|
|
exports._useShellSet = function(use) { useShell = use; };
|
|
|
|
exports.setPrint = function(fnc) { print = fnc; };
|
|
|
|
exports.setPrompt = function(newPrompt) {
|
|
/* jshint eqnull:true */
|
|
if (newPrompt != null) {
|
|
/* jshint eqnull:false */
|
|
promptText = newPrompt;
|
|
}
|
|
return promptText;
|
|
};
|
|
|
|
exports.setEncoding = function(newEncoding) {
|
|
if (typeof newEncoding === 'string') {
|
|
encoding = newEncoding;
|
|
}
|
|
return encoding;
|
|
};
|
|
|
|
exports.setBufferSize = function(newBufSize) {
|
|
if (typeof newBufSize === 'number') {
|
|
bufSize = newBufSize;
|
|
}
|
|
return bufSize;
|
|
};
|
|
|
|
exports.prompt = function(options) {
|
|
options = options || {};
|
|
options.display = promptText + '';
|
|
options.keyIn = false;
|
|
return _readlineSync(options);
|
|
};
|
|
|
|
exports.question = function(query, options) {
|
|
options = options || {};
|
|
/* jshint eqnull:true */
|
|
options.display = query != null ? query + '' : '';
|
|
/* jshint eqnull:false */
|
|
options.keyIn = false;
|
|
return _readlineSync(options);
|
|
};
|
|
|
|
exports.keyIn = function(query, options) {
|
|
options = options || {};
|
|
/* jshint eqnull:true */
|
|
options.display = query != null ? query + '' : '';
|
|
/* jshint eqnull:false */
|
|
options.keyIn = true;
|
|
options.noEchoBack = false;
|
|
return _readlineSync(options);
|
|
};
|