machinio/solrb

require 'json'
require 'faraday'
require 'addressable/uri'
require 'solr/support'
require 'solr/version'
require 'solr/configuration'
require 'solr/connection'
require 'solr/document'
require 'solr/document_collection'
require 'solr/grouped_document_collection'
require 'solr/response'
require 'solr/spatial_point'
require 'solr/spatial_rectangle'
require 'solr/request/http_request'
require 'solr/request/runner'
require 'solr/query/request'
require 'solr/update/request'

require 'solr/cloud/helper_methods'
require 'solr/master_slave/helper_methods'
require 'solr/helper_methods'
require 'solr/commands'

module Solr
  class << self
    include Solr::Commands
    include Solr::Cloud::HelperMethods
    include Solr::MasterSlave::HelperMethods
    include Solr::HelperMethods

    CURRENT_CORE_CONFIG_VARIABLE_NAME = :solrb_current_core_config
    SOLR_NODE_URL_OVERRIDE_CONFIG = :solrb_node_url_override_config

    attr_accessor :configuration

    Solr.configuration = Configuration.new

    def configure
      yield configuration
      configuration.validate!
      if configuration.zookeeper_url
        enable_solr_cloud!
      elsif configuration.master_url
        enable_master_slave!
      end
      configuration
    end

    def current_core_config
      Thread.current[CURRENT_CORE_CONFIG_VARIABLE_NAME] || Solr.configuration.default_core_config
    end

    def with_core(core)
      core_config = Solr.configuration.core_config_by_name(core)
      old_core_config = Thread.current[CURRENT_CORE_CONFIG_VARIABLE_NAME]
      Thread.current[CURRENT_CORE_CONFIG_VARIABLE_NAME] = core_config
      yield
    ensure
      Thread.current[CURRENT_CORE_CONFIG_VARIABLE_NAME] = old_core_config
    end

    def with_node_url(url)
      Thread.current[SOLR_NODE_URL_OVERRIDE_CONFIG] = url
      yield
    ensure
      Thread.current[SOLR_NODE_URL_OVERRIDE_CONFIG] = nil
    end

    def node_url_override
      Thread.current[SOLR_NODE_URL_OVERRIDE_CONFIG]
    end

    def solr_url(path = '')
      Solr::Support::UrlHelper.solr_url(path)
    end

    def instrument(name:, data: {})
      if defined? ActiveSupport::Notifications
        # Create a copy of data to avoid modifications on the original object by rails
        # https://github.com/rails/rails/blob/master/activesupport/lib/active_support/notifications.rb#L66-L70
        ActiveSupport::Notifications.instrument(name, data.dup) do
          yield if block_given?
        end
      else
        yield if block_given?
      end
    end
  end
end