mongous 0.1.6 → 0.4.0

data/lib/mongous/base.rb

@@ -50,12 +50,45 @@ module Mongous
       self.class_variable_set( :@@client, _client )
     end
 
+    def attach!( *names )
+      raise  Mongous::Error, "missing argument."   if names.empty?
+
+      names.each do |name|
+        case  name
+        when  String
+        when  Symbol
+          name  =  name.to_s
+        else
+          raise  Mongous::Error, "type invalid. :  #{ name }"
+        end
+
+        raise  Mongous::Error, "missing argument."   unless  /^[A-Z]/.match(name)
+
+        if  Object.const_defined?( name )
+          if  Object.const_get( name ).include?( Mongous::Document )
+            Object.class_eval{ remove_const( name ) }
+          else
+            raise  Mongous::Error, "type invalid. :  #{ Object.class_eval(name) }"
+          end
+        end
+
+        Object.class_eval <<-CLASS
+          class #{name}
+            include Mongous::Document
+          end
+        CLASS
+      end
+    end
+ 
     def client
-      self.class_variable_get( :@@client )    rescue  nil
+      if not self.class_variable_defined?( :@@client )
+        connect!
+      end
+      self.class_variable_get( :@@client )
     end
 
     def client=( _client )
-      if  !_client.is_a?( ::Mongo::Client )
+      if not _client.is_a?( ::Mongo::Client )
         raise  Mongous::Error, "type invalid. :  #{ _client }"
       end
       self.class_variable_set( :@@client, _client )

data/lib/mongous/document.rb

@@ -26,33 +26,66 @@ module Mongous
       end
     end
 
-    def having?( value )
-      case  value
+    def having?( label )
+      case  label
       when  NilClass
         false
       when  String
-        !value.strip.empty?
+        !label.strip.empty?
       else
         true
       end
     end
 
+    def getvalue_or_callproc( value_or_proc )
+      case  value_or_proc
+      when  Proc
+        value_or_proc.call
+      else
+        value_or_proc
+      end
+    end
+
     def save
+      if  @doc["_id"].nil? || self.class.collection.find( { "_id"=> @doc["_id"] } ).count == 0
+        savemode  =  :create
+      else
+        savemode  =  :update
+      end
+
       self.class.fields.each do |label, field|
-        _must  =  field[:_args].include?(:must)
+        default_value  =  getvalue_or_callproc( field[:default] )
+        _must  =  field[:_attrs].include?(:must)
         if  @doc.has_key?(label)
-          if  _must  &&  !having?( @doc[label] )
-            raise  Mongous::Error, "must and not having field. : #{ label }"
+          if  !having?( @doc[label] )
+            if  default_value
+              self[label]  =  default_value
+            elsif  _must
+              raise  Mongous::Error, "must but unassigned field. : #{ label }"
+            elsif  self.class.symbols[:strict]
+              self[label]  =  nil
+            end
           end
         else
-          if  block  =  field[:_block]
-            self[label]  =  block.call
+          if  default_value
+            self[label]  =  default_value
           elsif  _must
-            raise  Mongous::Error, "must and unassigned field. : #{ label }"
+            raise  Mongous::Error, "must but unassigned field. : #{ label }"
           elsif  self.class.symbols[:strict]
             self[label]  =  nil
           end
         end
+
+        case  savemode
+        when  :create
+          if  create_value  =  getvalue_or_callproc( field[:create] )
+            self[label]  =  create_value
+          end
+        when  :update
+          if  update_value  =  getvalue_or_callproc( field[:update] )
+            self[label]  =  update_value
+          end
+        end
       end
 
       self.class.blocks.each do |call_from, block|
@@ -61,23 +94,29 @@ module Mongous
         end
       end
 
-      if  @doc["_id"]
-        _filter  =  { "_id"=> @doc["_id"] }
-        if  self.class.collection.find( _filter ).first
-          self.class.collection.update_one( _filter, { '$set' => @doc } )
-          return  self
-        end
+      case  savemode
+      when  :create
+        self.class.collection.insert_one( @doc )
+      when  :update
+        self.class.collection.update_one( { "_id"=> @doc["_id"] }, { '$set' => @doc } )
       end
-
-      self.class.collection.insert_one( @doc )
       self
     end
 
+    def to_hash
+      @doc.dup
+    end
+
+    def to_json
+      @doc.to_json
+    end
+
     def []( label )
       label  =  label.to_s
 
       if  self.class.symbols[:strict]
-        if  !self.class.fields.keys.include?(label)
+        labels  =  ["_id"] + self.class.fields.keys
+        if  !labels.include?( label )
           raise  Mongous::Error, "undefined field. : #{ label }"
         end
       end
@@ -108,18 +147,18 @@ module Mongous
       return  @doc[label]  =  value    if field.nil?
 
       types  =  []
-      if  args  =  field[:_args]  ||  []
-        args.each do |arg|
-          if  arg.class == Class
-            types  <<  arg
-            args  -=  [arg]
+      if  attrs  =  field[:_attrs]  ||  []
+        attrs.each do |attr|
+          if  attr.class == Class
+            types  <<  attr
+            attrs  -=  [attr]
             break
           end
         end
       end
 
       if  !types.empty?
-        if  !(args & [:must, :not_null]).empty?
+        if  !(attrs & [:must, :not_null]).empty?
           if  !types.include?( value.class )
             raise  Mongous::Error, "invalid type. : #{ label } : #{ value.class }"
           end
@@ -129,7 +168,7 @@ module Mongous
           end
         end
       else
-        if  !(args & [:must, :not_null]).empty?
+        if  !(attrs & [:must, :not_null]).empty?
           if  [NilClass].include?( value.class )
             raise  Mongous::Error, "invalid type. : #{ label } : #{ value.class }"
           end
@@ -138,19 +177,23 @@ module Mongous
 
       @doc[label]  =  value
 
-      args.each do |arg|
-        case  arg
+      attrs.each do |attr|
+        case  attr
         when  Proc
-          if  !self.instance_eval( &arg )
+          if  !self.instance_eval( &attr )
             raise  Mongous::Error, "violation detected. : #{ label } : #{ value }"
           end
         when  Array
-          if  !arg.include?( value )
-            raise  Mongous::Error, "not include. : #{ label } :#{ value }"
+          if  !attr.include?( value )
+            raise  Mongous::Error, "not include. : #{ label } : #{ value }"
           end
         when  Range
-          if  !arg.cover?( value )
-            raise  Mongous::Error, "out of range. : #{ label } :#{ value }"
+          if  !attr.cover?( value )
+            raise  Mongous::Error, "out of range. : #{ label } : #{ value }"
+          end
+        when  Regexp
+          if  !attr.match( value )
+            raise  Mongous::Error, "unmatch regexp. : #{ label } : #{ value }"
           end
         end
       end

data/lib/mongous/extention.rb

@@ -10,7 +10,7 @@ module Mongous
       end
     end
 
-    def set_client( _client )
+    def client=( _client )
       m  =  /(.*?):(\d+)/.match( caller()[0] )
       call_from  =  [ m[1], m[2] ].join(":")
       if  !_client.is_a?( Mongo::Client )
@@ -21,28 +21,31 @@ module Mongous
 
     def collection_name
       if self.class_variable_defined?( :@@collection_name )
-        self.class_variable_get( :@@collection_name )
-      else
-        _client_name  =  self.name
-        self.class_variable_set( :@@collection_name, _client_name )
+        value  =  self.class_variable_get( :@@collection_name )
+        return  value    if value
       end
+
+      self.class_variable_set( :@@collection_name, self.name )
     end
 
-    def set_collection_name( _collection_name )
+    def collection_name=( _collection_name )
       self.class_variable_set( :@@collection_name, _collection_name )
       if self.class_variable_defined?( :@@collection )
         self.remove_class_variable( :@@collection )
       end
     end
 
-    def collection
-      if self.class_variable_defined?( :@@collection )
-        if  _collection  =  self.class_variable_get( :@@collection )
-          return  _collection
+    def collection( temp_collection_name = nil )
+      if  temp_collection_name.nil?
+        if self.class_variable_defined?( :@@collection )
+          if  _collection  =  self.class_variable_get( :@@collection )
+            return  _collection
+          end
         end
+        _collection_name  =  collection_name
+      else
+        _collection_name  =  temp_collection_name
       end
-
-      _collection_name  =  collection_name
       _client  =  client
 
       if  _client.database.collection_names.include?( _collection_name )
@@ -56,38 +59,41 @@ module Mongous
         _collection.indexes.create_one( keys, opts )    rescue  nil
       end
 
-      self.class_variable_set( :@@collection, _collection )
+      self.class_variable_set( :@@collection, _collection )    if temp_collection_name.nil?
+      _collection
     end
 
     def fields
-      if self.class_variable_defined?( :@@fields )
-        self.class_variable_get( :@@fields )
-      else
-        self.class_variable_set( :@@fields, {} )
-      end
+      setup_class_variable( :@@fields, {} )
     end
 
     def symbols
-      if self.class_variable_defined?( :@@symbols )
-        self.class_variable_get( :@@symbols )
-      else
-        self.class_variable_set( :@@symbols, {} )
-      end
+      setup_class_variable( :@@symbols, {} )
     end
 
     def blocks
-      if self.class_variable_defined?( :@@blocks )
-        self.class_variable_get( :@@blocks )
-      else
-        self.class_variable_set( :@@blocks, {} )
-      end
+      setup_class_variable( :@@blocks, {} )
     end
 
     def indexes
-      if self.class_variable_defined?( :@@indexes )
-        self.class_variable_get( :@@indexes )
+      setup_class_variable( :@@indexes, [] )
+    end
+
+    def filters
+      setup_class_variable( :@@filters, {} )
+    end
+
+    def defaults
+      setup_class_variable( :@@defaults, {} )
+    end
+
+    def setup_class_variable( symbol, default = {}, &block )
+      if self.class_variable_defined?( symbol )
+        self.class_variable_get( symbol )
+      elsif block_given?
+        self.class_variable_set( symbol, block.call )
       else
-        self.class_variable_set( :@@indexes, [] )
+        self.class_variable_set( symbol, default )
       end
     end
 
@@ -99,53 +105,62 @@ module Mongous
       self.collection.find( conditios, options )
     end
 
-    def field( label, *args, **opts, &block )
+    def field( symbol, *attrs, **items )
       m  =  /(.*?):(\d+)/.match( caller()[0] )
       call_from  =  [ m[1], m[2] ].join(":")
 
-      args.each do |arg|
-        if klass  =  arg.class 
-          if ![Class, Range, Array, Proc, Symbol].include?(klass)
-            raise  Mongous::Error, "field error. :  #{arg} on #{ label } at #{ call_from }"
+      attrs.each do |attr|
+        if klass  =  attr.class 
+          if ![Class, Range, Array, Regexp, Proc, Symbol].include?(klass)
+            raise  Mongous::Error, "'field' arguments error. : #{ attr } on #{ symbol } at #{ call_from }"
           end
         end
       end
 
-      opts.each do |key, value|
-        case  key
-        when  :default
-          case  value
-          when  Proc, String, Numeric
-          else
-            raise  Mongous::Error, "field error. : #{key} on #{ label } at #{ call_from }"
-          end
-        end
+      items.each do |key, value|
+        next    if [:default, :create, :update].include?(key) && [Proc, String, Numeric].include?(value.class)
+
+        raise  Mongous::Error, "'field' options error. : #{key} on #{ symbol } at #{ call_from }"
       end
 
-      opts[:_args]  =  args
-      opts[:_block]  =  block
-      fields[label.to_s]  =  opts
+      items[:_attrs]  =  attrs
+      fields[symbol.to_s]  =  items
     end
 
-    def verify( *syms, &block )
-      if !syms.empty?
-        syms.each do |sym|
-          symbols[sym]  =  true
+    def verify( *directives, &block )
+      if !directives.empty?
+        directives.each do |directive|
+          symbols[directive]  =  true
         end
       elsif block
         m  =  /(.*?):(\d+)/.match( caller()[0] )
         call_from  =  [ m[1], m[2] ].join(":")
         blocks[call_from]  =  block
+      else
+        raise  Mongous::Error, "'verify' arguments error. need directives or block."
       end
     end
 
-    def index( *syms, **opts )
-      opts[:background]  =  true    unless opts.has_key?(:background)
+    def index( *symbols, **options )
+      options[:background]  =  true    unless  options.has_key?(:background)
       keys  =  {}
-      syms.each do |sym|
-        keys[sym]  =  1
+      symbols.each do |symbol|
+        keys[symbol]  =  1
+      end
+      indexes.push  <<  [keys, options]
+    end
+
+    def filter( symbol, filter_or_condition )
+      case  filter_or_condition
+      when  Filter
+        filters[symbol]  =  filter_or_condition.to_condition
+      when  Hash
+        filters[symbol]  =  filter_or_condition
+      else
+        m  =  /(.*?):(\d+)/.match( caller()[0] )
+        call_from  =  [ m[1], m[2] ].join(":")
+        raise  Mongous::Error, "'filter' arguments error. : #{symbol}, #{filter_or_condition} at #{ call_from }"
       end
-      indexes.push  <<  [keys, opts]
     end
   end
 end

data/lib/mongous/filter.rb

@@ -1,208 +1,273 @@
-
-module Mongous
-  module Extention
-    def first
-      doc  =  self.collection.find.first
-      self.new( **doc )    if doc
-    end
-
-    def all
-      self.collection.find.map do |doc|
-        self.new( **doc )
-      end
-    end
-
-    def each( &block )
-      all.each( &block )
-    end
-
-    def delete
-      self.collection.delete_many({})
-    end
-
-    def filter( **conditions )
-      Filter.new( self ).filter( conditions )
-    end
-
-    def not( filter = nil, **conditions )
-      raise  Mongous::Error, "Unset args for #{self}.not."    if filter.nil? && conditions.empty?
-
-      filter  ||=  conditions
-      condition  =  case filter
-      when  Hash
-        Filter.new( self ).filter( filter ).to_condition
-      when  Filter
-        filter.to_condition
-      else
-        raise  Mongous::Error, "Invalid args for #{self}.not. : #{filter}"
-      end
-      Filter.new( self ).filter({"$nor" => [condition]})
-    end
-
-    def and( *filters )
-      raise  Mongous::Error, "Unset args for #{self}.and."    if filters.empty?
-
-      conditions  =  filters.map do |filter|
-        case  filter
-        when  Hash
-          filter
-        when  Filter
-          filter.to_condition
-        else
-          raise  Mongous::Error, "Invalid args for #{self}.and. : #{filter}"
-        end
-      end
-      Filter.new( self ).filter({"$and" => conditions})
-    end
-
-    def or( *filters )
-      raise  Mongous::Error, "Unset args for #{self}.or."    if filters.empty?
-
-      conditions  =  filters.map do |filter|
-        case  filter
-        when  Hash
-          filter
-        when  Filter
-          filter.to_condition
-        else
-          raise  Mongous::Error, "Invalid args for #{self}.or. : #{filter}"
-        end
-      end
-      Filter.new( self ).filter({"$or" => conditions})
-    end
-  end
-end
-
-module Mongous
-  class Filter
-    def initialize( klass )
-      @klass  =  klass
-      @filter  =  {}
-      @option  =  {}
-    end
-
-    def filter( conditions )
-      hash  =  {}
-      conditions.each do |key, item|
-        case  key
-        when  /\$(and|or|nor)/
-          hash[key]  =  item
-
-        else
-          case  item
-          when  Array
-            hash[key]  =  {"$in"=>item}
-
-          when  Range
-            _begin_oper  =  "$gte"
-            _end_oper  =  item.exclude_end?  ?  "$lt"  :  "$lte"
-
-            if      item.begin  &&   item.end
-              hash[key]  =  { _begin_oper => item.begin, _end_oper => item.end }
-
-            elsif  !item.begin  &&   item.end
-              hash[key]  =  { _end_oper => item.end }
-
-            elsif   item.begin  &&  !item.end
-              hash[key]  =  { _begin_oper => item.begin }
-
-            else
-              raise  Mongous::Error, "invalid range. :  #{ item }"
-
-            end
-
-          else
-            hash[key]  =  item
-
-          end
-        end
-      end
-      @filter.merge!( hash )
-      self.dup
-    end
-
-    def to_condition
-      @filter.dup
-    end
-
-    def option( _option )
-      self.option!( _option )
-      self.dup
-    end
-
-    def option!( _option )
-      @option.merge!( _option )
-    end
-
-    def projection( _projection )
-      self.projection!( _projection )
-      self.dup
-    end
-    alias  :select  :projection
-
-    def projection!( _projection )
-      @projection  =  _projection
-    end
-
-    def sort( _sort )
-      self.sort!( _sort )
-      self.dup
-    end
-    alias  :order  :sort
-
-    def sort!( _sort )
-      @sort  =  _sort
-    end
-
-    def skip( _skip )
-      self.skip!( _skip )
-      self.dup
-    end
-    alias  :offset  :skip
-
-    def skip!( _skip )
-      @skip  =  _skip
-    end
-
-    def limit( _limit )
-      self.limit!( _limit )
-      self.dup
-    end
-
-    def limit!( _limit )
-      @limit  =  _limit
-    end
-
-    def do_find
-      _filter  =  @filter
-      _option  =  @option
-      _option[:projection]  =  @projection    if @projection
-      found  =  @klass.collection.find( _filter, _option )
-      found  =  found.sort( @sort )    if  @sort
-      found  =  found.skip( @skip )    if  @skip
-      found  =  found.limit( @limit )    if  @limit
-      found
-    end
-
-    def first
-      doc  =  do_find.first
-      @klass.new( **doc )    if doc
-    end
-
-    def all
-      do_find.map do |doc|
-        @klass.new( **doc )
-      end
-    end
-
-    def each( &block )
-      all.each( &block )
-    end
-
-    def delete
-      _filter  =  @filter
-      @klass.collection.delete_many( _filter )
-    end
-  end
-end
-
+
+module Mongous
+  module Extention
+    def count
+      self.collection.estimated_document_count
+    end
+
+    def first
+      doc  =  self.collection.find.first
+      self.new( **doc )    if doc
+    end
+
+    def all
+      self.collection.find.map do |doc|
+        self.new( **doc )
+      end
+    end
+
+    def each( &block )
+      all.each( &block )
+    end
+
+    def map( &block )
+      all.map( &block )
+    end
+
+    def delete
+      self.collection.delete_many({})
+    end
+
+    def attach( collection_name )
+      Filter.new( self ).attach( collection_name )
+    end
+
+    def select( *keys, **hash )
+      Filter.new( self ).select( *keys, **hash )
+    end
+
+    def where( filter = nil, **conditions )
+      condition  =  normalize( filter, conditions )
+      Filter.new( self ).where( condition )
+    end
+
+    def not( filter = nil, **conditions )
+      raise  Mongous::Error, "Unset args for #{self}.not."    if filter.nil? && conditions.empty?
+
+      condition  =  normalize( filter, conditions )
+      Filter.new( self ).where({"$nor" => [condition]})
+    end
+
+    def and( *filters )
+      raise  Mongous::Error, "Unset args for #{self}.and."    if filters.empty?
+
+      conditions  =  filters.map do |filter|
+        normalize( filter, {} )
+      end
+      Filter.new( self ).where({"$and" => conditions})
+    end
+
+    def or( *filters )
+      raise  Mongous::Error, "Unset args for #{self}.or."    if filters.empty?
+
+      conditions  =  filters.map do |filter|
+        normalize( filter, {} )
+      end
+      Filter.new( self ).where({"$or" => conditions})
+    end
+
+    def normalize( filter, conditions )
+      condition  =  case  filter
+      when  Filter
+        filter.to_condition
+      when  Symbol
+        case  _filter  =  filters[filter]
+        when  Filter
+          _filter.to_condition
+        when  Hash
+          _filter
+        end
+      when  NilClass
+        Filter.new( self ).where( **conditions ).to_condition
+      else
+        caller_method  =  /`(.*?)'/.match( caller()[0] )[1]
+        raise  Mongous::Error, "Invalid args for #{self}.#{ caller_method }. : #{filter}, #{conditions}"
+      end
+    end
+  end
+end
+
+module Mongous
+  class Filter
+    def initialize( klass )
+      @klass  =  klass
+      @filter  =  {}
+      @option  =  {}
+    end
+
+    def attach( collection_name )
+      w  =  self.dup
+      w.instance_variable_set( :@collection_name, collection_name.to_s )
+      w
+    end
+
+    def where( conditions )
+      hash  =  {}
+      conditions.each do |key, item|
+        case  key
+        when  /\$(and|or|nor)/
+          hash[key]  =  item
+
+        else
+          case  item
+          when  Array
+            hash[key]  =  {"$in"=>item}
+
+          when  Range
+            _begin_oper  =  "$gte"
+            _end_oper  =  item.exclude_end?  ?  "$lt"  :  "$lte"
+
+            if      item.begin  &&   item.end
+              hash[key]  =  { _begin_oper => item.begin, _end_oper => item.end }
+
+            elsif  !item.begin  &&   item.end
+              hash[key]  =  { _end_oper => item.end }
+
+            elsif   item.begin  &&  !item.end
+              hash[key]  =  { _begin_oper => item.begin }
+
+            else
+              raise  Mongous::Error, "invalid range. :  #{ item }"
+
+            end
+
+          else
+            hash[key]  =  item
+
+          end
+        end
+      end
+      w  =  self.dup
+      w.instance_variable_set( :@filter, @filter.merge( hash ) )
+      w
+    end
+
+    def to_condition
+      @filter.dup
+    end
+
+    def option( _option )
+      w  =  self.dup
+      w.instance_variable_set( :@option, @option.merge( _option ) )
+      w
+    end
+
+    def projection( *keys, **hash )
+      if not keys.empty?
+        _projection  =  Hash[ keys.zip( Array.new(keys.length, 1) ) ]
+      elsif not hash.empty?
+        _projection  =  hash
+      else
+        _projection  =  nil
+      end
+      w  =  self.dup
+      w.instance_variable_set( :@projection, _projection )
+      w
+    end
+    alias  :select  :projection
+
+    def sort( *keys, **hash )
+      if not keys.empty?
+        _sort  =  Hash[ keys.zip( Array.new( keys.length, 1 ) ) ]
+      elsif not hash.empty?
+        _sort  =  hash
+      else
+        _sort  =  nil
+      end
+      w  =  self.dup
+      w.instance_variable_set( :@sort, _sort )
+      w
+    end
+    alias  :order  :sort
+
+    def []( nth_or_range, len = nil )
+      case  nth_or_range
+      when  Integer
+        _skip  =  nth_or_range
+        _limit  =  nil
+
+        if  len
+          raise  Mongous::Error, "invalid len. :  #{ len }"    if  !len.is_a? Integer || len <= 0
+          _limit  =  len
+        end
+
+      when  Range
+        from  =  nth_or_range.begin
+        raise  Mongous::Error, "invalid range. :  #{ nth_or_range }"    unless  from.is_a? Integer
+
+        to    =  nth_or_range.end
+        raise  Mongous::Error, "invalid range. :  #{ nth_or_range }"    unless  to.is_a? Integer
+
+        to  -=  1    if nth_or_range.exclude_end?
+        _skip  =  from
+        _limit  =  to - from + 1
+
+      else
+        raise  Mongous::Error, "invalid class. :  #{ nth_or_range }"
+
+      end
+
+      w  =  self.dup
+      w.instance_variable_set( :@skip, _skip )
+      w.instance_variable_set( :@limit, _limit )
+      w
+    end
+
+    def exec_query
+      _filter  =  @filter
+      _option  =  @option.dup
+      _option[:projection]  =  @projection    if @projection
+      found  =  @klass.collection( @collection_name ).find( _filter, _option )
+      found  =  found.sort( @sort )    if  @sort
+      found  =  found.skip( @skip )    if  @skip
+      found  =  found.limit( @limit )    if  @limit
+      found
+    end
+
+    def count
+      found  =  @klass.collection.find( @filter )
+      found  =  found.skip( @skip )    if  @skip
+      found  =  found.limit( @limit )    if  @limit
+      _count  =  found.count_documents
+      if  @skip
+        if  @skip > _count
+          0
+        elsif  @limit
+          [_count - @skip, @limit].min
+        else
+          _count - @skip
+        end
+      else
+        if  @limit
+          [_count, @limit].min
+        else
+          _count
+        end
+      end
+    end
+
+    def first
+      doc  =  exec_query.first
+      @klass.new( **doc )    if doc
+    end
+
+    def all
+      exec_query.map do |doc|
+        @klass.new( **doc )
+      end
+    end
+
+    def each( &block )
+      all.each( &block )
+    end
+
+    def map( &block )
+      all.map( &block )
+    end
+
+    def delete
+      @klass.collection.delete_many( @filter )
+    end
+  end
+end
+