elastic-enterprise-search 0.3.0 → 7.13.0.pre

data/lib/elastic/enterprise-search/api/health.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Elastic
+  module EnterpriseSearch
+    module Actions
+      # Health - Get information on the health of a deployment and basic statistics around resource usage
+      # Get information on the health of a deployment and basic statistics around resource usage
+      #
+      # @option arguments [Hash] :headers optional HTTP headers to send with the request
+      #
+      # @see https://www.elastic.co/guide/en/enterprise-search/current/monitoring-apis.html#health-api-example
+      #
+      def health(arguments = {})
+        headers = arguments.delete(:headers) || {}
+
+        request(
+          :get,
+          'api/ent/v1/internal/health/',
+          arguments,
+          {},
+          headers
+        )
+      end
+    end
+  end
+end

data/lib/elastic/enterprise-search/api/put_read_only.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Elastic
+  module EnterpriseSearch
+    module Actions
+      # Read-Only - Update the read-only flag's state
+      # Update the read-only flag's state
+      #
+      # @option arguments [Object] :body
+      # @option body [Boolean] :enabled
+      # @option arguments [Hash] :headers optional HTTP headers to send with the request
+      #
+      # @see https://www.elastic.co/guide/en/enterprise-search/current/read-only-api.html#setting-read-only-state
+      #
+      def put_read_only(arguments = {})
+        body = arguments.delete(:body) || {}
+        headers = arguments.delete(:headers) || {}
+
+        request(
+          :put,
+          'api/ent/v1/internal/read_only_mode/',
+          arguments,
+          body,
+          headers
+        )
+      end
+    end
+  end
+end

data/lib/elastic/enterprise-search/api/read_only.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Elastic
+  module EnterpriseSearch
+    module Actions
+      # Read-Only - Get the read-only flag's state
+      # Get the read-only flag's state
+      #
+      # @option arguments [Hash] :headers optional HTTP headers to send with the request
+      #
+      # @see https://www.elastic.co/guide/en/enterprise-search/current/read-only-api.html#getting-read-only-state
+      #
+      def read_only(arguments = {})
+        headers = arguments.delete(:headers) || {}
+
+        request(
+          :get,
+          'api/ent/v1/internal/read_only_mode/',
+          arguments,
+          {},
+          headers
+        )
+      end
+    end
+  end
+end

data/lib/elastic/enterprise-search/api/stats.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Elastic
+  module EnterpriseSearch
+    module Actions
+      # Stats - Get information about the resource usage of the application, the state of different internal queues, etc.
+      # Get information about the resource usage of the application, the state of different internal queues, etc.
+      #
+      # @param arguments [Hash] endpoint arguments
+      # @option arguments [Array] :include Comma-separated list of stats to return
+      # @option arguments [Hash] :headers optional HTTP headers to send with the request
+      #
+      # @see https://www.elastic.co/guide/en/enterprise-search/current/monitoring-apis.html#stats-api-example
+      #
+      def stats(arguments = {})
+        headers = arguments.delete(:headers) || {}
+
+        request(
+          :get,
+          'api/ent/v1/internal/stats/',
+          arguments,
+          {},
+          headers
+        )
+      end
+    end
+  end
+end

data/lib/elastic/enterprise-search/api/version.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Elastic
+  module EnterpriseSearch
+    module Actions
+      # Version - Get version information for this server
+      # Get version information for this server
+      #
+      # @option arguments [Hash] :headers optional HTTP headers to send with the request
+      #
+      # @see https://www.elastic.co/guide/en/enterprise-search/current/management-apis.html
+      #
+      def version(arguments = {})
+        headers = arguments.delete(:headers) || {}
+
+        request(
+          :get,
+          'api/ent/v1/internal/version/',
+          arguments,
+          {},
+          headers
+        )
+      end
+    end
+  end
+end

data/lib/elastic/enterprise-search/client.rb

@@ -1,37 +1,82 @@
-require 'set'
-require 'elastic/enterprise-search/configuration'
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
 require 'elastic/enterprise-search/request'
 require 'elastic/enterprise-search/utils'
+require 'elasticsearch-transport'
 
 module Elastic
   module EnterpriseSearch
-    # API client for the {Elastic Enterprise Search API}[https://swiftype.com/enterprise-search].
+    # API client for the {Elastic Enterprise Search API}[https://www.elastic.co/enterprise-search].
+    # This is the main client from which the Workplace Search and App Search clients inherit.
     class Client
-      autoload :ContentSourceDocuments, 'elastic/enterprise-search/client/content_source_documents.rb'
-      autoload :Permissions, 'elastic/enterprise-search/client/permissions.rb'
-
       DEFAULT_TIMEOUT = 15
 
       include Elastic::EnterpriseSearch::Request
+      include Elastic::EnterpriseSearch::Actions
+      include Elastic::EnterpriseSearch::Utils
+
+      def workplace_search(options = {})
+        @workplace_search ||= Elastic::EnterpriseSearch::WorkplaceSearch::Client.new(
+          host: host,
+          http_auth: options[:http_auth],
+          transport: @transport
+        )
+      end
 
-      def self.configure(&block)
-        Elastic::EnterpriseSearch.configure &block
+      def app_search(options = {})
+        @app_search ||= Elastic::EnterpriseSearch::AppSearch::Client.new(
+          host: host,
+          http_auth: options[:http_auth],
+          transport: @transport
+        )
       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 [String] :host Enterprise Search host
+      # @option options [Hash] :basic_auth a username and password for Basic Authentication
       # @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 [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"
+      # @option options [Boolean] :log Use the default logger (disabled by default)
+      # @option arguments [Object] :logger An instance of a Logger-compatible object
+      # @option enable_meta_header [Boolean] :enable_meta_header Enable sending the meta data header to Cloud.
+      #                                                          (Default: true)
       def initialize(options = {})
         @options = options
+        @transport = transport
       end
 
-      def access_token
-        @options[:access_token] || Elastic::EnterpriseSearch.access_token
+      def transport
+        @options[:transport] ||
+          Elasticsearch::Client.new(
+            host: host,
+            log: log,
+            logger: logger,
+            request_timeout: overall_timeout,
+            transport_options: {
+              request: { open_timeout: open_timeout }
+            },
+            enable_meta_header: @options[:enable_meta_header] || true
+          )
       end
 
       def open_timeout
@@ -46,8 +91,29 @@ module Elastic
         (@options[:overall_timeout] || DEFAULT_TIMEOUT).to_f
       end
 
-      include Elastic::EnterpriseSearch::Client::ContentSourceDocuments
-      include Elastic::EnterpriseSearch::Client::Permissions
+      def http_auth
+        @options[:http_auth] || { user: 'elastic', password: 'changeme' }
+      end
+
+      def http_auth=(http_auth)
+        @options[:http_auth] = http_auth
+      end
+
+      def log
+        @options[:log] || false
+      end
+
+      def logger
+        @options[:logger]
+      end
+
+      def host
+        return DEFAULT_HOST unless @options[:host]
+
+        raise URI::InvalidURIError unless @options[:host] =~ /\A#{URI::DEFAULT_PARSER.make_regexp}\z/
+
+        @options[:host]
+      end
     end
   end
 end

data/lib/elastic/enterprise-search/configuration.rb

@@ -1,18 +1,35 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
 require 'uri'
-require 'elastic/enterprise-search/version'
+require 'elastic/workplace-search/version'
 
 module Elastic
   module EnterpriseSearch
+    # Configuratin module
     module Configuration
-      DEFAULT_ENDPOINT = "http://localhost:3002/api/v1/"
+      # TODO: Endpoint for EE
+      # DEFAULT_ENDPOINT = 'http://localhost:3002/api/ws/v1/'
 
-      VALID_OPTIONS_KEYS = [
-        :access_token,
-        :user_agent,
-        :endpoint
-      ].freeze
+      VALID_OPTIONS_KEYS = [:access_token, :endpoint].freeze
 
-      attr_accessor *VALID_OPTIONS_KEYS
+      attr_accessor(*VALID_OPTIONS_KEYS)
 
       def self.extended(base)
         base.reset
@@ -22,11 +39,10 @@ module Elastic
       def reset
         self.access_token = nil
         self.endpoint = DEFAULT_ENDPOINT
-        self.user_agent = nil
         self
       end
 
-      # Yields the Elastic::EnterpriseSearch::Configuration module which can be used to set configuration options.
+      # Yields the Elastic::WorkplaceSearch::Configuration module which can be used to set configuration options.
       #
       # @return self
       def configure
@@ -37,17 +53,17 @@ module Elastic
       # Return a hash of the configured options.
       def options
         options = {}
-        VALID_OPTIONS_KEYS.each{ |k| options[k] = send(k) }
+        VALID_OPTIONS_KEYS.each { |k| options[k] = send(k) }
         options
       end
 
       # setter for endpoint that ensures it always ends in '/'
       def endpoint=(endpoint)
-        if endpoint.end_with?('/')
-          @endpoint = endpoint
-        else
-          @endpoint = "#{endpoint}/"
-        end
+        @endpoint = if endpoint.end_with?('/')
+                      endpoint
+                    else
+                      "#{endpoint}/"
+                    end
       end
     end
   end

data/lib/elastic/enterprise-search/exceptions.rb

@@ -1,3 +1,22 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
 module Elastic
   module EnterpriseSearch
     class ClientException < StandardError; end