require 'zip'
require 'rubyXL/objects/relationships'

module RubyXL

  class WorkbookRoot
    attr_accessor :filepath
    attr_accessor :thumbnail, :core_properties, :document_properties, :custom_properties, :workbook
    attr_accessor :content_types, :rels_hash

    include RubyXL::RelationshipSupport

    def related_objects
      [ content_types, thumbnail, core_properties, document_properties, workbook ]
    end

    def relationship_file_class
      RubyXL::RootRelationships
    end

    def attach_relationship(rid, rf)
      case rf
      when RubyXL::ThumbnailFile          then self.thumbnail = rf
      when RubyXL::CorePropertiesFile     then self.core_properties = rf
      when RubyXL::DocumentPropertiesFile then self.document_properties = rf
      when RubyXL::CustomPropertiesFile   then self.custom_properties = rf
      when RubyXL::Workbook               then self.workbook = rf
      else store_relationship(rf, :unknown)
      end
    end

    def self.default
      obj = self.new
      obj.document_properties    = RubyXL::DocumentPropertiesFile.new
      obj.core_properties        = RubyXL::CorePropertiesFile.new
      obj.relationship_container = RubyXL::RootRelationships.new
      obj.content_types          = RubyXL::ContentTypes.new
      obj
    end

    def self.parse_file(xl_file_path, opts)
      raise 'Not .xlsx or .xlsm excel file' unless opts[:skip_filename_check] || 
                                              %w{.xlsx .xlsm}.include?(File.extname(xl_file_path))

      ::Zip::File.open(xl_file_path) { |zip_file|
        root = self.new
        root.filepath = xl_file_path
        root.content_types = RubyXL::ContentTypes.parse_file(zip_file)
        root.load_relationships(zip_file)
        root.workbook.root = root
        root
      }
    end

  end

end