piotrmurach/github

# encoding: utf-8

require 'faraday'

module Github
  # Contains methods and attributes that act on the response returned from the 
  # request
  class Response < Faraday::Response::Middleware
    CONTENT_TYPE = 'Content-Type'.freeze

    class << self
      attr_accessor :parser
    end

    def self.define_parser(&block)
      @parser = block
    end

    def initialize(app, options = {})
      super(app)
      @content_types = Array(options[:content_type])
    end

    def process_body(env)
      env[:body] = parse(env[:body])
    end

    def parse_body?(env)
      parse_response_type?(response_type(env)) and parse_response?(env)
    end

    def response_type(env)
      type = env[:response_headers][CONTENT_TYPE].to_s
      type = type.split(';', 2).first if type.index(';')
      type
    end

    def parse_response_type?(type)
      @content_types.empty? || @content_types.any? { |pattern|
        pattern.is_a?(Regexp) ? type =~ pattern : type == pattern
      }
    end

    def parse_response?(env)
      env[:body].respond_to?(:to_str)
    end
  end # Response
end # Github