TINYhr/tiny_client

require 'active_support/json/decoding'
require 'active_support/gzip'

module TinyClient
  # Wrap the curl request response.
  class Response
    attr_reader :url, :body_str, :header_str, :status, :code

    def initialize(curb)
      @url = curb.url
      @body_str = curb.body_str
      @header_str = curb.header_str
      @status = curb.status
      @code = @status.to_i
    end

    # Convert the response json body into an hash.
    # @return the parsed response body
    def parse_body
      body = gzip? ? gzip_decompress : body_str
      ActiveSupport::JSON.decode(body) if body.present?
    end

    # Parse the X-Total-Count header
    # @return [Integer] the value of the X-Total-Count header, or nil if not present
    def total_count
      count = header_str[/X-Total-Count: ([0-9]+)/, 1]
      count.present? ? count.to_i : nil
    end

    # @return true if this response Content-Encoding is gzip
    def gzip?
      /Content-Encoding: gzip/ =~ header_str
    end

    # @return true if the http request has been successful.
    def success?
      (200..299).cover?(@code)
    end

    # @return true if the HTTP status code of this response correspond to an client or server error.
    def error?
      @code >= 400
    end

    # @return true if the HTTP status code of this response is 404
    def not_found_error?
      @code == 404
    end

    # @return true if the HTTP status code of this response correspond to an client error.
    def client_error?
      (400..499).cover?(@code)
    end

    # @return true if the HTTP status code of this response correspond to a server error.
    def server_error?
      @code >= 500
    end

    # @return true if the HTTP status code of this response correspond to a redirect.
    def redirect?
      (300..399).cover?(@code)
    end

    # @return Hash with url, status, body and headers fields
    def to_hash
      {
        'url'     => url,
        'status'  => status,
        'body'    => (parse_body rescue body_str),
        'headers' => (parse_headers rescue header_str)
      }
    end

    # @return String of #to_hash
    def to_s
      to_hash.to_s
    end

    protected

    def parse_headers
      {}.tap do |headers|
        header_str.to_s.each_line do |header|
          next if header.index(':').nil?
          key, value = header.split(':', 2)
          headers[key] = value.to_s.strip
        end
      end
    end

    def gzip_decompress
      ActiveSupport::Gzip.decompress(body_str)
    end
  end
end