Stanko K.R.
58 lines
1.3 KiB
Ruby
58 lines
1.3 KiB
Ruby
class Account::DataTransfer::ZipFile
|
|
class << self
|
|
def create
|
|
raise ArgumentError, "No block given" unless block_given?
|
|
|
|
Tempfile.new([ "export", ".zip" ]).tap do |tempfile|
|
|
Zip::File.open(tempfile.path, create: true) do |zip|
|
|
yield new(zip)
|
|
end
|
|
end
|
|
end
|
|
|
|
def open(path)
|
|
raise ArgumentError, "No block given" unless block_given?
|
|
|
|
Zip::File.open(path.to_s) do |zip|
|
|
yield new(zip)
|
|
end
|
|
end
|
|
end
|
|
|
|
def initialize(zip)
|
|
@zip = zip
|
|
end
|
|
|
|
def add_file(path, content = nil, compress: true, &block)
|
|
if block_given?
|
|
compression = compress ? nil : Zip::Entry::STORED
|
|
zip.get_output_stream(path, nil, nil, compression, &block)
|
|
else
|
|
zip.get_output_stream(path) { |f| f.write(content) }
|
|
end
|
|
end
|
|
|
|
def glob(pattern)
|
|
zip.glob(pattern).map(&:name).sort
|
|
end
|
|
|
|
def read(file_path, &block)
|
|
entry = zip.find_entry(file_path)
|
|
raise ArgumentError, "File not found in zip: #{file_path}" unless entry
|
|
raise ArgumentError, "Cannot read directory entry: #{file_path}" if entry.directory?
|
|
|
|
if block_given?
|
|
yield entry.get_input_stream
|
|
else
|
|
entry.get_input_stream.read
|
|
end
|
|
end
|
|
|
|
def exists?(file_path)
|
|
zip.find_entry(file_path).present?
|
|
end
|
|
|
|
private
|
|
attr_reader :zip
|
|
end
|