conversions.rb in mongoid-2.0.0.beta.16

- old
+ new
@@ -1,14 +1,56 @@
 # encoding: utf-8
 module Mongoid #:nodoc:
   module Extensions #:nodoc:
     module ObjectID #:nodoc:
       module Conversions #:nodoc:
+
         def set(value)
-          value
+          if value.is_a?(::String)
+            BSON::ObjectID.from_string(value) unless value.blank?
+          else
+            value
+          end
         end
+
         def get(value)
           value
+        end
+
+        # If the document is using BSON::ObjectIDs the convert the argument to
+        # either an object id or an array of them if the supplied argument is an
+        # Array. Otherwise just return.
+        #
+        # Options:
+        #  args: A +String+ or an +Array+ convert to +BSON::ObjectID+
+        #  cast: A +Boolean+ define if we can or not cast to BSON::ObjectID.
+        #        If false, we use the default type of args
+        #
+        # Example:
+        #
+        # <tt>Mongoid.cast_ids!("4ab2bc4b8ad548971900005c", true)</tt>
+        # <tt>Mongoid.cast_ids!(["4ab2bc4b8ad548971900005c"])</tt>
+        #
+        # Returns:
+        #
+        # If using object ids:
+        #   An +Array+ of +BSON::ObjectID+ of each element if params is an +Array+
+        #   A +BSON::ObjectID+ from params if params is +String+
+        # Otherwise:
+        #   <tt>args</tt>
+        def cast!(klass, args, cast = true)
+          if !klass.using_object_ids? || args.is_a?(::BSON::ObjectID) || !cast
+            return args
+          end
+          if args.is_a?(::String)
+            ::BSON::ObjectID(args)
+          elsif args.is_a?(::Array)
+            args.map{ |a|
+              a.is_a?(::BSON::ObjectID) ? a : ::BSON::ObjectID(a)
+            }
+          else
+            args
+          end
         end
       end
     end
   end
 end