GSA/i14y

class DocumentSearch
  NO_HITS = { "hits" => { "total" => 0, "hits" => [] }}

  attr_reader :doc_query, :offset, :size, :indices

  def initialize(options)
    @offset = options[:offset] || 0
    @size = options[:size]
    @doc_query = DocumentQuery.new(options)
    @indices = options[:handles].map { |handle| DocumentRepository.index_namespace(handle) }
  end

  def search
    i14y_search_results = execute_client_search
    if i14y_search_results.total.zero? && i14y_search_results.suggestion.present?
      suggestion = i14y_search_results.suggestion
      doc_query.query = suggestion['text']
      i14y_search_results = execute_client_search
      i14y_search_results.override_suggestion(suggestion) if i14y_search_results.results.present?
    end
    i14y_search_results
  rescue StandardError => error
    Rails.logger.error <<~ERROR_DETAILS
      Problem in DocumentSearch#search(): #{error}
      Query: #{doc_query.body.to_json}
      Backtrace: #{error.backtrace}
    ERROR_DETAILS
    NewRelic::Agent.notice_error(error, options: { custom_params: { indices: indices }})
    DocumentSearchResults.new(NO_HITS)
  end

  private

  def execute_client_search
    params = {
      index: indices,
      body: doc_query.body,
      from: offset,
      size: size,
      # For compatibility with ES 6. This parameter will be removed in ES 8.
      # https://www.elastic.co/guide/en/elasticsearch/reference/current/breaking-changes-7.0.html#hits-total-now-object-search-response
      rest_total_hits_as_int: true
    }
    Rails.logger.debug "Query: *****\n#{doc_query.body.to_json}\n*****"
    result = ES.client.search(params)
    DocumentSearchResults.new(result, offset)
  end
end