103 lines
2.5 KiB
JavaScript
103 lines
2.5 KiB
JavaScript
var signalExit = require('signal-exit')
|
||
var spawn = require('child_process').spawn
|
||
if (process.platform === 'win32') {
|
||
spawn = require('cross-spawn')
|
||
}
|
||
|
||
module.exports = function (program, args, cb) {
|
||
var arrayIndex = arguments.length
|
||
|
||
if (typeof args === 'function') {
|
||
cb = args
|
||
args = undefined
|
||
} else {
|
||
cb = Array.prototype.slice.call(arguments).filter(function (arg, i) {
|
||
if (typeof arg === 'function') {
|
||
arrayIndex = i
|
||
return true
|
||
}
|
||
})[0]
|
||
}
|
||
|
||
cb = cb || function (done) {
|
||
return done()
|
||
}
|
||
|
||
if (Array.isArray(program)) {
|
||
args = program.slice(1)
|
||
program = program[0]
|
||
} else if (!Array.isArray(args)) {
|
||
args = [].slice.call(arguments, 1, arrayIndex)
|
||
}
|
||
|
||
var spawnOpts = { stdio: [0, 1, 2] }
|
||
|
||
if (process.send) {
|
||
spawnOpts.stdio.push('ipc')
|
||
}
|
||
|
||
var child = spawn(program, args, spawnOpts)
|
||
|
||
var childExited = false
|
||
var unproxySignals = proxySignals(child)
|
||
process.on('exit', childHangup)
|
||
function childHangup () {
|
||
child.kill('SIGHUP')
|
||
}
|
||
|
||
child.on('close', function (code, signal) {
|
||
// Allow the callback to inspect the child’s exit code and/or modify it.
|
||
process.exitCode = signal ? 128 + signal : code
|
||
|
||
cb(function () {
|
||
unproxySignals()
|
||
process.removeListener('exit', childHangup)
|
||
childExited = true
|
||
if (signal) {
|
||
// If there is nothing else keeping the event loop alive,
|
||
// then there's a race between a graceful exit and getting
|
||
// the signal to this process. Put this timeout here to
|
||
// make sure we're still alive to get the signal, and thus
|
||
// exit with the intended signal code.
|
||
setTimeout(function () {}, 200)
|
||
process.kill(process.pid, signal)
|
||
} else {
|
||
// Equivalent to process.exit() on Node.js >= 0.11.8
|
||
process.exit(process.exitCode)
|
||
}
|
||
})
|
||
})
|
||
|
||
if (process.send) {
|
||
process.removeAllListeners('message')
|
||
|
||
child.on('message', function (message, sendHandle) {
|
||
process.send(message, sendHandle)
|
||
})
|
||
|
||
process.on('message', function (message, sendHandle) {
|
||
child.send(message, sendHandle)
|
||
})
|
||
}
|
||
|
||
return child
|
||
}
|
||
|
||
function proxySignals (child) {
|
||
var listeners = {}
|
||
signalExit.signals().forEach(function (sig) {
|
||
listeners[sig] = function () {
|
||
child.kill(sig)
|
||
}
|
||
process.on(sig, listeners[sig])
|
||
})
|
||
|
||
return unproxySignals
|
||
|
||
function unproxySignals () {
|
||
for (var sig in listeners) {
|
||
process.removeListener(sig, listeners[sig])
|
||
}
|
||
}
|
||
}
|