95 lines
2.4 KiB
JavaScript
95 lines
2.4 KiB
JavaScript
|
|
|||
|
|
var fs = require('fs');
|
|||
|
|
|
|||
|
|
// Collectable Mixin
|
|||
|
|
//
|
|||
|
|
// Provides utility methods to read from stdin and remaining arguments.
|
|||
|
|
var collectable = module.exports;
|
|||
|
|
|
|||
|
|
// XXX: tomdocify, generate in readme
|
|||
|
|
|
|||
|
|
collectable.stdin = function stdin(force, done) {
|
|||
|
|
if(!done) done = force, force = false;
|
|||
|
|
var argv = this.nopt.argv;
|
|||
|
|
var self = this;
|
|||
|
|
done = done || function(err) { err && self.emit('error', err); };
|
|||
|
|
|
|||
|
|
this._readFromStdin = true;
|
|||
|
|
|
|||
|
|
// not parsed, register done to be read when parse is called
|
|||
|
|
if(!argv) {
|
|||
|
|
this.once('stdin', done);
|
|||
|
|
return this;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// only read from stdin when no reamining args and not forced
|
|||
|
|
if(!argv.remain.length || force) {
|
|||
|
|
this.readStdin(done);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return this;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// Read files from remaining args, concat the result and call back the `done`
|
|||
|
|
// function with the concatanated result and the list of files.
|
|||
|
|
collectable.files = function files(done) {
|
|||
|
|
var argv = this.nopt.argv;
|
|||
|
|
var self = this;
|
|||
|
|
done = done || function(err) { err && self.emit('error', err); };
|
|||
|
|
|
|||
|
|
// not parsed, register done to be read when parse is called
|
|||
|
|
if(!argv) {
|
|||
|
|
this.once('files', done);
|
|||
|
|
return this;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// only read files when we actually have files to read from
|
|||
|
|
if(argv.remain.length) {
|
|||
|
|
this.readFiles(argv.remain, done);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return this;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
collectable.readStdin = function readStdin(done) {
|
|||
|
|
var data = '';
|
|||
|
|
var self = this;
|
|||
|
|
process.stdin.setEncoding('utf8');
|
|||
|
|
process.stdin.on('error', done);
|
|||
|
|
process.stdin.on('data', function(chunk){
|
|||
|
|
data += chunk;
|
|||
|
|
self.emit('stdin:data', chunk);
|
|||
|
|
}).on('end', function(){
|
|||
|
|
self.emit('stdin', null, data);
|
|||
|
|
done(null, data);
|
|||
|
|
}).resume();
|
|||
|
|
return this;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// Asynchronous walk of the remaining args, reading the content and returns
|
|||
|
|
// the concatanated result.
|
|||
|
|
collectable.readFiles = function readFiles(filepaths, done) {
|
|||
|
|
var data = '';
|
|||
|
|
var self = this;
|
|||
|
|
var files = filepaths.slice(0);
|
|||
|
|
(function read(file) {
|
|||
|
|
if(!file) {
|
|||
|
|
self.emit('files', null, data, filepaths);
|
|||
|
|
return done(null, data, filepaths);
|
|||
|
|
}
|
|||
|
|
fs.readFile(file, 'utf8', function(err, body) {
|
|||
|
|
if(err) return done(err);
|
|||
|
|
data += body;
|
|||
|
|
self.emit('files:data', body);
|
|||
|
|
read(files.shift());
|
|||
|
|
});
|
|||
|
|
})(files.shift());
|
|||
|
|
return this;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// Collect data either from stdin or the list of remaining args
|
|||
|
|
collectable.collect = function collect(done) {
|
|||
|
|
return this.stdin(done).files(done);
|
|||
|
|
};
|
|||
|
|
|