add aimonb/aviator to 3rdparty

[dsa-puppet.git] / 3rdparty / modules / aviator / feature / parts.rb

diff --git a/3rdparty/modules/aviator/feature/parts.rb b/3rdparty/modules/aviator/feature/parts.rb
new file mode 100644 (file)
index 0000000..c06cbd9
--- /dev/null
+++ b/3rdparty/modules/aviator/feature/parts.rb
@@ -0,0 +1,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