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