1 files changed, 10 insertions, 0 deletions
diff --git a/lib/gitlab/audit/type/definition.rb b/lib/gitlab/audit/type/definition.rb
index 81c88a3a0ae..772023616b8 100644
--- a/lib/gitlab/audit/type/definition.rb
+++ b/lib/gitlab/audit/type/definition.rb
@@ -13,6 +13,10 @@ module Gitlab
         validate :validate_schema
         validate :validate_file_name
 
+        def self.declarative_policy_class
+          'AuditEvents::DefinitionPolicy'
+        end
+
         InvalidAuditEventTypeError = Class.new(StandardError)
 
         AUDIT_EVENT_TYPE_SCHEMA_PATH = Rails.root.join('config', 'audit_events', 'types', 'type_schema.json')
@@ -78,6 +82,12 @@ module Gitlab
             definitions.keys.map(&:to_s)
           end
 
+          def names_with_category
+            definitions.map do |event_name, value|
+              { event_name: event_name, feature_category: value.attributes[:feature_category] }
+            end
+          end
+
           def defined?(key)
             get(key).present?
           end