1 files changed, 24 insertions, 0 deletions
diff --git a/lib/banzai/filter/references/vulnerability_reference_filter.rb b/lib/banzai/filter/references/vulnerability_reference_filter.rb
new file mode 100644
index 00000000000..e5f2408eda4
--- /dev/null
+++ b/lib/banzai/filter/references/vulnerability_reference_filter.rb
@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Banzai
+  module Filter
+    module References
+      # The actual filter is implemented in the EE mixin
+      class VulnerabilityReferenceFilter < IssuableReferenceFilter
+        self.reference_type = :vulnerability
+
+        def self.object_class
+          Vulnerability
+        end
+
+        private
+
+        def project
+          context[:project]
+        end
+      end
+    end
+  end
+end
+
+Banzai::Filter::References::VulnerabilityReferenceFilter.prepend_if_ee('EE::Banzai::Filter::References::VulnerabilityReferenceFilter')