forgeplus/app/queries/projects/list_query.rb

112 lines
3.7 KiB
Ruby

class Projects::ListQuery < ApplicationQuery
include CustomSortable
attr_reader :params, :current_user_id
sort_columns :updated_on, :created_on, :forked_count, :praises_count, default_by: :updated_on, default_direction: :desc, default_table: 'projects'
def initialize(params, current_user_id=nil)
@params = params
@current_user_id = current_user_id
end
def call
collection = main_collection
collection = filter_projects(collection)
sort = params[:sort_by] || "updated_on"
sort_direction = params[:sort_direction] || "desc"
collection = optimize_sorting(collection, sort) if params[:category_id].present?
# 如果有搜索关键字根据ES搜索结果排序
if params[:search].present? && @ids.present?
collection.reorder(Arel.sql("FIELD(projects.id,#{@ids.join(',')})"))
else
custom_sort(collection, sort, sort_direction)
end
end
def filter_projects(collection)
# collection = by_pinned(collection)
collection = by_search(collection) if params[:search].present?
collection = collection.where(id: params[:ids].to_s.split(",")) if params[:ids].present?
collection = by_project_type(collection)
collection = by_project_category(collection)
collection = by_project_language(collection)
collection = by_project_topic(collection)
collection
end
def main_collection
collection = Project.visible.where(status: 1)
# 增加私有组织中项目过滤
collection = collection.joins("left join organization_extensions on organization_extensions.organization_id = projects.user_id")
.where("organization_extensions.visibility is null or organization_extensions.visibility in (0,1)")
.where("projects.user_id > 0")
collection
end
def by_search(items)
@ids = Projects::ElasticsearchService.call(params[:search])
items = items.where(platform: 'forge')
if @ids.present?
# items = items.where(id: @ids).by_name_or_identifier(params[:search])
items = items.where(id: @ids)
else
# 表情处理
keywords = params[:search].to_s.each_char.select { |c| c.bytes.first < 240 }.join('')
items = items.by_name_or_identifier(keywords).or(main_collection.where(user_id: Owner.like(keywords).pluck(:id)))
end
items
end
def by_project_type(items)
items.with_project_type(params[:project_type])
end
def by_project_category(items)
items.with_project_category(params[:category_id])
end
def by_project_language(items)
items.with_project_language(params[:language_id])
end
def by_pinned(items)
(params[:pinned].present? && params[:category_id].present?) ? items.pinned : items
end
def by_project_topic(items)
if params[:topic_name].present?
items.with_project_topic_name(params[:topic_name].to_s.split(","))
else
items.with_project_topic(params[:topic_id])
end
end
# 优化排序
def optimize_sorting(relations, sort_by)
if sort_by == "updated_on"
relations.where("projects.updated_on>'2010-01-01'")
elsif sort_by == "created_on"
relations.where("projects.created_on>'2010-01-01'")
elsif sort_by == "forked_count"
relations.where("projects.forked_count>=0")
elsif sort_by == "praises_count"
relations.where("projects.praises_count>=0")
else
relations
end
end
def by_recommend(items)
params[:recommend].to_s == "true" ? items.where(recommend: true) : items
end
def exclude_fork(items)
return items if params[:exclude_forked].blank?
params[:exclude_forked].to_s == "true" ? items.where("forked_from_project_id is null") : items.where("forked_from_project_id is not null")
end
end