require 'hanami/utils/hash' require_relative 'body_parser/class_interface' module Hanami module Middleware # @since 1.3.0 # @api private class BodyParser # @since 1.3.0 # @api private CONTENT_TYPE = 'CONTENT_TYPE'.freeze # @since 1.3.0 # @api private MEDIA_TYPE_MATCHER = /\s*[;,]\s*/.freeze # @since 1.3.0 # @api private RACK_INPUT = 'rack.input'.freeze # @since 1.3.0 # @api private ROUTER_PARAMS = 'router.params'.freeze # @api private ROUTER_PARSED_BODY = 'router.parsed_body'.freeze # @api private FALLBACK_KEY = '_'.freeze extend ClassInterface def initialize(app, parsers) @app = app @parsers = build_parsers(parsers) end def call(env) body = env[RACK_INPUT].read return @app.call(env) if body.empty? env[RACK_INPUT].rewind # somebody might try to read this stream if (parser = @parsers[media_type(env)]) env[ROUTER_PARSED_BODY] = parser.parse(body) env[ROUTER_PARAMS] = _symbolize(env[ROUTER_PARSED_BODY]) end @app.call(env) end private def build_parsers(parser_names) parser_names = Array(parser_names) return {} if parser_names.empty? parser_names.each_with_object({}) { |name, parsers| parser = self.class.for(name) parser.mime_types.each do |mime| parsers[mime] = parser end } end # @api private def _symbolize(body) if body.is_a?(Hash) Utils::Hash.deep_symbolize(body) else { FALLBACK_KEY => body } end end # @api private def _parse(env, body) @parsers[ media_type(env) ].parse(body) end # @api private def media_type(env) if ct = content_type(env) ct.split(MEDIA_TYPE_MATCHER, 2).first.downcase end end # @api private def content_type(env) content_type = env[CONTENT_TYPE] content_type.nil? || content_type.empty? ? nil : content_type end end end end