masson/src/commands/run.coffee

Module execution engine for cluster deployments.

readline = require 'readline'

pad = require 'pad'
params = require '../params'
config = require '../config'
run = require '../run'

print_time = (time) ->
  if time > 1000*60
    "#{time / 1000}m"
  if time > 1000
    "#{time / 1000}s"
  else
    "#{time}ms"

module.exports = ->
  # running = []
  rl = readline.createInterface process.stdin, process.stdout
  rl.setPrompt ''
  rl.on 'SIGINT', process.exit
  hostlength = 20
  for s in config.servers then hostlength = Math.max(hostlength, s.host.length+2)
  params = params.parse()
  multihost = params.hosts?.length isnt 1 and config.servers.length isnt 1
  times = {}
  run(config, params)
  .on 'context', (ctx) ->
    ctx
    .on 'action', (status) ->
      return if ctx.action.hidden
      status ?= ctx.DISABLED
      if status is ctx.STARTED
        times[ctx.config.host] = Date.now()
      else
        time = Date.now() - times[ctx.config.host]
      line = ''
      line += "#{pad ctx.config.host, hostlength}"
      line += "#{pad ctx.action.name, 40}"
      statusmsg = switch status
        when ctx.PASS then "\x1b[36m#{ctx.PASS_MSG}\x1b[39m"
        when ctx.OK then "\x1b[36m#{ctx.OK_MSG}\x1b[39m"
        when ctx.FAILED then "\x1b[35m#{ctx.FAILED_MSG}\x1b[39m"
        when ctx.DISABLED then "\x1b[36m#{ctx.DISABLED_MSG}\x1b[39m"
        when ctx.TODO then "\x1b[36m#{ctx.TODO_MSG}\x1b[39m"
        when ctx.PARTIAL then "\x1b[36m#{ctx.PARTIAL_MSG}\x1b[39m"
        when ctx.STOP then "\x1b[35m#{ctx.STOP_MSG}\x1b[39m"
        when ctx.TIMEOUT then "\x1b[35m#{ctx.TIMEOUT_MSG}\x1b[39m"
        when ctx.WARN then "\x1b[33m#{ctx.WARN_MSG}\x1b[39m"
        when ctx.STARTED then "\x1b[36m#{ctx.STARTED_MSG}\x1b[39m"
        when ctx.INAPPLICABLE then "\x1b[36m#{ctx.INAPPLICABLE_MSG}\x1b[39m"
        else (if typeof status is 'number' then "INVALID CODE" else "\x1b[36m#{status}\x1b[39m")
      line += "#{pad statusmsg, 20}"
      line += "#{print_time time}" if time
      if status is ctx.STARTED
        rl.write line unless multihost
      else
        rl.cursor = 0
        rl.line = ''
        rl._refreshLine()
        if status isnt ctx.DISABLED
          rl.write line
          rl.write '\n'
    .on 'wait', ->
      return if multihost
      statusmsg = "\x1b[36mWAIT\x1b[39m"
      line = ''
      line += "#{pad ctx.config.host, hostlength}"
      line += "#{pad ctx.action.name, 40}"
      line += "#{pad statusmsg, 20}"
      rl.cursor = 0
      rl.line = ''
      rl._refreshLine()
      rl.write line
    .on 'waited', ->
      return if multihost
      statusmsg = "\x1b[36m#{ctx.STARTED_MSG}\x1b[39m"
      line = ''
      line += "#{pad ctx.config.host, hostlength}"
      line += "#{pad ctx.action.name, 40}"
      line += "#{pad statusmsg, 20}"
      rl.cursor = 0
      rl.line = ''
      rl._refreshLine()
      rl.write line
  .on 'server', (ctx, status) ->
    return unless config.servers.length
    line = ''
    line += "#{pad ctx.config.host, hostlength+40}"
    line += switch status
      when ctx.OK then "\x1b[36m#{ctx.OK_MSG}\x1b[39m"
      when ctx.FAILED then "\x1b[36m#{ctx.FAILED_MSG}\x1b[39m"
      else "INVALID CODE"
    rl.write line
    rl.write '\n'
  .on 'end', ->
    rl.write "\x1b[32mInstallation is finished\x1b[39m\n"
    rl.close()
  .on 'error', (err) ->
    rl.write '\n'
    rl.write "\x1b[31m#{err.stack or err.message}\x1b[39m"
    rl.close()