readline-sync/lib/readline-sync.js
2013-08-30 02:26:22 +09:00

52 lines
1.1 KiB
JavaScript

/*
* readlineSync
* https://github.com/anseki/readline-sync
*
* Copyright (c) 2013 anseki
* Licensed under the MIT license.
*/
'use strict';
var promptText = '> ',
encoding = 'utf8',
BUF_SIZE = 256,
fs = require('fs'),
stdin = process.stdin,
stdout = process.stdout,
buffer = new Buffer(BUF_SIZE);
var _readlineSync = function(display) {
var input = '', rsize;
if (display) { stdout.write(display, encoding); }
stdin.resume();
while ((rsize = fs.readSync(stdin.fd, buffer, 0, BUF_SIZE)) > 0) {
input += buffer.toString(encoding, 0, rsize);
if (/[\r\n]$/.test(input)) { break; }
}
stdin.pause();
return input.trim();
};
exports.setPrompt = function(newPrompt) {
if (typeof newPrompt === 'string') {
promptText = newPrompt;
}
};
exports.setEncoding = function(newEncoding) {
if (typeof newEncoding === 'string') {
encoding = newEncoding;
}
};
exports.prompt = function() {
return _readlineSync(promptText);
};
exports.question = function(query) {
return _readlineSync(typeof query === 'string' ? query : '');
};