# frozen_string_literal: true

module XlsxWriter

  class Worksheet
    # Last row number written with #add_row
    attr_reader :current_row, :col_auto_widths

    # Thiner characters list used for column width logic mimicking axlsx behaviour
    THIN_CHARS = '^.acfijklrstxzFIJL()-'.freeze

    # Write a +row+. If no +types+ passed XlsxWriter tries to deduce them automatically.
    #
    # Both +types+ and +style+ may be an array as well as a symbol.
    # In the latter case they are applied to all cells in the +row+.
    #
    # +height+ is a Numeric that specifies the row height.
    #
    # If +skip_empty+ is set to +true+ empty cells are not added to the sheet.
    # Otherwise they are added with type +:blank+.
    def add_row(row, style: nil, height: nil, types: nil, skip_empty: false)
      row_idx = @current_row ||= 0
      @current_row += 1

      style_ary = style if style.is_a?(Array)
      types_ary = types if types.is_a?(Array)

      row.each_with_index do |value, idx|
        cell_style = style_ary ? style_ary[idx] : style
        cell_type  = types_ary ? types_ary[idx] : types

        case cell_type && cell_type.to_sym
        when :string
          value = value.to_s
          write_string(row_idx, idx, value, cell_style)
          update_col_auto_width(idx, value, cell_style)
        when :rich_string
          write_rich_string(row_idx, idx, value, cell_style)
        when :number
          write_number(row_idx, idx, value.to_f, cell_style)
        when :formula
          write_formula(row_idx, idx, value, cell_style)
        when :datetime, :date, :time
          write_datetime(row_idx, idx, value.to_time, cell_style)
        when :url
          write_url(row_idx, idx, value, cell_style)
          update_col_auto_width(idx, value.to_s, cell_style)
        when :boolean
          write_boolean(row_idx, idx, value, cell_style)
        when :blank
          write_blank(row_idx, idx, cell_style)
        when :skip, :empty
          # write nothing
        when nil
          case value
          when Numeric
            write_number(row_idx, idx, value, cell_style)
          when TrueClass, FalseClass
            write_boolean(row_idx, idx, value, cell_style)
          when Time, (defined?(Date) ? Date : Time)
            write_datetime(row_idx, idx, value, cell_style)
          when '', nil
            write_blank(row_idx, idx, cell_style) unless skip_empty
          when /\A=/
            write_formula(row_idx, idx, value, cell_style)
          when String
            write_string(row_idx, idx, value, cell_style)
            update_col_auto_width(idx, value, cell_style)
          when RichString
            write_rich_string(row_idx, idx, value, cell_style)
          else # assume string
            value = value.to_s
            write_string(row_idx, idx, value, cell_style)
            update_col_auto_width(idx, value, cell_style)
          end
        else
          raise ArgumentError, "Unknown cell type #{cell_type}."
        end
      end

      set_row(row_idx, height: height) if height

      nil
    end

    # Apply cols automatic widths calculated by #add_row.
    def apply_auto_widths
      @col_auto_widths.each_with_index do |width, idx|
        set_column(idx, idx, width: width) if width
      end
    end

    def get_column_auto_width(col)
      @col_auto_widths[extract_column(col)]
    end

    private

    # Updates the col auto width value to fit the string.
    def update_col_auto_width(idx, val, format)
      font_scale = (@workbook.font_sizes[format] || 11) / 10.0
      width = (val.count(THIN_CHARS) + 3) * font_scale
      width = 255 if width > 255 # Max xlsx column width is 255 characters
      @col_auto_widths[idx] = [@col_auto_widths[idx] || 0, width].max
    end
  end
end