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

226 lines
4.4 KiB
Ruby
Raw Normal View History

2015-04-26 12:48:37 +05:30
# IssuableFinder
2014-09-02 18:07:02 +05:30
#
# Used to filter Issues and MergeRequests collections by set of params
#
# Arguments:
# klass - actual class like Issue or MergeRequest
# current_user - which user use
# params:
# scope: 'created-by-me' or 'assigned-to-me' or 'all'
# state: 'open' or 'closed' or 'all'
# group_id: integer
# project_id: integer
2015-09-11 14:41:01 +05:30
# milestone_title: string
2014-09-02 18:07:02 +05:30
# assignee_id: integer
# search: string
# label_name: string
# sort: string
#
2015-04-26 12:48:37 +05:30
require_relative 'projects_finder'
class IssuableFinder
NONE = '0'
2014-09-02 18:07:02 +05:30
attr_accessor :current_user, :params
2015-09-11 14:41:01 +05:30
def initialize(current_user, params)
2014-09-02 18:07:02 +05:30
@current_user = current_user
@params = params
2015-09-11 14:41:01 +05:30
end
2014-09-02 18:07:02 +05:30
2015-09-11 14:41:01 +05:30
def execute
2014-09-02 18:07:02 +05:30
items = init_collection
items = by_scope(items)
items = by_state(items)
items = by_group(items)
items = by_project(items)
items = by_search(items)
items = by_milestone(items)
items = by_assignee(items)
2015-04-26 12:48:37 +05:30
items = by_author(items)
2014-09-02 18:07:02 +05:30
items = by_label(items)
items = sort(items)
end
2015-09-11 14:41:01 +05:30
def group
return @group if defined?(@group)
@group =
if params[:group_id].present?
Group.find(params[:group_id])
else
nil
end
end
def project
return @project if defined?(@project)
@project =
if params[:project_id].present?
Project.find(params[:project_id])
else
nil
end
end
def search
params[:search].presence
end
def milestones?
params[:milestone_title].present?
end
def milestones
return @milestones if defined?(@milestones)
@milestones =
if milestones? && params[:milestone_title] != Milestone::None.title
Milestone.where(title: params[:milestone_title])
else
nil
end
end
def assignee?
params[:assignee_id].present?
end
def assignee
return @assignee if defined?(@assignee)
@assignee =
if assignee? && params[:assignee_id] != NONE
User.find(params[:assignee_id])
else
nil
end
end
def author?
params[:author_id].present?
end
def author
return @author if defined?(@author)
@author =
if author? && params[:author_id] != NONE
User.find(params[:author_id])
else
nil
end
end
2014-09-02 18:07:02 +05:30
private
def init_collection
table_name = klass.table_name
if project
2015-04-26 12:48:37 +05:30
if Ability.abilities.allowed?(current_user, :read_project, project)
2014-09-02 18:07:02 +05:30
project.send(table_name)
else
[]
end
2015-04-26 12:48:37 +05:30
elsif current_user && params[:authorized_only].presence && !current_user_related?
2014-09-02 18:07:02 +05:30
klass.of_projects(current_user.authorized_projects).references(:project)
else
klass.of_projects(ProjectsFinder.new.execute(current_user)).references(:project)
end
end
def by_scope(items)
case params[:scope]
when 'created-by-me', 'authored' then
items.where(author_id: current_user.id)
when 'all' then
items
when 'assigned-to-me' then
items.where(assignee_id: current_user.id)
else
raise 'You must specify default scope'
end
end
def by_state(items)
case params[:state]
when 'closed'
items.closed
2015-09-11 14:41:01 +05:30
when 'merged'
items.respond_to?(:merged) ? items.merged : items.closed
2014-09-02 18:07:02 +05:30
when 'all'
items
when 'opened'
items.opened
else
raise 'You must specify default state'
end
end
def by_group(items)
2015-09-11 14:41:01 +05:30
items = items.of_group(group) if group
2014-09-02 18:07:02 +05:30
items
end
def by_project(items)
2015-09-11 14:41:01 +05:30
items = items.of_projects(project.id) if project
2014-09-02 18:07:02 +05:30
items
end
def by_search(items)
2015-09-11 14:41:01 +05:30
items = items.search(search) if search
2014-09-02 18:07:02 +05:30
items
end
def sort(items)
items.sort(params[:sort])
end
def by_milestone(items)
2015-09-11 14:41:01 +05:30
if milestones?
items = items.where(milestone_id: milestones.try(:pluck, :id))
2014-09-02 18:07:02 +05:30
end
items
end
def by_assignee(items)
2015-09-11 14:41:01 +05:30
if assignee?
items = items.where(assignee_id: assignee.try(:id))
2015-04-26 12:48:37 +05:30
end
items
end
def by_author(items)
2015-09-11 14:41:01 +05:30
if author?
items = items.where(author_id: author.try(:id))
2014-09-02 18:07:02 +05:30
end
items
end
def by_label(items)
if params[:label_name].present?
label_names = params[:label_name].split(",")
item_ids = LabelLink.joins(:label).
where('labels.title in (?)', label_names).
where(target_type: klass.name).pluck(:target_id)
items = items.where(id: item_ids)
end
items
end
2015-04-26 12:48:37 +05:30
def current_user_related?
params[:scope] == 'created-by-me' || params[:scope] == 'authored' || params[:scope] == 'assigned-to-me'
end
2014-09-02 18:07:02 +05:30
end