lib/packer/binary/compressor.rb in packer-binary-0.1.1 vs lib/packer/binary/compressor.rb in packer-binary-0.2.0

@@ -1,32 +1,26 @@
-# 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"
-#   output_file = "/tmp/out.zip"
-#   zf = Zip.new(directoryToZip, output_file)
-#   zf.write()
 require 'zip'
 
 module Packer
   module Binary
+    # This class is used to zip and unzip files and directories
     class Compressor
       class << self
         include Packer::Binary::Helpers
 
         # Zip the input directory.
         def write(input_dir, output_file)
-          entries = Dir.entries(input_dir); entries.delete("."); entries.delete("..")
-          io = Zip::File.open(output_file, Zip::File::CREATE);
+          entries = Dir.entries(input_dir)
+          entries.delete('.')
+          entries.delete('..')
+          io = Zip::File.open(output_file, Zip::File::CREATE)
 
-          write_entries(entries, "", io, input_dir, output_file)
-          io.close();
+          write_entries(entries, '', io, input_dir, output_file)
+          io.close
         end
 
+        # Unzip the input zipfile
         def extract(input_file, output_dir)
           Zip::File.open(input_file) do |zip_file|
             # Handle entries one by one
             zip_file.each do |entry|
               # Extract to file/directory/symlink
@@ -39,22 +33,23 @@
 
         private
 
         # A helper method to make the recursion work.
         def write_entries(entries, path, io, input_dir, output_file)
-
-          entries.each { |e|
-            zip_file_path = path == "" ? e : File.join(path, e)
+          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)
+            if File.directory?(disk_file_path)
               io.mkdir(zip_file_path)
-              subdir =Dir.entries(disk_file_path); subdir.delete("."); subdir.delete("..")
+              subdir = Dir.entries(disk_file_path)
+              subdir.delete('.')
+              subdir.delete('..')
               write_entries(subdir, zip_file_path, io, input_dir, output_file)
             else
-              io.get_output_stream(zip_file_path) { |f| f.puts(File.open(disk_file_path, "rb").read())}
+              io.get_output_stream(zip_file_path) { |f| f.puts(File.open(disk_file_path, 'rb').read) }
             end
-          }
+          end
         end
       end
     end
   end
 end