tenacity 0.3.0 → 0.4.0

data/lib/tenacity/orm_ext/helpers.rb

@@ -0,0 +1,36 @@
+module Tenacity
+  module OrmExt
+    module Helpers #:nodoc:
+
+      def id_class_for(association)
+        if association.polymorphic?
+          String
+        elsif association.type == :belongs_to
+          association.source._t_id_type
+        else
+          association.associate_class._t_id_type
+        end
+      end
+
+      def _t_serialize_ids(ids)
+        if ids.respond_to?(:map)
+          ids.map { |id| _t_serialize(id) }
+        else
+          _t_serialize(ids)
+        end
+      end
+
+      def _t_serialize_id_for_sql(id)
+        return id if id.nil?
+
+        serialized_id = _t_serialize(id)
+        if serialized_id.class == String
+          "'#{serialized_id}'"
+        else
+          serialized_id
+        end
+      end
+
+    end
+  end
+end

data/lib/tenacity/orm_ext/mongo_mapper.rb

@@ -1,105 +1,123 @@
 module Tenacity
-  # Tenacity relationships on MongoMapper objects require no special keys
-  # defined on the object.  Tenacity will define the keys that it needs
-  # to support the relationships.  Take the following class for example:
-  #
-  #   class Car < ActiveRecord::Base
-  #     include MongoMapper::Document
-  #     include Tenacity
-  #
-  #     t_has_many    :wheels
-  #     t_has_one     :dashboard
-  #     t_belongs_to  :driver
-  #   end
-  #
-  # == t_belongs_to
-  #
-  # The +t_belongs_to+ association will define a key named after the association.
-  # The example above will create a key named <tt>:driver_id</tt>
-  #
-  #
-  # == t_has_one
-  #
-  # The +t_has_one+ association will not define any new keys on the object, since
-  # the associated object holds the foreign key.
-  #
-  #
-  # == t_has_many
-  #
-  # The +t_has_many+ association will define a key named after the association.
-  # The example above will create a key named <tt>:wheels_ids</tt>
-  #
-  module MongoMapper
-
-    def self.setup(model) #:nodoc:
-      require 'mongo_mapper'
-      if model.included_modules.include?(::MongoMapper::Document)
-        model.send :include, MongoMapper::InstanceMethods
-        model.extend MongoMapper::ClassMethods
-      end
-    rescue LoadError
-      # MongoMapper not available
-    end
+  module OrmExt
+    # Tenacity relationships on MongoMapper objects require no special keys
+    # defined on the object.  Tenacity will define the keys that it needs
+    # to support the relationships.  Take the following class for example:
+    #
+    #   class Car < ActiveRecord::Base
+    #     include MongoMapper::Document
+    #     include Tenacity
+    #
+    #     t_has_many    :wheels
+    #     t_has_one     :dashboard
+    #     t_belongs_to  :driver
+    #   end
+    #
+    # == t_belongs_to
+    #
+    # The +t_belongs_to+ association will define a key named after the association.
+    # The example above will create a key named <tt>:driver_id</tt>
+    #
+    #
+    # == t_has_one
+    #
+    # The +t_has_one+ association will not define any new keys on the object, since
+    # the associated object holds the foreign key.
+    #
+    #
+    # == t_has_many
+    #
+    # The +t_has_many+ association will define a key named after the association.
+    # The example above will create a key named <tt>:wheels_ids</tt>
+    #
+    module MongoMapper
 
-    module ClassMethods #:nodoc:
-      def _t_find(id)
-        find(id)
+      def self.setup(model) #:nodoc:
+        require 'mongo_mapper'
+        if model.included_modules.include?(::MongoMapper::Document)
+          model.send :include, MongoMapper::InstanceMethods
+          model.extend MongoMapper::ClassMethods
+        end
+      rescue LoadError
+        # MongoMapper not available
       end
 
-      def _t_find_bulk(ids=[])
-        find(ids)
-      end
+      module ClassMethods #:nodoc:
+        include Tenacity::OrmExt::Helpers
 
-      def _t_find_first_by_associate(property, id)
-        first(property => id.to_s)
-      end
+        def _t_id_type
+          String
+        end
 
-      def _t_find_all_by_associate(property, id)
-        all(property => id.to_s)
-      end
+        def _t_find(id)
+          find(_t_serialize(id))
+        end
 
-      def _t_initialize_has_many_association(association)
-        unless self.respond_to?(association.foreign_keys_property)
-          key association.foreign_keys_property, Array
-          after_save { |record| _t_save_associates(record, association) }
+        def _t_find_bulk(ids=[])
+          find(_t_serialize_ids(ids))
         end
-      end
 
-      def _t_initialize_belongs_to_association(association)
-        unless self.respond_to?(association.foreign_key)
-          key association.foreign_key, String
-          before_save { |record| _t_stringify_belongs_to_value(record, association) }
+        def _t_find_first_by_associate(property, id)
+          first(property => _t_serialize(id))
         end
-      end
 
-      def _t_delete(ids, run_callbacks=true)
-        if run_callbacks
-          destroy(ids)
-        else
-          delete(ids)
+        def _t_find_all_by_associate(property, id)
+          all(property => _t_serialize(id))
         end
-      end
-    end
 
-    module InstanceMethods #:nodoc:
-      def _t_reload
-        reload
-      rescue ::MongoMapper::DocumentNotFound
-        nil
-      end
+        def _t_initialize_tenacity
+          after_save { |record| record._t_save_autosave_associations }
+        end
 
-      def _t_associate_many(association, associate_ids)
-        self.send(association.foreign_keys_property + '=', associate_ids.map { |associate_id| associate_id.to_s })
-      end
+        def _t_initialize_has_one_association(association)
+          after_destroy { |record| record._t_cleanup_has_one_association(association) }
+        end
+
+        def _t_initialize_has_many_association(association)
+          unless self.respond_to?(association.foreign_keys_property)
+            key association.foreign_keys_property, Array
+            after_save { |record| _t_save_associates(record, association) }
+            after_destroy { |record| record._t_cleanup_has_many_association(association) }
+          end
+        end
+
+        def _t_initialize_belongs_to_association(association)
+          unless self.respond_to?(association.foreign_key)
+            key association.foreign_key, id_class_for(association)
+            key association.polymorphic_type, String if association.polymorphic?
+            after_destroy { |record| record._t_cleanup_belongs_to_association(association) }
+          end
+        end
 
-      def _t_get_associate_ids(association)
-        self.send(association.foreign_keys_property)
+        def _t_delete(ids, run_callbacks=true)
+          if run_callbacks
+            destroy(_t_serialize_ids(ids))
+          else
+            delete(_t_serialize_ids(ids))
+          end
+        end
       end
 
-      def _t_clear_associates(association)
-        self.send(association.foreign_keys_property + '=', [])
+      module InstanceMethods #:nodoc:
+        def _t_reload
+          reload
+        rescue ::MongoMapper::DocumentNotFound
+          nil
+        end
+
+        def _t_associate_many(association, associate_ids)
+          self.send(association.foreign_keys_property + '=', associate_ids)
+        end
+
+        def _t_get_associate_ids(association)
+          self.send(association.foreign_keys_property)
+        end
+
+        def _t_clear_associates(association)
+          self.send(association.foreign_keys_property + '=', [])
+        end
       end
-    end
 
+    end
   end
 end

data/lib/tenacity/orm_ext/mongoid.rb

@@ -1,108 +1,127 @@
 module Tenacity
-  # Tenacity relationships on Mongoid objects require no special keys
-  # defined on the object.  Tenacity will define the keys that it needs
-  # to support the relationships.  Take the following class for example:
-  #
-  #   class Car
-  #     include Mongoid::Document
-  #     include Tenacity
-  #
-  #     t_has_many    :wheels
-  #     t_has_one     :dashboard
-  #     t_belongs_to  :driver
-  #   end
-  #
-  # == t_belongs_to
-  #
-  # The +t_belongs_to+ association will define a key named after the association.
-  # The example above will create a key named <tt>:driver_id</tt>
-  #
-  #
-  # == t_has_one
-  #
-  # The +t_has_one+ association will not define any new keys on the object, since
-  # the associated object holds the foreign key.
-  #
-  #
-  # == t_has_many
-  #
-  # The +t_has_many+ association will define a key named after the association.
-  # The example above will create a key named <tt>:wheels_ids</tt>
-  #
-  module Mongoid
-
-    def self.setup(model) #:nodoc:
-      require 'mongoid'
-      if model.included_modules.include?(::Mongoid::Document)
-        model.send :include, Mongoid::InstanceMethods
-        model.extend Mongoid::ClassMethods
-      end
-    rescue LoadError
-      # Mongoid not available
-    end
+  module OrmExt
+    # Tenacity relationships on Mongoid objects require no special keys
+    # defined on the object.  Tenacity will define the keys that it needs
+    # to support the relationships.  Take the following class for example:
+    #
+    #   class Car
+    #     include Mongoid::Document
+    #     include Tenacity
+    #
+    #     t_has_many    :wheels
+    #     t_has_one     :dashboard
+    #     t_belongs_to  :driver
+    #   end
+    #
+    # == t_belongs_to
+    #
+    # The +t_belongs_to+ association will define a key named after the association.
+    # The example above will create a key named <tt>:driver_id</tt>
+    #
+    #
+    # == t_has_one
+    #
+    # The +t_has_one+ association will not define any new keys on the object, since
+    # the associated object holds the foreign key.
+    #
+    #
+    # == t_has_many
+    #
+    # The +t_has_many+ association will define a key named after the association.
+    # The example above will create a key named <tt>:wheels_ids</tt>
+    #
+    module Mongoid
 
-    module ClassMethods #:nodoc:
-      def _t_find(id)
-        (id.nil? || id.to_s.strip == "") ? nil : find(id)
-      rescue ::Mongoid::Errors::DocumentNotFound
-        nil
+      def self.setup(model) #:nodoc:
+        require 'mongoid'
+        if model.included_modules.include?(::Mongoid::Document)
+          model.send :include, Mongoid::InstanceMethods
+          model.extend Mongoid::ClassMethods
+        end
+      rescue LoadError
+        # Mongoid not available
       end
 
-      def _t_find_bulk(ids)
-        find(ids)
-      rescue ::Mongoid::Errors::DocumentNotFound
-        []
-      end
+      module ClassMethods #:nodoc:
+        include Tenacity::OrmExt::Helpers
 
-      def _t_find_first_by_associate(property, id)
-        find(:first, :conditions => { property => id })
-      end
+        def _t_id_type
+          String
+        end
 
-      def _t_find_all_by_associate(property, id)
-        find(:all, :conditions => { property => id })
-      end
+        def _t_find(id)
+          (id.nil? || id.to_s.strip == "") ? nil : find(_t_serialize(id))
+        rescue ::Mongoid::Errors::DocumentNotFound
+          nil
+        end
 
-      def _t_initialize_has_many_association(association)
-        unless self.respond_to?(association.foreign_keys_property)
-          field association.foreign_keys_property, :type => Array
-          after_save { |record| self.class._t_save_associates(record, association) }
+        def _t_find_bulk(ids)
+          docs = find(_t_serialize_ids(ids))
+          docs.respond_to?(:each) ? docs : [docs]
+        rescue ::Mongoid::Errors::DocumentNotFound
+          []
         end
-      end
 
-      def _t_initialize_belongs_to_association(association)
-        unless self.respond_to?(association.foreign_key)
-          field association.foreign_key, :type => String
-          before_save { |record| self.class._t_stringify_belongs_to_value(record, association) }
+        def _t_find_first_by_associate(property, id)
+          find(:first, :conditions => { property => _t_serialize(id) })
         end
-      end
 
-      def _t_delete(ids, run_callbacks=true)
-        docs = _t_find_bulk(ids)
-        if run_callbacks
-          docs.each { |doc| doc.destroy }
-        else
-          docs.each { |doc| doc.delete }
+        def _t_find_all_by_associate(property, id)
+          find(:all, :conditions => { property => _t_serialize(id) })
         end
-      end
-    end
 
-    module InstanceMethods #:nodoc:
-      def _t_reload
-        reload
-      end
+        def _t_initialize_tenacity
+          after_save { |record| record._t_save_autosave_associations }
+        end
 
-      def _t_associate_many(association, associate_ids)
-        self.send(association.foreign_keys_property + '=', associate_ids.map { |associate_id| associate_id.to_s })
-      end
+        def _t_initialize_has_one_association(association)
+          after_destroy { |record| record._t_cleanup_has_one_association(association) }
+        end
+
+        def _t_initialize_has_many_association(association)
+          unless self.respond_to?(association.foreign_keys_property)
+            field association.foreign_keys_property, :type => Array
+            after_save { |record| self.class._t_save_associates(record, association) }
+            after_destroy { |record| record._t_cleanup_has_many_association(association) }
+          end
+        end
+
+        def _t_initialize_belongs_to_association(association)
+          unless self.respond_to?(association.foreign_key)
+            field association.foreign_key, :type => id_class_for(association)
+            field association.polymorphic_type, :type => String if association.polymorphic?
+            after_destroy { |record| record._t_cleanup_belongs_to_association(association) }
+          end
+        end
 
-      def _t_get_associate_ids(association)
-        self.send(association.foreign_keys_property)
+        def _t_delete(ids, run_callbacks=true)
+          docs = _t_find_bulk(ids)
+          if run_callbacks
+            docs.each { |doc| doc.destroy }
+          else
+            docs.each { |doc| doc.delete }
+          end
+        end
       end
 
-      def _t_clear_associates(association)
-        self.send(association.foreign_keys_property + '=', [])
+      module InstanceMethods #:nodoc:
+        def _t_reload
+          reload
+        end
+
+        def _t_associate_many(association, associate_ids)
+          self.send(association.foreign_keys_property + '=', associate_ids)
+        end
+
+        def _t_get_associate_ids(association)
+          self.send(association.foreign_keys_property)
+        end
+
+        def _t_clear_associates(association)
+          self.send(association.foreign_keys_property + '=', [])
+        end
       end
-    end
 
+    end
   end
 end