Contents

require 'ostruct'
require 'active_support/core_ext/string'
module Relationships

  def self.included(base)
    base.extend(ClassMethods)
  end

  module ClassMethods
    attr_reader :has_and_belongs_to_many, :belongs_to, :has_one, :has_many
  end

  def relationships
    OpenStruct.new({has_many:   @has_many   ||= [],
                    has_one:    @has_one    ||= [],
                    belongs_to: @belongs_to ||= [],
                    has_and_belongs_to_many: @has_and_belongs_to_many ||= []})
  end

  def single_relationships
    belongs_to + has_one
  end

  def collections
    has_and_belongs_to_many + has_many
  end

  class Relationship
    attr_reader :name
    def initialize(name, options={})
      @name = name
      @options = options.reduce(HashWithIndifferentAccess.new, :merge)
    end

    def options
      @options.symbolize_keys
    end

    def through
      options[:through]
    end

    def class_name
      options[:class_name] || name.to_s.camelize.singularize
    end

    def foreign_key
      options[:foreign_key] || name.to_s.foreign_key
    end

    def join_table
      options[:join_table]
    end
  end

  class HasMany < Relationship

    attr_reader :klass_name

    def initialize(name, klass_name, options={})
      @klass_name = klass_name
      super(name, options)
    end

    def foreign_key
      options[:foreign_key] || klass_name.to_s.foreign_key
    end

  end

  def has_many(*args)
    @has_many ||= []
    @has_many.push HasMany.new(args.shift, self.name, args)
  end

  class HasOne < Relationship
  end

  def has_one(*args)
    @has_one ||= []
    @has_one.push HasOne.new(args.shift, args)
  end

  class BelongsTo < Relationship
  end

  def belongs_to(*args)
    @belongs_to ||= []
    @belongs_to.push BelongsTo.new(args.shift, args)
  end


  class HasAndBelongsToMany < Relationship
  end

  def has_and_belongs_to_many(*args)
    @has_and_belongs_to_many ||= []
    @has_and_belongs_to_many.push HasAndBelongsToMany.new(args.shift, args)
  end

end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
active_mocker-1.5.2	lib/active_mocker/active_record/relationships.rb
active_mocker-1.5.1	lib/active_mocker/active_record/relationships.rb
active_mocker-1.5	lib/active_mocker/active_record/relationships.rb
active_mocker-1.4.2	lib/active_mocker/active_record/relationships.rb
active_mocker-1.4.1	lib/active_mocker/active_record/relationships.rb