class PhusionPassenger::Config::DownloadAgentCommand

def self.create_option_parser(options)
  OptionParser.new do |opts|
    nl = "\n" + ' ' * 37
    opts.banner = "Usage: passenger-config download-agent [OPTIONS]\n"
    opts.separator ""
    opts.separator "  Download a precompiled #{PROGRAM_NAME} agent binary from the"
    opts.separator "  #{PROGRAM_NAME} website. The agent binary is required for #{PROGRAM_NAME}"
    opts.separator "  to function properly. Precompiled binaries are only available for Linux and"
    opts.separator "  OS X."
    opts.separator ""
    opts.separator "Options:"
    opts.on("--url-root URL", String, "Download the binary from a custom URL") do |value|
      options[:url_root] = value
    end
    opts.on("--log-prefix PREFIX", String, "Prefix all logs with the given string") do |value|
      options[:log_prefix] = value
    end
    opts.on("--log-level LEVEL", String, "Set log level (fatal,error,warn,info,#{nl}" +
      "debug). Default: info") do |value|
      case value
      when "fatal"
        options[:log_level] = Logger::FATAL
      when "error"
        options[:log_level] = Logger::ERROR
      when "warn"
        options[:log_level] = Logger::WARN
      when "info"
        options[:log_level] = Logger::INFO
      when "debug"
        options[:log_level] = Logger::DEBUG
      else
        abort "Invalid log level #{value.inspect}"
      end
    end
    opts.on("-f", "--force", "Skip sanity checks") do
      options[:force] = true
    end
    opts.on("--no-colors", "Never output colors") do
      options[:colors] = false
    end
    opts.on("--no-error-colors", "Do not colorized error messages") do
      options[:error_colors] = false
    end
    opts.on("--no-download-progress", "Never show download progress") do
      options[:show_download_progress] = false
    end
    opts.on("--no-compilation-tip", "Do not present compilation as an#{nl}" +
      "alternative way to install the agent") do
      options[:compilation_tip] = false
    end
    opts.on("--no-force-tip", "Do not print any tips regarding the#{nl}" +
      "--force parameter") do
      options[:force_tip] = false
    end
    opts.on("--skip-cache", "Do not copy the agent binary from cache") do
      options[:use_cache] = false
    end
    opts.on("--suppress-binary-unusable-message",
      "Do not print anything if the downloaded#{nl}" +
      "binary turns out to be unusable") do
      options[:suppress_binary_unusable_message] = true
    end
    opts.on("--dry-run", "Do everything except actually installing#{nl}" +
      "the binary") do
      options[:dry_run] = true
    end
    opts.on("--connect-timeout SECONDS", Integer,
      "The maximum amount of time to spend on DNS#{nl}" +
      "lookup and establishing the TCP connection.#{nl}" +
      "Default: 30") do |val|
      options[:connect_timeout] = val
    end
    opts.on("--idle-timeout SECONDS", Integer, "The maximum idle read time. Default: 30") do |val|
      options[:idle_timeout] = val
    end
    opts.on("-h", "--help", "Show this help") do
      options[:help] = true
    end
  end
end

def compile_tip_message
  return "" if !@options[:compilation_tip]
  if PhusionPassenger.build_system_dir
    result = " Please compile the agent from source instead, by running:\n\n"
    result << "  passenger-config compile-agent"
  else
    result = " Furthermore, this #{PROGRAM_NAME} installation does not "
    result << "come with any source code, so the agent binary cannot "
    result << "be compiled either. Please contact the person or "
    result << "organization who packaged #{PROGRAM_NAME} for help on this "
    result << "problem."
  end
  return result
end

def download_and_extract
  destdir = find_or_create_writable_support_binaries_dir!
  exit if @options[:dry_run]
  PhusionPassenger::Utils.mktmpdir("passenger-install.", PlatformInfo.tmpexedir) do |tmpdir|
    basename = "agent-#{PlatformInfo.cxx_binary_compatibility_id}.tar.gz"
    tarball  = "#{tmpdir}/#{basename}"
    if !download_support_file(basename, tarball)
      @logger.error "#{@colors.reset}------------------------------------------"
      @logger.fatal("Sorry, no precompiled agent binary is available for " +
        "your platform." + compile_tip_message)
      abort
    end
    @logger.info "Extracting precompiled agent binary to #{destdir}..."
    e_tmpdir = Shellwords.escape(tmpdir)
    e_tarball = Shellwords.escape(tarball)
    if !system("cd #{e_tmpdir} && tar xzf #{e_tarball}")
      @logger.fatal "The downloaded archive file could not be extracted."
      abort
    end
    if !File.exist?("#{tmpdir}/#{AGENT_EXE}")
      @logger.fatal "The downloaded archive file does not seem to " +
        "contain an agent binary. This is probably a problem in " +
        "the #{PROGRAM_NAME} website. Please report this problem to " +
        "the #{PROGRAM_NAME} authors."
      abort
    end
    @logger.info "Checking whether the downloaded binary is usable..."
    if test_binary("#{tmpdir}/#{AGENT_EXE}")
      @logger.info "The downloaded binary is usable."
    else
      if !@options[:suppress_binary_unusable_message]
        @logger.fatal "Sorry, the precompiled agent binary can not be run " +
          "your system.#{compile_tip_message}"
      end
      exit(BINARY_NOT_USABLE_EXIT_CODE)
    end
    FileUtils.mv("#{tmpdir}/#{AGENT_EXE}", "#{destdir}/#{AGENT_EXE}")
    @logger.info "#{@colors.green}Agent binary successfully download and installed."
  end
end

def download_support_file(name, output)
  @sites.each_with_index do |site, i|
    if real_download_support_file(site, name, output)
      if i > 0
        @logger.warn "#{@colors.green}Download OK!"
      else
        @logger.info "#{@colors.green}Download OK!"
      end
      return true
    elsif i != @sites.size - 1
      @logger.warn "Trying next mirror..."
    end
  end
  return false
end

def help
  puts @parser
end

def initialize_objects
  if @options[:url_root]
    @sites = [{ :url => @options[:url_root] }]
  else
    @sites = PhusionPassenger.binaries_sites
  end
  @colors = Utils::AnsiColors.new(@options[:colors])
  @logger = Logger.new(STDOUT)
  @logger.level = @options[:log_level]
  @logger.formatter = proc do |severity, datetime, progname, msg|
    if @options[:error_colors] && (severity == "FATAL" || severity == "ERROR")
      color = @colors.red
    else
      color = nil
    end
    result = ""
    msg.split("\n", -1).map do |line|
      result << "#{color}#{@options[:log_prefix]}#{line}#{@colors.reset}\n"
    end
    result
  end
end

def print_installation_error_header
  @logger.warn "------------------------------------------"
  @logger.fatal "Cannot store agent binary"
  @logger.fatal ""
end

def real_download_support_file(site, name, output)
  url = "#{site[:url]}/#{VERSION_STRING}/#{name}"
  options = {
    :cacert => site[:cacert],
    :logger => @logger,
    :use_cache => @options[:use_cache]
  }
  # connect_timeout and idle_timeout may be nil or 0, which means
  # that the default Utils::Download timeouts should be used.
  if @options[:connect_timeout] && @options[:connect_timeout] != 0
    options[:connect_timeout] = @options[:connect_timeout]
  end
  if @options[:idle_timeout] && @options[:idle_timeout] != 0
    options[:idle_timeout] = @options[:idle_timeout]
  end
  return PhusionPassenger::Utils::Download.download(url, output, options)
end

def run
  @options = {
    :log_level => Logger::INFO,
    :colors => :auto,
    :error_colors => true,
    :show_download_progress => STDOUT.tty?,
    :compilation_tip => true,
    :force_tip => true,
    :use_cache => true,
    :connect_timeout => 30,
    :idle_timeout => 30
  }
  parse_options
  initialize_objects
  sanity_check
  download_and_extract
end

def sanity_check
  return if @options[:force]
  if PhusionPassenger.find_support_binary(AGENT_EXE)
    @logger.warn "#{@colors.green}The #{PROGRAM_NAME} agent is already installed."
    if @options[:force_tip]
      @logger.warn "If you want to redownload it, re-run this program with the --force parameter."
    end
    exit
  end
  if !PhusionPassenger.installed_from_release_package?
    @logger.fatal("#{PROGRAM_NAME} was not installed from an official release " +
      "package, so you cannot download our precompiled agent binary." +
      compile_tip_message)
    if @options[:force_tip]
      @logger.warn "If you want to download it anyway, re-run this program with the --force parameter."
    end
    abort
  end
  check_for_download_tool!
end

def test_binary(filename)
  output = `env LD_BIND_NOW=1 DYLD_BIND_AT_LAUNCH=1 #{Shellwords.escape(filename)} test-binary`
  return $? && $?.exitstatus == 0 && output == "PASS\n"
end