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