# rubocop:disable all
# => disabling rubocop to preserve original style
require 'serverspec'
require 'pathname'

raise "Unexpected Serverspec version #{Serverspec::VERSION}" unless Serverspec::VERSION == '0.7.10'
module Serverspec::Backend
  class Exec
    # ORIGINAL
    #def run_command(cmd, opts={})
    #  cmd = build_command(cmd)
    #  cmd = add_pre_command(cmd)
    #  stdout = `#{build_command(cmd)} 2>&1`
    #  # In ruby 1.9, it is possible to use Open3.capture3, but not in 1.8
    #  #stdout, stderr, status = Open3.capture3(cmd)
    #
    #  if @example
    #    @example.metadata[:command] = cmd
    #    @example.metadata[:stdout]  = stdout
    #  end
    #
    #  { :stdout => stdout, :stderr => nil,
    #    :exit_status => $?, :exit_signal => nil }
    #end
    #/ORIGINAL

    def run_command(cmd, opts={})
      cmd = build_command(cmd)
      cmd = add_pre_command(cmd)
      cmd = build_command(cmd)

      if use_chroot?
        chroot_stdout = `#{chroot_cmd(cmd)} 2>&1`

        stdout = get_stdout(chroot_stdout)
        exit_status = get_exit_status(chroot_stdout)
      else
        stdout = `#{cmd} 2>&1`
        exit_status = $?
      end
      # In ruby 1.9, it is possible to use Open3.capture3, but not in 1.8
      #stdout, stderr, status = Open3.capture3(cmd)

      if @example
        @example.metadata[:command] = cmd
        @example.metadata[:stdout] = stdout
      end

      { :stdout => stdout, :stderr => nil,
        :exit_status => exit_status, :exit_signal => nil }
    end

    attr_accessor :chroot_dir

    private

    def get_stdout(chroot_stdout)
      chroot_stdout.gsub(/#{exit_code_regexp}/, '')
    end

    def get_exit_status(chroot_stdout)
      chroot_command_exit_status = chroot_stdout.match(/#{exit_code_regexp}/)[1]
      chroot_command_exit_status.to_i
    end

    def use_chroot?
      chroot_dir
    end

    def exit_code_token
      'EXIT_CODE='
    end

    def exit_code_regexp
      "#{exit_code_token}(\\d+)\s*\n"
    end

    def chroot_cmd(cmd)
      %Q{
sudo chroot #{chroot_dir} /bin/bash <<CHROOT_CMD
  #{cmd} 2>&1; echo #{exit_code_token}\\$?
CHROOT_CMD
}
    end
  end
end