require 'hanami/middleware/body_parser/parser'
require 'hanami/utils/hash'

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

      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

        env[ROUTER_PARAMS] ||= {} # prepare params
        env[ROUTER_PARSED_BODY] = _parse(env, body)
        env[ROUTER_PARAMS]      = _symbolize(env[ROUTER_PARSED_BODY]).merge(env[ROUTER_PARAMS])

        @app.call(env)
      end

      private

      def build_parsers(parsers)
        result  = Hash.new
        args    = Array(parsers)
        return result if args.empty?

        args.each do |arg|
          parser = Parser.for(arg)

          parser.mime_types.each do |mime|
            result[mime] = parser
          end
        end

        result.default = Parser.new
        result
      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