pulibrary / orangelight / 0e37073d-109c-440c-949f-49d2aa86311b

class SearchBuilder < Blacklight::SearchBuilder

  include Blacklight::Solr::SearchBuilderBehavior

  include BlacklightRangeLimit::RangeLimitBuilder

  include BlacklightHelper

  default_processor_chain.unshift(:conditionally_configure_json_query_dsl)

  self.default_processor_chain += %i[parslet_trick cleanup_boolean_operators

                                     cjk_mm wildcard_char_strip

                                     only_home_facets prepare_left_anchor_search

                                     series_title_results pul_holdings html_facets

                                     numismatics_advanced

                                     adjust_mm remove_unneeded_facets]

  def cleanup_boolean_operators(solr_parameters)

    transform_queries!(solr_parameters) { |query| cleaned_query(query) }

  end

  def parslet_trick(_solr_parameters)

    return unless blacklight_params[:q].is_a?(String)

    return unless blacklight_params[:q].strip.end_with?("()")

    blacklight_params[:q] = blacklight_params[:q].strip.gsub("()", "")

  end

  def numismatics_advanced(solr_parameters)

    return unless blacklight_params[:advanced_type] == 'numismatics'

    solr_parameters[:fq] ||= []

    solr_parameters[:fq] << "format:Coin"

  end

  def only_home_facets(solr_parameters)

    return if search_parameters? || advanced_search?

    solr_parameters['facet.field'] = blacklight_config.facet_fields.select { |_, v| v[:home] }.keys

    solr_parameters['facet.pivot'] = []

  end

  def advanced_search?

    blacklight_params[:advanced_type] == 'advanced' ||

      search_state.controller.try(:params).try(:[], :action) == 'advanced_search' ||

      blacklight_params[:advanced_type] == 'numismatics' ||

      blacklight_params[:search_field] == 'advanced' ||

      blacklight_params[:action] == 'numismatics'

  end

  def search_parameters?

    search_query_present? || facet_query_present?

  end

  def conditionally_configure_json_query_dsl(_solr_parameters)

    advanced_fields = %w[all_fields title author subject left_anchor publisher in_series notes series_title isbn issn]

    add_edismax(advanced_fields:)

  end

  def adjust_mm(solr_parameters)

    return unless includes_written_boolean?

    solr_parameters['mm'] = 0

  end

  def includes_written_boolean?

    if advanced_search? && search_query_present?

      json_query_dsl_clauses&.any? { |clause| clause&.dig('query')&.include?('OR') }

    else

      blacklight_params[:q].to_s.split.include? 'OR'

    end

  end

  def remove_unneeded_facets(solr_parameters)

    return unless facet

    remove_unneeded_stats(solr_parameters)

    solr_parameters.delete('facet.pivot') unless solr_parameters['facet.pivot']&.split(',')&.include? facet

    solr_parameters.delete('facet.query') unless solr_parameters['facet.query']&.any? { |query| query.partition(':').first == facet }

  end

  def wildcard_char_strip(solr_parameters)

    transform_queries!(solr_parameters) { |query| query.delete('?') }

  end

  private

    def search_query_present?

      !blacklight_params[:q].nil? || json_query_dsl_clauses&.any? { |clause| clause.dig('query')&.present? }

    end

    def facet_query_present?

      blacklight_params[:f].present? || blacklight_params[:action] == 'facet'

    end

    def json_query_dsl_clauses

      blacklight_params.dig('clause')&.values

    end

    def add_edismax(advanced_fields:)

      advanced_fields.each do |field|

        solr_params = blacklight_config.search_fields[field]['solr_parameters']

        edismax = solr_params.present? ? solr_params.dup : {}

        blacklight_config.search_fields[field]['clause_params'] = { edismax: }

      end

    end

    def remove_unneeded_stats(solr_parameters)

      return if solr_parameters['stats.field'].to_a.include? facet

      solr_parameters.delete('stats')

      solr_parameters.delete('stats.field')

    end

    def transform_queries!(solr_parameters)

      solr_parameters[:q] = yield solr_parameters[:q] if solr_parameters[:q]

      solr_parameters.dig('json', 'query', 'bool', 'must')&.map! do |search_element|

        search_element[:edismax][:query] = yield search_element[:edismax][:query]

        search_element

      end

    end

    def cleaned_query(query)

      return query if query.nil?

      query.gsub(/([A-Z]) (NOT|OR|AND) ([A-Z])/) do

        "#{Regexp.last_match(1)} #{Regexp.last_match(2).downcase} #{Regexp.last_match(3)}"

      end

    end

end