Class: Waivers::ExcelFileReader

Inherits:

Object

• Object
• Waivers::ExcelFileReader

Defined in:

lib/inspec/utils/waivers/excel_file_reader.rb

Class Method Summary

• .fetch_data(path) ⇒ Object
• .headers ⇒ Object
• .resolve(path) ⇒ Object

Class Method Details

.fetch_data(path) ⇒ Object

# File 'lib/inspec/utils/waivers/excel_file_reader.rb', line 13

def self.fetch_data(path)
  waiver_data_hash = {}
  file_extension = File.extname(path) == ".xlsx" ? :xlsx : :xls
  excel_file = Roo::Spreadsheet.open(path, extension: file_extension)
  excel_file.sheet(0).parse(headers: true).each_with_index do |row, index|
    if index == 0
      @headers = row.keys
    else
      row_hash = row
      control_id = row_hash["control_id"]
      # delete keys and values not required in final hash
      row_hash.delete("control_id")
      row_hash.delete_if { |k, v| k.nil? || v.nil? }
    end

    waiver_data_hash[control_id] = row_hash if control_id && !row_hash.blank?
  end
  waiver_data_hash
rescue Exception => e
  raise "Error reading InSpec waivers in Excel: #{e}"
end

.headers ⇒ Object

# File 'lib/inspec/utils/waivers/excel_file_reader.rb', line 35

def self.headers
  @headers
end

.resolve(path) ⇒ Object

# File 'lib/inspec/utils/waivers/excel_file_reader.rb', line 6

def self.resolve(path)
  return nil unless File.file?(path)

  @headers ||= []
  fetch_data(path)
end