module Mixlib::ShellOut::Helper

# File lib/mixlib/shellout/helper.rb, line 38
def shell_out(*args, **options)
  options = options.dup
  options = __maybe_add_timeout(self, options)
  if options.empty?
    shell_out_compacted(*__clean_array(*args))
  else
    shell_out_compacted(*__clean_array(*args), **options)
  end
end

# File lib/mixlib/shellout/helper.rb, line 48
def shell_out!(*args, **options)
  options = options.dup
  options = __maybe_add_timeout(self, options)
  if options.empty?
    shell_out_compacted!(*__clean_array(*args))
  else
    shell_out_compacted!(*__clean_array(*args), **options)
  end
end

# File lib/mixlib/shellout/helper.rb, line 81
def __apply_default_env(options)
  options = options.dup
  default_env = options.delete(:default_env)
  default_env = true if default_env.nil?
  if default_env
    env_key = options.key?(:env) ? :env : :environment
    options[env_key] = {
      "LC_ALL" => __config[:internal_locale],
      "LANGUAGE" => __config[:internal_locale],
      "LANG" => __config[:internal_locale],
      __env_path_name => default_paths,
    }.update(options[env_key] || {})
  end
  options
end

# File lib/mixlib/shellout/helper.rb, line 140
def __clean_array(*args)
  args.flatten.compact.map(&:to_s)
end

# File lib/mixlib/shellout/helper.rb, line 201
def __env_path_name
  if ChefUtils.windows?
    "Path"
  else
    "PATH"
  end
end

# File lib/mixlib/shellout/helper.rb, line 193
def __io_for_live_stream
  if !STDOUT.closed? && __log.trace?
    STDOUT
  else
    nil
  end
end

# File lib/mixlib/shellout/helper.rb, line 152
def __join_whitespace(*args, quote: false)
  args.map do |arg|
    if arg.is_a?(Array)
      __join_whitespace(*arg, quote: arg.count > 1)
    else
      arg = arg.include?(" ") ? sprintf('"%s"', arg) : arg if quote
      arg.strip
    end
  end.join(" ")
end

# File lib/mixlib/shellout/helper.rb, line 64
def __maybe_add_timeout(obj, options)
  options = options.dup
  # historically resources have not properly declared defaults on their timeouts, so a default default of 900s was enforced here
  default_val = 900
  return options if options.key?(:timeout)

  # FIXME: need to nuke descendent tracker out of Chef::Provider so we can just define that class here without requiring the
  # world, and then just use symbol lookup
  if obj.class.ancestors.map(&:name).include?("Chef::Provider") && obj.respond_to?(:new_resource) && obj.new_resource.respond_to?(:timeout) && !options.key?(:timeout)
    options[:timeout] = obj.new_resource.timeout ? obj.new_resource.timeout.to_f : default_val
  end
  options
end

# File lib/mixlib/shellout/helper.rb, line 163
def __shell_out_command(*args, **options)
  if __transport_connection
    command = __join_whitespace(args)
    unless ChefUtils.windows?
      if options[:cwd]
        # as `timeout` is used, commands need to be executed in a subshell
        command = "sh -c 'cd #{options[:cwd]}; #{command}'"
      end

      if options[:input]
        command.concat "<<'COMMANDINPUT'\n"
        command.concat __join_whitespace(options[:input])
        command.concat "\nCOMMANDINPUT\n"
      end
    end

    # FIXME: train should accept run_command(*args)
    FakeShellOut.new(args, options, __transport_connection.run_command(command, options))
  else
    cmd = if options.empty?
            Mixlib::ShellOut.new(*args)
          else
            Mixlib::ShellOut.new(*args, **options)
          end
    cmd.live_stream ||= __io_for_live_stream
    cmd.run_command
    cmd
  end
end

# File lib/mixlib/shellout/helper.rb, line 114
def shell_out_compacted(*args, **options)
  options = __apply_default_env(options)
  if options.empty?
    __shell_out_command(*args)
  else
    __shell_out_command(*args, **options)
  end
end

# File lib/mixlib/shellout/helper.rb, line 123
def shell_out_compacted!(*args, **options)
  options = __apply_default_env(options)
  cmd = if options.empty?
          __shell_out_command(*args)
        else
          __shell_out_command(*args, **options)
        end
  cmd.error!
  cmd
end