require_relative 'chart_axis'
require_relative 'chart_cells_range'
require_relative 'chart_legend'
require_relative 'chart_point'
require_relative 'display_labels_properties'
require_relative 'chart_style/office_2007_chart_style'
require_relative 'chart_style/office_2010_chart_style'

module OoxmlParser
  class Chart < OOXMLDocumentObject
    attr_accessor :type, :data, :grouping, :title, :legend, :display_labels, :axises, :alternate_content, :shape_properties

    def initialize
      @type = ''
      @data = []
      @grouping = nil
      @title = nil
      @legend = nil
      @axises = []
    end

    def parse_properties(chart_prop_node)
      chart_prop_node.xpath('*').each do |chart_props_node_child|
        case chart_props_node_child.name
        when 'grouping'
          @grouping = chart_props_node_child.attribute('val').value.to_sym
        when 'ser'
          case @type
          when :bar, :line, :area, :pie, :doughnut, :radar, :stock, :surface_3d, :column
            val = chart_props_node_child.xpath('c:val')[0]
          when :point, :bubble
            val = chart_props_node_child.xpath('c:yVal')[0]
          else
            next
          end
          next if val.xpath('c:numRef').empty?
          @data << ChartCellsRange.parse(val.xpath('c:numRef').first).dup
        when 'dLbls'
          @display_labels = DisplayLabelsProperties.parse(chart_props_node_child)
        end
      end
    end

    def self.parse
      chart = Chart.new
      chart_xml = Nokogiri::XML(File.open(OOXMLDocumentObject.current_xml))
      chart_xml.xpath('c:chartSpace/*', 'xmlns:c' => 'http://schemas.openxmlformats.org/drawingml/2006/chart').each do |chart_space_node_child|
        case chart_space_node_child.name
        when 'AlternateContent'
          chart.alternate_content = AlternateContent.parse(chart_space_node_child)
        when 'spPr'
          chart.shape_properties = DocxShapeProperties.parse chart_space_node_child
        when 'chart'
          chart_space_node_child.xpath('*').each do |chart_node_child|
            case chart_node_child.name
            when 'plotArea'
              chart_node_child.xpath('*').each do |plot_area_node_child|
                next unless chart.type.empty?
                case plot_area_node_child.name
                when 'barChart'
                  chart.type = :bar
                  bar_dir_node = plot_area_node_child.xpath('c:barDir')
                  unless bar_dir_node.first.nil?
                    chart.type = :column if bar_dir_node.first.attribute('val').value == 'col'
                  end
                  chart.parse_properties(plot_area_node_child)
                when 'lineChart'
                  chart.type = :line
                  chart.parse_properties(plot_area_node_child)
                when 'areaChart'
                  chart.type = :area
                  chart.parse_properties(plot_area_node_child)
                when 'bubbleChart'
                  chart.type = :bubble
                  chart.parse_properties(plot_area_node_child)
                when 'doughnutChart'
                  chart.type = :doughnut
                  chart.parse_properties(plot_area_node_child)
                when 'pieChart'
                  chart.type = :pie
                  chart.parse_properties(plot_area_node_child)
                when 'scatterChart'
                  chart.type = :point
                  chart.parse_properties(plot_area_node_child)
                when 'radarChart'
                  chart.type = :radar
                  chart.parse_properties(plot_area_node_child)
                when 'stockChart'
                  chart.type = :stock
                  chart.parse_properties(plot_area_node_child)
                when 'surface3DChart'
                  chart.type = :surface_3d
                  chart.parse_properties(plot_area_node_child)
                when 'line3DChart'
                  chart.type = :line_3d
                  chart.parse_properties(plot_area_node_child)
                when 'bar3DChart'
                  chart.type = :bar_3d
                  chart.parse_properties(plot_area_node_child)
                when 'pie3DChart'
                  chart.type = :pie_3d
                  chart.parse_properties(plot_area_node_child)
                end
              end
              chart_node_child.xpath('*').each do |plot_area_node_child|
                case plot_area_node_child.name
                when 'catAx'
                  chart.axises << ChartAxis.parse(plot_area_node_child)
                when 'valAx'
                  chart.axises << ChartAxis.parse(plot_area_node_child)
                end
              end
            when 'title'
              chart.title = ChartAxisTitle.parse(chart_node_child)
            when 'legend'
              chart.legend = ChartLegend.parse(chart_node_child)
            end
          end
        end
      end
      chart
    end
  end
end