mrkamel/tempfile_for

require "tempfile"

module TempfileFor
  class Tempfile < ::Tempfile
    def self.build(options = {})
      open options[:suffix] ? ["tempfile", options[:suffix]] : "tempfile", :encoding => options[:encoding]
    end

    def copy(options = {})
      tempfile = self.class.build(options)

      File.open(path, :encoding => options[:encoding]) { |stream| tempfile.write_ext stream }

      tempfile.rewind
      tempfile
    end

    def write_ext(io_or_data)
      if io_or_data.respond_to?(:read)
        while data = io_or_data.read(1024)
          write data
        end
      else
        write io_or_data
      end
    end
  end
end