Remove nodes_visible_to_user from ProjectParser and memoize readable_project_ids
This commit is contained in:
parent
a08b5144fe
commit
8cc646fa0e
1 changed files with 10 additions and 17 deletions
|
@ -1,35 +1,28 @@
|
|||
module Banzai
|
||||
module ReferenceParser
|
||||
class ProjectParser < BaseParser
|
||||
include Gitlab::Utils::StrongMemoize
|
||||
|
||||
self.reference_type = :project
|
||||
|
||||
def references_relation
|
||||
Project
|
||||
end
|
||||
|
||||
def nodes_visible_to_user(user, nodes)
|
||||
nodes_projects_hash = lazy { projects_for_nodes(nodes) }
|
||||
project_attr = 'data-project'
|
||||
|
||||
readable_project_ids = projects_readable_by_user(nodes_projects_hash.values, user)
|
||||
|
||||
nodes.select do |node|
|
||||
if node.has_attribute?(project_attr)
|
||||
readable_project_ids.include?(nodes_projects_hash[node].try(:id))
|
||||
else
|
||||
true
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
private
|
||||
|
||||
# Returns an Array of Project ids that can be read by the given user.
|
||||
#
|
||||
# projects - The projects to reduce down to those readable by the user.
|
||||
# user - The User for which to check the projects
|
||||
def projects_readable_by_user(projects, user)
|
||||
Project.public_or_visible_to_user(user).where("projects.id IN (?)", projects.map(&:id)).pluck(:id)
|
||||
def readable_project_ids_for(projects, user)
|
||||
strong_memoize(:readable_project_ids_for) do
|
||||
Project.public_or_visible_to_user(user).where("projects.id IN (?)", projects.map(&:id)).pluck(:id)
|
||||
end
|
||||
end
|
||||
|
||||
def can_read_reference?(user, ref_project, node)
|
||||
readable_project_ids_for(@projects_for_nodes.values, user).include?(ref_project.try(:id))
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue