66 lines
1.5 KiB
JavaScript
66 lines
1.5 KiB
JavaScript
'use strict';
|
|
var exec = require('child_process').exec;
|
|
var chalk = require('chalk');
|
|
|
|
module.exports = function (grunt) {
|
|
grunt.registerMultiTask('shell', 'Run shell commands', function () {
|
|
var cb = this.async();
|
|
var options = this.options({
|
|
stdout: true,
|
|
stderr: true,
|
|
stdin: true,
|
|
failOnError: true,
|
|
stdinRawMode: false
|
|
});
|
|
var cmd = this.data.command;
|
|
|
|
if (cmd === undefined) {
|
|
throw new Error('`command` required');
|
|
}
|
|
|
|
cmd = grunt.template.process(typeof cmd === 'function' ? cmd.apply(grunt, arguments) : cmd);
|
|
|
|
var cp = exec(cmd, options.execOptions, function (err, stdout, stderr) {
|
|
if (typeof options.callback === 'function') {
|
|
options.callback.call(this, err, stdout, stderr, cb);
|
|
} else {
|
|
if (err && options.failOnError) {
|
|
grunt.warn(err);
|
|
}
|
|
cb();
|
|
}
|
|
}.bind(this));
|
|
|
|
var captureOutput = function (child, output) {
|
|
if (grunt.option('color') === false) {
|
|
child.on('data', function (data) {
|
|
output.write(chalk.stripColor(data));
|
|
});
|
|
} else {
|
|
child.pipe(output);
|
|
}
|
|
};
|
|
|
|
grunt.verbose.writeln('Command:', chalk.yellow(cmd));
|
|
|
|
if (options.stdout || grunt.option('verbose')) {
|
|
captureOutput(cp.stdout, process.stdout);
|
|
}
|
|
|
|
if (options.stderr || grunt.option('verbose')) {
|
|
captureOutput(cp.stderr, process.stderr);
|
|
}
|
|
|
|
if (options.stdin) {
|
|
process.stdin.resume();
|
|
process.stdin.setEncoding('utf8');
|
|
|
|
if (options.stdinRawMode && process.stdin.isTTY) {
|
|
process.stdin.setRawMode(true);
|
|
}
|
|
|
|
process.stdin.pipe(cp.stdin);
|
|
}
|
|
});
|
|
};
|