1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#--
# 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
|
