kemal/src/kemal/param_parser.cr

require "json"

# ParamParser parses the request contents including query_params and body
# and converts them into a params hash which you can within the environment
# context.
alias AllParamTypes = Nil | String | Int64 | Float64 | Bool | Hash(String, JSON::Type) | Array(JSON::Type)

class Kemal::ParamParser
  URL_ENCODED_FORM = "application/x-www-form-urlencoded"
  APPLICATION_JSON = "application/json"

  def initialize(@request : HTTP::Request)
    @url = {} of String => String
    @query = HTTP::Params.new({} of String => Array(String))
    @body = HTTP::Params.new({} of String => Array(String))
    @json = {} of String => AllParamTypes
    @url_parsed = false
    @query_parsed = false
    @body_parsed = false
    @json_parsed = false
  end

  {% for method in %w(url query body json) %}
  def {{method.id}}
    # check memoization
    return @{{method.id}} if @{{method.id}}_parsed

    parse_{{method.id}}
    # memoize
    @{{method.id}}_parsed = true
    @{{method.id}}
  end
  {% end %}

  def parse_body
    return if (@request.headers["Content-Type"]? =~ /#{URL_ENCODED_FORM}/).nil?
    @body = parse_part(@request.body)
  end

  def parse_query
    @query = parse_part(@request.query)
  end

  def parse_url
    if params = @request.url_params
      params.each do |key, value|
        @url[key.as(String)] = value.as(String)
      end
    end
  end

  # Parses JSON request body if Content-Type is `application/json`.
  # If request body is a JSON Hash then all the params are parsed and added into `params`.
  # If request body is a JSON Array it's added into `params` as `_json` and can be accessed
  # like params["_json"]
  def parse_json
    return unless @request.body && @request.headers["Content-Type"]? == APPLICATION_JSON

    body = @request.body.as(String)
    case json = JSON.parse(body).raw
    when Hash
      json.each do |key, value|
        @json[key.as(String)] = value.as(AllParamTypes)
      end
    when Array
      @json["_json"] = json
    end
  end

  def parse_part(part)
    HTTP::Params.parse(part || "")
  end
end