Merge branch 'user-lists-strategies-unleash' into 'master'

Return User List Strategy to Unleash Clients

See merge request gitlab-org/gitlab!30650

ee/app/models/operations/feature_flag.rb

@@ -53,7 +53,7 @@ module Operations
       end
 
       def for_unleash_client(project, environment)
-        includes(strategies: :scopes)
+        includes(strategies: [:scopes, :user_list])
           .where(project: project)
           .merge(Operations::FeatureFlags::Scope.on_environment(environment))
           .reorder(:id)

ee/app/models/operations/feature_flags/strategy.rb

@@ -4,10 +4,12 @@ module Operations
   module FeatureFlags
     class Strategy < ApplicationRecord
       STRATEGY_DEFAULT = 'default'
+      STRATEGY_GITLABUSERLIST = 'gitlabUserList'
       STRATEGY_GRADUALROLLOUTUSERID = 'gradualRolloutUserId'
       STRATEGY_USERWITHID = 'userWithId'
       STRATEGIES = {
         STRATEGY_DEFAULT => [].freeze,
+        STRATEGY_GITLABUSERLIST => [].freeze,
         STRATEGY_GRADUALROLLOUTUSERID => %w[groupId percentage].freeze,
         STRATEGY_USERWITHID => ['userIds'].freeze
       }.freeze
@@ -17,6 +19,8 @@ module Operations
 
       belongs_to :feature_flag
       has_many :scopes, class_name: 'Operations::FeatureFlags::Scope'
+      has_one :strategy_user_list
+      has_one :user_list, through: :strategy_user_list
 
       validates :name,
         inclusion: {
@@ -25,11 +29,20 @@ module Operations
       }
 
       validate :parameters_validations, if: -> { errors[:name].blank? }
+      validates :user_list, presence: true, if: -> { name == STRATEGY_GITLABUSERLIST }
+      validates :user_list, absence: true, if: -> { name != STRATEGY_GITLABUSERLIST }
+      validate :same_project_validation, if: -> { user_list.present? }
 
       accepts_nested_attributes_for :scopes, allow_destroy: true
 
       private
 
+      def same_project_validation
+        unless user_list.project_id == feature_flag.project_id
+          errors.add(:user_list, 'must belong to the same project')
+        end
+      end
+
       def parameters_validations
         validate_parameters_type &&
           validate_parameters_keys &&

ee/app/models/operations/feature_flags/strategy_user_list.rb

+# frozen_string_literal: true
+
+module Operations
+  module FeatureFlags
+    class StrategyUserList < ApplicationRecord
+      self.table_name = 'operations_strategies_user_lists'
+
+      belongs_to :strategy
+      belongs_to :user_list
+    end
+  end
+end

ee/lib/ee/api/entities/unleash_feature.rb

@@ -7,7 +7,27 @@ module EE
         expose :name
         expose :description, unless: ->(feature) { feature.description.nil? }
         expose :active, as: :enabled
-        expose :strategies, using: UnleashStrategy
+        expose :strategies do |flag|
+          flag.strategies.map do |strategy|
+            if legacy_strategy?(strategy)
+              UnleashLegacyStrategy.represent(strategy)
+            elsif gitlab_user_list_strategy?(strategy)
+              UnleashGitlabUserListStrategy.represent(strategy)
+            else
+              UnleashStrategy.represent(strategy)
+            end
+          end
+        end
+
+        private
+
+        def legacy_strategy?(strategy)
+          !strategy.respond_to?(:name)
+        end
+
+        def gitlab_user_list_strategy?(strategy)
+          strategy.name == ::Operations::FeatureFlags::Strategy::STRATEGY_GITLABUSERLIST
+        end
       end
     end
   end

ee/lib/ee/api/entities/unleash_gitlab_user_list_strategy.rb

+# frozen_string_literal: true
+
+module EE
+  module API
+    module Entities
+      class UnleashGitlabUserListStrategy < Grape::Entity
+        expose :name do |_strategy|
+          ::Operations::FeatureFlags::Strategy::STRATEGY_USERWITHID
+        end
+        expose :parameters do |strategy|
+          { userIds: strategy.user_list.user_xids }
+        end
+      end
+    end
+  end
+end

ee/lib/ee/api/entities/unleash_legacy_strategy.rb

+# frozen_string_literal: true
+
+module EE
+  module API
+    module Entities
+      class UnleashLegacyStrategy < Grape::Entity
+        expose :name do |strategy|
+          strategy['name']
+        end
+        expose :parameters do |strategy|
+          strategy['parameters']
+        end
+      end
+    end
+  end
+end

ee/lib/ee/api/entities/unleash_strategy.rb

@@ -4,20 +4,8 @@ module EE
   module API
     module Entities
       class UnleashStrategy < Grape::Entity
-        expose :name do |strategy|
-          if strategy.respond_to?(:name)
-            strategy.name
-          else
-            strategy['name']
-          end
-        end
-        expose :parameters do |strategy|
-          if strategy.respond_to?(:parameters)
-            strategy.parameters
-          else
-            strategy['parameters']
-          end
-        end
+        expose :name
+        expose :parameters
       end
     end
   end

ee/spec/requests/api/unleash_spec.rb

@@ -494,6 +494,28 @@ describe API::Unleash do
             }]
           }])
         end
+
+        it 'returns a userWithId strategy for a gitlabUserList strategy' do
+          feature_flag = create(:operations_feature_flag, :new_version_flag, project: project,
+                                name: 'myfeature', active: true)
+          user_list = create(:operations_feature_flag_user_list, project: project,
+                             name: 'My List', user_xids: 'user1,user2')
+          strategy = create(:operations_strategy, feature_flag: feature_flag,
+                            name: 'gitlabUserList', parameters: {}, user_list: user_list)
+          create(:operations_scope, strategy: strategy, environment_scope: 'production')
+
+          get api(features_url), headers: { 'UNLEASH-INSTANCEID' => client.token, 'UNLEASH-APPNAME' => 'production' }
+
+          expect(response).to have_gitlab_http_status(:ok)
+          expect(json_response['features']).to eq([{
+            'name' => 'myfeature',
+            'enabled' => true,
+            'strategies' => [{
+              'name' => 'userWithId',
+              'parameters' => { 'userIds' => 'user1,user2' }
+            }]
+          }])
+        end
       end
 
       context 'when mixing version 1 and version 2 feature flags' do