gitlab-foss/lib/feature/kas.rb

# frozen_string_literal: true

module Feature
  class Kas
    PREFIX = "kas_"

    class << self
      def enabled_for_any?(feature_flag, *actors)
        return false unless Feature::FlipperFeature.table_exists?

        actors = actors.compact
        # rubocop:disable Gitlab/FeatureFlagKeyDynamic -- this is a helper function and we need a variable argument here
        # rubocop:disable Gitlab/FeatureFlagWithoutActor -- we explicitly don't want an actor here
        return Feature.enabled?(feature_flag, type: :undefined, default_enabled_if_undefined: false) if actors.empty?

        # rubocop:enable Gitlab/FeatureFlagWithoutActor

        actors.any? do |actor|
          Feature.enabled?(feature_flag, actor, type: :undefined, default_enabled_if_undefined: false)
        end
        # rubocop:enable Gitlab/FeatureFlagKeyDynamic
      rescue ActiveRecord::NoDatabaseError, PG::ConnectionBad
        false
      end

      def server_feature_flags_for_grpc_request(user: nil, project: nil, group: nil)
        server_feature_flags(
          ->(f) { "kas-feature-#{f.delete_prefix(PREFIX).tr('_', '-')}" },
          user: user,
          project: project,
          group: group
        )
      end

      def server_feature_flags_for_http_response(user: nil, project: nil, group: nil)
        server_feature_flags(
          ->(f) { f.delete_prefix(PREFIX) },
          user: user,
          project: project,
          group: group
        )
      end

      def user_actor(user = nil)
        return ::User.actor_from_id(user.id) if user.is_a?(::User)

        user_id = Gitlab::ApplicationContext.current_context_attribute(:user_id)
        ::User.actor_from_id(user_id) if user_id
      end

      def project_actor(container)
        return unless container

        return ::Project.actor_from_id(container.project.id) if container.is_a?(::Clusters::Agent)

        ::Project.actor_from_id(container.id) if container.is_a?(::Project)
      end

      def group_actor(container)
        return unless container

        return ::Group.actor_from_id(container.id) if container.is_a?(::Group)
        return ::Group.actor_from_id(container.project.namespace_id) if container.is_a?(::Clusters::Agent)

        ::Group.actor_from_id(container.namespace_id) if container.is_a?(::Project)
      end

      private

      def server_feature_flags(key_transform, user: nil, project: nil, group: nil)
        # We need to check that both the DB connection and table exists
        return {} unless FlipperFeature.database.cached_table_exists?

        # The order of actors here is significant.
        # Percentage-based actor selection may not work as expected if this order changes.
        actors = [user, project, group].compact

        Feature.persisted_names
          .select { |f| f.start_with?(PREFIX) }
          .to_h do |f|
            key = key_transform ? key_transform.call(f) : f
            [key, enabled_for_any?(f, *actors).to_s]
          end
      end
    end
  end
end