module Searchkick
  class Index
    attr_reader :name

    def initialize(name)
      @name = name
    end

    def create(options = {})
      client.indices.create index: name, body: options
    end

    def delete
      client.indices.delete index: name
    end

    def exists?
      client.indices.exists index: name
    end

    def refresh
      client.indices.refresh index: name
    end

    def store(record)
      client.index(
        index: name,
        type: record.document_type,
        id: record.id,
        body: record.as_indexed_json
      )
    end

    def remove(record)
      client.delete(
        index: name,
        type: record.document_type,
        id: record.id
      )
    end

    def import(records)
      if records.any?
        client.bulk(
          index: name,
          type: records.first.document_type,
          body: records.map{|r| data = r.as_indexed_json; {index: {_id: data["_id"] || data["id"] || r.id, data: data}} }
        )
      end
    end

    def retrieve(document_type, id)
      client.get(
        index: name,
        type: document_type,
        id: id
      )["_source"]
    end

    protected

    def client
      Searchkick.client
    end

  end
end