diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2021-10-20 11:43:02 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2021-10-20 11:43:02 +0300 |
commit | d9ab72d6080f594d0b3cae15f14b3ef2c6c638cb (patch) | |
tree | 2341ef426af70ad1e289c38036737e04b0aa5007 /lib/gitlab/feature_categories.rb | |
parent | d6e514dd13db8947884cd58fe2a9c2a063400a9b (diff) |
Add latest changes from gitlab-org/gitlab@14-4-stable-eev14.4.0-rc42
Diffstat (limited to 'lib/gitlab/feature_categories.rb')
-rw-r--r-- | lib/gitlab/feature_categories.rb | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/lib/gitlab/feature_categories.rb b/lib/gitlab/feature_categories.rb new file mode 100644 index 00000000000..d06f3b14fed --- /dev/null +++ b/lib/gitlab/feature_categories.rb @@ -0,0 +1,38 @@ +# frozen_string_literal: true + +module Gitlab + class FeatureCategories + FEATURE_CATEGORY_DEFAULT = 'unknown' + + attr_reader :categories + + def self.default + @default ||= self.load_from_yaml + end + + def self.load_from_yaml + categories = YAML.load_file(Rails.root.join('config', 'feature_categories.yml')) + + new(categories) + end + + def initialize(categories) + @categories = categories.to_set + end + + # If valid, returns a feature category from the given request. + def from_request(request) + category = request.headers["HTTP_X_GITLAB_FEATURE_CATEGORY"].presence + + return unless category && valid?(category) + + return unless ::Gitlab::RequestForgeryProtection.verified?(request.env) + + category + end + + def valid?(category) + categories.include?(category.to_s) + end + end +end |