Sha256: 69be7a7a1d7a5a3c0679a269763137e94db336fe8c349412484fadb2f3f1908f

Contents?: true

Size: 1.32 KB

Versions: 24

Compression:

Stored size: 1.32 KB

Contents

require 'fileutils'

module TmpFile

  TMPDIR = "/tmp/#{ENV['USER']}/tmpfiles" 
  FileUtils.mkdir_p TMPDIR unless File.exist? TMPDIR

  def self.tmpdir=(tmpdir)
    TMPDIR.replace tmpdir
    FileUtils.mkdir_p TMPDIR unless File.exist? TMPDIR
  end

  def self.tmpdir
    TMPDIR
  end


  # Creates a random file name, with the given suffix and a random number
  # up to +max+
  def self.random_name(s = "", max = 10000000)
    n = rand(max)
    s + n.to_s
  end

  # Creates a random filename in the temporary directory
  def self.tmp_file(s = "",max=10000000)
    File.join(TMPDIR, random_name(s,max))
  end

  def self.with_file(content = nil, erase = true, options = {})
    prefix = options[:prefix] || ""
    tmpfile = tmp_file prefix
    if options[:extension]
      tmpfile += ".#{options[:extension]}"
    end

    if IO === content
      Misc.consume_stream(content, false, tmpfile)
    else
      File.open(tmpfile, 'w') do |f| f.write content end if content != nil
    end

    result = yield(tmpfile)

    FileUtils.rm_rf tmpfile if File.exist?(tmpfile) and erase

    result
  end

  def self.with_dir(erase = true, options = {})
    prefix = options[:prefix] || ""
    tmpdir = tmp_file prefix

    FileUtils.mkdir_p tmpdir

    result = yield(tmpdir)

    FileUtils.rm_rf tmpdir if File.exist?(tmpdir) and erase

    result
  end
end

Version data entries

24 entries across 24 versions & 1 rubygems

Version Path
rbbt-util-5.21.121 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.120 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.119 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.118 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.116 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.115 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.114 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.113 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.112 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.111 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.110 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.109 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.107 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.106 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.104 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.103 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.102 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.101 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.100 lib/rbbt/util/tmpfile.rb
rbbt-util-5.21.99 lib/rbbt/util/tmpfile.rb