require_relative 'workbook/chartsheet'
require_relative 'workbook/style_sheet'
require_relative 'workbook/worksheet'
require_relative 'workbook/workbook_helpers'
# Class for storing XLSX Workbook
module OoxmlParser
  class XLSXWorkbook < CommonDocumentStructure
    include WorkbookHelpers
    attr_accessor :worksheets
    # @return [PresentationTheme] theme of Workbook
    attr_accessor :theme
    # @return [Relationships] rels of book
    attr_accessor :relationships
    # @return [StyleSheet] styles of book
    attr_accessor :style_sheet

    def initialize(worksheets = [])
      @worksheets = worksheets
    end

    def cell(column, row, sheet = 0)
      column = Coordinates.new(row, column).get_column_number unless StringHelper.numeric?(column.to_s)

      if StringHelper.numeric?(sheet.to_s)
        row = @worksheets[sheet].rows[row.to_i - 1]
        return nil if row.nil?
        return row.cells[column.to_i - 1]
      elsif sheet.is_a?(String)
        @worksheets.each do |worksheet|
          if worksheet.name == sheet
            return worksheet.rows[row.to_i - 1].cells[column.to_i - 1] unless worksheet.rows[row.to_i - 1].nil?
          end
        end
        return nil
      end
      raise "Error. Wrong sheet value: #{sheet}"
    end

    def difference(other)
      Hash.object_to_hash(self).diff(Hash.object_to_hash(other))
    end

    # Get all values of formulas.
    # @param [Fixnum] precision of formulas counting
    # @return [Array, String] all formulas
    def all_formula_values(precision = 14)
      formulas = []
      worksheets.each do |c_sheet|
        next unless c_sheet
        c_sheet.rows.each do |c_row|
          next unless c_row
          c_row.cells.each do |c_cell|
            next unless c_cell
            next unless c_cell.formula
            text = c_cell.raw_text
            if StringHelper.numeric?(text)
              text = text.to_f.round(10).to_s[0..precision]
            elsif StringHelper.complex?(text)
              complex_number = Complex(text.tr(',', '.'))
              real_part = complex_number.real
              real_rounded = real_part.to_f.round(10).to_s[0..precision].to_f

              imag_part = complex_number.imag
              imag_rounded = imag_part.to_f.round(10).to_s[0..precision].to_f
              complex_rounded = Complex(real_rounded, imag_rounded)
              text = complex_rounded.to_s
            end
            formulas << text
          end
        end
      end
      formulas
    end

    def self.parse
      workbook = XLSXWorkbook.new
      workbook.relationships = Relationships.parse_rels("#{OOXMLDocumentObject.path_to_folder}xl/_rels/workbook.xml.rels")
      OOXMLDocumentObject.xmls_stack = []
      OOXMLDocumentObject.root_subfolder = 'xl/'
      self.shared_strings = nil
      OOXMLDocumentObject.add_to_xmls_stack('xl/workbook.xml')
      doc = Nokogiri::XML.parse(File.open(OOXMLDocumentObject.current_xml))
      XLSXWorkbook.styles_node = Nokogiri::XML(File.open("#{OOXMLDocumentObject.path_to_folder}/#{OOXMLDocumentObject.root_subfolder}/styles.xml"))
      workbook.theme = PresentationTheme.parse("xl/#{link_to_theme_xml}") if link_to_theme_xml
      workbook.style_sheet = StyleSheet.new(parent: self).parse
      doc.xpath('xmlns:workbook/xmlns:sheets/xmlns:sheet').each do |sheet|
        file = workbook.relationships.target_by_id(sheet.attribute('id').value)
        if file.start_with?('worksheets')
          workbook.worksheets << Worksheet.parse(file, parent: workbook)
          workbook.worksheets.last.name = sheet.attribute('name').value
        elsif file.start_with?('chartsheets')
          workbook.worksheets << Chartsheet.new(parent: workbook).parse(file)
        end
      end
      OOXMLDocumentObject.xmls_stack.pop
      workbook
    end

    class << self
      # @return [Array, Nokogiri::XML::Eelement] list of shared strings
      attr_writer :shared_strings
      attr_accessor :styles_node

      # Accessor for shared string. Initialization for this array
      def shared_strings
        if @shared_strings.nil?
          shared_strings_file = "#{OOXMLDocumentObject.path_to_folder}/#{OOXMLDocumentObject.root_subfolder}/sharedStrings.xml"
          @shared_strings = Nokogiri::XML(File.open(shared_strings_file)).xpath('//xmlns:si') if File.exist?(shared_strings_file)
        end
        @shared_strings
      end

      def link_to_theme_xml
        file = File.open(OOXMLDocumentObject.path_to_folder + 'xl/_rels/workbook.xml.rels')
        relationships = Relationships.parse_rels(file)
        relationships.target_by_type('theme')
      end
    end
  end
end