3rdparty/modules/aviator/feature/faraday/request/multipart.rb

   1 require File.expand_path("../url_encoded", __FILE__)
   2
   3 module Faraday
   4   class Request::Multipart < Request::UrlEncoded
   5     self.mime_type = 'multipart/form-data'.freeze
   6     DEFAULT_BOUNDARY = "-----------RubyMultipartPost".freeze unless defined? DEFAULT_BOUNDARY
   7
   8     def call(env)
   9       match_content_type(env) do |params|
  10         env.request.boundary ||= DEFAULT_BOUNDARY
  11         env.request_headers[CONTENT_TYPE] += "; boundary=#{env.request.boundary}"
  12         env.body = create_multipart(env, params)
  13       end
  14       @app.call env
  15     end
  16
  17     def process_request?(env)
  18       type = request_type(env)
  19       env.body.respond_to?(:each_key) and !env.body.empty? and (
  20         (type.empty? and has_multipart?(env.body)) or
  21         type == self.class.mime_type
  22       )
  23     end
  24
  25     def has_multipart?(obj)
  26       # string is an enum in 1.8, returning list of itself
  27       if obj.respond_to?(:each) && !obj.is_a?(String)
  28         (obj.respond_to?(:values) ? obj.values : obj).each do |val|
  29           return true if (val.respond_to?(:content_type) || has_multipart?(val))
  30         end
  31       end
  32       false
  33     end
  34
  35     def create_multipart(env, params)
  36       boundary = env.request.boundary
  37       parts = process_params(params) do |key, value|
  38         Faraday::Parts::Part.new(boundary, key, value)
  39       end
  40       parts << Faraday::Parts::EpiloguePart.new(boundary)
  41
  42       body = Faraday::CompositeReadIO.new(parts)
  43       env.request_headers[Faraday::Env::ContentLength] = body.length.to_s
  44       return body
  45     end
  46
  47     def process_params(params, prefix = nil, pieces = nil, &block)
  48       params.inject(pieces || []) do |all, (key, value)|
  49         key = "#{prefix}[#{key}]" if prefix
  50
  51         case value
  52         when Array
  53           values = value.inject([]) { |a,v| a << [nil, v] }
  54           process_params(values, key, all, &block)
  55         when Hash
  56           process_params(value, key, all, &block)
  57         else
  58           all << block.call(key, value)
  59         end
  60       end
  61     end
  62   end
  63 end