require "httpi/adapter/base"
require "httpi/response"

module HTTPI
  module Adapter

    # = HTTPI::Adapter::HTTPClient
    #
    # Adapter for the HTTPClient client.
    # http://rubygems.org/gems/httpclient
    class HTTPClient < Base

      register :httpclient, :deps => %w(httpclient)

      def initialize(request)
        @request = request
        @client = ::HTTPClient.new
      end

      attr_reader :client

      # Executes arbitrary HTTP requests.
      # @see HTTPI.request
      def request(method)
        setup_client
        respond_with @client.request(method, @request.url, nil, @request.body, @request.headers)
      rescue OpenSSL::SSL::SSLError
        raise SSLError
      rescue Errno::ECONNREFUSED   # connection refused
        $!.extend ConnectionError
        raise
      end

      private

      def setup_client
        basic_setup
        setup_auth if @request.auth.http?
        setup_ssl_auth if @request.auth.ssl?
      end

      def basic_setup
        @client.proxy = @request.proxy if @request.proxy
        @client.connect_timeout = @request.open_timeout if @request.open_timeout
        @client.receive_timeout = @request.read_timeout if @request.read_timeout
      end

      def setup_auth
        @client.set_auth @request.url, *@request.auth.credentials
      end

      def setup_ssl_auth
        ssl = @request.auth.ssl

        unless ssl.verify_mode == :none
          @client.ssl_config.client_cert = ssl.cert
          @client.ssl_config.client_key = ssl.cert_key
          @client.ssl_config.add_trust_ca(ssl.ca_cert_file) if ssl.ca_cert_file
        end

        @client.ssl_config.verify_mode = ssl.openssl_verify_mode
        @client.ssl_config.ssl_version = ssl.ssl_version if ssl.ssl_version
      end

      def respond_with(response)
        Response.new response.code, Hash[*response.header.all.flatten], response.content
      end

    end
  end
end