search-kit 0.0.2 → 0.0.3

data/lib/search_kit/clients/scaffold.rb

@@ -0,0 +1,36 @@
+require 'faraday'
+require 'json'
+
+module SearchKit
+  module Clients
+    class Scaffold
+      attr_reader :connection, :token
+
+      def initialize
+        uri = [SearchKit.config.app_uri, "scaffold"].join("/")
+        @connection = Faraday.new(uri)
+        @token      = SearchKit.config.app_token
+      end
+
+      def create(name, documents)
+        options = {
+          token: token,
+          data: {
+            type:          'indices',
+            attributes:    { name: name },
+            relationships: { documents: documents }
+          }
+        }
+
+        response = connection.post('', options)
+        body     = JSON.parse(response.body, symbolize_names: true)
+
+        fail Errors::BadRequest    if response.status == 400
+        fail Errors::Unauthorized  if response.status == 401
+        fail Errors::Unprocessable if response.status == 422
+
+        body
+      end
+    end
+  end
+end

data/lib/search_kit/clients/search.rb

@@ -0,0 +1,33 @@
+require 'faraday'
+require 'json'
+
+module SearchKit
+  module Clients
+    class Search
+      attr_reader :connection, :token
+
+      def initialize
+        uri = [SearchKit.config.app_uri, "search"].join("/")
+        @connection = Faraday.new(uri)
+        @token      = SearchKit.config.app_token
+      end
+
+      def search(slug, options)
+        params = {
+          token: token, data: { type: "searches", attributes: options }
+        }
+
+        response = connection.post(slug, params)
+        body     = JSON.parse(response.body, symbolize_names: true)
+
+        fail Errors::BadRequest    if response.status == 400
+        fail Errors::Unauthorized  if response.status == 401
+        fail Errors::IndexNotFound if response.status == 404
+        fail Errors::Unprocessable if response.status == 422
+
+        SearchKit::Models::Search.new(body.fetch(:data, {}))
+      end
+
+    end
+  end
+end

data/lib/search_kit/clients/subscribers.rb

@@ -0,0 +1,54 @@
+require 'faraday'
+require 'json'
+
+module SearchKit
+  module Clients
+    class Subscribers
+      attr_reader :connection, :token
+
+      def initialize
+        uri = [ SearchKit.config.app_uri, "subscribers" ].join("/")
+        @connection = Faraday.new(uri)
+        @token      = SearchKit.config.app_token
+      end
+
+      def create(options = {})
+        options  = { data: { type: 'subscribers', attributes: options } }
+        response = connection.post("", options)
+        body     = JSON.parse(response.body, symbolize_names: true)
+
+        fail Errors::BadRequest    if response.status == 400
+        fail Errors::Unprocessable if response.status == 422
+
+        SearchKit::Models::Subscriber.new body.fetch(:data, {})
+      end
+
+      def info
+        response = connection.get("", token: token)
+        body     = JSON.parse(response.body, symbolize_names: true)
+
+        fail Errors::Unauthorized       if response.status == 401
+        fail Errors::SubscriberNotFound if response.status == 404
+
+        SearchKit::Models::Subscriber.new body.fetch(:data, {})
+      end
+
+      def update(options = {})
+        options  = {
+          token: token, data: { type: 'subscribers', attributes: options }
+        }
+
+        response = connection.patch("", options)
+        body     = JSON.parse(response.body, symbolize_names: true)
+
+        fail Errors::BadRequest         if response.status == 400
+        fail Errors::Unauthorized       if response.status == 401
+        fail Errors::SubscriberNotFound if response.status == 404
+        fail Errors::Unprocessable      if response.status == 422
+
+        SearchKit::Models::Subscriber.new body.fetch(:data, {})
+      end
+
+    end
+  end
+end

data/lib/search_kit/configuration.rb

@@ -1,6 +1,5 @@
 require 'ostruct'
 require 'yaml'
-require 'uri'
 require 'user_config'
 
 module SearchKit

data/lib/search_kit/errors.rb

@@ -1,9 +1,12 @@
 module SearchKit
   module Errors
     class BadRequest < StandardError; end
+    class EventNotFound < StandardError; end
     class IndexNotFound < StandardError; end
+    class KeyNotFound < StandardError; end
     class PublicationFailed < StandardError; end
-    class EventNotFound < StandardError; end
+    class SubscriberNotFound < StandardError; end
+    class Unauthorized < StandardError; end
     class Unprocessable < StandardError; end
   end
 end

data/lib/search_kit/messages.rb

@@ -0,0 +1,41 @@
+require 'ansi'
+
+module SearchKit
+  class Messages
+    autoload :Messaging, 'search_kit/messages/messaging'
+
+    include Messaging
+
+    def unauthorized
+      warning(I18n.t('http.401'))
+    end
+
+    def bad_request
+      warning(I18n.t('http.400'))
+    end
+
+    def not_found(type = 'Resource')
+      message = I18n.t('http.404', type: type)
+      warning(message)
+    end
+
+    def json_parse_error(type = 'Argument')
+      message = I18n.t('cli.errors.json_parse', type: type)
+      warning(message)
+    end
+
+    def no_service
+      message = I18n.t('cli.errors.no_service', uri: SearchKit.config.app_uri)
+      warning(message)
+    end
+
+    def unprocessable
+      warning(I18n.t('http.422'))
+    end
+
+    def unreadable(error)
+      warning(I18n.t('cli.errors.unreadable', error: error))
+    end
+
+  end
+end

data/lib/search_kit/messages/messaging.rb

@@ -0,0 +1,64 @@
+require 'ansi'
+
+module SearchKit
+  class Messages
+    # The goal of the Messaging module is to provide an easy to include internal
+    # interface which will allow a SearchKit gem to dutifully log and provide
+    # output of what it's up to and how it may be doing.
+    #
+    module Messaging
+      def info(message)
+        Message.new(message).info
+      end
+
+      def warning(message)
+        Message.new(message).warn
+      end
+
+      private
+
+      # Most of the logic for the Messaging module exists in this (not so)
+      # private class.  This lets more complex handling of message logic enter
+      # into the module gracefully, for example silence or logging level.
+      #
+      class Message
+        attr_reader :message
+
+        def initialize(message)
+          @message = message
+        end
+
+        def warn
+          Kernel.warn(Prefixed(message.ansi(:red))) if SearchKit.config.verbose
+          SearchKit.logger.warn message
+        end
+
+        def info
+          Kernel.puts(Prefixed(message.ansi(:cyan))) if SearchKit.config.verbose
+          SearchKit.logger.info message
+        end
+
+        private
+
+        def Prefixed(*messages)
+          Prefixed.new.join(*messages)
+        end
+
+        class Prefixed
+          attr_reader :body
+
+          def initialize
+            env   = SearchKit.config.app_env.to_s.ansi(:magenta)
+            @body = "--> [ #{env} ]: "
+          end
+
+          def join(*messages)
+            [body, *messages].join(" ")
+          end
+
+        end
+      end
+
+    end
+  end
+end

data/lib/search_kit/models.rb

@@ -0,0 +1,14 @@
+require 'virtus'
+
+module SearchKit
+  module Models
+    autoload :Document,   'search_kit/models/document'
+    autoload :Documents,  'search_kit/models/documents'
+    autoload :Event,      'search_kit/models/event'
+    autoload :Events,     'search_kit/models/events'
+    autoload :Key,        'search_kit/models/key'
+    autoload :Keys,       'search_kit/models/keys'
+    autoload :Search,     'search_kit/models/search'
+    autoload :Subscriber, 'search_kit/models/subscriber'
+  end
+end

data/lib/search_kit/models/document.rb

@@ -0,0 +1,30 @@
+module SearchKit
+  module Models
+    class Document
+      class AttributeNotFound < StandardError; end
+
+      include Virtus.model
+
+      attribute :id
+      attribute :source
+      attribute :score
+
+      def initialize(document_data = {})
+        attributes = document_data.fetch(:attributes, {})
+
+        super(
+          source: attributes,
+          id:     attributes.fetch(:id, nil),
+          score:  attributes.fetch(:score, nil)
+        )
+      end
+
+      def get(field)
+        source.fetch(field.to_sym)
+      rescue KeyError
+        fail AttributeNotFound, field
+      end
+
+    end
+  end
+end

data/lib/search_kit/models/documents.rb

@@ -0,0 +1,31 @@
+module SearchKit
+  module Models
+    class Documents
+      include Enumerable
+
+      def self.[](*arguments)
+        new(arguments)
+      end
+
+      attr_reader :contents, :member_class
+
+      def initialize(contents = [])
+        @contents     = contents
+        @member_class = SearchKit::Models::Document
+      end
+
+      def <<(new_doc)
+        case new_doc
+        when Hash         then contents << member_class.new(new_doc)
+        when member_class then contents << new_doc
+        else contents
+        end
+      end
+
+      def each(&block)
+        contents.each(&block)
+      end
+
+    end
+  end
+end

data/lib/search_kit/models/event.rb

@@ -0,0 +1,18 @@
+module SearchKit
+  module Models
+    class Event
+      include Virtus.model
+
+      attribute :id
+      attribute :channel
+      attribute :state
+      attribute :payload, Hash
+
+      def initialize(event_data = {})
+        attributes = event_data.fetch(:attributes, {})
+        super(attributes)
+      end
+
+    end
+  end
+end

data/lib/search_kit/models/events.rb

@@ -0,0 +1,31 @@
+module SearchKit
+  module Models
+    class Events
+      include Enumerable
+
+      def self.[](*arguments)
+        new(arguments)
+      end
+
+      attr_reader :contents, :member_class
+
+      def initialize(contents = [])
+        @contents     = contents
+        @member_class = SearchKit::Models::Event
+      end
+
+      def <<(new_event)
+        case new_event
+        when Hash         then contents << member_class.new(new_event)
+        when member_class then contents << new_event
+        else contents
+        end
+      end
+
+      def each(&block)
+        contents.each(&block)
+      end
+
+    end
+  end
+end

data/lib/search_kit/models/key.rb

@@ -0,0 +1,26 @@
+require 'virtus'
+
+module SearchKit
+  module Models
+    class Key
+      include Virtus.model
+
+      attribute :id,        String
+      attribute :name,      String
+      attribute :privilege, String
+      attribute :token,     String
+      attribute :uri,       String
+
+      def initialize(key_data = {})
+        attributes = key_data.fetch(:attributes, {})
+        uri        = key_data.fetch(:links, {}).fetch(:self, '')
+
+        super(attributes.merge(uri: uri))
+      end
+
+      def creator?
+        privilege == 'creator'
+      end
+    end
+  end
+end

data/lib/search_kit/models/keys.rb

@@ -0,0 +1,38 @@
+module SearchKit
+  module Models
+    class Keys
+      include Enumerable
+
+      def self.[](*arguments)
+        new(arguments)
+      end
+
+      attr_reader :contents, :member_class
+
+      def initialize(contents = [])
+        @contents     = contents
+        @member_class = SearchKit::Models::Key
+      end
+
+      def <<(new_key)
+        case new_key
+        when Hash         then contents << member_class.new(new_key)
+        when member_class then contents << new_key
+        else contents
+        end
+      end
+
+      def each(&block)
+        contents.each(&block)
+      end
+
+      def creator
+        self.class.new(select(&:creator?))
+      end
+
+      def tokens
+        contents.map(&:token)
+      end
+    end
+  end
+end