class ForemanOpenscap::BulkUpload

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

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

def files_from_guide(package)
  `rpm -ql #{package} | grep ds.xml`.split
end

def initialize
  @result = OpenStruct.new(:errors => [], :results => [])
end

def original_filename(file)
  file.split('/').last
end

def package_installed?(package)
  `rpm -qa | grep #{package}`.present?
end

def security_guide_packages
  %w(scap-security-guide)
end

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

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.read(datastream, mode: 'rb')
    title = content_name(datastream, from_scap_guide)
    filename = original_filename(datastream)
    scap_content = ScapContent.where(:title => title).first_or_initialize
    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

def upload_from_scap_guide
  package = security_guide_packages.find { |p| package_installed? p }
  unless package
    joined_packages = security_guide_packages.join(', ')
    msg = _("Can't find %{packages} RPM(s), are you sure it is installed on your server?")
    @result.errors.push(msg % {packages: joined_packages})
    return @result
  end
  upload_from_files(files_from_guide(package), true)
end