Reuse LabelsFinder on Banzai::Filter::LabelReferenceFilter

1 files changed, 5 insertions, 7 deletions

diff --git a/lib/banzai/filter/label_reference_filter.rb b/lib/banzai/filter/label_reference_filter.rb
index 649c697b415..2ce33aa1d15 100644
--- a/lib/banzai/filter/label_reference_filter.rb
+++ b/lib/banzai/filter/label_reference_filter.rb
@@ -39,13 +39,7 @@ module Banzai
       end
 
       def find_labels(project)
-        label_ids = []
-        label_ids << project.group.labels.select(:id) if project.group.present?
-        label_ids << project.labels.select(:id)
-
-        union = Gitlab::SQL::Union.new(label_ids)
-
-        object_class.where("labels.id IN (#{union.to_sql})")
+        LabelsFinder.new(user, project_id: project.id).execute
       end
 
       # Parameters to pass to `Label.find_by` based on the given arguments
@@ -91,6 +85,10 @@ module Banzai
         object.is_a?(ProjectLabel) && object.project == project
       end
 
+      def user
+        context[:current_user] || context[:author]
+      end
+
       def project
         context[:project]
       end