module Elasticity
  module Strategies
    class SingleIndex
      STATUSES = [:missing, :ok]

      def initialize(client, index_name)
        @client     = client
        @index_name = index_name
      end

      def remap!
        raise NotImplementedError
      end

      def missing?
        not @client.index_exists(index: @index_name)
      end

      def create(index_def)
        if missing?
          @client.index_create(index: @index_name, body: index_def)
        else
          raise IndexError.new(@index_name, "index already exist")
        end
      end

      def create_if_undefined(index_def)
        create(index_def) if missing?
      end

      def delete
        @client.index_delete(index: @index_name)
      end

      def delete_if_defined
        delete unless missing?
      end

      def recreate(index_def)
        delete_if_defined
        create(index_def)
      end

      def index_document(type, id, attributes)
        res = @client.index(index: @index_name, type: type, id: id, body: attributes)

        if id = res["_id"]
          [id, res["created"]]
        else
          raise IndexError.new(@update_alias, "failed to index document")
        end
      end

      def delete_document(type, id)
        @client.delete(index: @index_name, type: type, id: id)
      end

      def get_document(type, id)
        @client.get(index: @index_name, type: type, id: id)
      end

      def search(type, body)
        Search::Facade.new(@client, Search::Definition.new(@index_name, type, body))
      end

      def delete_by_query(type, body)
        @client.delete_by_query(index: @index_name, type: type, body: body)
      end

      def bulk
        b = Bulk::Index.new(@client, @index_name)
        yield b
        b.execute
      end

      def settings
        args = { index: @index_name }
        settings = @client.index_get_settings(index: @index_name)
        settings[@index_name]["settings"]
      rescue Elasticsearch::Transport::Transport::Errors::NotFound
        nil
      end

      def mappings
        args = { index: @index_name }
        mapping = @client.index_get_mapping(index: @index_name)
        mapping[@index_name]["mappings"]
      rescue Elasticsearch::Transport::Transport::Errors::NotFound
        nil
      end

      def flush
        @client.index_flush(index: @index_name)
      end
    end
  end
end