require 'set'
require 'elastic/enterprise-search/configuration'
require 'elastic/enterprise-search/request'
require 'elastic/enterprise-search/utils'

module Elastic
  module EnterpriseSearch
    # API client for the {Elastic Enterprise Search API}[https://swiftype.com/enterprise-search].
    class Client
      DEFAULT_TIMEOUT = 15

      include Elastic::EnterpriseSearch::Request

      def self.configure(&block)
        Elastic::EnterpriseSearch.configure &block
      end

      # Create a new Elastic::EnterpriseSearch::Client client
      #
      # @param options [Hash] a hash of configuration options that will override what is set on the Elastic::EnterpriseSearch class.
      # @option options [String] :access_token an Access Token to use for this client
      # @option options [Numeric] :overall_timeout overall timeout for requests in seconds (default: 15s)
      # @option options [Numeric] :open_timeout the number of seconds Net::HTTP (default: 15s)
      #   will wait while opening a connection before raising a Timeout::Error
      # @option options [String] :proxy url of proxy to use, ex: "http://localhost:8888"
      def initialize(options = {})
        @options = options
      end

      def access_token
        @options[:access_token] || Elastic::EnterpriseSearch.access_token
      end

      def open_timeout
        @options[:open_timeout] || DEFAULT_TIMEOUT
      end

      def proxy
        @options[:proxy]
      end

      def overall_timeout
        (@options[:overall_timeout] || DEFAULT_TIMEOUT).to_f
      end

      # Documents have fields that can be searched or filtered.
      #
      # For more information on indexing documents, see the {Content Source documentation}[https://swiftype.com/documentation/enterprise-search/guides/content-sources].
      module ContentSourceDocuments

        # Index a batch of documents using the {Content Source API}[https://swiftype.com/documentation/enterprise-search/api/custom-sources].
        #
        # @param [String] content_source_key the unique Content Source key as found in your Content Sources dashboard
        # @param [Array] documents an Array of Document Hashes
        #
        # @return [Array<Hash>] an Array of Document indexing Results
        #
        # @raise [Elastic::EnterpriseSearch::InvalidDocument] when a single document is missing required fields or contains unsupported fields
        # @raise [Timeout::Error] when timeout expires waiting for results
        def index_documents(content_source_key, documents)
          documents = Array(documents).map! { |document| normalize_document(document) }

          async_create_or_update_documents(content_source_key, documents)
        end

        # Destroy a batch of documents given a list of external IDs
        #
        # @param [Array<String>] document_ids an Array of Document External IDs
        #
        # @return [Array<Hash>] an Array of Document destroy result hashes
        #
        # @raise [Timeout::Error] when timeout expires waiting for results
        def destroy_documents(content_source_key, document_ids)
          document_ids = Array(document_ids)
          post("ent/sources/#{content_source_key}/documents/bulk_destroy.json", document_ids)
        end

        private

        def async_create_or_update_documents(content_source_key, documents)
          post("ent/sources/#{content_source_key}/documents/bulk_create.json", documents)
        end

        def normalize_document(document)
          Utils.stringify_keys(document)
        end
      end

      include Elastic::EnterpriseSearch::Client::ContentSourceDocuments
    end
  end
end