Sha256: 7a54d1325ebfceec8e311ac5c071354e6670377483775c7ab7d44508284d1b8c

Contents?: true

Size: 1.23 KB

Versions: 2

Compression:

Stored size: 1.23 KB

Contents

class Dir
  Backports.make_block_optional self, :each, :test_on => Dir.new(".")
  class << self
    Backports.make_block_optional self, :foreach, :test_on => Dir, :arg => "."
    
    def mktmpdir(prefix_suffix=nil, tmpdir=nil)
      raise NoMethodError, "undefined method `mktmpdir' for Dir:Class; you must require 'tmpdir'" unless method_defined? :tmpdir
      
      case prefix_suffix
      when nil
        prefix = "d"
        suffix = ""
      when String
        prefix = prefix_suffix
        suffix = ""
      when Array
        prefix = prefix_suffix[0]
        suffix = prefix_suffix[1]
      else
        raise ArgumentError, "unexpected prefix_suffix: #{prefix_suffix.inspect}"
      end
      tmpdir ||= Dir.tmpdir
      t = Time.now.strftime("%Y%m%d")
      n = nil
      begin
        path = "#{tmpdir}/#{prefix}#{t}-#{$$}-#{rand(0x100000000).to_s(36)}"
        path << "-#{n}" if n
        path << suffix
        Dir.mkdir(path, 0700)
      rescue Errno::EEXIST
        n ||= 0
        n += 1
        retry
      end

      if block_given?
        begin
          yield path
        ensure
          FileUtils.remove_entry_secure path
        end
      else
        path
      end
    end unless method_defined? :mktmpdir
    
  end
end
  

Version data entries

2 entries across 2 versions & 2 rubygems

Version Path
marcandre-backports-1.8.3 lib/backports/1.8.7/dir.rb
backports-1.8.3 lib/backports/1.8.7/dir.rb