1 files changed, 42 insertions, 0 deletions
diff --git a/app/models/import/resource.rb b/app/models/import/resource.rb
index e6de935ae..27dc367a8 100644
--- a/app/models/import/resource.rb
+++ b/app/models/import/resource.rb
@@ -7,14 +7,56 @@ class Import::Resource < ApplicationModel
   belongs_to :referential
   has_many :messages, class_name: "Import::Message", foreign_key: :resource_id
 
+  scope :main_resources, ->{ where(resource_type: "referential") }
+
   def root_import
     import = self.import
     import = import.parent while import.parent
     import
   end
 
+  def next_step
+    if root_import.class == Import::Workbench
+
+      # XXX
+      # return unless netex_import&.successful?
+
+      if workbench.import_compliance_control_set_id.present? && workbench_import_control_set.nil?
+        ComplianceControlSetCopyWorker.perform_async workbench.import_compliance_control_set_id, referential_id
+        return
+      end
+
+      # XXX
+      # return if workbench_import_control_set && !workbench_import_control_set.successful?
+
+      if workgroup.import_compliance_control_set_id.present? && workgroup_import_control_set.nil?
+        ComplianceControlSetCopyWorker.perform_async workgroup.import_compliance_control_set_id, referential_id
+      end
+    end
+  end
+
+  def workbench
+    import.workbench
+  end
+
+  def workgroup
+    workbench.workgroup
+  end
+
   def netex_import
     return unless self.resource_type == "referential"
     import.children.where(name: self.reference).last
   end
+
+  def workbench_import_control_set
+    return unless referential.present?
+    return unless referential.workbench.import_compliance_control_set_id.present?
+    referential.compliance_check_sets.where(compliance_control_set_id: referential.workbench.import_compliance_control_set_id, referential_id: referential_id).last
+  end
+
+  def workgroup_import_control_set
+    return unless referential.present?
+    return unless referential.workbench.workgroup.import_compliance_control_set_id.present?
+    referential.compliance_check_sets.where(compliance_control_set_id: referential.workbench.workgroup.import_compliance_control_set_id, referential_id: referential_id).last
+  end
 end