diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-05-20 14:34:42 +0000 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-05-20 14:34:42 +0000 |
commit | 9f46488805e86b1bc341ea1620b866016c2ce5ed (patch) | |
tree | f9748c7e287041e37d6da49e0a29c9511dc34768 /app/services/alert_management/update_alert_status_service.rb | |
parent | dfc92d081ea0332d69c8aca2f0e745cb48ae5e6d (diff) | |
download | gitlab-ce-9f46488805e86b1bc341ea1620b866016c2ce5ed.tar.gz |
Add latest changes from gitlab-org/gitlab@13-0-stable-ee
Diffstat (limited to 'app/services/alert_management/update_alert_status_service.rb')
-rw-r--r-- | app/services/alert_management/update_alert_status_service.rb | 63 |
1 files changed, 63 insertions, 0 deletions
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 |