module Stretchy

# File lib/stretchy.rb, line 23
def client
  @client ||= Elasticsearch::Client.new
end

# File lib/stretchy.rb, line 27
def client=(client)
  @client = client
end

# File lib/stretchy.rb, line 56
def count_index(name, params = {})
  client.count({index: name}.merge(params))['count']
end

# File lib/stretchy.rb, line 48
def create_index(name, params = {})
  client.indices.create({index: name}.merge(params)) unless index_exists? name
end

# File lib/stretchy.rb, line 44
def delete_index(name)
  client.indices.delete(index: name) if index_exists? name
end

# File lib/stretchy.rb, line 74
def fake_results
  Results.fake
end

# File lib/stretchy.rb, line 60
def index_document(params = {})
  Utils.require_params!(:index_document, params, :index, :body)

  raise IndexDoesNotExistError.new(
    "index #{params[:index]} does not exist"
  ) unless index_exists? params[:index]

  client.index(params)
end

# File lib/stretchy.rb, line 40
def index_exists?(name)
  client.indices.exists? index: name
end

# File lib/stretchy.rb, line 78
def method_missing(method, *args, &block)
  if client.respond_to?(method)
    client.send(method, *args, &block)
  else
    super
  end
end

# File lib/stretchy.rb, line 70
def query(options = {})
  API.new(root: options)
end

# File lib/stretchy.rb, line 52
def refresh_index(name, params = {})
  client.indices.refresh({index: name}.merge(params)) if index_exists? name
end

# File lib/stretchy.rb, line 31
def search(options = {})
  client.search(options)
rescue Elasticsearch::Transport::Transport::Errors::BadRequest => bre
  msg = bre.message[-150..-1]
  msg << "\n\n"
  msg << JSON.pretty_generate(options)
  raise msg
end