big_index.rb

require 'rubygems'
require 'pathname'
require 'set'

dir = File.expand_path(File.join(File.dirname(__FILE__), 'big_index')) + '/'
vendor_dir = File.expand_path(File.join(File.dirname(__FILE__), '..', 'vendor')) + '/'

require (vendor_dir + 'solr').to_s
require dir + 'support'
require dir + 'adapters'
require dir + 'repository'
require dir + 'model'
require dir + 'resource'
require dir + 'index_field'


module BigIndex
  extend Assertions

  def self.root
    @root ||= File.expand_path(File.join(File.dirname(__FILE__), '..'))
  end

  def self.vendor_root
    @vendor_root ||= File.expand_path(File.join(File.dirname(__FILE__), '..', 'vendor')) + '/'
  end

  def self.setup(name, options)
    assert_kind_of 'name',      name,       Symbol, String
    assert_kind_of 'options',   options,    Hash

    name = name.to_sym if name.is_a?(String)

    adapter_name = options[:adapter].to_s

    class_name = adapter_name.capitalize + 'Adapter'

    unless Adapters::const_defined?(class_name)
      lib_name = "#{adapter_name}_adapter"
      begin
        require root + 'lib' + 'big_index' + 'adapters' + lib_name
      rescue LoadError => e
        begin
          require lib_name
        rescue Exception
          # library not found, raise the original error
          raise e
        end
      end
    end

    Repository.adapters[name] = Adapters::const_get(class_name).new(name, options)
  end

  def self.configurations
    Repository.adapters || {}
  end

  def self.configurations=(configurations)
    assert_kind_of 'configurations', configurations, Hash

    Repository.clear_adapters

    @configurations = symbolize_keys(configurations)
    @configurations.each do |key, value|
      assert_kind_of 'value', value, Hash

      setup(key, value)
    end

    BigIndex::Repository.default_name = @configurations.keys[0] if @configurations.size == 1
  end

  def self.repository(name = nil) # :yields: current_context
    current_repository = if name
      raise ArgumentError, "First optional argument must be a Symbol, but was #{name.inspect}" unless name.is_a?(Symbol)
      Repository.context.detect { |r| r.name == name } || Repository.new(name)
    else
      Repository.context.last || Repository.new(Repository.default_name)
    end

    if block_given?
      current_repository.scope { |*block_args| yield(*block_args) }
    else
      current_repository
    end
  end

  def self.symbolize_keys(h)
    config = {}

    h.each do |k, v|
      if k == 'port'
        config[k.to_sym] = v.to_i
      elsif v.is_a?(Hash)
        config[k.to_sym] = symbolize_keys(v)
      else
        config[k.to_sym] = v
      end
    end

    config
  end

end # module BigIndex