class ForemanOpenscap::BulkUpload

Public Class Methods

new() click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 6
def initialize
  @result = OpenStruct.new(:errors => [], :results => [])
end

Public Instance Methods

files_from_guide() click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 10
def files_from_guide
  `rpm -ql scap-security-guide | grep ds.xml`.split
end
scap_guide_installed?() click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 14
def scap_guide_installed?
  `rpm -qa | grep scap-security-guide`.present?
end
upload_from_directory(directory_path) click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 64
def upload_from_directory(directory_path)
  unless directory_path && Dir.exist?(directory_path)
    @result[:errors].push(_("No such directory: %s. Please check the path you have provided.") % directory_path)
    return @result
  end

  files_array = Dir["#{directory_path}/*-ds.xml"]
  upload_from_files(files_array)
end
upload_from_files(files_array, from_scap_guide = false) click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 27
def upload_from_files(files_array, from_scap_guide = false)
  unless files_array.is_a? Array
    @result.errors.push(_("Expected an array of files to upload, got: %s.") % files_array)
    return @result
  end

  files_array.each do |datastream|
    if File.directory?(datastream)
      @result.errors.push(_("%s is a directory, expecting file.") % datastream)
      next
    end

    unless File.file?(datastream)
      @result.errors.push(_("%s does not exist, skipping.") % datastream)
      next
    end

    file = File.open(datastream, 'rb').read
    digest = Digest::SHA2.hexdigest(datastream)
    title = content_name(datastream, from_scap_guide)
    filename = original_filename(datastream)
    scap_content = ScapContent.where(:title => title, :digest => digest).first_or_initialize
    next if scap_content.persisted?
    scap_content.scap_file = file
    scap_content.original_filename = filename
    scap_content.location_ids = Location.all.pluck(:id)
    scap_content.organization_ids = Organization.all.pluck(:id)

    if scap_content.save
      @result.results.push(scap_content)
    else
      @result.errors.push(_("Failed saving %s:") % datastream + " #{scap_content.errors.full_messages.uniq.join(',')}")
    end
  end
  @result
end
upload_from_scap_guide() click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 18
def upload_from_scap_guide
  unless scap_guide_installed?
    @result.errors.push(_("Can't find scap-security-guide RPM, are you sure it is installed on your server?"))
    return @result
  end

  upload_from_files(files_from_guide, true)
end

Private Instance Methods

content_name(datastream, from_scap_guide) click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 86
def content_name(datastream, from_scap_guide)
  os_name = extract_name_from_file(datastream)
  from_scap_guide ? (_("Red Hat %s default content") % os_name) : (_("%s content") % os_name)
end
extract_name_from_file(file) click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 76
def extract_name_from_file(file)
  # SCAP datastream files are in format of ssg-<OS>-ds.xml
  # We wish to extract the <OS> and create a name of it
  original_filename(file).gsub('ssg-', '').gsub('-ds.xml', '')
end
original_filename(file) click to toggle source
# File lib/foreman_openscap/bulk_upload.rb, line 82
def original_filename(file)
  file.split('/').last
end