From 9f46488805e86b1bc341ea1620b866016c2ce5ed Mon Sep 17 00:00:00 2001
From: GitLab Bot <gitlab-bot@gitlab.com>
Date: Wed, 20 May 2020 14:34:42 +0000
Subject: Add latest changes from gitlab-org/gitlab@13-0-stable-ee

---
 .../update_alert_status_service.rb                 | 63 ++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 app/services/alert_management/update_alert_status_service.rb

(limited to 'app/services/alert_management/update_alert_status_service.rb')

diff --git a/app/services/alert_management/update_alert_status_service.rb b/app/services/alert_management/update_alert_status_service.rb
new file mode 100644
index 00000000000..a7ebddb82e0
--- /dev/null
+++ b/app/services/alert_management/update_alert_status_service.rb
@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+module AlertManagement
+  class UpdateAlertStatusService
+    include Gitlab::Utils::StrongMemoize
+
+    # @param alert [AlertManagement::Alert]
+    # @param user [User]
+    # @param status [Integer] Must match a value from AlertManagement::Alert::STATUSES
+    def initialize(alert, user, status)
+      @alert = alert
+      @user = user
+      @status = status
+    end
+
+    def execute
+      return error_no_permissions unless allowed?
+      return error_invalid_status unless status_key
+
+      if alert.update(status_event: status_event)
+        success
+      else
+        error(alert.errors.full_messages.to_sentence)
+      end
+    end
+
+    private
+
+    attr_reader :alert, :user, :status
+
+    delegate :project, to: :alert
+
+    def allowed?
+      user.can?(:update_alert_management_alert, project)
+    end
+
+    def status_key
+      strong_memoize(:status_key) do
+        AlertManagement::Alert::STATUSES.key(status)
+      end
+    end
+
+    def status_event
+      AlertManagement::Alert::STATUS_EVENTS[status_key]
+    end
+
+    def success
+      ServiceResponse.success(payload: { alert: alert })
+    end
+
+    def error_no_permissions
+      error(_('You have no permissions'))
+    end
+
+    def error_invalid_status
+      error(_('Invalid status'))
+    end
+
+    def error(message)
+      ServiceResponse.error(payload: { alert: alert }, message: message)
+    end
+  end
+end
-- 
cgit v1.2.1