module Selenium::WebDriver::Zipper

def add_zip_entry(zip, file, entry_name)
  entry = Zip::Entry.new(zip.name, entry_name)
  entry.follow_symlinks = true
  zip.add entry, file
end

def unzip(path)
  destination = Dir.mktmpdir('webdriver-unzip')
  FileReaper << destination
  Zip::File.open(path) do |zip|
    zip.each do |entry|
      to      = File.join(destination, entry.name)
      dirname = File.dirname(to)
      FileUtils.mkdir_p dirname
      zip.extract(entry, to)
    end
  end
  destination
end

def with_tmp_zip(&blk)
  # Don't use Tempfile since it lacks rb_file_s_rename permission on Windows.
  Dir.mktmpdir do |tmp_dir|
    zip_path = File.join(tmp_dir, 'webdriver-zip')
    Zip::File.open(zip_path, Zip::File::CREATE, &blk)
  end
end

def zip(path)
  with_tmp_zip do |zip|
    ::Find.find(path) do |file|
      add_zip_entry zip, file, file.sub("#{path}/", '') unless File.directory?(file)
    end
    zip.commit
    File.open(zip.name, 'rb') { |io| Base64.strict_encode64 io.read }
  end
end

def zip_file(path)
  with_tmp_zip do |zip|
    add_zip_entry zip, path, File.basename(path)
    zip.commit
    File.open(zip.name, 'rb') { |io| Base64.strict_encode64 io.read }
  end
end