rubyでディレクトリをzip圧縮する

rubyでzip圧縮といえばrubyzipというgemですが、githubに載っているサンプルスクリプトだと、日本語が文字化けしたり、なんと大容量ファイルを圧縮しようとした途端...その解決方法たちをご紹介します。

スポンサーリンク

rubyでディレクトリをzip圧縮する

rubyでzip圧縮をする必要があり、rubyzipというgemを使用することに決めました。また、ディレクトリを圧縮するスクリプトも、rubyzipのGitHubに載っていたので、簡単にできると思っていたのですが...

以下の2箇所でつまづく事になりました。。

1. 日本語のディレクトリ名を使った場合に文字化けする
2. ディレクトリの合計容量が大きかった場合に、処理が途中でコケる

これらを踏まえ、修正したスクリプトがこちらです。

require 'zip'

# 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:
#   directory_to_zip = "/tmp/input"
#   output_file = "/tmp/out.zip"
#   zf = ZipFileGenerator.new(directory_to_zip, output_file)
#   zf.write()
class ZipFileGenerator
    # Initialize with the directory to zip and the location of the output archive.
    def initialize(input_dir, output_file)
        @input_dir = input_dir
        @output_file = output_file
    end
  
    # Zip the input directory.
    def write
        entries = Dir.entries(@input_dir) - %w(. ..)
  
        Zip.unicode_names = true
        ::Zip::File.open(@output_file, ::Zip::File::CREATE) do |zipfile|
            write_entries entries, '', zipfile
        end
    end
  
    private
  
    # A helper method to make the recursion work.
    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
  
    def recursively_deflate_directory(disk_file_path, zipfile, zipfile_path)
        zipfile.mkdir zipfile_path.encode('cp932')
        subdir = Dir.entries(disk_file_path) - %w(. ..)
        write_entries subdir, zipfile_path, zipfile
    end
  
    def put_into_archive(disk_file_path, zipfile, zipfile_path)
        zipfile.get_output_stream(zipfile_path.encode('cp932')) do |f|
            #f.write(File.open(disk_file_path, 'rb').read)
            IO.copy_stream(File.open(disk_file_path, 'rb'), f)
        end
    end
end

サンプルコードでは、大容量ファイルでメモリリークが発生していたみたいですね...！