documents.rb in goldendocx-0.2.3

- old
+ new

@@ -2,41 +2,54 @@
 
 # Parts hold entries at directory word/
 module Goldendocx
   module Parts
     class Documents
+      include Goldendocx::HasAssociations
+
       XML_PATH = 'word/'
+      RELATIONSHIPS_XML_PATH = 'word/_rels/document.xml.rels'
 
       attr_reader :document, # document.xml
-                  :styles, # styles.xml
-                  :relationships, # _rels/document.xml.rels
                   :charts, # charts/
                   :medias # New Medias to media/
 
       attr_accessor :media_amount # Entries amount in directory media/ for generating relationship id
 
+      relationships_at RELATIONSHIPS_XML_PATH
+      associate :styles, class_name: 'Goldendocx::Documents::Styles'
+      associate :settings, class_name: 'Goldendocx::Documents::Settings'
+      associate :document, class_name: 'Goldendocx::Documents::Document', isolate: true
+
       class << self
         def read_from(docx_file)
           parts = Goldendocx::Parts::Documents.new
-          parts.document.read_from(docx_file)
-          parts.styles.read_from(docx_file)
-          parts.relationships.read_from(docx_file)
+          parts.read_relationships(docx_file)
+          parts.read_associations(docx_file)
           parts.media_amount = docx_file.entries.count { |entry| entry.name.start_with?('word/media/') }
           parts
         end
       end
 
       def initialize
-        @document = Goldendocx::Documents::Document.new
-        @styles = Goldendocx::Documents::Styles.new
-        @relationships = Goldendocx::Documents::Relationships.new
+        associations.each do |association, options|
+          association_class = options[:class_name].constantize
+          instance_variable_set("@#{association}", association_class.new)
+          add_relationship association_class::TYPE, association_class::XML_PATH.gsub(XML_PATH, '') unless options[:isolate]
+        end
+
         @medias = []
+        @media_amount = 0
       end
 
       def write_stream(zos)
-        styles.write_to(zos)
-        relationships.write_to(zos)
+        write_relationships(zos)
+
+        associations.each_key do |association|
+          send(association).write_to(zos)
+        end
+
         document.write_to(zos)
         medias.each { |media| media.write_to(zos) }
         document.body.charts.each { |chart| chart.write_to(zos) }
       end
 
@@ -58,18 +71,18 @@
         table
       end
 
       def create_image(image_data, options = {})
         image_media = add_image_media(image_data, options)
-        relationship_id = relationships.add_relationship(image_media.type, image_media.target)
+        relationship_id = add_relationship(image_media.type, image_media.target)
 
         document.body.create_image(relationship_id, options)
       end
 
       def create_embed_image(image_data, options = {})
         image_media = add_image_media(image_data, options)
-        relationship_id = relationships.add_relationship(image_media.type, image_media.target)
+        relationship_id = add_relationship(image_media.type, image_media.target)
 
         document.body.create_embed_image(relationship_id, options)
       end
 
       def add_image_media(image_data, _options = {})
@@ -85,10 +98,10 @@
       def create_chart(chart_type, width, height)
         document.body.ensure_chart_type(chart_type)
 
         chart_id = document.body.charts.size + 1
 
-        relationship_id = relationships.add_relationship(
+        relationship_id = add_relationship(
           Goldendocx::Charts::RELATIONSHIP_TYPE,
           format(Goldendocx::Charts::RELATIONSHIP_NAME_PATTERN, id: chart_id)
         )
 
         document.body.create_chart(chart_type, chart_id, relationship_id, width: width, height: height)