response_parser.rb

# frozen_string_literal: true

require_relative 'response_body_parsers'

module OpenapiFirst
  ParsedResponse = Data.define(:body, :headers)

  # Parse a response
  class ResponseParser
    def initialize(headers:, content_type:)
      @headers_parser = build_headers_parser(headers)
      @body_parser = ResponseBodyParsers[content_type]
    end

    def parse(rack_response)
      ParsedResponse.new(
        body: @body_parser.call(read_body(rack_response)),
        headers: @headers_parser&.call(rack_response.headers) || {}
      )
    end

    private

    def read_body(rack_response)
      buffered_body = +''

      if rack_response.body.respond_to?(:each)
        rack_response.body.each { |chunk| buffered_body << chunk }
        return buffered_body
      end
      rack_response.body
    rescue TypeError
      raise Error, "Cannot not read response body. Response is not string-like, but is a #{rack_response.body.class}."
    end

    def build_headers_parser(headers)
      return unless headers&.any?

      headers_as_parameters = headers.map do |header|
        {
          'name' => header.name,
          'explode' => false,
          'in' => 'header',
          'schema' => header.resolved_schema
        }
      end
      OpenapiParameters::Header.new(headers_as_parameters).method(:unpack)
    end
  end
end