svyatov/http_wrapper

# frozen_string_literal: true

require 'net/https'

class HTTPWrapper
  KNOWN_OPTIONS_KEYS = %i[timeout verify_cert logger max_redirects user_agent].freeze

  attr_accessor :timeout, :verify_cert, :logger, :max_redirects, :user_agent

  def initialize(options = {})
    Util.validate_hash_keys options, KNOWN_OPTIONS_KEYS

    @timeout       = options.fetch(:timeout) { 10 }
    @verify_cert   = options.fetch(:verify_cert) { true }
    @logger        = options.fetch(:logger) { nil }
    @max_redirects = options.fetch(:max_redirects) { 10 }
    @user_agent    = options.fetch(:user_agent) { USER_AGENT }
  end

  %i[get post put delete].each do |method_as_symbol|
    define_method method_as_symbol do |url, params = {}|
      params[:user_agent] ||= @user_agent
      get_response Request.new(url, method_as_symbol, params)
    end

    method_as_string = method_as_symbol.to_s

    %w[ajax json ajax_json].each do |request_type|
      define_method "#{method_as_string}_#{request_type}" do |url, params = {}|
        (params[:headers] ||= {}).merge!(headers_specific_for(request_type))
        public_send method_as_symbol, url, params
      end
    end

    alias_method "#{method_as_string}_json_ajax", "#{method_as_string}_ajax_json"
  end

  def post_and_get_cookie(url, params = {})
    response = post url, params
    response['set-cookie']
  end

  def execute(request, uri)
    connection = create_connection uri
    connection.request request
  end

  private

  def get_response(request, redirects_limit = @max_redirects)
    raise TooManyRedirectsError, 'Too many redirects!' if redirects_limit == 0

    response = execute request.create, request.uri

    if response.is_a? Net::HTTPRedirection
      request.uri = response['location']
      response = get_response request, redirects_limit - 1
    end

    response
  end

  def headers_specific_for(request_type)
    self.class.const_get "#{request_type.upcase}_HEADER"
  end

  def create_connection(uri)
    connection = Net::HTTP.new uri.host, uri.port
    connection.read_timeout = @timeout
    connection.open_timeout = @timeout

    if uri.is_a? URI::HTTPS
      connection.use_ssl = true
      connection.verify_mode = OpenSSL::SSL::VERIFY_NONE unless @verify_cert
    end

    connection.set_debug_output(@logger)
    connection
  end
end