chhean-mongoid 2.0.1.beta1

data/lib/mongoid/extensions/array/accessors.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      module Accessors #:nodoc:
+        # If the attributes already exists in the array then they will be
+        # updated, otherwise they will be appended.
+        def update(attributes)
+          delete_if { |e| attributes["_id"] && (e["_id"] == attributes["_id"]) }
+          self.<< attributes
+        end
+
+        alias :merge! :update
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/array/aliasing.rb

@@ -0,0 +1,4 @@
+# encoding: utf-8
+class Array
+  alias :count :size
+end

data/lib/mongoid/extensions/array/assimilation.rb

@@ -0,0 +1,26 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      module Assimilation #:nodoc:
+        # Introduces a child object into the +Document+ object graph. This will
+        # set up the relationships between the parent and child and update the
+        # attributes of the parent +Document+.
+        #
+        # Options:
+        #
+        # parent: The +Document+ to assimilate into.
+        # options: The association +Options+ for the child.
+        #
+        # Example:
+        #
+        # <tt>[{:street => "Queen St."}, {:street => "King St."}].assimilate(person, options)</tt>
+        #
+        # Returns: The child +Document+.
+        def assimilate(parent, options)
+          each { |child| child.assimilate(parent, options) }
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/array/conversions.rb

@@ -0,0 +1,27 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      # This module converts arrays into mongoid related objects.
+      module Conversions #:nodoc:
+        extend ActiveSupport::Concern
+
+        module InstanceMethods #:nodoc:
+          # Converts this array into an array of hashes.
+          def mongoidize
+            collect { |obj| obj.attributes }
+          end
+        end
+
+        module ClassMethods #:nodoc:
+          def get(value)
+            value
+          end
+          def set(value)
+            value
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/array/parentization.rb

@@ -0,0 +1,13 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      module Parentization #:nodoc:
+        # Adds the parent document to each element in the array.
+        def parentize(parent, association_name)
+          each { |obj| obj.parentize(parent, association_name) }
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/big_decimal/conversions.rb

@@ -0,0 +1,19 @@
+# encoding: utf-8
+require "bigdecimal"
+
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module BigDecimal #:nodoc:
+      module Conversions #:nodoc:
+        # Get the string as a +BigDecimal+
+        def get(value)
+          value ? ::BigDecimal.new(value) : value
+        end
+        # Set the value in the hash as a string.
+        def set(value)
+          value ? value.to_s : value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/binary/conversions.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Binary #:nodoc:
+      module Conversions #:nodoc:
+        # Get the value from the db hash.
+        def get(value)
+          value
+        end
+        # Set the value in the db hash.
+        def set(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/boolean/conversions.rb

@@ -0,0 +1,22 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Boolean #:nodoc:
+      module Conversions #:nodoc:
+
+        BOOLEAN_MAP = {
+          true => true, "true" => true, "TRUE" => true, "1" => true, 1 => true, 1.0 => true,
+          false => false, "false" => false, "FALSE" => false, "0" => false, 0 => false, 0.0 => false
+        }
+
+        def set(value)
+          value = BOOLEAN_MAP[value]
+          value.nil? ? nil : value
+        end
+        def get(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/date/conversions.rb

@@ -0,0 +1,24 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Date #:nodoc:
+      module Conversions #:nodoc:
+        def get(value)
+          return nil if value.blank?
+          if Mongoid::Config.instance.use_utc?
+            value.to_date
+          else
+            ::Date.new(value.year, value.month, value.day)
+          end
+        end
+
+        protected
+
+        def convert_to_time(value)
+          value = ::Date.parse(value) if value.is_a?(::String)
+          ::Time.utc(value.year, value.month, value.day)
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/datetime/conversions.rb

@@ -0,0 +1,12 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module DateTime #:nodoc:
+      module Conversions #:nodoc:
+        def get(value)
+          super.try(:to_datetime)
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/float/conversions.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Float #:nodoc:
+      module Conversions #:nodoc:
+        def set(value)
+          return nil if value.blank?
+          begin
+            Float(value)
+          rescue ArgumentError => e
+            puts(e.message); value
+          end
+        end
+        def get(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/accessors.rb

@@ -0,0 +1,38 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Accessors #:nodoc:
+
+        # Remove a set of attributes from a hash. If the attributes are
+        # contained in an array it will remove it from the array, otherwise it
+        # will delete the child attribute completely.
+        def remove(key, attrs)
+          elements = self[key]
+          if elements
+            key.singular? ? self[key] = nil : elements.delete(attrs)
+          end
+        end
+
+        # Inserts new attributes into the hash. If the elements are present in
+        # the hash it will update them, otherwise it will add the new
+        # attributes into the hash as either a child hash or child array of
+        # hashes.
+        def insert(key, attrs)
+          elements = self[key]
+          if elements
+            elements.merge!(attrs)
+          else
+            self[key] = key.singular? ? attrs : [attrs]
+          end
+        end
+
+        # If a _type key exists in the hash, return the class for the value.
+        def klass
+          class_name = self["_type"]
+          class_name ? class_name.constantize : nil
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/assimilation.rb

@@ -0,0 +1,39 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Assimilation #:nodoc:
+        # Introduces a child object into the +Document+ object graph. This will
+        # set up the relationships between the parent and child and update the
+        # attributes of the parent +Document+.
+        #
+        # Options:
+        #
+        # parent: The +Document+ to assimilate into.
+        # options: The association +Options+ for the child.
+        #
+        # Example:
+        #
+        # <tt>{:first_name => "Hank", :last_name => "Moody"}.assimilate(name, options)</tt>
+        #
+        # Returns: The child +Document+.
+        def assimilate(parent, options, type = nil)
+          klass = self.klass || (type ? type : options.klass)
+          child = klass.instantiate("_id" => self["_id"])
+          self.merge("_type" => klass.name) if klass.hereditary
+          init(parent, child, options)
+        end
+
+        protected
+
+        def init(parent, child, options)
+          child._parent = parent
+          child.write_attributes(self)
+          child.identify
+          child.reset_modifications
+          child.assimilate(parent, options)
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/conversions.rb

@@ -0,0 +1,45 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Conversions #:nodoc:
+        extend ActiveSupport::Concern
+
+        module InstanceMethods #:nodoc:
+
+          # Get the difference between 2 hashes. This will give back a new hash
+          # with the keys and pairs of [ old, new ] values.
+          #
+          # Example:
+          #
+          #   first = { :field => "value" }
+          #   second = { :field => "new" }
+          #   first.difference(second) # => { :field => [ "value", "new" ] }
+          #
+          # Returns:
+          #
+          # A +Hash+ of modifications.
+          def difference(other)
+            changes = {}
+            each_pair do |key, value|
+              if other.has_key?(key)
+                new_value = other[key]
+                changes[key] = [ value, new_value ] if new_value != value
+              end
+            end
+            changes
+          end
+        end
+
+        module ClassMethods #:nodoc:
+          def get(value)
+            value
+          end
+          def set(value)
+            value
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/criteria_helpers.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module CriteriaHelpers #:nodoc:
+        def expand_complex_criteria
+          hsh = {}
+          self.each_pair do |k,v|
+            if k.class == Mongoid::Criterion::Complex
+              hsh[k.key] = {"$#{k.operator}" => v}
+            else
+              hsh[k] = v
+            end
+          end
+          hsh
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/scoping.rb

@@ -0,0 +1,12 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Scoping #:nodoc:
+        def scoped(*args)
+          self
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/integer/conversions.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Integer #:nodoc:
+      module Conversions #:nodoc:
+        def set(value)
+          return nil if value.blank?
+          begin
+            Integer(value)
+          rescue ArgumentError => e
+            puts(e.message); value
+          end
+        end
+        def get(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/nil/assimilation.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Nil #:nodoc:
+      module Assimilation #:nodoc:
+        # Will remove the child object from the parent.
+        def assimilate(parent, options, type = nil)
+          parent.remove_attribute(options.name); self
+        end
+
+        def collectionize
+          to_s.collectionize
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/object/conversions.rb

@@ -0,0 +1,27 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Object #:nodoc:
+      # This module converts objects into mongoid related objects.
+      module Conversions #:nodoc:
+        extend ActiveSupport::Concern
+        module InstanceMethods
+          # Converts this object to a hash of attributes
+          def mongoidize
+            self.raw_attributes
+          end
+        end
+
+        module ClassMethods
+          def set(value)
+            value.respond_to?(:raw_attributes) ? value.raw_attributes : value
+          end
+
+          def get(value)
+            value ? self.instantiate(value) : value
+          end
+        end
+      end
+    end
+  end
+end