ManageIQ/manageiq-ui-classic

class AutomateImportService
  def cancel_import(import_file_upload_id)
    import_file_upload = ImportFileUpload.find(import_file_upload_id)

    destroy_queued_deletion(import_file_upload.id)
    import_file_upload.destroy
  end

  def import_datastore(import_file_upload,
                       domain_name_to_import_from,
                       domain_name_to_import_to,
                       namespace_or_class_list)
    Tempfile.open(['automate_temporary_zip', '.zip']) do |zip_file|
      zip_file.binmode
      zip_file.write(import_file_upload.binary_blob.binary)
      zip_file.flush

      import_options = {
        "import_as" => domain_name_to_import_to.presence || domain_name_to_import_from,
        "overwrite" => true,
        "zip_file"  => zip_file.path,
        "tenant_id" => User.current_user.current_tenant.id
      }
      ae_import = MiqAeImport.new(domain_name_to_import_from, import_options)

      namespace_list = namespace_or_class_list.reject do |namespace_or_class|
        namespace_or_class.match(/\.class/)
      end

      class_list = namespace_or_class_list.select do |namespace_or_class|
        namespace_or_class.match(/\.class/)
      end

      ae_import.remove_unrelated_entries(domain_name_to_import_from)
      reject_unrelated_namespaces(ae_import, domain_name_to_import_from, namespace_list)
      reject_unrelated_classes(ae_import, domain_name_to_import_from, class_list)

      result = ae_import.import

      result.nil? ? nil : ae_import.import_stats
    end
  end

  def store_for_import(file_contents)
    import_file_upload = ImportFileUpload.create
    import_file_upload.store_binary_data_as_yml(file_contents, "Automation import")
    import_file_upload
  ensure
    queue_deletion(import_file_upload.id)
  end

  private

  def reject_unrelated_classes(ae_import, domain_name, class_list)
    domain_name_related_files = ae_import.all_class_files.select do |class_file|
      class_file.name.match(domain_name)
    end

    remove_unrelated_files(ae_import, domain_name_related_files, class_list)
  end

  def reject_unrelated_namespaces(ae_import, domain_name, namespace_list)
    domain_name_related_files = ae_import.all_namespace_files.select do |namespace_file|
      namespace_file.name.match(domain_name)
    end

    remove_unrelated_files(ae_import, domain_name_related_files, namespace_list)
  end

  def remove_unrelated_files(ae_import, domain_name_related_files, list_of_files)
    files_to_reject = domain_name_related_files.reject do |domain_name_file|
      list_of_files.include?(File.dirname(domain_name_file.name).split("/")[1..-1].join("/"))
    end

    files_to_reject.each do |file|
      ae_import.remove_entry(file)
    end

    ae_import.update_sorted_entries
  end

  def destroy_queued_deletion(import_file_upload_id)
    MiqQueue.unqueue(
      :class_name  => "ImportFileUpload",
      :instance_id => import_file_upload_id,
      :method_name => "destroy"
    )
  end

  def queue_deletion(import_file_upload_id)
    MiqQueue.put(
      :class_name  => "ImportFileUpload",
      :instance_id => import_file_upload_id,
      :deliver_on  => 1.day.from_now,
      :method_name => "destroy"
    )
  end
end