Module: Util::Tar

Included in:
Cloudspin::Stack::Artefact::Builder
Defined in:
lib/util/tar.rb

Instance Method Summary collapse

Instance Method Details

#gzip(tarfile) ⇒ Object

gzips the underlying string in the given StringIO, returning a new StringIO representing the compressed file.



48
49
50
51
52
53
54
55
56
57
 
# File 'lib/util/tar.rb', line 48

def gzip(tarfile)
  gz = StringIO.new('')
  z = Zlib::GzipWriter.new(gz)
  z.write tarfile.string
  z.close # this is necessary!
  
  # z was closed to write the gzip footer, so
  # now we need a new StringIO
  StringIO.new gz.string
end

#tar(path) ⇒ Object

Creates a tar file in memory recursively from the given path.

Returns a StringIO whose underlying String is the contents of the tar file.



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
 
# File 'lib/util/tar.rb', line 24

def tar(path)
  tarfile = StringIO.new('')
  Gem::Package::TarWriter.new(tarfile) do |tar|
    Dir[File.join(path, '**/*')].each do |file|
      mode = File.stat(file).mode
      relative_file = file.sub /^#{Regexp::escape path}\/?/, ''
      
      if File.directory?(file)
        tar.mkdir relative_file, mode
      else
        tar.add_file relative_file, mode do |tf|
          File.open(file, 'rb') { |f| tf.write f.read }
        end
      end
    end
  end
  
  tarfile.rewind
  tarfile
end

#ungzip(tarfile) ⇒ Object

un-gzips the given IO, returning the decompressed version as a StringIO



61
62
63
64
65
66
 
# File 'lib/util/tar.rb', line 61

def ungzip(tarfile)
  z = Zlib::GzipReader.new(tarfile)
  unzipped = StringIO.new(z.read)
  z.close
  unzipped
end

#untar(io, destination) ⇒ Object

untars the given IO into the specified directory



70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/util/tar.rb', line 70

def untar(io, destination)
  Gem::Package::TarReader.new io do |tar|
    tar.each do |tarfile|
      destination_file = File.join destination, tarfile.full_name
      
      if tarfile.directory?
        FileUtils.mkdir_p destination_file
      else
        destination_directory = File.dirname(destination_file)
        FileUtils.mkdir_p destination_directory unless File.directory?(destination_directory)
        File.open destination_file, 'wb' do |f|
          f.write tarfile.read
        end
      end
    end
  end
end