Support closed, merged and reopened pull/merge requests in OBS workflows

Co-authored-by: Victor Pereira <vpereira@suse.com>

src/api/app/models/token/workflow.rb

@@ -13,13 +13,67 @@ def call(options)
 
     yaml_file = Workflows::YAMLDownloader.new(scm_webhook.payload, token: self).call
     workflows = Workflows::YAMLToWorkflowsService.new(yaml_file: yaml_file, scm_webhook: scm_webhook, token: self).call
+
+    case
+    when scm_webhook.closed_merged_pull_request?
+      destroy_target_projects(workflows)
+    when scm_webhook.reopened_pull_request?
+      restore_target_projects(workflows)
+    when scm_webhook.new_pull_request?, scm_webhook.updated_pull_request?
+      call_steps(workflows)
+    end
+  rescue Octokit::Unauthorized, Gitlab::Error::Unauthorized => e
+    raise Token::Errors::SCMTokenInvalid, e.message
+  end
+
+  private
+
+  def destroy_target_projects(workflows)
+    workflows.each do |workflow|
+      # Do not process steps for which there's nothing to do
+      workflow_steps = workflow.steps.reject { |step| step.instance_of?(::Workflow::Step::ConfigureRepositories) }
+      target_packages = workflow_steps.map(&:target_package).uniq.compact
+      delete_subscriptions(target_packages)
+
+      target_project_names = workflow_steps.map(&:target_project_name).uniq.compact
+      destroy_all_target_projects(target_project_names)
+    end
+  end
+
+  # We want the callbacks to run after destroy, so we can't use delete_all
+  def destroy_all_target_projects(target_project_names)
+    Project.where(name: target_project_names).destroy_all
+  end
+
+  def delete_subscriptions(packages)
+    EventSubscription.where(channel: 'scm', token: self, package: packages).delete_all
+  end
+
+  def restore_target_projects(workflows)
+    token_user_login = user.login
+
+    workflows.each do |workflow|
+      # Do not process steps for which there's nothing to do
+      workflow_steps = workflow.steps.reject { |step| step.instance_of?(::Workflow::Step::ConfigureRepositories) }
+      target_project_names = workflow_steps.map(&:target_project_name).uniq.compact
+      target_project_names.each do |target_project_name|
+        Project.restore(target_project_name, user: token_user_login)
+      end
+
+      target_packages = workflow_steps.map(&:target_package).uniq.compact
+      target_packages.each do |target_package|
+        # FIXME: We shouldn't rely on a workflow step to be able to create/update subscriptions
+        workflow_steps.first.create_or_update_subscriptions(target_package, workflow.filters)
+      end
+    end
+  end
+
+  def call_steps(workflows)
     workflows.each do |workflow|
       workflow.steps.each do |step|
         step.call({ workflow_filters: workflow.filters })
       end
     end
-  rescue Octokit::Unauthorized, Gitlab::Error::Unauthorized => e
-    raise Token::Errors::SCMTokenInvalid, e.message
   end
 end
 

src/api/app/models/workflow/step.rb

@@ -16,6 +16,29 @@ def call(_options)
     raise AbstractMethodCalled
   end
 
+  def target_project_name
+    "home:#{@token.user.login}:#{source_project_name}:PR-#{scm_webhook.payload[:pr_number]}"
+  end
+
+  def target_package
+    Package.find_by_project_and_name(target_project_name, target_package_name)
+  end
+
+  # FIXME: Remove this and use create_subscriptions and update_subscriptions as soon as BranchPackageStep and Token::Workflow are refactored
+  #        This method is public since it has to be called from Token::Workflow. This shouldn't be needed.
+  def create_or_update_subscriptions(package, workflow_filters)
+    ['Event::BuildFail', 'Event::BuildSuccess'].each do |build_event|
+      subscription = EventSubscription.find_or_create_by!(eventtype: build_event,
+                                                          receiver_role: 'reader', # We pass a valid value, but we don't need this.
+                                                          user: @token.user,
+                                                          channel: 'scm',
+                                                          enabled: true,
+                                                          token: @token,
+                                                          package: package)
+      subscription.update!(payload: scm_webhook.payload.merge({ workflow_filters: workflow_filters }))
+    end
+  end
+
   protected
 
   def validate_step_instructions
@@ -28,10 +51,6 @@ def source_project_name
     step_instructions[:source_project]
   end
 
-  def target_project_name
-    "home:#{@token.user.login}:#{source_project_name}:PR-#{scm_webhook.payload[:pr_number]}"
-  end
-
   def source_package_name
     step_instructions[:source_package]
   end
@@ -44,10 +63,6 @@ def target_package_name
 
   private
 
-  def target_package
-    Package.find_by_project_and_name(target_project_name, target_package_name)
-  end
-
   def remote_source?
     Project.find_remote_project(source_project_name).present?
   end
@@ -84,20 +99,6 @@ def branch_request_content_gitlab
       object_attributes: { source: { default_branch: scm_webhook.payload[:commit_sha] } } }.to_json
   end
 
-  # FIXME: remove this and use create_subscriptions and update_subscriptions as soon as BranchPackageStep is refactored
-  def create_or_update_subscriptions(package, workflow_filters)
-    ['Event::BuildFail', 'Event::BuildSuccess'].each do |build_event|
-      subscription = EventSubscription.find_or_create_by!(eventtype: build_event,
-                                                          receiver_role: 'reader', # We pass a valid value, but we don't need this.
-                                                          user: @token.user,
-                                                          channel: 'scm',
-                                                          enabled: true,
-                                                          token: @token,
-                                                          package: package)
-      subscription.update!(payload: scm_webhook.payload.merge({ workflow_filters: workflow_filters }))
-    end
-  end
-
   def create_subscriptions(package, workflow_filters)
     ['Event::BuildFail', 'Event::BuildSuccess'].each do |build_event|
       EventSubscription.create!(eventtype: build_event,

src/api/app/models/workflow/step/branch_package_step.rb

@@ -5,14 +5,17 @@ class Workflow::Step::BranchPackageStep < ::Workflow::Step
   def call(options = {})
     return unless valid?
 
-    # FIXME: Support closed/merged/reopened PRs
-    return if scm_webhook.closed_merged_pull_request? || scm_webhook.reopened_pull_request?
+    workflow_filters = options.fetch(:workflow_filters, {})
+    branch_package(workflow_filters)
+  end
+
+  private
 
+  def branch_package(workflow_filters = {})
     branched_package = find_or_create_branched_package
 
     add_or_update_branch_request_file(package: branched_package)
 
-    workflow_filters = options.fetch(:workflow_filters, {})
     create_or_update_subscriptions(branched_package, workflow_filters)
 
     workflow_repositories(target_project_name, workflow_filters).each do |repository|
@@ -27,14 +30,6 @@ def call(options = {})
     branched_package
   end
 
-  private
-
-  def find_or_create_branched_package
-    return target_package if scm_webhook.updated_pull_request? && target_package.present?
-
-    branch
-  end
-
   def check_source_access
     return if remote_source?
 
@@ -49,7 +44,9 @@ def check_source_access
     Pundit.authorize(@token.user, src_package, :create_branch?)
   end
 
-  def branch
+  def find_or_create_branched_package
+    return target_package if scm_webhook.updated_pull_request? && target_package.present?
+
     check_source_access
 
     begin

src/api/app/models/workflow/step/configure_repositories.rb

@@ -8,9 +8,6 @@ class Workflow::Step::ConfigureRepositories < Workflow::Step
   def call(_options = {})
     return unless valid?
 
-    # FIXME: Support closed/merged/reopened PRs
-    return if scm_webhook.closed_merged_pull_request? || scm_webhook.reopened_pull_request?
-
     step_instructions[:repositories].each do |repository_instructions|
       repository = Repository.includes(:architectures).find_or_create_by(name: repository_instructions[:name], project: Project.get_by_name(target_project_name))
 

src/api/app/models/workflow/step/link_package_step.rb

@@ -5,15 +5,11 @@ class Workflow::Step::LinkPackageStep < ::Workflow::Step
   def call(options = {})
     return unless valid?
 
-    # FIXME: Support closed/merged/reopened PRs
-    return if scm_webhook.closed_merged_pull_request? || scm_webhook.reopened_pull_request?
-
     workflow_filters = options.fetch(:workflow_filters, {})
 
-    # Updated PR
     if scm_webhook.updated_pull_request? && target_package.present?
       update_subscriptions(target_package, workflow_filters)
-    else # New PR
+    elsif scm_webhook.new_pull_request?
       create_target_package
       create_subscriptions(target_package, workflow_filters)
     end