forgeplus/app/queries/projects/list_query.rb

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 = 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
    # 增加私有组织中项目过滤
    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)
    items.with_project_topic(params[:topic_id])
  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