module Jets::Util
  module Sh
    def sh(command, options = {})
      quiet = options[:quiet]
      on_fail = options[:on_fail] || :raise

      puts "=> #{command}" unless quiet
      system(command)
      success = $?.success?

      case on_fail
      when :raise
        raise Jets::Error.new("Command failed: #{command}\n#{caller(1..1).first}") unless success
      when :exit
        unless success
          if quiet
            abort("Command failed: #{command}\n")
          else
            abort("Command failed: #{command}\n#{caller.join("\n")}")
          end
        end
      end

      success
    end

    def quiet_sh(command, options = {})
      options = options.merge(quiet: true) unless ENV["JETS_DEBUG"]
      sh(command, options)
    end

    def capture(command)
      out = `#{command}`.strip
      raise Jets::Error.new("Command failed: #{command}\n#{caller(1..1).first}") unless $?.success?
      out
    end
  end
end