# 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