SpeciesFileGroup/taxonworks

module Queries
  module GeographicArea
    class Autocomplete < Query::Autocomplete

      include Queries::Concerns::AlternateValues
      include Queries::Concerns::Tags

      def initialize(string, **params)
        set_tags_params(params)
        set_alternate_value(params)
        super
      end

      # TODO: use or_clauses
      # @return [String]
      def where_sql
        named.to_sql
      end

      def autocomplete_alternate_values_name
        matching_alternate_value_on(:name)
      end

      # @return Array
      def autocomplete
        #t = Time.now
        pr_id = project_id.join(',') if project_id.present?

        return [] if query_string.blank?
        queries = [
          [ ::GeographicArea.where(id: query_string).all, false ],
          [ ::GeographicArea.where(name: query_string).all, false ],
          [ ::GeographicArea.joins(:alternate_values).where('alternate_values.value ILIKE ?', query_string).all, false ],
          [ ::GeographicArea.joins(parent_child_join).where(Arel.sql(parent_child_where.to_sql)).limit(5).all, true ],
          [ ::GeographicArea.where(Arel.sql(where_sql)).limit(dynamic_limit).all, true ],
          [ autocomplete_exact_id, false ],
          [ autocomplete_identifier_cached_exact, false ],
          [ autocomplete_identifier_identifier_exact, false ],
          [ autocomplete_alternate_values_name.limit(dynamic_limit), true ]
        ]

        queries.compact!

        updated_queries = []

        queries.each_with_index do |q, i|
          a = q[0]
          if q[1] && query_string.length > 3 # do not use extended query for identifiers
            if project_id.present? && !a.nil?
               a = a.left_outer_joins(:asserted_distributions)
                 .left_outer_joins(:collecting_events)
                 .select("geographic_areas.*, (COUNT(collecting_events.id) + COUNT(asserted_distributions.id)) AS use_count, CASE WHEN asserted_distributions.project_id IN (#{pr_id}) THEN asserted_distributions.project_id WHEN collecting_events.project_id IN (#{pr_id}) THEN collecting_events.project_id ELSE NULL END AS in_project")
                 .group('geographic_areas.id, collecting_events.project_id, asserted_distributions.project_id')
                 .order('in_project, use_count DESC')
            end
          end
          updated_queries[i] = a
        end
        result = []
        updated_queries.compact!
        #t2 = Time.now.to_f - t.to_f
        #puts t2
        updated_queries.each do |q|
          result += q.includes(:geographic_areas_geographic_items).to_a
          result.uniq!
          break if result.count > 19
        end
        #t2 = Time.now.to_f - t.to_f
        #puts t2
        result = result[0..19]
      end

    end
  end
end