sam-dm-core 0.9.6

data/lib/dm-core/associations/one_to_one.rb

@@ -0,0 +1,61 @@
+module DataMapper
+  module Associations
+    module OneToOne
+      extend Assertions
+
+      # Setup one to one relationship between two models
+      # -
+      # @api private
+      def self.setup(name, model, options = {})
+        assert_kind_of 'name',    name,    Symbol
+        assert_kind_of 'model',   model,   Model
+        assert_kind_of 'options', options, Hash
+
+        repository_name = model.repository.name
+
+        model.class_eval <<-EOS, __FILE__, __LINE__
+          def #{name}
+            #{name}_association.first
+          end
+
+          def #{name}=(child_resource)
+            #{name}_association.replace(child_resource.nil? ? [] : [ child_resource ])
+          end
+
+          private
+
+          def #{name}_association
+            @#{name}_association ||= begin
+              unless relationship = model.relationships(#{repository_name.inspect})[:#{name}]
+                raise ArgumentError, "Relationship #{name.inspect} does not exist in \#{model}"
+              end
+              association = Associations::OneToMany::Proxy.new(relationship, self)
+              parent_associations << association
+              association
+            end
+          end
+        EOS
+
+        model.relationships(repository_name)[name] = if options.has_key?(:through)
+          RelationshipChain.new(
+            :child_model              => options.fetch(:class_name, Extlib::Inflection.classify(name)),
+            :parent_model             => model,
+            :repository_name          => repository_name,
+            :near_relationship_name   => options[:through],
+            :remote_relationship_name => options.fetch(:remote_name, name),
+            :parent_key               => options[:parent_key],
+            :child_key                => options[:child_key]
+          )
+        else
+          Relationship.new(
+            name,
+            repository_name,
+            options.fetch(:class_name, Extlib::Inflection.classify(name)),
+            model,
+            options
+          )
+        end
+      end
+    end # module HasOne
+  end # module Associations
+end # module DataMapper

data/lib/dm-core/associations/relationship.rb

@@ -0,0 +1,218 @@
+module DataMapper
+  module Associations
+    class Relationship
+      include Assertions
+
+      OPTIONS = [ :class_name, :child_key, :parent_key, :min, :max, :through ]
+
+      # @api private
+      attr_reader :name, :options, :query
+
+      # @api private
+      def child_key
+        @child_key ||= begin
+          child_key = nil
+          child_model.repository.scope do |r|
+            model_properties = child_model.properties(r.name)
+
+            child_key = parent_key.zip(@child_properties || []).map do |parent_property,property_name|
+              # TODO: use something similar to DM::NamingConventions to determine the property name
+              parent_name = Extlib::Inflection.underscore(Extlib::Inflection.demodulize(parent_model.base_model.name))
+              property_name ||= "#{parent_name}_#{parent_property.name}".to_sym
+
+              if model_properties.has_property?(property_name)
+                model_properties[property_name]
+              else
+                options = {}
+
+                [ :length, :precision, :scale ].each do |option|
+                  options[option] = parent_property.send(option)
+                end
+
+                # NOTE: hack to make each many to many child_key a true key,
+                # until I can figure out a better place for this check
+                if child_model.respond_to?(:many_to_many)
+                  options[:key] = true
+                end
+
+                child_model.property(property_name, parent_property.primitive, options)
+              end
+            end
+          end
+          PropertySet.new(child_key)
+        end
+      end
+
+      # @api private
+      def parent_key
+        @parent_key ||= begin
+          parent_key = nil
+          parent_model.repository.scope do |r|
+            parent_key = if @parent_properties
+              parent_model.properties(r.name).slice(*@parent_properties)
+            else
+              parent_model.key
+            end
+          end
+          PropertySet.new(parent_key)
+        end
+      end
+
+      # @api private
+      def parent_model
+        Class === @parent_model ? @parent_model : (Class === @child_model ? @child_model.find_const(@parent_model) : Object.find_const(@parent_model))
+      rescue NameError
+        raise NameError, "Cannot find the parent_model #{@parent_model} for #{@child_model}"
+      end
+
+      # @api private
+      def child_model
+        Class === @child_model ? @child_model : (Class === @parent_model ? @parent_model.find_const(@child_model) : Object.find_const(@child_model))
+      rescue NameError
+        raise NameError, "Cannot find the child_model #{@child_model} for #{@parent_model}"
+      end
+
+      # @api private
+      def get_children(parent, options = {}, finder = :all, *args)
+        bind_values   = parent_key.get(parent)
+        parent_values = bind_values.dup
+
+        with_repository(child_model) do |r|
+          parent_identity_map = parent.repository.identity_map(parent_model)
+          child_identity_map  = r.identity_map(child_model)
+
+          query_values = parent_identity_map.keys.flatten
+          query_values.reject! { |k| child_identity_map[[k]] }
+
+          bind_values = query_values unless query_values.empty?
+          query = child_key.map { |k| [ k, bind_values ] }.to_hash
+
+          collection = child_model.send(finder, *(args.dup << @query.merge(options).merge(query)))
+          return collection unless collection.kind_of?(Collection) && collection.any?
+
+          grouped_collection = Hash.new { |h,k| h[k] = [] }
+          collection.each do |resource|
+            grouped_collection[get_parent(resource, parent)] << resource
+          end
+
+          association_accessor = "#{self.name}_association"
+
+          ret = nil
+          grouped_collection.each do |parent, children|
+            association = parent.send(association_accessor)
+
+            query = collection.query.dup
+            query.conditions.map! do |operator, property, bind_value|
+              if operator != :raw && child_key.has_property?(property.name)
+                bind_value = *children.map { |child| property.get(child) }.uniq
+              end
+              [ operator, property, bind_value ]
+            end
+
+            parents_children = Collection.new(query)
+            children.each { |child| parents_children.send(:add, child) }
+
+            if parent_key.get(parent) == parent_values
+              ret = parents_children
+            else
+              association.instance_variable_set(:@children, parents_children)
+            end
+          end
+
+          ret || child_model.send(finder, *(args.dup << @query.merge(options).merge(child_key.zip(parent_values).to_hash)))
+        end
+      end
+
+      # @api private
+      def get_parent(child, parent = nil)
+        child_value = child_key.get(child)
+        return nil unless child_value.nitems == child_value.size
+
+        with_repository(parent || parent_model) do
+          parent_identity_map = (parent || parent_model).repository.identity_map(parent_model)
+          child_identity_map  = child.repository.identity_map(child_model)
+
+          if parent = parent_identity_map[child_value]
+            return parent
+          end
+
+          children = child_identity_map.values | [child]
+
+          bind_values = children.map { |c| child_key.get(c) }.uniq
+          query_values = bind_values.reject { |k| parent_identity_map[k] }
+
+          bind_values = query_values unless query_values.empty?
+          query = parent_key.zip(bind_values.transpose).to_hash
+          association_accessor = "#{self.name}_association"
+
+          collection = parent_model.send(:all, query)
+          unless collection.empty?
+            collection.send(:lazy_load)
+            children.each do |c|
+              c.send(association_accessor).instance_variable_set(:@parent, collection.get(*child_key.get(c)))
+            end
+            child.send(association_accessor).instance_variable_get(:@parent)
+          end
+        end
+      end
+
+      # @api private
+      def with_repository(object = nil, &block)
+        other_model = object.model == child_model ? parent_model : child_model if object.respond_to?(:model)
+        other_model = object       == child_model ? parent_model : child_model if object.kind_of?(DataMapper::Resource)
+
+        if other_model && other_model.repository == object.repository && object.repository.name != @repository_name
+          object.repository.scope(&block)
+        else
+          repository(@repository_name, &block)
+        end
+      end
+
+      # @api private
+      def attach_parent(child, parent)
+        child_key.set(child, parent && parent_key.get(parent))
+      end
+
+      private
+
+      # +child_model_name and child_properties refers to the FK, parent_model_name
+      # and parent_properties refer to the PK.  For more information:
+      # http://edocs.bea.com/kodo/docs41/full/html/jdo_overview_mapping_join.html
+      # I wash my hands of it!
+      def initialize(name, repository_name, child_model, parent_model, options = {})
+        assert_kind_of 'name',            name,            Symbol
+        assert_kind_of 'repository_name', repository_name, Symbol
+        assert_kind_of 'child_model',     child_model,     String, Class
+        assert_kind_of 'parent_model',    parent_model,    String, Class
+
+        if child_properties = options[:child_key]
+          assert_kind_of 'options[:child_key]', child_properties, Array
+        end
+
+        if parent_properties = options[:parent_key]
+          assert_kind_of 'options[:parent_key]', parent_properties, Array
+        end
+
+        @name              = name
+        @repository_name   = repository_name
+        @child_model       = child_model
+        @child_properties  = child_properties   # may be nil
+        @query             = options.reject { |k,v| OPTIONS.include?(k) }
+        @parent_model      = parent_model
+        @parent_properties = parent_properties  # may be nil
+        @options           = options
+
+        # attempt to load the child_key if the parent and child model constants are defined
+        if model_defined?(@child_model) && model_defined?(@parent_model)
+          child_key
+        end
+      end
+
+      # @api private
+      def model_defined?(model)
+        # TODO: figure out other ways to see if the model is loaded
+        model.kind_of?(Class)
+      end
+    end # class Relationship
+  end # module Associations
+end # module DataMapper

data/lib/dm-core/associations/relationship_chain.rb

@@ -0,0 +1,81 @@
+module DataMapper
+  module Associations
+    class RelationshipChain < Relationship
+      OPTIONS = [
+        :repository_name, :near_relationship_name, :remote_relationship_name,
+        :child_model, :parent_model, :parent_key, :child_key,
+        :min, :max
+      ]
+
+      undef_method :get_parent
+      undef_method :attach_parent
+
+      # @api private
+      def child_model
+        near_relationship.child_model
+      end
+
+      # @api private
+      def get_children(parent, options = {}, finder = :all, *args)
+        query = @query.merge(options).merge(child_key.to_query(parent_key.get(parent)))
+
+        query[:links]  = links
+        query[:unique] = true
+
+        with_repository(parent) do
+          results = grandchild_model.send(finder, *(args << query))
+          # FIXME: remove the need for the uniq.freeze
+          finder == :all ? (@mutable ? results : results.freeze) : results
+        end
+      end
+
+      private
+
+      # @api private
+      def initialize(options)
+        if (missing_options = OPTIONS - [ :min, :max ] - options.keys ).any?
+          raise ArgumentError, "The options #{missing_options * ', '} are required", caller
+        end
+
+        @repository_name          = options.fetch(:repository_name)
+        @near_relationship_name   = options.fetch(:near_relationship_name)
+        @remote_relationship_name = options.fetch(:remote_relationship_name)
+        @child_model              = options.fetch(:child_model)
+        @parent_model             = options.fetch(:parent_model)
+        @parent_properties        = options.fetch(:parent_key)
+        @child_properties         = options.fetch(:child_key)
+        @mutable                  = options.delete(:mutable) || false
+
+        @name        = near_relationship.name
+        @query       = options.reject{ |key,val| OPTIONS.include?(key) }
+        @extra_links = []
+        @options     = options
+      end
+
+      # @api private
+      def near_relationship
+        parent_model.relationships[@near_relationship_name]
+      end
+
+      # @api private
+      def links
+        if remote_relationship.kind_of?(RelationshipChain)
+          remote_relationship.instance_eval { links } + [remote_relationship.instance_eval { near_relationship } ]
+        else
+          [ remote_relationship ]
+        end
+      end
+
+      # @api private
+      def remote_relationship
+        near_relationship.child_model.relationships[@remote_relationship_name] ||
+          near_relationship.child_model.relationships[@remote_relationship_name.to_s.singularize.to_sym]
+      end
+
+      # @api private
+      def grandchild_model
+        Class === @child_model ? @child_model : (Class === @parent_model ? @parent_model.find_const(@child_model) : Object.find_const(@child_model))
+      end
+    end # class Relationship
+  end # module Associations
+end # module DataMapper

data/lib/dm-core/auto_migrations.rb

@@ -0,0 +1,113 @@
+# TODO: move to dm-more/dm-migrations
+
+module DataMapper
+  class AutoMigrator
+    ##
+    # Destructively automigrates the data-store to match the model.
+    # First migrates all models down and then up.
+    # REPEAT: THIS IS DESTRUCTIVE
+    #
+    # @param Symbol repository_name the repository to be migrated
+    def self.auto_migrate(repository_name = nil, *descendants)
+      auto_migrate_down(repository_name, *descendants)
+      auto_migrate_up(repository_name, *descendants)
+    end
+
+    ##
+    # Destructively automigrates the data-store down
+    # REPEAT: THIS IS DESTRUCTIVE
+    #
+    # @param Symbol repository_name the repository to be migrated
+    # @calls DataMapper::Resource#auto_migrate_down!
+    # @api private
+    def self.auto_migrate_down(repository_name = nil, *descendants)
+      descendants = DataMapper::Resource.descendants.to_a if descendants.empty?
+      descendants.reverse.each do |model|
+        model.auto_migrate_down!(repository_name)
+      end
+    end
+
+    ##
+    # Automigrates the data-store up
+    #
+    # @param Symbol repository_name the repository to be migrated
+    # @calls DataMapper::Resource#auto_migrate_up!
+    # @api private
+    def self.auto_migrate_up(repository_name = nil, *descendants)
+      descendants = DataMapper::Resource.descendants.to_a if descendants.empty?
+      descendants.each do |model|
+        model.auto_migrate_up!(repository_name)
+      end
+    end
+
+    ##
+    # Safely migrates the data-store to match the model
+    # preserving data already in the data-store
+    #
+    # @param Symbol repository_name the repository to be migrated
+    # @calls DataMapper::Resource#auto_upgrade!
+    def self.auto_upgrade(repository_name = nil)
+      DataMapper::Resource.descendants.each do |model|
+        model.auto_upgrade!(repository_name)
+      end
+    end
+  end # class AutoMigrator
+
+  module AutoMigrations
+    ##
+    # Destructively automigrates the data-store to match the model
+    # REPEAT: THIS IS DESTRUCTIVE
+    #
+    # @param Symbol repository_name the repository to be migrated
+    def auto_migrate!(repository_name = self.repository_name)
+      auto_migrate_down!(repository_name)
+      auto_migrate_up!(repository_name)
+    end
+
+    ##
+    # Destructively migrates the data-store down, which basically
+    # deletes all the models.
+    # REPEAT: THIS IS DESTRUCTIVE
+    #
+    # @param Symbol repository_name the repository to be migrated
+    # @api private
+    def auto_migrate_down!(repository_name = self.repository_name)
+      # repository_name ||= default_repository_name
+      if self.superclass != Object
+        self.superclass.auto_migrate!(repository_name)
+      else
+        repository(repository_name) do |r|
+          r.adapter.destroy_model_storage(r, self)
+        end
+      end
+    end
+
+    ##
+    # Auto migrates the data-store to match the model
+    #
+    # @param Symbol repository_name the repository to be migrated
+    # @api private
+    def auto_migrate_up!(repository_name = self.repository_name)
+      if self.superclass != Object
+        self.superclass.auto_migrate!(repository_name)
+      else
+        repository(repository_name) do |r|
+          r.adapter.create_model_storage(r, self)
+        end
+      end
+    end
+
+    ##
+    # Safely migrates the data-store to match the model
+    # preserving data already in the data-store
+    #
+    # @param Symbol repository_name the repository to be migrated
+    def auto_upgrade!(repository_name = self.repository_name)
+      repository(repository_name) do |r|
+        r.adapter.upgrade_model_storage(r, self)
+      end
+    end
+
+    Model.send(:include, self)
+  end # module AutoMigrations
+end # module DataMapper