39 lines
847 B
Ruby
39 lines
847 B
Ruby
|
# 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
|