eepub 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 025305fc54a605c35c60cfc8cfe7dda9046e7a5d
-  data.tar.gz: 7d2ff2df5a02f29e0c6611f43993fa08b92f10cc
+  metadata.gz: 47898b927d644865e6bfc5acafa1e338b105cade
+  data.tar.gz: 8caef5facc55165930418b10b81d476fb7b91dec
 SHA512:
-  metadata.gz: 5cb94e81780bf8eb68665fb899f40d6bac235a350012b0caca545edaa48b12d8ed329ae80aefeac78e570c4f57f94aa698e478d76486fd80ce76aa6d2ccd181d
-  data.tar.gz: 890c918e8afb1ef46254f46597cb0cb1ff5432934a1a59bd630b3609797bf820fb6332d4f7150dd6dbbbd0582135770a85fd472738e3f2772b0827a85d7cbd6c
+  metadata.gz: e859a6ee9cf51421654bc57e32d1894427481fab98d9d7d31a404f23312ae22321b4afdc103a195f43b620295fb450b403f8d97481ac776101feeb3231693a18
+  data.tar.gz: 1a8cba1bb4e01126b8c70a24b6ee1906151a2aaaea906535cdf0b9a8fb6c478d116467081990dc962b2edda91b38c2ed8b227e2ca80e67505586caf6a2cb018e

data/lib/eepub/epub.rb

@@ -2,8 +2,23 @@ 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
@@ -11,10 +26,10 @@ module Eepub
     attr_reader :path
 
     def title
-      @title ||= Zip::File.open(@path) {|zip|
-        package_xml = REXML::Document.new(package_entry(zip).get_input_stream)
+      @title ||= Zip::File.open(path) {|zip|
+        rootfile_doc = REXML::Document.new(rootfile_entry(zip).get_input_stream)
 
-        REXML::XPath.first(package_xml, '//dc:title', 'dc' => 'http://purl.org/dc/elements/1.1/').text
+        REXML::XPath.first(rootfile_doc, '//dc:title', XMLNS.slice('dc')).text
       }
     end
 
@@ -24,14 +39,13 @@ module Eepub
       FileUtils.cp path, to unless path == to
 
       Zip::File.open to do |zip|
-        entry      = package_entry(zip)
-        xml        = REXML::Document.new(entry.get_input_stream)
-        title_node = REXML::XPath.first(xml, '//dc:title', 'dc' => 'http://purl.org/dc/elements/1.1/')
+        rootfile_entry = rootfile_entry(zip)
+        rootfile_doc   = rootfile_entry.get_input_stream(&REXML::Document.method(:new))
 
-        title_node.text = title
+        REXML::XPath.first(rootfile_doc, '//dc:title', XMLNS.slice('dc')).text = title
 
-        zip.get_output_stream entry.name do |stream|
-          stream.write xml.to_s
+        zip.get_output_stream rootfile_entry.name do |stream|
+          stream.write rootfile_doc.to_s
         end
 
         zip.commit
@@ -40,12 +54,11 @@ module Eepub
 
     private
 
-    def package_entry(zip)
-      container_entry = zip.find_entry('META-INF/container.xml')
-      container_xml   = REXML::Document.new(container_entry.get_input_stream)
-      path            = REXML::XPath.first(container_xml, '//container:rootfile/@full-path', 'container' => 'urn:oasis:names:tc:opendocument:xmlns:container').value
+    def rootfile_entry(zip)
+      container_doc = zip.find_entry('META-INF/container.xml').get_input_stream(&REXML::Document.method(:new))
+      rootfile_path = REXML::XPath.first(container_doc, '//container:rootfile/@full-path', XMLNS.slice('container')).value
 
-      zip.find_entry(path)
+      zip.find_entry(rootfile_path)
     end
   end
 end

data/lib/eepub/version.rb

@@ -1,3 +1,3 @@
 module Eepub
-  VERSION = '0.3.1'
+  VERSION = '0.3.2'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: eepub
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - Keita Urashima
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-10-21 00:00:00.000000000 Z
+date: 2017-10-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubyzip