class PolicyManager::ZipGenerator

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/policy_manager/exporter/zip_generator.rb, line 17
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/policy_manager/exporter/zip_generator.rb, line 23
def write
  entries = Dir.entries(@input_dir) - %w(. ..)

  ::Zip::File.open(@output_file, ::Zip::File::CREATE) do |zipfile|
    write_entries entries, '', zipfile
  end
end

Private Instance Methods

put_into_archive(disk_file_path, zipfile, zipfile_path) click to toggle source
# File lib/policy_manager/exporter/zip_generator.rb, line 54
def put_into_archive(disk_file_path, zipfile, zipfile_path)
  zipfile.get_output_stream(zipfile_path) do |f|
    f.write(File.open(disk_file_path, 'rb').read)
  end
end
recursively_deflate_directory(disk_file_path, zipfile, zipfile_path) click to toggle source
# File lib/policy_manager/exporter/zip_generator.rb, line 48
def recursively_deflate_directory(disk_file_path, zipfile, zipfile_path)
  zipfile.mkdir zipfile_path
  subdir = Dir.entries(disk_file_path) - %w(. ..)
  write_entries subdir, zipfile_path, zipfile
end
write_entries(entries, path, zipfile) click to toggle source

A helper method to make the recursion work.

# File lib/policy_manager/exporter/zip_generator.rb, line 34
def write_entries(entries, path, zipfile)
  entries.each do |e|
    zipfile_path = path == '' ? e : File.join(path, e)
    disk_file_path = File.join(@input_dir, zipfile_path)
    puts "Deflating #{disk_file_path}"

    if File.directory? disk_file_path
      recursively_deflate_directory(disk_file_path, zipfile, zipfile_path)
    else
      put_into_archive(disk_file_path, zipfile, zipfile_path)
    end
  end
end