1 files changed, 15 insertions, 2 deletions
diff --git a/lib/banzai/reference_parser/snippet_parser.rb b/lib/banzai/reference_parser/snippet_parser.rb
index b86c259efbd..bd42f6e6ed4 100644
--- a/lib/banzai/reference_parser/snippet_parser.rb
+++ b/lib/banzai/reference_parser/snippet_parser.rb
@@ -9,10 +9,23 @@ module Banzai
         Snippet
       end
 
+      # Returns all the nodes that are visible to the given user.
+      def nodes_visible_to_user(user, nodes)
+        snippets = lazy { grouped_objects_for_nodes(nodes, references_relation, self.class.data_attribute) }
+
+        nodes.select do |node|
+          if node.has_attribute?(self.class.data_attribute)
+            can_read_reference?(user, snippets[node])
+          else
+            true
+          end
+        end
+      end
+
       private
 
-      def can_read_reference?(user, ref_project, node)
-        can?(user, :read_snippet, referenced_by([node]).first)
+      def can_read_reference?(user, snippet)
+        can?(user, :read_snippet, snippet)
       end
     end
   end