Class: RestClient::Payload::Multipart

Inherits:
Base
  • Object
show all
Defined in:
lib/restclient/payload.rb

Constant Summary collapse

EOL =
"\r\n"

Instance Method Summary collapse

Methods inherited from Base

#flatten_params, #flatten_params_array, #initialize, #inspect, #read, #short_inspect, #size

Constructor Details

This class inherits a constructor from RestClient::Payload::Base

Instance Method Details

#boundaryObject



202
203
204
# File 'lib/restclient/payload.rb', line 202

def boundary
  @boundary ||= rand(1_000_000).to_s
end

#build_stream(params) ⇒ Object



146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
# File 'lib/restclient/payload.rb', line 146

def build_stream(params)
  b = "--#{boundary}"

  @stream = Tempfile.new("RESTClient.Stream.#{rand(1000)}")
  @stream.binmode
  @stream.write(b + EOL)

  if params.is_a? Hash
    x = flatten_params(params)
  else
    x = params
  end

  last_index = x.length - 1
  x.each_with_index do |a, index|
    k, v = * a
    if v.respond_to?(:read) && v.respond_to?(:path)
      create_file_field(@stream, k, v)
    else
      create_regular_field(@stream, k, v)
    end
    @stream.write(EOL + b)
    @stream.write(EOL) unless last_index == index
  end
  @stream.write('--')
  @stream.write(EOL)
  @stream.seek(0)
end

#closeObject



215
216
217
# File 'lib/restclient/payload.rb', line 215

def close
  @stream.close!
end

#create_file_field(s, k, v) ⇒ Object



182
183
184
185
186
187
188
189
190
191
192
193
194
195
# File 'lib/restclient/payload.rb', line 182

def create_file_field(s, k, v)
  begin
    s.write("Content-Disposition: form-data;")
    s.write(" name=\"#{k}\";") unless (k.nil? || k=='')
    s.write(" filename=\"#{v.respond_to?(:original_filename) ? v.original_filename : File.basename(v.path)}\"#{EOL}")
    s.write("Content-Type: #{v.respond_to?(:content_type) ? v.content_type : mime_for(v.path)}#{EOL}")
    s.write(EOL)
    while data = v.read(8124)
      s.write(data)
    end
  ensure
    v.close if v.respond_to?(:close)
  end
end

#create_regular_field(s, k, v) ⇒ Object



175
176
177
178
179
180
# File 'lib/restclient/payload.rb', line 175

def create_regular_field(s, k, v)
  s.write("Content-Disposition: form-data; name=\"#{k}\"")
  s.write(EOL)
  s.write(EOL)
  s.write(v)
end

#handle_key(key) ⇒ Object

for Multipart do not escape the keys



207
208
209
# File 'lib/restclient/payload.rb', line 207

def handle_key key
  key
end

#headersObject



211
212
213
# File 'lib/restclient/payload.rb', line 211

def headers
  super.merge({'Content-Type' => %Q{multipart/form-data; boundary=#{boundary}}})
end

#mime_for(path) ⇒ Object



197
198
199
200
# File 'lib/restclient/payload.rb', line 197

def mime_for(path)
  mime = MIME::Types.type_for path
  mime.empty? ? 'text/plain' : mime[0].content_type
end