gitlab-org--gitlab-foss/app/finders/group_children_finder.rb
2017-10-04 22:46:49 +02:00

92 lines
2.4 KiB
Ruby

class GroupChildrenFinder
include Gitlab::Allowable
attr_reader :current_user, :parent_group, :params
def initialize(current_user = nil, parent_group:, params: {})
@current_user = current_user
@parent_group = parent_group
@params = params
end
def execute
Kaminari.paginate_array(children)
end
# This allows us to fetch only the count without loading the objects. Unless
# the objects were already loaded.
def total_count
@total_count ||= subgroup_count + project_count
end
def subgroup_count
@subgroup_count ||= if defined?(@children)
children.count { |child| child.is_a?(Group) }
else
subgroups.count
end
end
def project_count
@project_count ||= if defined?(@children)
children.count { |child| child.is_a?(Project) }
else
projects.count
end
end
private
def children
@children ||= subgroups + projects
end
def base_groups
GroupsFinder.new(current_user,
parent: parent_group,
all_available: true).execute
end
def all_subgroups
Gitlab::GroupHierarchy.new(Group.where(id: parent_group)).all_groups
end
def subgroups_matching_filter
all_subgroups.search(params[:filter])
end
def subgroups
return Group.none unless Group.supports_nested_groups?
return Group.none unless can?(current_user, :read_group, parent_group)
groups = if params[:filter]
subgroups_matching_filter
else
base_groups
end
groups = groups.includes(:route).includes(:children)
groups.sort(params[:sort])
end
def base_projects
GroupProjectsFinder.new(group: parent_group, params: params, current_user: current_user).execute
end
def projects_matching_filter
ProjectsFinder.new(current_user: current_user).execute
.search(params[:filter])
.where(namespace: all_subgroups)
end
def projects
return Project.none unless can?(current_user, :read_group, parent_group)
projects = if params[:filter]
projects_matching_filter
else
base_projects
end
projects = projects.includes(:route)
projects.sort(params[:sort])
end
end