lib/rpub/xml_file.rb in rpub-0.4.0 vs lib/rpub/xml_file.rb in rpub-0.5.0

@@ -1,11 +1,14 @@
 module Rpub
   class XmlFile
     # @return [Builder::XmlMarkup]
     attr_reader :xml
 
-    def initialize
+    attr_reader :book
+
+    def initialize(book = nil)
       @xml = Builder::XmlMarkup.new :indent => 2
+      @book = book
     end
 
     # @return [String] render this file and output as string
     def to_s
       render