mongoid-locomotive 2.0.0.beta9

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/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,23 @@
+# 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 ClassMethods #:nodoc:
+          def raise_or_return(value)
+            unless value.nil? || value.is_a?(::Array)
+              raise Mongoid::Errors::InvalidType.new(::Array, value)
+            end
+            value
+          end
+
+          alias :get :raise_or_return
+          alias :set :raise_or_return
+        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,27 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Boolean #:nodoc:
+      module Conversions #:nodoc:
+        extend ActiveSupport::Concern
+
+        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
+        }
+
+        module ClassMethods #:nodoc
+
+          def set(value)
+            value = BOOLEAN_MAP[value]
+            value.nil? ? nil : value
+          end
+
+          def get(value)
+            value
+          end
+        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/false_class/equality.rb

@@ -0,0 +1,13 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module FalseClass #:nodoc:
+      module Equality #:nodoc:
+        def is_a?(other)
+          return true if other.name == "Boolean"
+          super(other)
+        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
+            value
+          end
+        end
+        def get(value)
+          value
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+# 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
+            if elements.kind_of?(::Array)
+              elements.merge!(attrs)
+            else
+              self[key] = attrs.reverse_merge!(elements)
+            end
+          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,40 @@
+# 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.parentize(parent, options.name)
+          child.write_attributes(self)
+          child.identify
+          child.reset_modifications
+          child.notify
+          child
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Conversions #:nodoc:
+        extend ActiveSupport::Concern
+
+        # 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
+
+        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
+            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,21 @@
+# 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 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