add aimonb/aviator to 3rdparty

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

diff --git a/3rdparty/modules/aviator/feature/multipartable.rb b/3rdparty/modules/aviator/feature/multipartable.rb
new file mode 100644 (file)
index 0000000..28fa41e
--- /dev/null
+++ b/3rdparty/modules/aviator/feature/multipartable.rb
@@ -0,0 +1,29 @@
+#--
+# Copyright (c) 2007-2013 Nick Sieger.
+# See the file README.txt included with the distribution for
+# software license details.
+#++
+
+require 'parts'
+  module Multipartable
+    DEFAULT_BOUNDARY = "-----------RubyMultipartPost"
+    def initialize(path, params, headers={}, boundary = DEFAULT_BOUNDARY)
+      headers = headers.clone # don't want to modify the original variable
+      parts_headers = headers.delete(:parts) || {}
+      super(path, headers)
+      parts = params.map do |k,v|
+        case v
+        when Array
+          v.map {|item| Parts::Part.new(boundary, k, item, parts_headers[k]) }
+        else
+          Parts::Part.new(boundary, k, v, parts_headers[k])
+        end
+      end.flatten
+      parts << Parts::EpiloguePart.new(boundary)
+      ios = parts.map {|p| p.to_io }
+      self.set_content_type(headers["Content-Type"] || "multipart/form-data",
+                            { "boundary" => boundary })
+      self.content_length = parts.inject(0) {|sum,i| sum + i.length }
+      self.body_stream = CompositeReadIO.new(*ios)
+    end
+  end