Action "execute.assert"

const {$status} = await nikita.execute.assert({
  command: 'exit 0'
})
console.info(`Command was succeeded: ${$status}`)

const {$status} = await nikita.execute.assert({
  command: 'echo hello',
  assert: 'hello'
})
console.info(`Stdout was asserted: ${$status}`)

on_action = ({config, metadata}) ->
  config.code ?= [0] unless config.content

definitions =
  config:
    type: 'object'
    properties:
      'code':
        type: 'array'
        items:
          type: 'integer'
        description: '''
        Expected exit code, activated by default unless content is provided.
        '''
      'content':
        oneOf: [{type: 'string'}, {instanceof: 'Buffer'}, {instanceof: 'RegExp'}]
        description: '''
        Content to match, optional.
        '''
      'error':
        type: 'string'
        description: '''
        The error message to throw if assert failed.
        '''
      'not':
        $ref: 'module://@nikitajs/core/src/actions/assert#/definitions/config/properties/not'
      'trim':
        type: 'boolean'
        default: false
        description: '''
        Trim the expected content as well as the command output before
        matching.
        '''

handler = ({config}) ->
  config.content = config.content.toString() if Buffer.isBuffer config.content
  config.content = config.content.trim() if config.content and config.trim
  # Command exit code
  if config.code?
    {code} = await @execute config, $relax: true
    unless config.not
      unless code in config.code
        throw utils.error 'NIKITA_EXECUTE_ASSERT_EXIT_CODE', [
          'an unexpected exit code was encountered,'
          "got #{JSON.stringify code}"
          if config.code.length is 1
          then "while expecting #{config.code}."
          else "while expecting one of #{JSON.stringify config.code}."
        ]
    else
      if code in config.code
        throw utils.error 'NIKITA_EXECUTE_ASSERT_NOT_EXIT_CODE', [
          'an unexpected exit code was encountered,'
          "got #{JSON.stringify code}"
          if config.code.length is 1
          then "while expecting anything but #{config.code}."
          else "while expecting anything but one of #{JSON.stringify config.code}."
        ]
  # Content is a string or a buffer
  if config.content? and typeof config.content is 'string'
    res = await @execute config
    {stdout} = res
    stdout = stdout.trim() if config.trim
    unless config.not
      unless stdout is config.content
        throw utils.error 'NIKITA_EXECUTE_ASSERT_CONTENT', [
          'the command output is not matching the content,'
          "got #{JSON.stringify stdout}"
          "while expecting to match #{JSON.stringify config.content}."
        ]
    else
      if stdout is config.content
        throw utils.error 'NIKITA_EXECUTE_ASSERT_NOT_CONTENT', [
          'the command output is unfortunately matching the content,'
          "got #{JSON.stringify stdout}."
        ]
  # Content is a regexp
  if config.content? and utils.regexp.is config.content
    {stdout} = await @execute config
    stdout = stdout.trim() if config.trim
    unless config.not
      unless config.content.test stdout
        throw utils.error 'NIKITA_EXECUTE_ASSERT_CONTENT_REGEX', [
          'the command output is not matching the content regexp,'
          "got #{JSON.stringify stdout}"
          "while expecting to match #{JSON.stringify config.content}."
        ]
    else
      if config.content.test stdout
        throw utils.error 'NIKITA_EXECUTE_ASSERT_NOT_CONTENT_REGEX', [
          'the command output is unfortunately matching the content regexp,'
          "got #{JSON.stringify stdout}"
          "matching #{JSON.stringify config.content}."
        ]

module.exports =
  handler: handler
  hooks:
    on_action: on_action
  metadata:
    definitions: definitions

utils = require '../../utils'