ThunderNetworkRaD/readline-sync
2
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions

Change a method to open TTY.

Browse source
This commit is contained in:
anseki 2015-03-08 00:19:40 +09:00
parent 0e22bc9f28
commit f5d4a8231f
2 changed files with 141 additions and 62 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
lib/read.bat
View file

@ -7,7 +7,6 @@ if "%~1"=="noechoback" (
if ERRORLEVEL 1 exit /b 1
) else (
set /p INPUT=<CON >CON
if ERRORLEVEL 1 exit /b 1
)
set /p ="'%INPUT%'"<NUL
endlocal

202
lib/readline-sync.js
View file

@ -18,90 +18,163 @@ var
fs = require('fs'),
childProc = require('child_process'),
stdin = process.stdin,
stdout = process.stdout,
promptText = '> ',
encoding = 'utf8',
bufSize = 1024,
useShell = false, print, tempdir, salt = 0;
print,
tempdir, salt = 0, useShell = false;
function _readlineSync(display, options) {
var input = '', res, buffer = new Buffer(bufSize), rsize;
options = options || {};
function _readlineSync(options) { // options.display is string
var input = '', fd, buffer, rsize, res, isOpened, fsBind, constBind;
if (display !== '') { // null and undefined were excluded.
if (typeof print === 'function') { print(display, encoding); }
stdout.write(display + '', encoding);
}
if (useShell || options.noEchoBack) { // Try reading via shell
if (options.display !== '' && typeof print === 'function')
{ print(options.display, encoding); }
if (useShell || options.noEchoBack) {
res = _readlineShell(options);
if (res.error) {
if (display !== '') { stdout.write('\n', encoding); } // Return from prompt line.
throw res.error;
}
if (res.error) { throw res.error; }
input = res.input;
} else {
stdin.resume();
while (true) {
rsize = 0;
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 {
rsize = fs.readSync(stdin.fd, buffer, 0, bufSize);
} catch (e) {
if (e.code === 'EOF') { break; }
// Try reading via shell
res = _readlineShell(options);
if (res.error) {
if (display !== '') { stdout.write('\n', encoding); } // Return from prompt line.
throw res.error;
console.warn('/dev/tty TRY');
fd = fs.openSync('/dev/tty', 'rs+');
isOpened = true;
if (options.display !== '') {
fs.writeSync(fd, options.display);
options.display = '';
}
input += res.input;
break;
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; }
}
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;
}
stdin.pause();
if (isOpened && !process.stdin.isTTY) { fs.closeSync(fd); }
}
return options.noTrim ? input.replace(/[\r\n]+$/, '') : input.trim();
}
function _readlineShell(options) {
var args = [], res = {},
var cmdArgs = [], execArgs, res = {},
execOptions = {
env: process.env,
stdio: [stdin], // ScriptPW needs piped stdin
// 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) {
args.push('noechoback');
cmdArgs.push('noechoback');
}
stdin.pause(); // re-start in child process
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,
(IS_WIN ? ['/C', SHELL_CMD] : [SHELL_CMD]).concat(args), execOptions);
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 = args;
res.error.args = cmdArgs;
res.error.shellMessage = e.stderr.trim();
}
} else {
res = _execSyncByFile(args, execOptions);
console.warn('_execSyncByFile');
res = _execSyncByFile(cmdArgs, execOptions);
}
if (!res.error) { res.input = res.input.replace(/^'|'$/g, ''); }
@ -109,7 +182,7 @@ function _readlineShell(options) {
}
// piping via files (node v0.10-)
function _execSyncByFile(args, execOptions) {
function _execSyncByFile(cmdArgs, execOptions) {
function getTempfile(name) {
var path = require('path'), filepath, suffix = '', fd;
@ -133,7 +206,7 @@ function _execSyncByFile(args, execOptions) {
return filepath;
}
var cmdArgs, res = {},
var execArgs, res = {},
pathStdout = getTempfile('readline-sync.stdout'),
pathStderr = getTempfile('readline-sync.stderr'),
pathStatus = getTempfile('readline-sync.status'),
@ -146,27 +219,28 @@ function _execSyncByFile(args, execOptions) {
decipher = crypto.createDecipher(ALGORITHM_CIPHER, password);
if (IS_WIN) {
// The quote (") is escaped by node before parsed by shell. Then use ENV{Q}.
process.env.Q = '"';
process.env.Q = '"'; // The quote (") that isn't escaped.
// `()` for ignore space by echo
cmdArgs = ['/V:ON', '/S', '/C',
'(' + SHELL_PATH + ' /V:ON /S /C %Q%%Q%' + SHELL_CMD + '%Q% ' + args.join(' ') +
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 {
cmdArgs = ['-c',
execArgs = ['-c',
// Use `()`, not `{}` for `-c` (text param)
'(' + SHELL_PATH + ' "' + SHELL_CMD + '" ' + args.join(' ') +
'(' + 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, cmdArgs, execOptions);
childProc.spawn(SHELL_PATH, execArgs, execOptions);
while (fs.readFileSync(pathDone, {encoding: encoding}).trim() !== '1') {}
if (fs.readFileSync(pathStatus, {encoding: encoding}).trim() === '0') {
@ -177,7 +251,7 @@ function _execSyncByFile(args, execOptions) {
res.error = new Error(DEFAULT_ERR_MSG);
res.error.method = '_execSyncByFile';
res.error.command = SHELL_CMD;
res.error.args = args;
res.error.args = cmdArgs;
res.error.shellMessage = fs.readFileSync(pathStderr, {encoding: encoding}).trim();
}
@ -218,21 +292,27 @@ exports.setBufferSize = function(newBufSize) {
};
exports.prompt = function(options) {
return _readlineSync(promptText, options);
options = options || {};
options.display = promptText + '';
options.keyIn = false;
return _readlineSync(options);
};
exports.question = function(query, options) {
return _readlineSync(
options = options || {};
/* jshint eqnull:true */
query != null ? query : '',
options.display = query != null ? query + '' : '';
/* jshint eqnull:false */
options);
options.keyIn = false;
return _readlineSync(options);
};
exports.keyIn = function(query) {
return _readlineSync(
exports.keyIn = function(query, options) {
options = options || {};
/* jshint eqnull:true */
query != null ? query : '',
options.display = query != null ? query + '' : '';
/* jshint eqnull:false */
{keyIn: true});
options.keyIn = true;
options.noEchoBack = false;
return _readlineSync(options);
};