SpeciesFileGroup/taxonworks

module Queries

  class Repository::Autocomplete < Query::Autocomplete

    include Queries::Concerns::AlternateValues
    include Queries::Concerns::Identifiers

    # @param [Hash] args
    def initialize(string, **params)
      set_identifier_params(params)
      set_alternate_value(params)
      super
    end

    # @return [Scope]
    def autocomplete_acronym_match
      base_query.where(
        table[:acronym].matches_any(terms).to_sql
      )
    end

    # @return [Scope]
    def autocomplete_exact_acronym
      base_query.where(
        table[:acronym].eq(query_string).to_sql
      )
    end

    def autocomplete_alternate_values_acronym
      matching_alternate_value_on(:acronym)
    end

    def autocomplete_alternate_values_name
      matching_alternate_value_on(:name)
    end

    # @return [Array]
    def autocomplete
      t = Time.now
      return [] if query_string.blank?
      queries = [
        [autocomplete_exact_id, nil ],
        [autocomplete_exactly_named, true],
        [autocomplete_exact_acronym, true],
        [autocomplete_identifier_identifier_exact, nil],
        [autocomplete_acronym_match.limit(10), true],
        [autocomplete_named, true ],
        [autocomplete_alternate_values_acronym.limit(20), true ],
        [autocomplete_alternate_values_name.limit(20), true ]
      ]

      queries.delete_if{|a,b| a.nil?} # Note this pattern differs because [[]] so we don't use compact. /lib/queries/source/autocomplete.rb follows same pattern

      result = []

      pr_id = project_id.join(',') if project_id
      queries.each do |q, scope|
        a = q
        if project_id && scope && query_string.length > 2
          a = a.select("repositories.*, COUNT(collection_objects.id) AS use_count, CASE WHEN collection_objects.project_id IN (#{pr_id}) THEN collection_objects.project_id ELSE NULL END AS in_project")
             .joins('LEFT OUTER JOIN collection_objects ON (repositories.id = collection_objects.repository_id OR repositories.id = collection_objects.current_repository_id)')
             .group('repositories.id, collection_objects.project_id')
             .order('in_project, use_count DESC')
        end
        a ||= q

        result += a.to_a
        result.uniq!
        break if result.count > 40
      end
      t2 = Time.now.to_f - t.to_f
      puts t2
      result[0..40]
    end
  end
end