try again, with puppetforge modules, correctly included now

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

diff --git a/3rdparty/modules/aviator/feature/composite_io.rb b/3rdparty/modules/aviator/feature/composite_io.rb
new file mode 100644 (file)
index 0000000..4ba7cf5
--- /dev/null
+++ b/3rdparty/modules/aviator/feature/composite_io.rb
@@ -0,0 +1,108 @@
+#--
+# Copyright (c) 2007-2012 Nick Sieger.
+# See the file README.txt included with the distribution for
+# software license details.
+#++
+
+# Concatenate together multiple IO objects into a single, composite IO object
+# for purposes of reading as a single stream.
+#
+# Usage:
+#
+#     crio = CompositeReadIO.new(StringIO.new('one'), StringIO.new('two'), StringIO.new('three'))
+#     puts crio.read # => "onetwothree"
+#
+class CompositeReadIO
+  # Create a new composite-read IO from the arguments, all of which should
+  # respond to #read in a manner consistent with IO.
+  def initialize(*ios)
+    @ios = ios.flatten
+    @index = 0
+  end
+
+  # Read from IOs in order until `length` bytes have been received.
+  def read(length = nil, outbuf = nil)
+    got_result = false
+    outbuf = outbuf ? outbuf.replace("") : ""
+
+    while io = current_io
+      if result = io.read(length)
+        got_result ||= !result.nil?
+        result.force_encoding("BINARY") if result.respond_to?(:force_encoding)
+        outbuf << result
+        length -= result.length if length
+        break if length == 0
+      end
+      advance_io
+    end
+    (!got_result && length) ? nil : outbuf
+  end
+
+  def rewind
+    @ios.each { |io| io.rewind }
+    @index = 0
+  end
+
+  private
+
+  def current_io
+    @ios[@index]
+  end
+
+  def advance_io
+    @index += 1
+  end
+end
+
+# Convenience methods for dealing with files and IO that are to be uploaded.
+class UploadIO
+  # Create an upload IO suitable for including in the params hash of a
+  # Net::HTTP::Post::Multipart.
+  #
+  # Can take two forms. The first accepts a filename and content type, and
+  # opens the file for reading (to be closed by finalizer).
+  #
+  # The second accepts an already-open IO, but also requires a third argument,
+  # the filename from which it was opened (particularly useful/recommended if
+  # uploading directly from a form in a framework, which often save the file to
+  # an arbitrarily named RackMultipart file in /tmp).
+  #
+  # Usage:
+  #
+  #     UploadIO.new("file.txt", "text/plain")
+  #     UploadIO.new(file_io, "text/plain", "file.txt")
+  #
+  attr_reader :content_type, :original_filename, :local_path, :io, :opts
+
+  def initialize(filename_or_io, content_type, filename = nil, opts = {})
+    io = filename_or_io
+    local_path = ""
+    if io.respond_to? :read
+      # in Ruby 1.9.2, StringIOs no longer respond to path
+      # (since they respond to :length, so we don't need their local path, see parts.rb:41)
+      local_path = filename_or_io.respond_to?(:path) ? filename_or_io.path : "local.path"
+    else
+      io = File.open(filename_or_io)
+      local_path = filename_or_io
+    end
+    filename ||= local_path
+
+    @content_type = content_type
+    @original_filename = File.basename(filename)
+    @local_path = local_path
+    @io = io
+    @opts = opts
+  end
+
+  def self.convert!(io, content_type, original_filename, local_path)
+    raise ArgumentError, "convert! has been removed. You must now wrap IOs using:\nUploadIO.new(filename_or_io, content_type, filename=nil)\nPlease update your code."
+  end
+
+  def method_missing(*args)
+    @io.send(*args)
+  end
+
+  def respond_to?(meth, include_all = false)
+    @io.respond_to?(meth, include_all) || super(meth, include_all)
+  end
+end