Contents

require 'fileutils'
require 'rexml/document'
require 'zip'

using Module.new {
  refine Hash do
    def slice(*keys)
      keys.each_with_object({}) {|k, h|
        h[k] = self[k]
      }
    end
  end
} unless {}.respond_to?(:slice)

module Eepub
  class Epub
    XMLNS = {
      'container' => 'urn:oasis:names:tc:opendocument:xmlns:container',
      'dc'        => 'http://purl.org/dc/elements/1.1/',
    }

    def initialize(path)
      @path = path
    end

    attr_reader :path

    def title
      @title ||= Zip::File.open(path) {|zip|
        rootfile_doc = parse_xml(find_rootfile_entry(zip))

        get_title_element(rootfile_doc).text
      }
    end

    attr_writer :title

    def save!(to: path)
      FileUtils.cp path, to unless path == to

      Zip::File.open to do |zip|
        rootfile_entry = find_rootfile_entry(zip)
        rootfile_doc   = parse_xml(rootfile_entry)

        get_title_element(rootfile_doc).text = title

        zip.get_output_stream rootfile_entry.name, &rootfile_doc.method(:write)
        zip.commit
      end
    end

    private

    def find_rootfile_entry(zip)
      container_doc = parse_xml(zip.find_entry('META-INF/container.xml'))
      rootfile_path = REXML::XPath.first(container_doc, '//container:rootfile/@full-path', XMLNS.slice('container')).value

      zip.find_entry(rootfile_path)
    end

    def parse_xml(entry)
      content = entry.get_input_stream.read
      REXML::Document.new(content)
    end

    def get_title_element(doc)
      REXML::XPath.first(doc, '//dc:title', XMLNS.slice('dc'))
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
eepub-0.3.4	lib/eepub/epub.rb