dolly 1.1.5 → 3.1.0

data/lib/dolly/property_manager.rb

@@ -0,0 +1,53 @@
+module Dolly
+  module PropertyManager
+    def build_property(attributes)
+      assign_identity_properties(attributes)
+      assign_rev_properties(attributes)
+
+      lambda do |property|
+        name = property.key.to_sym
+        next unless doc[name].nil?
+        write_attribute(name, attributes[name])
+      end
+    end
+
+    def update_attribute
+      lambda do |key, value|
+        raise InvalidProperty unless valid_property?(key)
+        write_attribute(key, value)
+      end
+    end
+
+    def write_attribute(key, value)
+      value = set_property_value(key, value)
+      instance_variable_set(:"@#{key}", value)
+      update_doc(key, value)
+    end
+
+    def valid_property?(name)
+      properties.include?(name)
+    end
+
+    def update_doc(key, value)
+      doc[key] = value
+    end
+
+    def properties
+      self.class.properties
+    end
+
+    def set_property_value(key, value)
+      properties[key].cast_value(value)
+    end
+
+    def assign_identity_properties(opts = {})
+      id_presence = opts[:id] || opts[:_id] || opts['id'] || opts['_id']
+      self.id = id_presence if id_presence
+    end
+
+    def assign_rev_properties(opts = {})
+      rev_presence = opts[:rev] || opts [:_rev] || opts['rev'] || opts['_rev']
+      self.rev = rev_presence if rev_presence
+    end
+  end
+end

data/lib/dolly/property_set.rb

@@ -0,0 +1,23 @@
+module Dolly
+  class PropertySet < Set
+    def include? key
+      keys.include?(key)
+    end
+
+    def <<(property)
+      return if include?(property.key)
+      super(property)
+    end
+
+    def [](key)
+      return detect {|property| property.key == key } if key.is_a?(Symbol)
+      super
+    end
+
+    private
+
+    def keys
+      map(&:key)
+    end
+  end
+end

data/lib/dolly/query.rb

@@ -1,85 +1,73 @@
-require "dolly/connection"
-require "dolly/collection"
-require "dolly/name_space"
-require "exceptions/dolly"
+require 'dolly/collection'
+require 'dolly/query_arguments'
+require 'dolly/document_type'
+require 'refinements/string_refinements'
 
 module Dolly
   module Query
-    module ClassMethods
-      include Dolly::NameSpace
-      include Dolly::Connection
-      attr_accessor :properties
-
-      DESIGN_DOC = "dolly"
-
-      def find *keys
-        query_hash = { keys: keys.map{|key| namespace key} }
-
-        if keys.count > 1
-          build_collection( query_hash )
-        else
-          self.new.from_json( database.all_docs(query_hash).parsed_response )
-        end
-      rescue NoMethodError => err
-        if err.message == "undefined method `[]' for nil:NilClass"
-          raise Dolly::ResourceNotFound
-        else
-          raise
-        end
-      end
-
-      def default_query_args
-        {startkey: "#{name_paramitized}/", endkey: "#{name_paramitized}/\ufff0"}
-      end
-
-      def all
-        build_collection default_query_args
-      end
-
-      def first limit = 1
-        res = build_collection default_query_args.merge( limit: 1)
-        limit == 1 ? res.first : res
-      end
-
-      def last limit = 1
-        res = build_collection({startkey: default_query_args[:endkey], endkey: default_query_args[:startkey], limit: limit, descending: true})
-        limit == 1 ? res.first : res
-      end
-
-      def build_collection q
-        res = database.all_docs(q)
-        Collection.new res.parsed_response, name_for_class
-      end
-
-      def find_with doc, view_name, opts = {}
-        res = view "_design/#{doc}/_view/#{view_name}", opts
-        Collection.new res.parsed_response, name_for_class
-      end
-
-      #TODO: new implementation for collection returning
-      # multiple types is failling when the class has a namespace
-      # as the namespace does not exists on the doc id
-      # we need to reimplement this through a references class method.
-      def name_for_class
-        if name.include? '::'
-          name.constantize
-        end
-      end
-
-      def view doc, options = {}
-        options.merge! include_docs: true
-        database.get doc, options
-      end
-
-      def raw_view doc, view, opts = {}
-        JSON.parse database.get "_design/#{doc}/_view/#{view}", opts
-      end
+    include QueryArguments
+    include DocumentType
 
+    using StringRefinements
+
+    def find *keys
+      query_hash = { keys: namespace_keys(keys) }
+
+      build_collection(query_hash).first_or_all&.itself ||
+        raise(Dolly::ResourceNotFound)
+    end
+
+    def bulk_find(*keys_to_find)
+      data = {
+        query: { include_docs: true },
+        keys: keys_to_find.map { |key| namespace_key(key) }
+      }
+
+      res = connection.post('_all_docs', data)
+      Collection.new(rows: res, options: { doc_type: self.class_name })
+    end
+
+    def find_all(*keys)
+      query_hash = { keys: namespace_keys(keys) }
+      return [] if query_hash[:keys].none?
+
+      keys_to_find_counter = query_hash[:keys].length
+      build_collection(query_hash).first_or_all(true)&.itself
+    end
+
+    def safe_find *keys
+      find *keys
+    rescue Dolly::ResourceNotFound
+      nil
     end
 
-    def self.included(base)
-      base.extend ClassMethods
+    def all
+      build_collection(default_query_args)
     end
 
+    def first limit = 1
+      query_hash = default_query_args.merge(limit: limit)
+      build_collection(query_hash).first_or_all(limit > 1)
+    end
+
+    def last limit = 1
+      query_hash = descending_query_args.merge(limit: limit)
+      build_collection(query_hash).first_or_all(limit > 1)
+    end
+
+    def find_with doc, view_name, opts = {}
+      opts          = opts.each_with_object({}) { |(k, v), h| h[k] = v }
+      query_results = raw_view(doc, view_name, opts)
+
+      Collection.new({ rows: query_results, options: {} }).first_or_all
+    end
+
+    def build_collection(query)
+      Collection.new({ rows: connection.get('_all_docs', query.merge(include_docs: true)), options: { doc_type: self.class_name }})
+    end
+
+    def bulk_document
+      BulkDocument.new(connection)
+    end
   end
 end

data/lib/dolly/query_arguments.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Dolly
+  module QueryArguments
+    def last_item_in_range
+      "\ufff0"
+    end
+
+    def default_query_args
+      {
+        startkey: "#{name_paramitized}/",
+        endkey: "#{name_paramitized}/#{last_item_in_range}"
+      }
+    end
+
+    def descending_query_args
+      {
+        startkey: default_query_args[:endkey],
+        endkey: default_query_args[:startkey],
+        descending: true
+      }
+    end
+
+    def escape_value(value)
+      return value                if value.is_a? Numeric
+      return escape_values(value) if value.is_a? Array
+      return CGI.escape(value)    if value.is_a? String
+      value
+    end
+
+    def escape_values *values
+      values.flatten.map { |value| escape_value(value) }
+    end
+  end
+end

data/lib/dolly/request.rb

@@ -1,117 +1,24 @@
-require "httparty"
-require "dolly/bulk_document"
-
 module Dolly
-
-  class Request
-    include HTTParty
-    DEFAULT_HOST = 'localhost'
-    DEFAULT_PORT = '5984'
-
-    attr_accessor :database_name, :host, :port, :bulk_document
-
-    def initialize options = {}
-      @host = options["host"] || DEFAULT_HOST
-      @port = options["port"] || DEFAULT_PORT
-
-      @database_name = options["name"]
-      @username      = options["username"]
-      @password      = options["password"]
-      @protocol      = options["protocol"]
-
-      @bulk_document = Dolly::BulkDocument.new []
-      self.class.base_uri "#{protocol}://#{host}:#{port}"
-    end
-
-    def get resource, data = nil
-      q = {query: values_to_json(data)} if data
-      request :get, full_path(resource), q
-    end
-
-    def stats
-      request :get, "/#{database_name}"
-    end
-
-    def put resource, data
-      request :put, full_path(resource), {body: data}
-    end
-
-    def post resource, data
-      request :post, full_path(resource), {body: data}
-    end
-
-    def delete resource
-      request :delete, full_path(resource), {}
+  module Request
+    def set_namespace name
+      @namespace = name
     end
 
-    def attach resource, attachment_name, data, headers = {}
-      data = StringIO.new(data) if data.is_a?(String)
-      request :put, attachment_path(resource, attachment_name), {body: data, headers: headers}
+    def set_app_env env
+      @app_env = env
     end
 
-    def protocol
-      @protocol || 'http'
+    def connection
+      @connection ||= Connection.new(namespace, app_env)
     end
 
-    def uuids opts = {}
-      tools("_uuids", opts)["uuids"]
+    def namespace
+      @namespace ||= :default
     end
 
-    def all_docs data = {}
-      data =  values_to_json data.merge( include_docs: true )
-      request :get, full_path('_all_docs'), {query: data}
-    end
-
-    def request method, resource, data = nil
-      data ||= {}
-      data.merge!(basic_auth: auth_info) if auth_info.present?
-      headers = { 'Content-Type' => 'application/json' }
-      headers.merge! data[:headers] if data[:headers]
-      response = self.class.send method, resource, data.merge(headers: headers)
-      log_request(resource, response.code) if Dolly.log_requests?
-      if response.code == 404
-        raise Dolly::ResourceNotFound
-      elsif (400..600).include? response.code
-        raise Dolly::ServerError.new( response )
-      else
-        response
-      end
-    end
-
-    private
-    def tools path, opts = nil
-      data = {}
-      q = "?#{CGI.unescape(opts.to_query)}" unless opts.blank?
-      data.merge!(basic_auth: auth_info) if auth_info.present?
-      JSON::parse self.class.get("/#{path}#{q}", data)
-    end
-
-    def auth_info
-      return nil unless @username.present?
-      {username: @username, password: @password}
-    end
-
-    def values_to_json hash
-      hash.reduce({}){|h, v| h[v.first] = v.last.to_json; h}
-    end
-
-    def full_path resource
-      "/#{database_name}/#{resource}"
-    end
-
-    def attachment_path resource, attachment_name
-      "#{full_path(resource)}/#{attachment_name}"
-    end
-
-    def log_request resource, response_code
-      log_value = ->(resource, response_code) { "Query: #{resource}, Response Code: #{response_code}" }
-      case response_code
-      when 200..399
-        Dolly.logger.info log_value[resource, response_code]
-      when 400..600
-        Dolly.logger.warn log_value[resource, response_code]
-      end
+    def app_env
+      return Rails.env if defined? Rails.env
+      @app_env ||= :development
     end
   end
-
 end

data/lib/dolly/request_header.rb

@@ -0,0 +1,26 @@
+require 'forwardable'
+
+module Dolly
+  class HeaderRequest
+    extend Forwardable
+
+    CONTENT_TYPE_KEY = 'Content-Type'
+    JSON_CONTENT = 'application/json'
+
+    def_delegators :@collection, :[], :[]=, :keys, :each, :present?, :merge!
+
+    def initialize hash = nil
+      @collection = hash || default_value
+    end
+
+    def json?
+      @collection[CONTENT_TYPE_KEY] == JSON_CONTENT
+    end
+
+    private
+
+    def default_value
+      { CONTENT_TYPE_KEY => JSON_CONTENT }
+    end
+  end
+end

data/lib/dolly/timestamp.rb

@@ -0,0 +1,24 @@
+module Dolly
+  module Timestamp
+    def write_timestamps(is_persisted)
+      return unless timestamped?
+      write_attribute(:created_at, Time.now) unless is_persisted
+      write_attribute(:updated_at, Time.now)
+    end
+
+    def timestamped?
+      respond_to?(:created_at) && respond_to?(:updated_at)
+    end
+
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      def timestamps!
+        property :created_at, :updated_at, class_name: Time
+      end
+    end
+  end
+end
+