tempfile_for 0.0.9 → 0.1.0

data/README.md

@@ -28,8 +28,7 @@ $ gem install tempfile_for
 
 ### for
 
-TempfileFor is a very tiny gem, with only a few lines of code.
-However, it can save you lines of ugly code.
+TempfileFor is a very tiny gem, but it can save you lines of ugly code.
 
 To get a quick introduction into what TempfileFor does, check this out:
 

data/lib/tempfile_for/tempfile.rb

@@ -0,0 +1,28 @@
+
+require "tempfile"
+
+module TempfileFor
+  class Tempfile < ::Tempfile
+    def copy(options = {})
+      encoding = options[:encoding]
+
+      tempfile = self.class.open("tempfile", :encoding => encoding)
+
+      File.open path, :encoding => encoding do |stream|
+        tempfile.write_ext stream
+      end
+
+      tempfile.rewind
+      tempfile
+    end
+
+    def write_ext(io_or_data)
+      if io_or_data.respond_to?(:read)
+        write(io_or_data.read(1024)) until io_or_data.eof?
+      else
+        write io_or_data
+      end
+    end
+  end
+end
+

data/lib/tempfile_for/version.rb

@@ -1,3 +1,3 @@
 module TempfileFor
-  VERSION = "0.0.9"
+  VERSION = "0.1.0"
 end

data/lib/tempfile_for.rb

@@ -1,28 +1,30 @@
 
 require "tempfile_for/version"
+require "tempfile_for/tempfile"
 require "tempfile"
 
 class Tempfile
-  def self.for(data, options = {})
-    blank :encoding => data.encoding, :read => options[:read] do |tempfile|
-      tempfile.write data
+  def self.for(io_or_data, options = {})
+    blank options.merge(:encoding => options[:encoding] || io_or_data.encoding) do |tempfile|
+      tempfile.write_ext io_or_data
       tempfile.flush
 
-      yield tempfile
+      yield tempfile if block_given?
     end
-  end
+  end 
 
   def self.blank(options = {})
-    tempfile = open("tempfile", :encoding => options[:encoding])
+    encoding = options[:encoding]
+
+    tempfile = TempfileFor::Tempfile.open("tempfile", :encoding => encoding)
 
-    yield tempfile
+    yield tempfile if block_given?
 
     tempfile.flush
-    tempfile.rewind
 
-    options[:read] != false ? tempfile.read : tempfile
+    options[:read] != false ? File.read(tempfile.path, :encoding => encoding) : tempfile.copy(:encoding => encoding)
   ensure
-    tempfile.close! if options[:read] != false
+    tempfile.close!
   end
 end
 

data/test/tempfile_for/tempfile_test.rb

@@ -0,0 +1,31 @@
+
+require File.expand_path("../../test_helper", __FILE__)
+require "stringio"
+
+class TempfileFor::TemfileTest < Minitest::Test
+  def test_copy
+    tempfile = TempfileFor::Tempfile.open("tempfile", :encoding => Encoding::BINARY)
+    tempfile.write "test"
+    tempfile.flush
+
+    copy = tempfile.copy(:encoding => Encoding::ISO_8859_1).read
+
+    assert_equal "test", copy
+    assert_equal Encoding::ISO_8859_1, copy.encoding
+  end
+
+  def test_write_ext
+    tempfile = TempfileFor::Tempfile.open("tempfile")
+    tempfile.write_ext StringIO.new("test")
+    tempfile.rewind
+
+    assert_equal "test", tempfile.read
+
+    tempfile = TempfileFor::Tempfile.open("tempfile")
+    tempfile.write_ext "test"
+    tempfile.rewind
+
+    assert_equal "test", tempfile.read
+  end
+end
+

data/test/tempfile_for_test.rb

@@ -1,8 +1,5 @@
 
-require File.expand_path("../../lib/tempfile_for", __FILE__)
-
-require "minitest"
-require "minitest/autorun"
+require File.expand_path("../test_helper", __FILE__)
 
 class TempfileForTest < Minitest::Test
   def test_tempfile_for

data/test/test_helper.rb

@@ -0,0 +1,6 @@
+
+require File.expand_path("../../lib/tempfile_for", __FILE__)
+
+require "minitest"
+require "minitest/autorun"
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tempfile_for
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -40,9 +40,12 @@ files:
 - README.md
 - Rakefile
 - lib/tempfile_for.rb
+- lib/tempfile_for/tempfile.rb
 - lib/tempfile_for/version.rb
 - tempfile_for.gemspec
+- test/tempfile_for/tempfile_test.rb
 - test/tempfile_for_test.rb
+- test/test_helper.rb
 homepage: https://github.com/mrkamel/tempfile_for
 licenses:
 - MIT