rom 5.4.2 → 6.0.0.alpha1

data/lib/rom/schema/associations_dsl.rb

@@ -0,0 +1,222 @@
+# frozen_string_literal: true
+
+require "rom/support/inflector"
+
+require "rom/associations/definitions"
+
+module ROM
+  class Schema
+    # Additional schema DSL for definition SQL associations
+    #
+    # This DSL is exposed in `associations do .. end` blocks in schema defintions.
+    #
+    # @api public
+    class AssociationsDSL < ::BasicObject
+      class << self
+        define_method(:const_missing, ::Object.method(:const_get))
+      end
+
+      include Associations::Definitions
+
+      # @!attribute [r] source
+      #   @return [Relation::Name] The source relation
+      attr_reader :source
+
+      # @!attribute [r] inflector
+      #   @return [Dry::Inflector] String inflector
+      attr_reader :inflector
+
+      # @!attribute [r] registry
+      #   @return [Hash]
+      attr_reader :registry
+
+      # @api private
+      def initialize(source, inflector = Inflector, &block)
+        @source = source
+        @inflector = inflector
+        @registry = {}
+        instance_exec(&block) if block
+      end
+
+      # Establish a one-to-many association
+      #
+      # @example using relation identifier
+      #   has_many :tasks
+      #
+      # @example setting custom foreign key name
+      #   has_many :tasks, foreign_key: :assignee_id
+      #
+      # @example with a :through option
+      #   # this establishes many-to-many association
+      #   has_many :tasks, through: :users_tasks
+      #
+      # @example using a custom view which overrides default one
+      #   has_many :posts, view: :published, override: true
+      #
+      # @example using aliased association with a custom view
+      #   has_many :posts, as: :published_posts, view: :published
+      #
+      # @example using custom target relation
+      #   has_many :user_posts, relation: :posts
+      #
+      # @example using custom target relation and an alias
+      #   has_many :user_posts, relation: :posts, as: :published, view: :published
+      #
+      # @param [Symbol] target The target relation identifier
+      # @param [Hash] options A hash with additional options
+      #
+      # @return [Associations::OneToMany]
+      #
+      # @see #many_to_many
+      #
+      # @api public
+      def one_to_many(target, **options)
+        if options[:through]
+          many_to_many(target, **options, inflector: inflector)
+        else
+          add(build(OneToMany, target, options))
+        end
+      end
+      alias_method :has_many, :one_to_many
+
+      # Establish a one-to-one association
+      #
+      # @example using relation identifier
+      #   one_to_one :addresses, as: :address
+      #
+      # @example with an intermediate join relation
+      #   one_to_one :tasks, as: :priority_task, through: :assignments
+      #
+      # @param [Symbol] target The target relation identifier
+      # @param [Hash] options A hash with additional options
+      #
+      # @return [Associations::OneToOne]
+      #
+      # @see #belongs_to
+      #
+      # @api public
+      def one_to_one(target, **options)
+        if options[:through]
+          one_to_one_through(target, **options)
+        else
+          add(build(OneToOne, target, options))
+        end
+      end
+
+      # Establish a one-to-one association with a :through option
+      #
+      # @example
+      #   one_to_one_through :users, as: :author, through: :users_posts
+      #
+      # @return [Associations::OneToOneThrough]
+      #
+      # @api public
+      def one_to_one_through(target, **options)
+        add(build(OneToOneThrough, target, options))
+      end
+
+      # Establish a many-to-many association
+      #
+      # @example using relation identifier
+      #   many_to_many :tasks, through: :users_tasks
+      #
+      # @param [Symbol] target The target relation identifier
+      # @param [Hash] options A hash with additional options
+      #
+      # @return [Associations::ManyToMany]
+      #
+      # @see #one_to_many
+      #
+      # @api public
+      def many_to_many(target, **options)
+        add(build(ManyToMany, target, options))
+      end
+
+      # Establish a many-to-one association
+      #
+      # @example using relation identifier
+      #   many_to_one :users, as: :author
+      #
+      # @param [Symbol] target The target relation identifier
+      # @param [Hash] options A hash with additional options
+      #
+      # @return [Associations::ManyToOne]
+      #
+      # @see #one_to_many
+      #
+      # @api public
+      def many_to_one(target, **options)
+        add(build(ManyToOne, target, options))
+      end
+
+      # Shortcut for many_to_one which sets alias automatically
+      #
+      # @example with an alias (relation identifier is inferred via pluralization)
+      #   belongs_to :user
+      #
+      # @example with an explicit alias
+      #   belongs_to :users, as: :author
+      #
+      # @see #many_to_one
+      #
+      # @return [Associations::ManyToOne]
+      #
+      # @api public
+      def belongs_to(target, **options)
+        many_to_one(dataset_name(target), as: target, **options)
+      end
+
+      # Shortcut for one_to_one which sets alias automatically
+      #
+      # @example with an alias (relation identifier is inferred via pluralization)
+      #   has_one :address
+      #
+      # @example with an explicit alias and a custom view
+      #   has_one :posts, as: :priority_post, view: :prioritized
+      #
+      # @see #one_to_one
+      #
+      # @return [Associations::OneToOne]
+      #
+      # @api public
+      def has_one(target, **options)
+        one_to_one(dataset_name(target), as: target, **options)
+      end
+
+      # Return an association set for a schema
+      #
+      # @return [Array<Association::Definition]
+      #
+      # @api private
+      def call
+        registry.values
+      end
+
+      private
+
+      # @api private
+      def build(definition, target, options)
+        definition.new(source, target, **options, inflector: inflector)
+      end
+
+      # @api private
+      def add(association)
+        key = association.as || association.name
+
+        if registry.key?(key)
+          ::Kernel.raise(
+            ::ArgumentError,
+            "association #{key.inspect} is already defined for #{source.to_sym.inspect} relation"
+          )
+        end
+
+        registry[key] = association
+      end
+
+      # @api private
+      def dataset_name(name)
+        inflector.pluralize(name).to_sym
+      end
+    end
+  end
+end

data/lib/rom/schema/inferrer.rb

@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+
+require "dry/core/class_attributes"
+
+module ROM
+  class Schema
+    # @api private
+    class Inferrer
+      extend Dry::Core::ClassAttributes
+      extend Initializer
+
+      # @!method self.attributes_inferrer
+      #   @overload attributes_inferrer
+      #     @return [Proc]
+      #
+      #   @overload attributes_inferrer(value)
+      #     @param value [Proc]
+      #     @return [Proc]
+      #
+      # @!method self.attr_class
+      #   @overload attr_class
+      #     @return [Class(ROM::Attribute)]
+      #
+      #   @overload attr_class(value)
+      #     @param value [Class(ROM::Attribute)]
+      #     @return [Class(ROM::Attribute)]
+      defines :attributes_inferrer, :attr_class
+
+      MissingAttributesError = Class.new(StandardError) do
+        def initialize(name, attributes)
+          super(
+            "Following attributes in #{Relation::Name[name].relation.inspect} schema cannot "\
+            "be inferred and have to be defined explicitly: #{attributes.map(&:inspect).join(", ")}"
+          )
+        end
+      end
+
+      DEFAULT_ATTRIBUTES = [EMPTY_ARRAY, EMPTY_ARRAY].freeze
+
+      attributes_inferrer -> * { DEFAULT_ATTRIBUTES }
+
+      attr_class Attribute
+
+      include Dry::Equalizer(:options)
+
+      # @!attribute [r] attr_class
+      #   @return [Class(ROM::Attribute)]
+      option :attr_class, default: -> { self.class.attr_class }
+
+      # @!attribute [r] enabled
+      #   @return [Boolean]
+      option :enabled, default: -> { true }
+
+      alias_method :enabled?, :enabled
+
+      # @!attribute [r] attributes_inferrer
+      #   @return [ROM::Schema::AttributesInferrer]
+      option :attributes_inferrer, default: -> { self.class.attributes_inferrer }
+
+      # @api private
+      def call(schema, gateway)
+        if enabled?
+          inferred, missing = attributes_inferrer.(schema, gateway, options)
+        else
+          inferred, missing = DEFAULT_ATTRIBUTES
+        end
+
+        attributes = merge_attributes(schema.attributes, inferred)
+
+        check_all_attributes_defined(schema, attributes, missing)
+
+        {attributes: attributes}
+      end
+
+      # @api private
+      def check_all_attributes_defined(schema, all_known, not_inferred)
+        not_defined = not_inferred - all_known.map(&:name)
+
+        raise MissingAttributesError.new(schema.name, not_defined) unless not_defined.empty?
+      end
+
+      # @api private
+      def merge_attributes(defined, inferred)
+        type_lookup = lambda do |attrs, name|
+          attrs.find { |a| a.name == name }.type
+        end
+        defined_with_type, defined_names =
+          defined.each_with_object([[], []]) do |attr, (attrs, names)|
+            attrs << if attr.type.nil?
+                       attr.class.new(
+                         type_lookup.(inferred, attr.name),
+                         **attr.options
+                       )
+                     else
+                       attr
+                     end
+            names << attr.name
+          end
+
+        defined_with_type + inferred.reject do |attr|
+          defined_names.include?(attr.name)
+        end
+      end
+    end
+  end
+end