Contents

require "csv"

module CodeInventory
  class CSVFile
    attr_reader :csv

    def initialize(csv_file)
      @csv = CSV.read(csv_file, { headers: true, converters: [ :integer ] })
      validate_headers
    end

    def projects
      @csv.collect do |row|
        csv_data = row.to_hash
        project = csv_data.inject({}) do |memo, pair|
          csv_header, csv_value = pair
          case
          when csv_header == "tags"
            new_pair = { "tags" => csv_value.split(",").collect { |tag| tag.strip } }
          when csv_header.include?(".")
            new_pair = dotted_to_nested(csv_header, csv_value)
          else
            new_pair = { csv_header => csv_value }
          end
          memo.merge(new_pair)
        end
        yield project if block_given?
        project
      end
    end

    private

    # Convert a dotted notation header and a value to a nested hash
    # e.g., "contact.email" header with value "[e-mail]" becomes
    # { "contact" => { "email" => "[e-mail]" } }
    def dotted_to_nested(path, value)
      path.split(".").reverse.inject(value) do |hash, element|
        { element => hash }
      end
    end

    def validate_headers
      required_headers = [ "name", "description", "license", "openSourceProject", "governmentWideReuseProject", "tags", "contact.email" ]
      required_headers.each do |required|
        raise FileFormatError unless @csv.headers.include? required
      end
    end
  end

  class FileFormatError < StandardError
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
codeinventory-0.3.1	lib/codeinventory/csv_file.rb
codeinventory-0.3.0	lib/codeinventory/csv_file.rb
codeinventory-0.2.1	lib/codeinventory/csv_file.rb
codeinventory-0.2.0	lib/codeinventory/csv_file.rb
codeinventory-0.1.3	lib/codeinventory/csv_file.rb