# frozen_string_literal: true

# Parts hold entries at directory word/
module Goldendocx
  module Parts
    class Documents
      XML_PATH = 'word/'

      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

      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.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
        @medias = []
      end

      def write_stream(zos)
        styles.write_to(zos)
        relationships.write_to(zos)
        document.write_to(zos)
        medias.each { |media| media.write_to(zos) }
        document.body.charts.each { |chart| chart.write_to(zos) }
      end

      def create_text(text, options = {})
        paragraph = document.body.create_text(text, options)

        style = styles.find_text_style(options[:style])
        paragraph.style = style.id if style

        paragraph
      end

      def create_table(options = {})
        table = document.body.create_table(options)

        style = styles.find_table_style(options[:style])
        table.style = style.id if style

        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)

        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)

        document.body.create_embed_image(relationship_id, options)
      end

      def add_image_media(image_data, _options = {})
        extension = 'png'
        # Add to media
        @media_amount += 1
        image_name = "image#{media_amount}.#{extension}"
        Goldendocx::Parts::Media.new(image_name, image_data).tap do |media|
          medias << media
        end
      end

      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(
          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)
      end

      def add_style(fragment)
        styles.add_style(fragment)
      end
    end
  end
end