153 lines
3.5 KiB
Ruby
153 lines
3.5 KiB
Ruby
class LabelsFinder < UnionFinder
|
|
prepend FinderWithCrossProjectAccess
|
|
include FinderMethods
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
requires_cross_project_access unless: -> { project? }
|
|
|
|
def initialize(current_user, params = {})
|
|
@current_user = current_user
|
|
@params = params
|
|
end
|
|
|
|
def execute(skip_authorization: false)
|
|
@skip_authorization = skip_authorization
|
|
items = find_union(label_ids, Label) || Label.none
|
|
items = with_title(items)
|
|
sort(items)
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :current_user, :params, :skip_authorization
|
|
|
|
def label_ids
|
|
label_ids = []
|
|
|
|
if project?
|
|
if project
|
|
if project.group.present?
|
|
labels_table = Label.arel_table
|
|
group_ids = group_ids_for(project.group)
|
|
|
|
label_ids << Label.where(
|
|
labels_table[:type].eq('GroupLabel').and(labels_table[:group_id].in(group_ids)).or(
|
|
labels_table[:type].eq('ProjectLabel').and(labels_table[:project_id].eq(project.id))
|
|
)
|
|
)
|
|
else
|
|
label_ids << project.labels
|
|
end
|
|
end
|
|
else
|
|
if group?
|
|
group = Group.find(params[:group_id])
|
|
label_ids << Label.where(group_id: group_ids_for(group))
|
|
end
|
|
|
|
label_ids << Label.where(group_id: projects.group_ids)
|
|
label_ids << Label.where(project_id: projects.select(:id)) unless only_group_labels?
|
|
end
|
|
|
|
label_ids
|
|
end
|
|
|
|
def sort(items)
|
|
items.reorder(title: :asc)
|
|
end
|
|
|
|
def with_title(items)
|
|
return items if title.nil?
|
|
return items.none if title.blank?
|
|
|
|
items.where(title: title)
|
|
end
|
|
|
|
# Gets redacted array of group ids
|
|
# which can include the ancestors and descendants of the requested group.
|
|
def group_ids_for(group)
|
|
strong_memoize(:group_ids) do
|
|
groups = groups_to_include(group)
|
|
|
|
groups_user_can_read_labels(groups).map(&:id)
|
|
end
|
|
end
|
|
|
|
def groups_to_include(group)
|
|
groups = [group]
|
|
|
|
groups += group.ancestors if include_ancestor_groups?
|
|
groups += group.descendants if include_descendant_groups?
|
|
|
|
groups
|
|
end
|
|
|
|
def include_ancestor_groups?
|
|
params[:include_ancestor_groups]
|
|
end
|
|
|
|
def include_descendant_groups?
|
|
params[:include_descendant_groups]
|
|
end
|
|
|
|
def group?
|
|
params[:group_id].present?
|
|
end
|
|
|
|
def project?
|
|
params[:project_id].present?
|
|
end
|
|
|
|
def projects?
|
|
params[:project_ids]
|
|
end
|
|
|
|
def only_group_labels?
|
|
params[:only_group_labels]
|
|
end
|
|
|
|
def title
|
|
params[:title] || params[:name]
|
|
end
|
|
|
|
def project
|
|
return @project if defined?(@project)
|
|
|
|
if project?
|
|
@project = Project.find(params[:project_id])
|
|
@project = nil unless authorized_to_read_labels?(@project)
|
|
else
|
|
@project = nil
|
|
end
|
|
|
|
@project
|
|
end
|
|
|
|
def projects
|
|
return @projects if defined?(@projects)
|
|
|
|
@projects = if skip_authorization
|
|
Project.all
|
|
else
|
|
ProjectsFinder.new(params: { non_archived: true }, current_user: current_user).execute
|
|
end
|
|
|
|
@projects = @projects.in_namespace(params[:group_id]) if group?
|
|
@projects = @projects.where(id: params[:project_ids]) if projects?
|
|
@projects = @projects.reorder(nil)
|
|
|
|
@projects
|
|
end
|
|
|
|
def authorized_to_read_labels?(label_parent)
|
|
return true if skip_authorization
|
|
|
|
Ability.allowed?(current_user, :read_label, label_parent)
|
|
end
|
|
|
|
def groups_user_can_read_labels(groups)
|
|
DeclarativePolicy.user_scope do
|
|
groups.select { |group| authorized_to_read_labels?(group) }
|
|
end
|
|
end
|
|
end
|