Contents

module HoneyFormat
  # Represents columns.
  class Columns
    # @return [Columns] a new instance of Columns.
    # @param [Array] header array of strings.
    # @param [Array] valid array of symbols representing valid columns.
    # @raise [MissingCSVHeaderColumnError] raised when header is missing
    # @raise [UnknownCSVHeaderColumnError] raised when column is not in valid list.
    def initialize(header, valid = :all)
      @columns = build_columns(header, valid)
    end

    # Returns columns as array.
    # @return [Array] of columns.
    def to_a
      @columns
    end

    private

    def build_columns(header, valid)
      header.map do |column|
        Sanitize.string!(column)
        validate_column_presence!(column)

        column = symnolize_string!(column)

        validate_column_name!(column, valid)
        column
      end
    end

    def symnolize_string!(column)
      column.downcase!
      column.gsub!(/ /, '')
      column.gsub!(/-/, '_')
      column.to_sym
    end

    def validate_column_presence!(col)
      if col.nil? || col.empty?
        fail(MissingCSVHeaderColumnError, "CSV header column can't be empty.")
      end
    end

    def validate_column_name!(column, valid)
      return if valid == :all

      valid.include?(column) ||
        begin
          err_msg = "column :#{column} not in #{valid.inspect}"
          fail(UnknownCSVHeaderColumnError, err_msg)
        end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
honey_format-0.2.1	lib/honey_format/columns.rb
honey_format-0.2.0	lib/honey_format/columns.rb