class ZipFileGenerator

This is a simple example which uses rubyzip to recursively generate a zip file from the contents of a specified directory. The directory itself is not included in the archive, rather just its contents.

Usage:

directoryToZip = "/tmp/input"
outputFile = "/tmp/out.zip"
zf = ZipFileGenerator.new(directoryToZip, outputFile)
zf.write()

Public Class Methods

new(input_dir, output_file) click to toggle source

Initialize with the directory to zip and the location of the output archive.

# File lib/helpers/zip_helpers.rb, line 15
def initialize(input_dir, output_file)
  @input_dir = input_dir
  @output_file = output_file
end

Public Instance Methods

write() click to toggle source

Zip the input directory.

# File lib/helpers/zip_helpers.rb, line 21
def write
  entries = Dir.entries(@input_dir)
  entries.delete('.')
  entries.delete('..')
  io = Zip::File.open(@output_file, Zip::File::CREATE)

  write_entries(entries, '', io)
  io.close
end

Private Instance Methods

write_entries(entries, path, io) click to toggle source

A helper method to make the recursion work.

# File lib/helpers/zip_helpers.rb, line 34
def write_entries(entries, path, io)
  entries.each do |e|
    zip_file_path = path == '' ? e : File.join(path, e)
    disk_file_path = File.join(@input_dir, zip_file_path)
    if File.directory?(disk_file_path)
      io.mkdir(zip_file_path)
      subdir = Dir.entries(disk_file_path)
      subdir.delete('.')
      subdir.delete('..')
      write_entries(subdir, zip_file_path, io)
    else
      io.get_output_stream(zip_file_path) { |f| f.puts(File.open(disk_file_path, 'rb').read) }
    end
  end
end