Add latest changes from gitlab-org/gitlab@15-10-stable-eev15.10.0-rc42

2 files changed, 95 insertions, 0 deletions

diff --git a/app/services/achievements/award_service.rb b/app/services/achievements/award_service.rb
new file mode 100644
index 00000000000..674bb8837fb
--- /dev/null
+++ b/app/services/achievements/award_service.rb
@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Achievements
+  class AwardService
+    attr_reader :current_user, :achievement_id, :recipient_id
+
+    def initialize(current_user, achievement_id, recipient_id)
+      @current_user = current_user
+      @achievement_id = achievement_id
+      @recipient_id = recipient_id
+    end
+
+    def execute
+      achievement = Achievements::Achievement.find(achievement_id)
+      return error_no_permissions unless allowed?(achievement)
+
+      recipient = User.find(recipient_id)
+
+      user_achievement = Achievements::UserAchievement.create(
+        achievement: achievement,
+        user: recipient,
+        awarded_by_user: current_user)
+      return error_awarding(user_achievement) unless user_achievement.persisted?
+
+      ServiceResponse.success(payload: user_achievement)
+    rescue ActiveRecord::RecordNotFound => e
+      error(e.message)
+    end
+
+    private
+
+    def allowed?(achievement)
+      current_user&.can?(:award_achievement, achievement)
+    end
+
+    def error_no_permissions
+      error('You have insufficient permissions to award this achievement')
+    end
+
+    def error_awarding(user_achievement)
+      error(user_achievement&.errors&.full_messages || 'Failed to award achievement')
+    end
+
+    def error(message)
+      ServiceResponse.error(message: Array(message))
+    end
+  end
+end
diff --git a/app/services/achievements/revoke_service.rb b/app/services/achievements/revoke_service.rb
new file mode 100644
index 00000000000..4601622f517
--- /dev/null
+++ b/app/services/achievements/revoke_service.rb
@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+module Achievements
+  class RevokeService
+    attr_reader :current_user, :user_achievement
+
+    def initialize(current_user, user_achievement)
+      @current_user = current_user
+      @user_achievement = user_achievement
+    end
+
+    def execute
+      return error_no_permissions unless allowed?(user_achievement.achievement)
+      return error_already_revoked if user_achievement.revoked?
+
+      user_achievement.assign_attributes({
+        revoked_by_user_id: current_user.id,
+        revoked_at: Time.zone.now
+      })
+      return error_awarding unless user_achievement.save
+
+      ServiceResponse.success(payload: user_achievement)
+    end
+
+    private
+
+    def allowed?(achievement)
+      current_user&.can?(:award_achievement, achievement)
+    end
+
+    def error_no_permissions
+      error('You have insufficient permissions to revoke this achievement')
+    end
+
+    def error_already_revoked
+      error('This achievement has already been revoked')
+    end
+
+    def error_awarding
+      error(user_achievement&.errors&.full_messages || 'Failed to revoke achievement')
+    end
+
+    def error(message)
+      ServiceResponse.error(message: Array(message))
+    end
+  end
+end