paulspringett/csv_shaper

require 'csv'

module CsvShaper
  # Encoder
  # Takes a Header and Array of Rows and converts them to a valid CSV formatted String
  # Example:
  # ```
  # CsvShaper::Encoder.new(@header, @rows).to_csv
  # ```
  class Encoder
    def initialize(header, rows = [])
      if header.nil?
        raise MissingHeadersError, 'you must define some headers using csv.headers ...'
      end

      @header = header
      @rows = rows
    end

    # Public: converts the Shaper mapped headers and rows into
    # a CSV String
    #
    # Returns a String
    def to_csv(local_config = nil)
      csv_options = options.merge(local_options(local_config))
      rows = padded_rows.map do |data|
        CSV::Row.new(@header.mapped_columns, data, false)
      end

      if csv_options[:write_headers] && rows.empty?
        rows << CSV::Row.new(@header.mapped_columns, [], true)
      end

      table = CSV::Table.new(rows)
      csv_options.except!(*custom_options.keys)
      table.to_csv(**csv_options)
    end

    private

    def options
      CsvShaper::Shaper.config && CsvShaper::Shaper.config.options || {}
    end

    def local_options(local_config)
      local_config && local_config.options || {}
    end

    def custom_options
      CsvShaper::Config::CUSTOM_DEFAULT_OPTIONS
    end

    # Internal: make use of `CSV#values_at` to pad out the
    # cells into the correct columns for the headers
    #
    # Returns an Array of Arrays
    def padded_rows
      rows = @rows.map do |row|
        CSV::Row.new(row.cells.keys, row.cells.values)
      end

      table = CSV::Table.new(rows)
      table.values_at(*@header.columns)
    end
  end
end