1 files changed, 72 insertions, 0 deletions
diff --git a/lib/tempfile.rb b/lib/tempfile.rb
new file mode 100644
index 0000000000..66d96200b5
--- /dev/null
+++ b/lib/tempfile.rb
@@ -0,0 +1,72 @@
+#
+# $Id$
+#
+# The class for temporary files.
+#  o creates a temporary file, which name is "basename.pid.n" with mode "w+".
+#  o Tempfile objects can be used like IO object.
+#  o with tmpfile.close(true) created temporary files are removed.
+#  o created files are also removed on script termination.
+#  o with Tempfile#open, you can reopen the temporary file.
+#  o file mode of the temporary files are 0600.
+
+require 'delegate'
+require 'final'
+
+class Tempfile < SimpleDelegater
+  Max_try = 10
+
+  def initialize(basename, tmpdir = '/tmp')
+    umask = File.umask(0177)
+    begin
+      n = 0
+      while true
+	begin
+	  @tmpname = sprintf('%s/%s.%d.%d', tmpdir, basename, $$, n)
+	  unless File.exist?(@tmpname)
+	    File.symlink(tmpdir, @tmpname + '.lock')
+	    break
+	  end
+	rescue
+	  raise "cannot generate tmpfile `%s'" % @tmpname if n >= Max_try
+	  #sleep(1)
+	end
+	n += 1
+      end
+
+      @clean_files = proc {|id| 
+	if File.exist?(@tmpname)
+	  File.unlink(@tmpname) 
+	end
+	if File.exist?(@tmpname + '.lock')
+	  File.unlink(@tmpname + '.lock')
+	end
+      }
+      ObjectSpace.define_finalizer(self, @clean_files)
+
+      @tmpfile = File.open(@tmpname, 'w+')
+      super(@tmpfile)
+      File.unlink(@tmpname + '.lock')
+    ensure
+      File.umask(umask)
+    end
+  end
+
+  def Tempfile.open(*args)
+    Tempfile.new(*args)
+  end
+
+  def open
+    @tmpfile.close if @tmpfile
+    @tmpfile = File.open(@tmpname, 'r+')
+    __setobj__(@tmpfile)
+  end
+
+  def close(real=false)
+    @tmpfile.close if @tmpfile
+    @tmpfile = nil
+    if real
+      @clean_files.call
+      ObjectSpace.undefine_finalizer(self)
+    end
+  end
+end