class Sensucronic

# File lib/sensucronic.rb, line 57
def cli(argv = ARGV)
  parse_options(argv)
  run
  issue_report
end

# File lib/sensucronic.rb, line 49
def dryrun
  config[:dryrun]
end

# File lib/sensucronic.rb, line 111
def exitcode
  # be like the bash on signals
  status.exited? ? status.exitstatus : 128 + exitsignal
end

# File lib/sensucronic.rb, line 107
def exitsignal
  status.termsig || status.stopsig # I think stop sig will hang?
end

# File lib/sensucronic.rb, line 125
def fields
  config[:field]
    .map { |f| f.split(/:\s*/,2) }
    .each_with_object({}) { |(k,v), h| h[k.to_sym] = v }
end

# File lib/sensucronic.rb, line 75
def file_socket
  "/dev/tcp/localhost/#{port}"
end

# File lib/sensucronic.rb, line 63
def issue_report
  if dryrun
    report_stdout
  else
    report_socket
  end
end

# File lib/sensucronic.rb, line 101
def output
  [stdout, stderr, output_status]
    .select {|msg| msg && msg != "" }
    .join("\n");
end

# File lib/sensucronic.rb, line 97
def output_status
  status.to_s unless status.exited?
end

# File lib/sensucronic.rb, line 53
def port
  config[:port]
end

# File lib/sensucronic.rb, line 131
def report
  {
    command:  Shellwords.shelljoin(cli_arguments),
    output:   output,
    status:   sensu_status,
    exitcode: exitcode,
    agent:    self.class.to_s.downcase,
  }.tap do |r|
    r[:exitsignal] = status.termsig  if status.termsig
    r[:source]     = config[:source] if config[:source]
    r.update(fields) { |k,o,n| o }
  end
end

# File lib/sensucronic.rb, line 79
def report_socket
  # TODO if needed actually connect via tcp to the
  # client input socket.
  File.open(file_socket, 'w') do |sock|
    sock.write(report.to_json)
  end
rescue Errno::ENOENT => e
  STDERR.puts "failed to submit to sensu client input socket"
  STDERR.puts e.message
  exit 127
end

# File lib/sensucronic.rb, line 71
def report_stdout
  puts JSON.pretty_generate(report)
end

# File lib/sensucronic.rb, line 91
def run
  @stdout, @stderr, @status = Open3.capture3(*cli_arguments)
rescue Errno::ENOENT => e
  @stdout, @stderr, @status = e.class, e.message, $?
end

# File lib/sensucronic.rb, line 116
def sensu_status
  case exitcode
  when 0..2
    exitcode
  else
    3
  end
end