boot-stacker/lib/run_cmd.coffee

Boot your stack. Simple, flexible, magical manager for running tasks and applications of all sorts.

fs = require 'fs'
_ = require 'lodash'
mexpect = require './mexpect'
util = require './util'
state_lib = require './state_lib'
proc_lib = require './proc_lib'
_log = (args...) -> util.debug_log.apply null, [__filename].concat args

module.exports.run_cmd =
  # Run a command
  # if id is passed in, will prefix output with that
  # ({cmd: [string], task_name: string, cwd: string, env: map, silent: boolean, pipe_output: boolean}) -> child_process
  ({cmd, id, cwd, env, silent, pipe_output, close_stdin, direct}) ->
    cmd = if _.isString(cmd) then [cmd] else cmd
    cwd ?= process.cwd()

    missing_dir_error = "This task has an invalid working directory (#{cwd}).  Please check your configuration."
    try
      unless fs.statSync(cwd).isDirectory()
        throw new Error missing_dir_error
    catch e
      _log e.stack
      throw new Error missing_dir_error

    silent ?= false
    pipe_output ?= true
    close_stdin ?= true
    direct ?= false

    shell_env = _.assign {}, state_lib.get_stacker_state().shell_env, env
    env = state_lib.get_shell_env shell_env

    mproc = mexpect.spawn
      id: id
      cmd: cmd
      cwd: cwd
      env: env
      silent: silent
      pipe_output: pipe_output

    stop_indicator = util.start_progress_indicator()
    mproc.proc.stdout.on 'readable', stop_indicator
    mproc.proc.stdout.on 'data', stop_indicator
    mproc.proc.stderr.on 'readable', stop_indicator
    mproc.proc.stderr.on 'data', stop_indicator

    child_id = id ? "#{util.regex_extract(/\/([\w-]+)$/, cwd)}-#{cmd.join('-')}-#{mproc.proc.pid}".replace(/\s/g, '-')

    mproc.on_close.then ([exit_code, signal]) ->
      proc_lib.remove_proc child_id
      unless silent
        util.print_process_status child_id, exit_code, signal
      util.kill_tree mproc.proc.pid
    .catch (error) ->
      console.log error
      _log error.stack

    proc_lib.add_proc child_id, mproc.proc

    if pipe_output
      util.prefix_pipe_output child_id, mproc.proc

    if close_stdin
      mproc.proc.stdin.end()

    unless silent
      util.print util.pretty_command_str cmd, shell_env

    mproc