amis-rpc-design/node_modules/readline/readline.js
2023-10-07 19:42:30 +08:00

61 lines
1.9 KiB
JavaScript

var fs = require('fs'),
EventEmitter = require('events').EventEmitter,
util = require('util');
var readLine = module.exports = function(file, opts) {
if (!(this instanceof readLine)) return new readLine(file, opts);
EventEmitter.call(this);
opts = opts || {};
opts.maxLineLength = opts.maxLineLength || 4096; // 4K
opts.retainBuffer = !!opts.retainBuffer; //do not convert to String prior to invoking emit 'line' event
var self = this,
lineBuffer = new Buffer(opts.maxLineLength),
lineLength = 0,
lineCount = 0,
byteCount = 0,
emit = function(lineCount, byteCount) {
try {
var line = lineBuffer.slice(0, lineLength);
self.emit('line', opts.retainBuffer? line : line.toString(), lineCount, byteCount);
} catch (err) {
self.emit('error', err);
} finally {
lineLength = 0; // Empty buffer.
}
};
this.input = ('string' === typeof file) ? fs.createReadStream(file, opts) : file;
this.input.on('open', function(fd) {
self.emit('open', fd);
})
.on('data', function(data) {
for (var i = 0; i < data.length; i++) {
if (data[i] == 10 || data[i] == 13) { // Newline char was found.
if (data[i] == 10) {
lineCount++;
emit(lineCount, byteCount);
}
} else {
lineBuffer[lineLength] = data[i]; // Buffer new line data.
lineLength++;
}
byteCount++;
}
})
.on('error', function(err) {
self.emit('error', err);
})
.on('end', function() {
// Emit last line if anything left over since EOF won't trigger it.
if (lineLength) {
lineCount++;
emit(lineCount, byteCount);
}
self.emit('end');
})
.on('close', function() {
self.emit('close');
});
};
util.inherits(readLine, EventEmitter);