manifest.rb in epub-parser-0.1.9

- old
+ new

@@ -1,7 +1,6 @@
 require 'set'
-require 'enumerabler'
 require 'rchardet'
 require 'epub/constants'
 require 'epub/parser/content_document'
 
 module EPUB
@@ -22,24 +21,38 @@
           item.manifest = self
           @items[item.id] = item
           self
         end
 
+        def each_nav
+          if block_given?
+            each_item do |item|
+              yield item if item.nav?
+            end
+          else
+            each_item.lazy.select(&:nav?)
+          end
+        end
+
         def navs
-          items.selector(&:nav?)
+          items.select(&:nav?)
         end
 
         def nav
           navs.first
         end
 
         def cover_image
-          items.selector(&:cover_image?).first
+          items.select(&:cover_image?).first
         end
 
         def each_item
-          @items.each_value do |item|
-            yield item
+          if block_given?
+            @items.each_value do |item|
+              yield item
+            end
+          else
+            @items.each_value
           end
         end
 
         def items
           @items.values