debian-mirror-gitlab/app/finders/labels_finder.rb

165 lines
3.6 KiB
Ruby
Raw Normal View History

2016-11-03 12:29:30 +05:30
class LabelsFinder < UnionFinder
2018-03-27 19:54:05 +05:30
prepend FinderWithCrossProjectAccess
include FinderMethods
2018-03-17 18:26:18 +05:30
include Gitlab::Utils::StrongMemoize
2018-03-27 19:54:05 +05:30
requires_cross_project_access unless: -> { project? }
2016-11-03 12:29:30 +05:30
def initialize(current_user, params = {})
@current_user = current_user
@params = params
end
def execute(skip_authorization: false)
@skip_authorization = skip_authorization
2016-11-24 13:41:30 +05:30
items = find_union(label_ids, Label) || Label.none
2016-11-03 12:29:30 +05:30
items = with_title(items)
2018-11-18 11:00:15 +05:30
items = by_search(items)
2016-11-03 12:29:30 +05:30
sort(items)
end
private
attr_reader :current_user, :params, :skip_authorization
def label_ids
label_ids = []
2016-11-24 13:41:30 +05:30
if project?
if project
2017-08-17 22:00:37 +05:30
if project.group.present?
labels_table = Label.arel_table
2018-05-09 12:01:36 +05:30
group_ids = group_ids_for(project.group)
2017-08-17 22:00:37 +05:30
label_ids << Label.where(
2018-05-09 12:01:36 +05:30
labels_table[:type].eq('GroupLabel').and(labels_table[:group_id].in(group_ids)).or(
2017-08-17 22:00:37 +05:30
labels_table[:type].eq('ProjectLabel').and(labels_table[:project_id].eq(project.id))
)
)
else
label_ids << project.labels
end
2016-11-24 13:41:30 +05:30
end
2016-11-03 12:29:30 +05:30
else
2018-05-09 12:01:36 +05:30
if group?
group = Group.find(params[:group_id])
label_ids << Label.where(group_id: group_ids_for(group))
end
2016-11-03 12:29:30 +05:30
label_ids << Label.where(group_id: projects.group_ids)
2018-05-09 12:01:36 +05:30
label_ids << Label.where(project_id: projects.select(:id)) unless only_group_labels?
2016-11-03 12:29:30 +05:30
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
2018-11-18 11:00:15 +05:30
def by_search(labels)
return labels unless search?
labels.search(params[:search])
end
2018-05-09 12:01:36 +05:30
# Gets redacted array of group ids
# which can include the ancestors and descendants of the requested group.
def group_ids_for(group)
2018-03-27 19:54:05 +05:30
strong_memoize(:group_ids) do
2018-05-09 12:01:36 +05:30
groups = groups_to_include(group)
groups_user_can_read_labels(groups).map(&:id)
2018-03-17 18:26:18 +05:30
end
end
2018-05-09 12:01:36 +05:30
def groups_to_include(group)
2018-03-27 19:54:05 +05:30
groups = [group]
2018-05-09 12:01:36 +05:30
groups += group.ancestors if include_ancestor_groups?
groups += group.descendants if include_descendant_groups?
2018-03-27 19:54:05 +05:30
groups
end
2018-05-09 12:01:36 +05:30
def include_ancestor_groups?
params[:include_ancestor_groups]
end
def include_descendant_groups?
params[:include_descendant_groups]
end
2016-11-24 13:41:30 +05:30
def group?
params[:group_id].present?
2016-11-03 12:29:30 +05:30
end
2016-11-24 13:41:30 +05:30
def project?
params[:project_id].present?
2016-11-03 12:29:30 +05:30
end
2016-11-24 13:41:30 +05:30
def projects?
2018-03-17 18:26:18 +05:30
params[:project_ids]
end
def only_group_labels?
params[:only_group_labels]
2016-11-03 12:29:30 +05:30
end
2018-11-18 11:00:15 +05:30
def search?
params[:search].present?
end
2016-11-03 12:29:30 +05:30
def title
params[:title] || params[:name]
end
def project
return @project if defined?(@project)
2016-11-24 13:41:30 +05:30
if project?
@project = Project.find(params[:project_id])
@project = nil unless authorized_to_read_labels?(@project)
2016-11-03 12:29:30 +05:30
else
@project = nil
end
@project
end
def projects
return @projects if defined?(@projects)
2017-09-10 17:25:29 +05:30
@projects = if skip_authorization
Project.all
else
ProjectsFinder.new(params: { non_archived: true }, current_user: current_user).execute
end
2016-11-24 13:41:30 +05:30
@projects = @projects.in_namespace(params[:group_id]) if group?
@projects = @projects.where(id: params[:project_ids]) if projects?
2016-11-03 12:29:30 +05:30
@projects = @projects.reorder(nil)
@projects
end
2018-03-17 18:26:18 +05:30
def authorized_to_read_labels?(label_parent)
2016-11-24 13:41:30 +05:30
return true if skip_authorization
2018-03-17 18:26:18 +05:30
Ability.allowed?(current_user, :read_label, label_parent)
2016-11-03 12:29:30 +05:30
end
2018-03-27 19:54:05 +05:30
def groups_user_can_read_labels(groups)
DeclarativePolicy.user_scope do
groups.select { |group| authorized_to_read_labels?(group) }
end
end
2016-11-03 12:29:30 +05:30
end