index.js 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. const {spawn} = require('child_process')
  2. const inferOwner = require('infer-owner')
  3. const isPipe = (stdio = 'pipe', fd) =>
  4. stdio === 'pipe' || stdio === null ? true
  5. : Array.isArray(stdio) ? isPipe(stdio[fd], fd)
  6. : false
  7. // 'extra' object is for decorating the error a bit more
  8. const promiseSpawn = (cmd, args, opts, extra = {}) => {
  9. const cwd = opts.cwd || process.cwd()
  10. const isRoot = process.getuid && process.getuid() === 0
  11. const { uid, gid } = isRoot ? inferOwner.sync(cwd) : {}
  12. return promiseSpawnUid(cmd, args, {
  13. ...opts,
  14. cwd,
  15. uid,
  16. gid
  17. }, extra)
  18. }
  19. const stdioResult = (stdout, stderr, {stdioString, stdio}) =>
  20. stdioString ? {
  21. stdout: isPipe(stdio, 1) ? Buffer.concat(stdout).toString() : null,
  22. stderr: isPipe(stdio, 2) ? Buffer.concat(stderr).toString() : null,
  23. }
  24. : {
  25. stdout: isPipe(stdio, 1) ? Buffer.concat(stdout) : null,
  26. stderr: isPipe(stdio, 2) ? Buffer.concat(stderr) : null,
  27. }
  28. const promiseSpawnUid = (cmd, args, opts, extra) => {
  29. let proc
  30. const p = new Promise((res, rej) => {
  31. proc = spawn(cmd, args, opts)
  32. const stdout = []
  33. const stderr = []
  34. const reject = er => rej(Object.assign(er, {
  35. cmd,
  36. args,
  37. ...stdioResult(stdout, stderr, opts),
  38. ...extra,
  39. }))
  40. proc.on('error', reject)
  41. if (proc.stdout) {
  42. proc.stdout.on('data', c => stdout.push(c)).on('error', reject)
  43. proc.stdout.on('error', er => reject(er))
  44. }
  45. if (proc.stderr) {
  46. proc.stderr.on('data', c => stderr.push(c)).on('error', reject)
  47. proc.stderr.on('error', er => reject(er))
  48. }
  49. proc.on('close', (code, signal) => {
  50. const result = {
  51. cmd,
  52. args,
  53. code,
  54. signal,
  55. ...stdioResult(stdout, stderr, opts),
  56. ...extra
  57. }
  58. if (code || signal)
  59. rej(Object.assign(new Error('command failed'), result))
  60. else
  61. res(result)
  62. })
  63. })
  64. p.stdin = proc.stdin
  65. p.process = proc
  66. return p
  67. }
  68. module.exports = promiseSpawn