Class: EPUB::OCF::PhysicalContainer::ArchiveZip

Inherits:
EPUB::OCF::PhysicalContainer show all
Defined in:
lib/epub/maker/ocf/physical_container/archive_zip.rb

Instance Method Summary collapse

Methods inherited from EPUB::OCF::PhysicalContainer

save, write

Instance Method Details

#write(path_name, content) ⇒ Object

TODO:

Write multiple files at once



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/epub/maker/ocf/physical_container/archive_zip.rb', line 8

def write(path_name, content)
  ::Dir.mktmpdir do |dir|
    tmp_archive_path = ::File.join(dir, ::File.basename(@container_path) + '.tmp')
    ::File.open @container_path do |archive_in|
      ::File.open tmp_archive_path, 'w' do |archive_out|
        archive_out.binmode
        Archive::Zip.open archive_in, :r do |z_in|
          Archive::Zip.open archive_out, :w do |z_out|
            updated = false
            z_in.each do |entry|
              if entry.zip_path == path_name
                entry.file_data = StringIO.new(content)
                updated = true
              end
              z_out << entry
            end
            unless updated
              entry = Archive::Zip::Entry::File.new(path_name)
              entry.file_data = StringIO.new(content)
              z_out << entry
            end
          end
        end
      end
    end
    ::File.chmod 0666 & ~::File.umask, tmp_archive_path
    ::File.rename tmp_archive_path, @container_path
  end
end