+++ /dev/null
-#--
-# Copyright (c) 2007-2013 Nick Sieger.
-# See the file README.txt included with the distribution for
-# software license details.
-#++
-
-module Parts
- module Part #:nodoc:
- def self.new(boundary, name, value, headers = {})
- headers ||= {} # avoid nil values
- if file?(value)
- FilePart.new(boundary, name, value, headers)
- else
- ParamPart.new(boundary, name, value, headers)
- end
- end
-
- def self.file?(value)
- value.respond_to?(:content_type) && value.respond_to?(:original_filename)
- end
-
- def length
- @part.length
- end
-
- def to_io
- @io
- end
- end
-
- class ParamPart
- include Part
- def initialize(boundary, name, value, headers = {})
- @part = build_part(boundary, name, value, headers)
- @io = StringIO.new(@part)
- end
-
- def length
- @part.bytesize
- end
-
- def build_part(boundary, name, value, headers = {})
- part = ''
- part << "--#{boundary}\r\n"
- part << "Content-Disposition: form-data; name=\"#{name.to_s}\"\r\n"
- part << "Content-Type: #{headers["Content-Type"]}\r\n" if headers["Content-Type"]
- part << "\r\n"
- part << "#{value}\r\n"
- end
- end
-
- # Represents a part to be filled from file IO.
- class FilePart
- include Part
- attr_reader :length
- def initialize(boundary, name, io, headers = {})
- file_length = io.respond_to?(:length) ? io.length : File.size(io.local_path)
- @head = build_head(boundary, name, io.original_filename, io.content_type, file_length,
- io.respond_to?(:opts) ? io.opts.merge(headers) : headers)
- @foot = "\r\n"
- @length = @head.bytesize + file_length + @foot.length
- @io = CompositeReadIO.new(StringIO.new(@head), io, StringIO.new(@foot))
- end
-
- def build_head(boundary, name, filename, type, content_len, opts = {}, headers = {})
- trans_encoding = opts["Content-Transfer-Encoding"] || "binary"
- content_disposition = opts["Content-Disposition"] || "form-data"
-
- part = ''
- part << "--#{boundary}\r\n"
- part << "Content-Disposition: #{content_disposition}; name=\"#{name.to_s}\"; filename=\"#{filename}\"\r\n"
- part << "Content-Length: #{content_len}\r\n"
- if content_id = opts["Content-ID"]
- part << "Content-ID: #{content_id}\r\n"
- end
-
- if headers["Content-Type"] != nil
- part << "Content-Type: " + headers["Content-Type"] + "\r\n"
- else
- part << "Content-Type: #{type}\r\n"
- end
-
- part << "Content-Transfer-Encoding: #{trans_encoding}\r\n"
- part << "\r\n"
- end
- end
-
- # Represents the epilogue or closing boundary.
- class EpiloguePart
- include Part
- def initialize(boundary)
- @part = "--#{boundary}--\r\n\r\n"
- @io = StringIO.new(@part)
- end
- end
-end